[Offline Pages] Adds new policy bits and reverse lookup.

chrome/browser/android/offline_pages/offline_page_utils.cc

 // Copyright 2015 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/android/offline_pages/offline_page_utils.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_util.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "chrome/browser/android/offline_pages/offline_page_mhtml_archiver.h"
 #include "chrome/browser/android/offline_pages/offline_page_model_factory.h"
 #include "chrome/browser/android/offline_pages/offline_page_tab_helper.h"
 #include "chrome/browser/android/tab_android.h"
 #include "components/offline_pages/client_namespace_constants.h"
+#include "components/offline_pages/client_policy_controller.h"
 #include "components/offline_pages/offline_page_feature.h"
 #include "components/offline_pages/offline_page_item.h"
 #include "components/offline_pages/offline_page_model.h"
 #include "components/offline_pages/request_header/offline_page_header.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/web_contents.h"
 #include "url/gurl.h"
 
 namespace offline_pages {
 namespace {
@@ skipping 22 matching lines @@
     const base::Callback<void(const GURL&)>& callback,
     const OfflinePageItem* item) {
   GURL result_url;
   if (item)
     result_url = item->url;
   callback.Run(result_url);
 }
 
 void OnGetPagesByOnlineURLDone(
     int tab_id,
+    const std::vector<std::string>& namespaces_to_show_in_original_tab,
     const base::Callback<void(const OfflinePageItem*)>& callback,
     const MultipleOfflinePageItemResult& pages) {
   const OfflinePageItem* selected_page = nullptr;
   std::string tab_id_str = base::IntToString(tab_id);
+
   for (const auto& offline_page : pages) {
-    if (offline_page.client_id.name_space != kLastNNamespace ||
-        offline_page.client_id.id == tab_id_str) {
-      if (!selected_page ||
-          offline_page.creation_time > selected_page->creation_time) {
-        selected_page = &offline_page;
-      }
+    auto result = std::find(namespaces_to_show_in_original_tab.begin(),
+                            namespaces_to_show_in_original_tab.end(),
+                            offline_page.client_id.name_space);
+    if (result != namespaces_to_show_in_original_tab.end() &&
+        offline_page.client_id.id != tab_id_str) {
+      continue;
+    }
+
+    if (!selected_page ||
+        offline_page.creation_time > selected_page->creation_time) {
+      selected_page = &offline_page;
     }
   }
   callback.Run(selected_page);
 }
 
 }  // namespace
 
 // static
 bool OfflinePageUtils::MightBeOfflineURL(const GURL& url) {
   // It has to be a file URL ending with .mhtml extension.
@@ skipping 23 matching lines @@
     const base::Callback<void(const OfflinePageItem*)>& callback) {
   OfflinePageModel* offline_page_model =
       OfflinePageModelFactory::GetForBrowserContext(browser_context);
   if (!offline_page_model) {
     base::ThreadTaskRunnerHandle::Get()->PostTask(
         FROM_HERE, base::Bind(callback, nullptr));
     return;
   }
 
   offline_page_model->GetPagesByOnlineURL(
-      online_url, base::Bind(&OnGetPagesByOnlineURLDone, tab_id, callback));
+      online_url, base::Bind(&OnGetPagesByOnlineURLDone, tab_id,
+                             offline_page_model->GetPolicyController()
+                                 ->GetNamespacesRestrictedToOriginalTab(),
+                             callback));
 }
 
 // static
 void OfflinePageUtils::GetOnlineURLForOfflineURL(
     content::BrowserContext* browser_context,
     const GURL& offline_url,
     const base::Callback<void(const GURL&)>& callback) {
   OfflinePageModel* offline_page_model =
       OfflinePageModelFactory::GetForBrowserContext(browser_context);
   if (!offline_page_model) {
@@ skipping 63 matching lines @@
 bool OfflinePageUtils::GetTabId(content::WebContents* web_contents,
                                 int* tab_id) {
   TabAndroid* tab_android = TabAndroid::FromWebContents(web_contents);
   if (!tab_android)
     return false;
   *tab_id = tab_android->GetAndroidId();
   return true;
 }
 
 }  // namespace offline_pages