// Copyright 2012 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/webui/devtools/devtools_ui.h"

#include "base/command_line.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "chrome/browser/devtools/devtools_availability_checker.h"
#include "chrome/browser/devtools/url_constants.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/webui/devtools/devtools_ui_data_source.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/url_constants.h"
#include "components/embedder_support/user_agent_utils.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/storage_partition.h"
#include "content/public/browser/url_data_source.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
#include "content/public/common/bindings_policy.h"

// static
GURL DevToolsUI::GetProxyURL(const std::string& frontend_url) {
  GURL url(frontend_url);
  if (url.GetScheme() == content::kChromeDevToolsScheme &&
      url.GetHost() == chrome::kChromeUIDevToolsHost) {
    return GURL();
  }
  if (!url.is_valid() || url.GetHost() != kRemoteFrontendDomain) {
    return GURL();
  }
  return GURL(base::StringPrintf(
      "%s://%s/%s/%s?%s", content::kChromeDevToolsScheme,
      chrome::kChromeUIDevToolsHost, chrome::kChromeUIDevToolsRemotePath,
      url.GetPath().substr(1).c_str(), url.GetQuery().c_str()));
}

// static
GURL DevToolsUI::GetRemoteBaseURL() {
  return GURL(
      base::StringPrintf("%s%s/%s/", kRemoteFrontendBase, kRemoteFrontendPath,
                         embedder_support::GetChromiumGitRevision().c_str()));
}

// static
bool DevToolsUI::IsFrontendResourceURL(const GURL& url) {
  if (url.host() == kRemoteFrontendDomain) {
    return true;
  }

  const base::CommandLine* cmd_line = base::CommandLine::ForCurrentProcess();
  if (cmd_line->HasSwitch(switches::kCustomDevtoolsFrontend)) {
    GURL custom_frontend_url =
        GURL(cmd_line->GetSwitchValueASCII(switches::kCustomDevtoolsFrontend));
    if (custom_frontend_url.is_valid() &&
        custom_frontend_url.scheme() == url.scheme() &&
        custom_frontend_url.host() == url.host() &&
        custom_frontend_url.EffectiveIntPort() == url.EffectiveIntPort() &&
        base::StartsWith(url.path(), custom_frontend_url.path(),
                         base::CompareCase::SENSITIVE)) {
      return true;
    }
  }
  return false;
}

DevToolsUI::DevToolsUI(content::WebUI* web_ui)
#if BUILDFLAG(ARKWEB_DEVTOOLS)
    : WebUIController(web_ui) {
#else
    : WebUIController(web_ui), bindings_(web_ui->GetWebContents()) {
#endif // BUILDFLAG(ARKWEB_DEVTOOLS)
  web_ui->SetBindings(content::BindingsPolicySet());
  content::BrowserContext* browser_context =
      web_ui->GetWebContents()->GetBrowserContext();
  if (IsInspectionAllowed(Profile::FromBrowserContext(browser_context),
                          static_cast<content::WebContents*>(nullptr))) {
    auto factory = browser_context->GetDefaultStoragePartition()
                       ->GetURLLoaderFactoryForBrowserProcess();
    content::URLDataSource::Add(
        browser_context,
        std::make_unique<DevToolsDataSource>(std::move(factory)));
  }
}

DevToolsUI::~DevToolsUI() = default;