[arc-intents] Handle redirects.

chrome/browser/chromeos/arc/arc_navigation_throttle.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/chromeos/arc/arc_navigation_throttle.h"
 
 #include <algorithm>
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted.h"
 #include "base/metrics/histogram_macros.h"
 #include "components/arc/arc_bridge_service.h"
 #include "components/arc/arc_service_manager.h"
 #include "components/arc/intent_helper/arc_intent_helper_bridge.h"
+#include "components/arc/intent_helper/local_activity_resolver.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_handle.h"
 #include "net/base/registry_controlled_domains/registry_controlled_domain.h"
 #include "ui/base/page_transition_types.h"
 
 namespace arc {
 
 namespace {
 
 constexpr int kMinInstanceVersion = 7;
@@ skipping 21 matching lines @@
     VLOG(1) << "ARC intent helper instance is too old.";
     return nullptr;
   }
   return intent_helper_instance;
 }
 
 }  // namespace
 
 ArcNavigationThrottle::ArcNavigationThrottle(
     content::NavigationHandle* navigation_handle,
-    const ShowDisambigDialogCallback& show_disambig_dialog_cb)
+    const ShowIntentPickerCallback& show_intent_picker_cb)
     : content::NavigationThrottle(navigation_handle),
-      show_disambig_dialog_callback_(show_disambig_dialog_cb),
+      show_intent_picker_callback_(show_intent_picker_cb),
+      previous_user_action_(CloseReason::INVALID),
       weak_ptr_factory_(this) {}
 
 ArcNavigationThrottle::~ArcNavigationThrottle() = default;
 
 content::NavigationThrottle::ThrottleCheckResult
 ArcNavigationThrottle::WillStartRequest() {
+  return HandleRequest();

[djacobo_, 2016/07/29 20:23:17]

question: Do we need to add DCHECK_CURRENTLY_ON(content::BrowserThread::UI);

here and in WillRedirectRequest() ? We have it on HandleRequest() but I don't
know if it suffices.

[Ben Kwa, 2016/07/29 22:21:00]

I moved the DCHECK thread checks back into WillHandleRequest and
WillRedirectRequest.

+}
+
+content::NavigationThrottle::ThrottleCheckResult
+ArcNavigationThrottle::WillRedirectRequest() {
+  switch (previous_user_action_) {
+    case CloseReason::ERROR:
+    case CloseReason::DIALOG_DEACTIVATED:
+      // User dismissed the dialog, or some error occurred before.  Don't
+      // repeatedly pop up the dialog.
+      return content::NavigationThrottle::PROCEED;
+
+    case CloseReason::ALWAYS_PRESSED:
+    case CloseReason::JUST_ONCE_PRESSED:
+    case CloseReason::PREFERRED_ACTIVITY_FOUND:
+      // Should never get here - if the user selected one of these previously,
+      // Chrome should not see a redirect.
+      NOTREACHED();
+
+    case CloseReason::INVALID:
+      // No picker has previously been popped up for this - continue.
+      break;
+  }
+  return HandleRequest();
+}
+
+content::NavigationThrottle::ThrottleCheckResult
+ArcNavigationThrottle::HandleRequest() {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 
+  const GURL& url = navigation_handle()->GetURL();
+
+  // Mask out any redirect qualifiers - this method handles navigation from
+  // redirect and non-redirect navigations equivalently.
   const ui::PageTransition transition =
-      navigation_handle()->GetPageTransition();
+      ui::PageTransitionFromInt(navigation_handle()->GetPageTransition() &
+                                ~ui::PAGE_TRANSITION_IS_REDIRECT_MASK);
 
   if (!ui::PageTransitionCoreTypeIs(transition, ui::PAGE_TRANSITION_LINK)) {
-    // If this navigation event wasn't spawned by the user clicking on a link.
+    // Allow navigation to proceed if this event wasn't spawned by the user
+    // clicking on a link.
     return content::NavigationThrottle::PROCEED;
   }
 
   if (ui::PageTransitionGetQualifier(transition) != 0) {
     // Qualifiers indicate that this navigation was the result of a click on a
-    // forward/back button, or a redirect, or typing in the URL bar, etc.  Don't
-    // pass any of those types of navigations to the intent helper (see
-    // crbug.com/630072).
+    // forward/back button, or typing in the URL bar, etc.  Don't pass any of
+    // those types of navigations to the intent helper (see crbug.com/630072).
+    // Note that redirects, which we do pass on, are masked out above.
     return content::NavigationThrottle::PROCEED;
   }
 
   if (!ShouldOverrideUrlLoading(navigation_handle()))
     return content::NavigationThrottle::PROCEED;
 
-  const GURL& url = navigation_handle()->GetURL();
+  arc::ArcServiceManager* arc_service_manager = arc::ArcServiceManager::Get();
+  DCHECK(arc_service_manager);
+  scoped_refptr<arc::LocalActivityResolver> local_resolver =
+      arc_service_manager->activity_resolver();
+  if (local_resolver->ShouldChromeHandleUrl(url)) {
+    // Allow navigation to proceed if there isn't an android app that handles
+    // the given URL.
+    return content::NavigationThrottle::PROCEED;
+  }
+
   mojom::IntentHelperInstance* bridge_instance = GetIntentHelper();
   if (!bridge_instance)
     return content::NavigationThrottle::PROCEED;
   bridge_instance->RequestUrlHandlerList(
       url.spec(), base::Bind(&ArcNavigationThrottle::OnAppCandidatesReceived,
                              weak_ptr_factory_.GetWeakPtr()));
   return content::NavigationThrottle::DEFER;
 }
 
-content::NavigationThrottle::ThrottleCheckResult
-ArcNavigationThrottle::WillRedirectRequest() {
-  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
-  return content::NavigationThrottle::PROCEED;
-}
-
 // We received the array of app candidates to handle this URL (even the Chrome
 // app is included).
 void ArcNavigationThrottle::OnAppCandidatesReceived(
     mojo::Array<mojom::UrlHandlerInfoPtr> handlers) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   if (handlers.empty() ||
       (handlers.size() == 1 && ArcIntentHelperBridge::IsIntentHelperPackage(
                                    handlers[0]->package_name))) {
     // This scenario shouldn't be accesed as ArcNavigationThrottle is created
     // iff there are ARC apps which can actually handle the given URL.
     DVLOG(1) << "There are no app candidates for this URL: "
              << navigation_handle()->GetURL().spec();
     navigation_handle()->Resume();
     return;
   }
 
   // If one of the apps is marked as preferred, use it right away without
   // showing the UI.
   for (size_t i = 0; i < handlers.size(); ++i) {
     if (!handlers[i]->is_preferred)
       continue;
     if (ArcIntentHelperBridge::IsIntentHelperPackage(
             handlers[i]->package_name)) {
       // If Chrome browser was selected as the preferred app, we should't
       // create a throttle.
       DVLOG(1)
           << "Chrome browser is selected as the preferred app for this URL: "
           << navigation_handle()->GetURL().spec();
     }
-    OnDisambigDialogClosed(std::move(handlers), i,
-                           CloseReason::PREFERRED_ACTIVITY_FOUND);
+    OnIntentPickerClosed(std::move(handlers), i,
+                         CloseReason::PREFERRED_ACTIVITY_FOUND);
     return;
   }
 
   // Swap Chrome app with any app in row |kMaxAppResults-1| iff his index is
   // bigger, thus ensuring the user can always see Chrome without scrolling.
   size_t chrome_app_index = 0;
   for (size_t i = 0; i < handlers.size(); ++i) {
     if (ArcIntentHelperBridge::IsIntentHelperPackage(
             handlers[i]->package_name)) {
       chrome_app_index = i;
@@ skipping 29 matching lines @@
   for (const auto& handler : handlers) {
     gfx::Image icon;
     const ActivityIconLoader::ActivityName activity(handler->package_name,
                                                     handler->activity_name);
     const auto it = icons->find(activity);
 
     app_info.emplace_back(
         handler->name, it != icons->end() ? it->second.icon20 : gfx::Image());
   }
 
-  show_disambig_dialog_callback_.Run(
+  show_intent_picker_callback_.Run(
       navigation_handle(), app_info,
-      base::Bind(&ArcNavigationThrottle::OnDisambigDialogClosed,
+      base::Bind(&ArcNavigationThrottle::OnIntentPickerClosed,
                  weak_ptr_factory_.GetWeakPtr(), base::Passed(&handlers)));
 }
 
-void ArcNavigationThrottle::OnDisambigDialogClosed(
+void ArcNavigationThrottle::OnIntentPickerClosed(
     mojo::Array<mojom::UrlHandlerInfoPtr> handlers,
     size_t selected_app_index,
     CloseReason close_reason) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   const GURL& url = navigation_handle()->GetURL();
   content::NavigationHandle* handle = navigation_handle();
 
+  previous_user_action_ = close_reason;
+
   mojom::IntentHelperInstance* bridge = GetIntentHelper();
   if (!bridge || selected_app_index >= handlers.size()) {
     close_reason = CloseReason::ERROR;
   }
 
   switch (close_reason) {
     case CloseReason::ERROR:
     case CloseReason::DIALOG_DEACTIVATED: {
       // If the user fails to select an option from the list, or the UI returned
       // an error or if |selected_app_index| is not a valid index, then resume
@@ skipping 12 matching lines @@
               handlers[selected_app_index]->package_name)) {
         handle->Resume();
       } else {
         bridge->HandleUrl(url.spec(),
                           handlers[selected_app_index]->package_name);
         handle->CancelDeferredNavigation(
             content::NavigationThrottle::CANCEL_AND_IGNORE);
       }
       break;
     }
-    case CloseReason::SIZE: {
+    case CloseReason::INVALID: {
       NOTREACHED();
       return;
     }
   }
 
   UMA_HISTOGRAM_ENUMERATION("Arc.IntentHandlerAction",
                             static_cast<int>(close_reason),
                             static_cast<int>(CloseReason::SIZE));
 }
 
 bool ArcNavigationThrottle::ShouldOverrideUrlLoading(
     content::NavigationHandle* navigation_handle) {
   GURL previous_url = navigation_handle->GetReferrer().url;
   GURL current_url = navigation_handle->GetURL();
   return !net::registry_controlled_domains::SameDomainOrHost(
       current_url, previous_url,
       net::registry_controlled_domains::INCLUDE_PRIVATE_REGISTRIES);
 }
 
 }  // namespace arc