// Copyright 2018 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/connection_help_tab_helper.h"

#include "base/feature_list.h"
#include "components/security_interstitials/content/urls.h"
#include "components/tabs/public/tab_interface.h"
#include "content/public/browser/navigation_controller.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/web_contents.h"
#include "content/public/common/referrer.h"
#include "net/base/net_errors.h"
#include "url/gurl.h"

namespace {
const char kHelpCenterConnectionHelpUrl[] =
    "https://support.google.com/chrome/answer/6098869";

void RedirectToBundledHelp(content::WebContents* web_contents) {
  GURL::Replacements replacements;
  std::string error_code = web_contents->GetLastCommittedURL().GetRef();
  replacements.SetRefStr(error_code);
  web_contents->GetController().LoadURL(
      GURL(security_interstitials::kChromeUIConnectionHelpURL)
          .ReplaceComponents(replacements),
      content::Referrer(), ui::PageTransition::PAGE_TRANSITION_AUTO_TOPLEVEL,
      std::string());
}
}  // namespace

ConnectionHelpTabHelper::~ConnectionHelpTabHelper() = default;

void ConnectionHelpTabHelper::DidFinishNavigation(
    content::NavigationHandle* navigation_handle) {
  // Ignore pre-rendering navigations.
  if (navigation_handle->IsInPrimaryMainFrame() &&
      web_contents()->GetLastCommittedURL().EqualsIgnoringRef(
          GetHelpCenterURL()) &&
      navigation_handle->IsErrorPage() &&
      net::IsCertificateError(navigation_handle->GetNetErrorCode())) {
    RedirectToBundledHelp(web_contents());
  }
}

void ConnectionHelpTabHelper::SetHelpCenterUrlForTesting(const GURL& url) {
  testing_url_ = url;
}

DEFINE_USER_DATA(ConnectionHelpTabHelper);

ConnectionHelpTabHelper::ConnectionHelpTabHelper(
    tabs::TabInterface& tab,
    content::WebContents* web_contents)
    : content::WebContentsObserver(web_contents),
      scoped_unowned_user_data_(tab.GetUnownedUserDataHost(), *this) {}

// static
ConnectionHelpTabHelper* ConnectionHelpTabHelper::From(
    tabs::TabInterface* tab) {
  return Get(tab->GetUnownedUserDataHost());
}

GURL ConnectionHelpTabHelper::GetHelpCenterURL() {
  if (testing_url_.is_valid()) {
    return testing_url_;
  }
  return GURL(kHelpCenterConnectionHelpUrl);
}