1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_GLIC_HOST_GLIC_ACTOR_CONTROLLER_H_
#define CHROME_BROWSER_GLIC_HOST_GLIC_ACTOR_CONTROLLER_H_
#include "base/functional/callback_forward.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/actor/task_id.h"
#include "chrome/browser/glic/host/glic.mojom.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/common/actor.mojom-forward.h"
#include "components/optimization_guide/proto/features/actions_data.pb.h"
#include "components/optimization_guide/proto/features/model_prototyping.pb.h"
namespace optimization_guide::proto {
class BrowserStartTaskResult;
}
namespace actor {
class ExecutionEngine;
class ActorTask;
} // namespace actor
namespace content {
class WebContents;
} // namespace content
namespace tabs {
class TabInterface;
} // namespace tabs
namespace glic {
// Controls the interaction with the actor to complete an action.
class GlicActorController {
public:
explicit GlicActorController(Profile* profile);
GlicActorController(const GlicActorController&) = delete;
GlicActorController& operator=(const GlicActorController&) = delete;
~GlicActorController();
// ActorKeyedService, the underlying framework, supports multi-tab actuation.
// But this class does not because it does not expose the concept of
// start/stop task. Instead it keeps track of any ongoing task, and implicitly
// creates one for Act() if one does not already exist.
// Invokes the actor to complete an action.
void Act(const optimization_guide::proto::BrowserAction& action,
const mojom::GetTabContextOptions& options,
mojom::WebClientHandler::ActInFocusedTabCallback callback);
void StopTask(actor::TaskId task_id);
void PauseTask(actor::TaskId task_id);
void ResumeTask(
actor::TaskId task_id,
const mojom::GetTabContextOptions& context_options,
glic::mojom::WebClientHandler::ResumeActorTaskCallback callback);
// These may not be necessarily generate actor tasks, but they are
// useful for recording in the ActorJournal.
void OnUserInputSubmitted();
void OnRequestStarted();
void OnResponseStarted();
void OnResponseStopped();
bool IsExecutionEngineActingOnTab(const content::WebContents* tab) const;
actor::ExecutionEngine& GetExecutionEngineForTesting(tabs::TabInterface* tab);
private:
void OnTaskStartedForAct(
const optimization_guide::proto::BrowserAction& action,
const mojom::GetTabContextOptions& options,
mojom::WebClientHandler::ActInFocusedTabCallback callback,
optimization_guide::proto::BrowserStartTaskResult result);
// Core logic to execute an action.
void ActImpl(const optimization_guide::proto::BrowserAction& action,
const mojom::GetTabContextOptions& options,
mojom::WebClientHandler::ActInFocusedTabCallback callback) const;
// Handles the result of the action, returning new page context if necessary.
void OnActionFinished(
actor::TaskId task_id,
const mojom::GetTabContextOptions& options,
mojom::WebClientHandler::ActInFocusedTabCallback callback,
actor::mojom::ActionResultPtr result) const;
actor::ExecutionEngine* GetExecutionEngine() const;
base::WeakPtr<const GlicActorController> GetWeakPtr() const;
base::WeakPtr<GlicActorController> GetWeakPtr();
class OngoingRequest;
raw_ptr<Profile> profile_;
// The most recently created task, or nullptr if no task has ever been
// created.
raw_ptr<actor::ActorTask> actor_task_ = nullptr;
// True if and only if a task is in the process of being started.
bool starting_task_ = false;
std::unique_ptr<OngoingRequest> current_request_;
base::WeakPtrFactory<GlicActorController> weak_ptr_factory_{this};
};
} // namespace glic
#endif // CHROME_BROWSER_GLIC_HOST_GLIC_ACTOR_CONTROLLER_H_
|