// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/web_applications/webui_web_app_navigation_throttle.h"

#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/browser_window/public/global_browser_collection.h"
#include "chrome/browser/ui/web_applications/app_browser_controller.h"
#include "chrome/browser/web_applications/web_app_constants.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/web_contents.h"
#include "content/public/common/url_utils.h"
#include "url/gurl.h"

namespace web_app {

WebUIWebAppNavigationThrottle::WebUIWebAppNavigationThrottle(
    content::NavigationThrottleRegistry& registry)
    : content::NavigationThrottle(registry) {}

WebUIWebAppNavigationThrottle::~WebUIWebAppNavigationThrottle() = default;

const char* WebUIWebAppNavigationThrottle::GetNameForLogging() {
  return "WebUIWebAppNavigationThrottle";
}

// static
void WebUIWebAppNavigationThrottle::MaybeCreateAndAdd(
    content::NavigationThrottleRegistry& registry) {
  content::NavigationHandle& handle = registry.GetNavigationHandle();
  if (!handle.IsInPrimaryMainFrame()) {
    return;
  }

  // Reloading the page should not cause the tab to change.
  if (handle.GetReloadType() != content::ReloadType::NONE) {
    return;
  }

  content::WebContents* web_contents = handle.GetWebContents();

  BrowserWindowInterface* browser =
      GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(web_contents);
  auto* app_controller =
      browser ? web_app::AppBrowserController::From(browser) : nullptr;
  if (!browser || !app_controller) {
    return;
  }

#if BUILDFLAG(IS_CHROMEOS)
  // Exclude system web apps.
  if (app_controller->system_app()) {
    return;
  }
#endif

  // Proceed only if the app is coming from Chrome WebUI.
  GURL start_url = app_controller->GetAppStartUrl();
  if (!content::HasWebUIScheme(start_url)) {
    return;
  }

  registry.AddThrottle(
      std::make_unique<WebUIWebAppNavigationThrottle>(registry));
}

content::NavigationThrottle::ThrottleCheckResult
WebUIWebAppNavigationThrottle::WillStartRequest() {
  GURL navigation_url = navigation_handle()->GetURL();

  content::WebContents* web_contents = navigation_handle()->GetWebContents();
  BrowserWindowInterface* browser =
      GlobalBrowserCollection::GetInstance()->FindBrowserWithTab(web_contents);
  DCHECK(browser);
  web_app::AppBrowserController* app_controller =
      web_app::AppBrowserController::From(browser);
  DCHECK(app_controller);
  GURL start_url = app_controller->GetAppStartUrl();

  if (content::HasWebUIScheme(navigation_url) &&
      !url::IsSameOriginWith(navigation_url, start_url)) {
    content::OpenURLParams params =
        content::OpenURLParams::FromNavigationHandle(navigation_handle());
    params.disposition = WindowOpenDisposition::NEW_FOREGROUND_TAB;
    // Clear the FrameTreeNode id, as the new navigation will be in a new tab
    // rather than the frame of the original navigation.
    params.frame_tree_node_id = content::FrameTreeNodeId();
    navigation_handle()->GetWebContents()->OpenURL(
        std::move(params), /*navigation_handle_callback=*/{});
    // Deactivate app window to foreground the browser with new tab.
    browser->GetWindow()->Deactivate();
    return content::NavigationThrottle::CANCEL_AND_IGNORE;
  } else {
    return content::NavigationThrottle::PROCEED;
  }
}

content::NavigationThrottle::ThrottleCheckResult
WebUIWebAppNavigationThrottle::WillRedirectRequest() {
  return WillStartRequest();
}

}  // namespace web_app