// Copyright 2021 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/ssl/https_only_mode_tab_helper.h"

#include "base/feature_list.h"
#include "chrome/browser/ssl/ask_before_http_dialog_controller.h"
#include "components/security_interstitials/core/features.h"
#include "components/tabs/public/tab_interface.h"
#include "content/public/browser/navigation_entry.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/reload_type.h"
#include "content/public/browser/web_contents.h"
#include "url/gurl.h"

HttpsOnlyModeTabHelper::~HttpsOnlyModeTabHelper() = default;

void HttpsOnlyModeTabHelper::DidStartNavigation(
    content::NavigationHandle* navigation_handle) {
  if (base::FeatureList::IsEnabled(
          security_interstitials::features::kHttpsFirstDialogUi)) {
    // Close the Ask-before-HTTP dialog if a new navigation begins in the
    // primary main frame. TabDialogManager has a parameter to close tab-modal
    // dialogs if a cross-site navigation occurs, but we need to be more strict
    // and close on *any* new navigation. (For example, without this when a user
    // clicks from badssl.com to http.badssl.com, sees the warning prompt, and
    // then clicks the back button, the warning prompt would still be showing.)
    if (navigation_handle->IsInPrimaryMainFrame() &&
        !navigation_handle->IsSameDocument()) {
      if (auto* tab = tabs::TabInterface::MaybeGetFromContents(
              navigation_handle->GetWebContents())) {
        auto* const dialog_controller =
            AskBeforeHttpDialogController::From(tab);
        if (dialog_controller && dialog_controller->HasOpenDialog()) {
          dialog_controller->CloseDialog();
        }
      }
    }
  }

  // If the user was on an exempt net error and the tab was reloaded, only
  // reset the exempt error state, but keep the upgrade state so the reload
  // will result in continuing to attempt the upgraded navigation (and if it
  // later fails, the fallback will be to the original fallback URL).
  bool should_maintain_upgrade_state =
      is_exempt_error() &&
      navigation_handle->GetReloadType() != content::ReloadType::NONE;
  set_is_exempt_error(false);
  if (!should_maintain_upgrade_state) {
    set_fallback_url(GURL());
    set_is_navigation_fallback(false);
    set_is_navigation_upgraded(false);
    set_is_typed_schemeless_upgrade(false);
    set_fallback_reason(
        security_interstitials::https_only_mode::FallbackReason::kNone);
  }
}

void HttpsOnlyModeTabHelper::DidFinishNavigation(
    content::NavigationHandle* navigation_handle) {
  if (!navigation_handle->IsInPrimaryMainFrame()) {
    return;
  }

  // TODO(crbug.com/351990829): Consider if this check could be made more
  // precise. One option mght be to communicate a "interstitial enabled" flag on
  // the helper which is set by the throttle.
  // If dialog UI is enabled and this was a navigation cancelled by the
  // throttle when we would interstitial the navigation, then trigger the
  // dialog UI.
  if (base::FeatureList::IsEnabled(
          security_interstitials::features::kHttpsFirstDialogUi) &&
      is_navigation_fallback_ &&
      !navigation_handle->GetURL().SchemeIsCryptographic()) {
    auto* tab = tabs::TabInterface::MaybeGetFromContents(
        navigation_handle->GetWebContents());
    if (!tab) {
      return;
    }
    auto* const dialog_controller = AskBeforeHttpDialogController::From(tab);
    if (!dialog_controller) {
      // If there is no dialog controller, then the tab is being destroyed
      // and there is nothing to show a modal on. Just return.
      return;
    }
    ukm::SourceId ukm_source_id = ukm::ConvertToSourceId(
        navigation_handle->GetNavigationId(), ukm::SourceIdType::NAVIGATION_ID);
    dialog_controller->ShowDialog(navigation_handle->GetWebContents(),
                                  navigation_handle->GetURL(), ukm_source_id,
                                  fallback_reason());
    // Make sure that the security indicator shows the correct icon.
    // TODO(crbug.com/351990829): Add the new icon and integration.
    navigation_handle->GetWebContents()->DidChangeVisibleSecurityState();
  }
}

HttpsOnlyModeTabHelper::HttpsOnlyModeTabHelper(
    content::WebContents* web_contents)
    : WebContentsObserver(web_contents),
      content::WebContentsUserData<HttpsOnlyModeTabHelper>(*web_contents) {}

WEB_CONTENTS_USER_DATA_KEY_IMPL(HttpsOnlyModeTabHelper);