// Copyright 2022 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/passwords/password_manager_navigation_throttle.h"

#include "base/task/sequenced_task_runner.h"
#include "components/password_manager/core/browser/manage_passwords_referrer.h"
#include "components/password_manager/core/browser/password_manager_constants.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/page_navigator.h"
#include "content/public/browser/storage_partition.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_contents_user_data.h"
#include "ui/base/page_transition_types.h"
#include "url/gurl.h"
#include "url/origin.h"

#if BUILDFLAG(IS_ANDROID)
#include "chrome/browser/password_manager/android/password_manager_launcher_android.h"
#else
#include "chrome/browser/password_manager/chrome_password_manager_client.h"
#endif

namespace {

using content::NavigationHandle;
using content::NavigationThrottle;
using content::NavigationThrottleRegistry;
using content::WebContents;

bool IsTriggeredOnGoogleOwnedUI(NavigationHandle* handle) {
  // Only cover cases when the user clicked on a link.
  if (!ui::PageTransitionCoreTypeIs(handle->GetPageTransition(),
                                    ui::PAGE_TRANSITION_LINK)) {
    return false;
  }

  // Referrer origin and target URL must match.
  if (handle->GetURL() != GURL(password_manager::kManageMyPasswordsURL)) {
    return false;
  }

  url::Origin origin = handle->GetInitiatorOrigin().value_or(url::Origin());
  if (origin != url::Origin::Create(GURL(password_manager::kReferrerURL)) &&
      origin !=
          url::Origin::Create(GURL(password_manager::kManageMyPasswordsURL))) {
    return false;
  }

  return true;
}

}  // namespace

PasswordManagerNavigationThrottle::PasswordManagerNavigationThrottle(
    NavigationThrottleRegistry& registry)
    : NavigationThrottle(registry) {}

PasswordManagerNavigationThrottle::~PasswordManagerNavigationThrottle() =
    default;

// static
void PasswordManagerNavigationThrottle::MaybeCreateAndAdd(
    NavigationThrottleRegistry& registry) {
  // Don't handle navigations in subframes or main frames that are in a nested
  // frame tree (e.g. fenced frames)
  auto& handle = registry.GetNavigationHandle();
  if (!handle.GetParentFrameOrOuterDocument() &&
      IsTriggeredOnGoogleOwnedUI(&handle)) {
    registry.AddThrottle(
        std::make_unique<PasswordManagerNavigationThrottle>(registry));
  }
}

NavigationThrottle::ThrottleCheckResult
PasswordManagerNavigationThrottle::WillStartRequest() {
  WebContents* web_contents = navigation_handle()->GetWebContents();
  if (!web_contents) {
    return NavigationThrottle::PROCEED;
  }

#if BUILDFLAG(IS_ANDROID)
  password_manager_launcher::ShowPasswordSettings(
      web_contents,
      password_manager::ManagePasswordsReferrer::kPasswordsGoogleWebsite,
      /*manage_passkeys=*/false);
#else
  ChromePasswordManagerClient* client =
      ChromePasswordManagerClient::FromWebContents(web_contents);
  if (!client) {
    // WebContents without a password manager client (non-tab surfaces such
    // as the presentation receiver window) have no passwords UI to redirect
    // to. Let the navigation proceed in place instead of cancelling it and
    // closing the WebContents below.
    return NavigationThrottle::PROCEED;
  }
  client->NavigateToManagePasswordsPage(
      password_manager::ManagePasswordsReferrer::kPasswordsGoogleWebsite);
#endif
  // Schedule a task to close current tab since on Android Password Manager is
  // shown in the Native UI, and on desktop NavigateToManagePasswordsPage()
  // handles creation or redirection to Passwords Manager tab.
  base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
      FROM_HERE,
      base::BindOnce(&content::WebContents::Close, web_contents->GetWeakPtr()));
  return NavigationThrottle::CANCEL_AND_IGNORE;
}

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