components/autofill_assistant/browser/headless/headless_script_controller_impl.cc

// Copyright 2022 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/autofill_assistant/browser/headless/headless_script_controller_impl.h"

#include "base/time/default_tick_clock.h"
#include "components/autofill_assistant/browser/desktop/starter_delegate_desktop.h"
#include "components/autofill_assistant/browser/public/password_change/website_login_manager.h"
#include "components/autofill_assistant/browser/starter.h"

namespace autofill_assistant {

HeadlessScriptControllerImpl::HeadlessScriptControllerImpl(
    content::WebContents* web_contents,
    ExternalActionDelegate* action_extension_delegate,
    WebsiteLoginManager* website_login_manager)
    : web_contents_(web_contents) {
  DCHECK(web_contents_);

  auto* starter = Starter::FromWebContents(web_contents_);
  if (starter) {
    client_ = std::make_unique<ClientHeadless>(
        web_contents, starter->GetCommonDependencies(),
        action_extension_delegate, website_login_manager, this);
  }
}

HeadlessScriptControllerImpl::~HeadlessScriptControllerImpl() = default;

void HeadlessScriptControllerImpl::StartScript(
    const base::flat_map<std::string, std::string>& script_parameters,
    base::OnceCallback<void(ScriptResult)> script_ended_callback) {
  // This HeadlessScriptController is currently executing a script, so we return
  // an error.
  if (script_ended_callback_) {
    std::move(script_ended_callback).Run({false});
    return;
  }
  auto* starter = Starter::FromWebContents(web_contents_);
  // The starter has not yet been initialized or was not initialized at the
  // time the constructor was called.
  if (!starter || !client_) {
    std::move(script_ended_callback).Run({false});
    return;
  }

  script_ended_callback_ = std::move(script_ended_callback);
  auto parameters = std::make_unique<ScriptParameters>(script_parameters);
  auto trigger_context = std::make_unique<TriggerContext>(
      std::move(parameters),
      /* experiment_ids = */ "",
      starter->GetPlatformDependencies()->IsCustomTab(*web_contents_),
      /*onboarding_shown = */ false,
      /*is_direct_action = */ false,
      /* initial_url = */ "",
      /* is_in_chrome_triggered = */ true,
      /* is_externally_triggered = */ true);
  starter->CanStart(
      std::move(trigger_context),
      base::BindOnce(&HeadlessScriptControllerImpl::OnReadyToStart,
                     weak_ptr_factory_.GetWeakPtr()));
}

void HeadlessScriptControllerImpl::OnReadyToStart(
    bool can_start,
    absl::optional<GURL> url,
    std::unique_ptr<TriggerContext> trigger_context) {
  if (!can_start || !url.has_value()) {
    std::move(script_ended_callback_).Run({false});
    return;
  }
  // TODO(b/201964911): At this point we should be sure no other Controller
  // exists on this tab. Add logic to the starter to check that's the case.
  client_->Start(*url, std::move(trigger_context));
}

void HeadlessScriptControllerImpl::NotifyScriptEnded(
    Metrics::DropOutReason reason) {
  std::move(script_ended_callback_)
      .Run({reason == Metrics::DropOutReason::SCRIPT_SHUTDOWN});
}

}  // namespace autofill_assistant