Showing previews UI for Offline Previews

chrome/browser/loader/chrome_navigation_data.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/loader/chrome_navigation_data.h"
 
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_data.h"
+#include "components/offline_pages/loaded_offline_page_info.h"
+#include "content/public/browser/navigation_handle.h"
 #include "net/url_request/url_request.h"
 
+namespace {
+
 const void* const kChromeNavigationDataUserDataKey =
     &kChromeNavigationDataUserDataKey;
+}
 
 ChromeNavigationData::ChromeNavigationData() {}
 
 ChromeNavigationData::~ChromeNavigationData() {}
 
 ChromeNavigationData* ChromeNavigationData::GetDataAndCreateIfNecessary(
     net::URLRequest* request) {
   if (!request)
     return nullptr;
   ChromeNavigationData* data = static_cast<ChromeNavigationData*>(
       request->GetUserData(kChromeNavigationDataUserDataKey));
   if (data)
     return data;
   data = new ChromeNavigationData();
   request->SetUserData(kChromeNavigationDataUserDataKey, data);
   return data;
 }
 
+ChromeNavigationData* ChromeNavigationData::GetForNavigationHandle(
+    content::NavigationHandle* navigation_handle) {
+  // As documented in content/public/browser/navigation_handle.h, this
+  // NavigationData is a clone of the NavigationData instance returned from
+  // ResourceDispatcherHostDelegate::GetNavigationData during commit.
+  // Because ChromeResourceDispatcherHostDelegate always returns a
+  // ChromeNavigationData, it is safe to static_cast here.
+  if (!navigation_handle)
+    return nullptr;
+  return static_cast<ChromeNavigationData*>(
+      navigation_handle->GetNavigationData());
+}
+
 std::unique_ptr<content::NavigationData> ChromeNavigationData::Clone() const {
   std::unique_ptr<ChromeNavigationData> copy(new ChromeNavigationData());
   if (data_reduction_proxy_data_)
     copy->SetDataReductionProxyData(data_reduction_proxy_data_->DeepCopy());
+  if (loaded_offline_page_info_)
+    copy->SetLoadedOfflinePageInfo(loaded_offline_page_info_->DeepCopy());
   return std::move(copy);
 }
+
+void ChromeNavigationData::SetDataReductionProxyData(
+    std::unique_ptr<data_reduction_proxy::DataReductionProxyData>
+        data_reduction_proxy_data) {
+  data_reduction_proxy_data_ = std::move(data_reduction_proxy_data);
+}
+
+void ChromeNavigationData::SetLoadedOfflinePageInfo(
+    std::unique_ptr<offline_pages::LoadedOfflinePageInfo>
+        loaded_offline_page_info) {
+  loaded_offline_page_info_ = std::move(loaded_offline_page_info);
+}