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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// 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.
#include "chrome/renderer/actor/tool_executor.h"
#include <cstdint>
#include <memory>
#include "base/functional/callback.h"
#include "base/memory/ptr_util.h"
#include "base/notreached.h"
#include "base/task/sequenced_task_runner.h"
#include "chrome/common/actor.mojom.h"
#include "chrome/common/actor/action_result.h"
#include "chrome/renderer/actor/click_tool.h"
#include "chrome/renderer/actor/drag_and_release_tool.h"
#include "chrome/renderer/actor/journal.h"
#include "chrome/renderer/actor/mouse_move_tool.h"
#include "chrome/renderer/actor/scroll_tool.h"
#include "chrome/renderer/actor/select_tool.h"
#include "chrome/renderer/actor/tool_utils.h"
#include "chrome/renderer/actor/type_tool.h"
#include "content/public/renderer/render_frame.h"
#include "third_party/blink/public/web/web_local_frame.h"
#include "third_party/blink/public/web/web_node.h"
using blink::WebLocalFrame;
using content::RenderFrame;
namespace actor {
ToolExecutor::ToolExecutor(RenderFrame* frame, Journal& journal)
: frame_(*frame), journal_(journal) {}
ToolExecutor::~ToolExecutor() = default;
void ToolExecutor::InvokeTool(mojom::ToolInvocationPtr request,
ToolExecutorCallback callback) {
CHECK(!completion_callback_);
completion_callback_ = std::move(callback);
journal_entry_ =
journal_->CreatePendingAsyncEntry(request->task_id, "InvokeTool", "");
WebLocalFrame* web_frame = frame_->GetWebFrame();
// Tool calls should only be routed to local root frames.
CHECK(!web_frame || web_frame->LocalRoot() == web_frame);
// Check LocalRoot in case the frame is a subframe.
if (!web_frame || !web_frame->FrameWidget()) {
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE,
base::BindOnce(&ToolExecutor::ToolFinished,
weak_ptr_factory_.GetWeakPtr(),
MakeResult(mojom::ActionResultCode::kFrameWentAway)));
return;
}
std::unique_ptr<ToolBase> tool;
switch (request->action->which()) {
case actor::mojom::ToolAction::Tag::kClick: {
// Check the mojom we received is in good shape.
CHECK(request->action->get_click());
tool = std::make_unique<ClickTool>(
frame_.get(), request->task_id, journal_.get(),
std::move(request->action->get_click()));
break;
}
case actor::mojom::ToolAction::Tag::kMouseMove: {
CHECK(request->action->get_mouse_move());
tool = std::make_unique<MouseMoveTool>(
frame_.get(), request->task_id, journal_.get(),
std::move(request->action->get_mouse_move()));
break;
}
case actor::mojom::ToolAction::Tag::kType: {
CHECK(request->action->get_type());
tool = std::make_unique<TypeTool>(frame_.get(), request->task_id,
journal_.get(),
std::move(request->action->get_type()));
break;
}
case actor::mojom::ToolAction::Tag::kScroll: {
CHECK(request->action->get_scroll());
tool = std::make_unique<ScrollTool>(
frame_.get(), request->task_id, journal_.get(),
std::move(request->action->get_scroll()));
break;
}
case actor::mojom::ToolAction::Tag::kSelect: {
CHECK(request->action->get_select());
tool = std::make_unique<SelectTool>(
frame_.get(), request->task_id, journal_.get(),
std::move(request->action->get_select()));
break;
}
case actor::mojom::ToolAction::Tag::kDragAndRelease: {
CHECK(request->action->get_drag_and_release());
tool = std::make_unique<DragAndReleaseTool>(
frame_.get(), request->task_id, journal_.get(),
std::move(request->action->get_drag_and_release()));
break;
}
default:
NOTREACHED();
}
page_stability_monitor_ = std::make_unique<PageStabilityMonitor>(*frame_);
auto execute_journal = journal_->CreatePendingAsyncEntry(
request->task_id, "ExecuteTool", tool->DebugString());
mojom::ActionResultPtr result = tool->Execute();
execute_journal.reset();
page_stability_monitor_->WaitForStable(
*tool, request->task_id, *journal_,
base::BindOnce(&ToolExecutor::ToolFinished,
weak_ptr_factory_.GetWeakPtr(), std::move(result)));
}
void ToolExecutor::ToolFinished(mojom::ActionResultPtr result) {
CHECK(completion_callback_);
page_stability_monitor_.reset();
std::move(completion_callback_).Run(std::move(result));
journal_entry_.reset();
}
} // namespace actor
|
