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
|
// 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 "third_party/blink/renderer/modules/ai/ai_interface_proxy.h"
namespace blink {
const char AIInterfaceProxy::kSupplementName[] = "AIInterfaceProxy";
// TODO(crbug.com/406770758): Consider refactoring to have this class own the
// execution context as a member.
AIInterfaceProxy::AIInterfaceProxy(ExecutionContext* execution_context)
: Supplement<ExecutionContext>(*execution_context),
task_runner_(
execution_context->GetTaskRunner(TaskType::kInternalDefault)),
language_detection_model_(
MakeGarbageCollected<LanguageDetectionModel>()) {}
AIInterfaceProxy::~AIInterfaceProxy() = default;
void AIInterfaceProxy::Trace(Visitor* visitor) const {
Supplement<ExecutionContext>::Trace(visitor);
visitor->Trace(translation_manager_remote_);
visitor->Trace(language_detection_driver_);
visitor->Trace(language_detection_model_);
visitor->Trace(ai_manager_remote_);
}
// static
scoped_refptr<base::SequencedTaskRunner> AIInterfaceProxy::GetTaskRunner(
ExecutionContext* execution_context) {
return AIInterfaceProxy::From(execution_context)->task_runner_;
}
// static
HeapMojoRemote<mojom::blink::TranslationManager>&
AIInterfaceProxy::GetTranslationManagerRemote(
ExecutionContext* execution_context) {
return From(execution_context)
->GetTranslationManagerRemoteImpl(execution_context);
}
// static
void AIInterfaceProxy::GetLanguageDetectionModelStatus(
ExecutionContext* execution_context,
GetLanguageDetectionModelStatusCallback callback) {
From(execution_context)
->GetLanguageDetectionDriverRemote(execution_context)
->GetLanguageDetectionModelStatus(std::move(callback));
}
// static
void AIInterfaceProxy::GetLanguageDetectionModel(
ExecutionContext* execution_context,
GetLanguageDetectionModelCallback callback) {
From(execution_context)
->GetLanguageDetectionModelImpl(execution_context, std::move(callback));
}
// static
HeapMojoRemote<mojom::blink::AIManager>& AIInterfaceProxy::GetAIManagerRemote(
ExecutionContext* execution_context) {
return From(execution_context)->GetAIManagerRemoteImpl(execution_context);
}
// static
AIInterfaceProxy* AIInterfaceProxy::From(ExecutionContext* execution_context) {
AIInterfaceProxy* translation_manager_proxy =
Supplement<ExecutionContext>::From<AIInterfaceProxy>(*execution_context);
if (!translation_manager_proxy) {
translation_manager_proxy =
MakeGarbageCollected<AIInterfaceProxy>(execution_context);
ProvideTo(*execution_context, translation_manager_proxy);
}
return translation_manager_proxy;
}
HeapMojoRemote<mojom::blink::TranslationManager>&
AIInterfaceProxy::GetTranslationManagerRemoteImpl(
ExecutionContext* execution_context) {
if (!translation_manager_remote_.is_bound()) {
execution_context->GetBrowserInterfaceBroker().GetInterface(
translation_manager_remote_.BindNewPipeAndPassReceiver(task_runner_));
}
return translation_manager_remote_;
}
HeapMojoRemote<
language_detection::mojom::blink::ContentLanguageDetectionDriver>&
AIInterfaceProxy::GetLanguageDetectionDriverRemote(
ExecutionContext* execution_context) {
if (!language_detection_driver_.is_bound()) {
execution_context->GetBrowserInterfaceBroker().GetInterface(
language_detection_driver_.BindNewPipeAndPassReceiver(task_runner_));
}
return language_detection_driver_;
}
void AIInterfaceProxy::GetLanguageDetectionModelImpl(
ExecutionContext* execution_context,
GetLanguageDetectionModelCallback callback) {
GetLanguageDetectionDriverRemote(execution_context)
->GetLanguageDetectionModel(WTF::BindOnce(
[](LanguageDetectionModel* language_detection_model,
GetLanguageDetectionModelCallback callback, base::File model) {
language_detection_model->LoadModelFile(std::move(model),
std::move(callback));
},
WrapPersistent(language_detection_model_.Get()),
std::move(callback)));
}
HeapMojoRemote<mojom::blink::AIManager>&
AIInterfaceProxy::GetAIManagerRemoteImpl(ExecutionContext* execution_context) {
if (!ai_manager_remote_.is_bound()) {
execution_context->GetBrowserInterfaceBroker().GetInterface(
ai_manager_remote_.BindNewPipeAndPassReceiver(task_runner_));
}
return ai_manager_remote_;
}
} // namespace blink
|
