// 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/extensions/chrome_extension_frame_host.h"

#include "chrome/browser/extensions/activity_log/activity_log.h"
#include "chrome/browser/extensions/error_console/error_console.h"
#include "chrome/browser/extensions/extension_action_runner.h"
#include "chrome/browser/extensions/tab_helper.h"
#include "chrome/common/extensions/extension_constants.h"
#include "components/crx_file/id_util.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/web_contents.h"
#include "extensions/browser/bad_message.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/extension_util.h"
#include "extensions/buildflags/buildflags.h"
#include "extensions/common/extension_id.h"
#include "extensions/common/extension_set.h"
#include "extensions/common/extension_urls.h"
#include "third_party/blink/public/common/logging/logging_utils.h"
#include "url/gurl.h"

static_assert(BUILDFLAG(ENABLE_EXTENSIONS_CORE));

namespace extensions {

ChromeExtensionFrameHost::ChromeExtensionFrameHost(
    content::WebContents* web_contents)
    : ExtensionFrameHost(web_contents) {}

ChromeExtensionFrameHost::~ChromeExtensionFrameHost() = default;

void ChromeExtensionFrameHost::RequestScriptInjectionPermission(
    const ExtensionId& extension_id,
    mojom::InjectionType script_type,
    mojom::RunLocation run_location,
    RequestScriptInjectionPermissionCallback callback) {
  if (!crx_file::id_util::IdIsValid(extension_id)) {
    content::RenderProcessHost* render_process =
        receivers_.GetCurrentTargetFrame()->GetProcess();
    if (render_process) {
      bad_message::ReceivedBadMessage(
          render_process,
          bad_message::CEFH_INVALID_EXTENSION_ID_FOR_SCRIPT_INJECT_REQUEST);
    }
    return;
  }

  ExtensionActionRunner* runner =
      ExtensionActionRunner::GetForWebContents(web_contents_);
  if (!runner) {
    std::move(callback).Run(false);
    return;
  }
  runner->OnRequestScriptInjectionPermission(extension_id, script_type,
                                             run_location, std::move(callback));
}

void ChromeExtensionFrameHost::GetAppInstallState(
    const GURL& requestor_url,
    GetAppInstallStateCallback callback) {
  ExtensionRegistry* registry =
      ExtensionRegistry::Get(web_contents_->GetBrowserContext());
  const ExtensionSet& extensions = registry->enabled_extensions();
  const ExtensionSet& disabled_extensions = registry->disabled_extensions();

  std::string state;
  if (extensions.GetHostedAppByURL(requestor_url))
    state = extension_misc::kAppStateInstalled;
  else if (disabled_extensions.GetHostedAppByURL(requestor_url))
    state = extension_misc::kAppStateDisabled;
  else
    state = extension_misc::kAppStateNotInstalled;

  std::move(callback).Run(state);
}

void ChromeExtensionFrameHost::WatchedPageChange(
    const std::vector<std::string>& css_selectors) {
  TabHelper* tab_helper = TabHelper::FromWebContents(web_contents_);
  if (!tab_helper)
    return;
  tab_helper->OnWatchedPageChanged(css_selectors);
}

void ChromeExtensionFrameHost::DetailedConsoleMessageAdded(
    const std::u16string& message,
    const std::u16string& source,
    const StackTrace& stack_trace,
    blink::mojom::ConsoleMessageLevel level) {
  if (!IsSourceFromAnExtension(source))
    return;

  content::RenderFrameHost* render_frame_host =
      receivers_.GetCurrentTargetFrame();
  ExtensionId extension_id = util::GetExtensionIdFromFrame(render_frame_host);
  if (extension_id.empty())
    extension_id = GURL(source).GetHost();

  content::BrowserContext* browser_context = web_contents_->GetBrowserContext();
  ErrorConsole::Get(browser_context)
      ->ReportError(std::unique_ptr<ExtensionError>(new RuntimeError(
          extension_id, browser_context->IsOffTheRecord(), source, message,
          stack_trace, web_contents_->GetLastCommittedURL(),
          blink::ConsoleMessageLevelToLogSeverity(level),
          render_frame_host->GetRoutingID(),
          render_frame_host->GetProcess()->GetDeprecatedID())));
}

void ChromeExtensionFrameHost::ContentScriptsExecuting(
    const base::flat_map<ExtensionId, std::vector<std::string>>&
        extension_id_to_scripts,
    const GURL& frame_url) {
  ActivityLog::GetInstance(web_contents_->GetBrowserContext())
      ->OnScriptsExecuted(web_contents_, extension_id_to_scripts, frame_url);
}

}  // namespace extensions