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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#import "components/translate/ios/browser/translate_java_script_feature.h"
#import "base/check_op.h"
#import "base/functional/callback_helpers.h"
#import "base/strings/utf_string_conversions.h"
#import "components/translate/ios/browser/translate_controller.h"
#import "ios/web/public/js_messaging/script_message.h"
namespace {
constexpr char kScriptName[] = "translate_ios";
constexpr char kScriptMessageName[] = "TranslateMessage";
} // namespace
namespace translate {
// static
TranslateJavaScriptFeature* TranslateJavaScriptFeature::GetInstance() {
static base::NoDestructor<TranslateJavaScriptFeature> instance;
return instance.get();
}
TranslateJavaScriptFeature::TranslateJavaScriptFeature()
: web::JavaScriptFeature(
web::ContentWorld::kIsolatedWorld,
{FeatureScript::CreateWithFilename(
kScriptName,
FeatureScript::InjectionTime::kDocumentStart,
FeatureScript::TargetFrames::kMainFrame,
FeatureScript::ReinjectionBehavior::kInjectOncePerWindow)}) {}
TranslateJavaScriptFeature::~TranslateJavaScriptFeature() = default;
void TranslateJavaScriptFeature::StartTranslation(
web::WebFrame* frame,
const std::string& source_language,
const std::string& target_language) {
if (!frame) {
return;
}
base::Value::List parameters;
parameters.Append(source_language);
parameters.Append(target_language);
CallJavaScriptFunction(frame, "translate.startTranslation", parameters);
}
void TranslateJavaScriptFeature::RevertTranslation(web::WebFrame* frame) {
if (!frame) {
return;
}
CallJavaScriptFunction(frame, "translate.revertTranslation", {});
}
void TranslateJavaScriptFeature::InjectTranslateScript(
web::WebFrame* frame,
const std::string script) {
if (!frame) {
return;
}
// Always revert translation since this page could have been loaded from the
// WebKit page cache.
RevertTranslation(frame);
ExecuteJavaScript(frame, base::UTF8ToUTF16(script),
base::DoNothingAs<void(const base::Value*, NSError*)>());
}
std::optional<std::string>
TranslateJavaScriptFeature::GetScriptMessageHandlerName() const {
return kScriptMessageName;
}
void TranslateJavaScriptFeature::ScriptMessageReceived(
web::WebState* web_state,
const web::ScriptMessage& message) {
if (!message.is_main_frame() || !message.body() ||
!message.body()->is_dict()) {
return;
}
TranslateController* translate_controller =
TranslateController::FromWebState(web_state);
translate_controller->OnJavascriptCommandReceived(message.body()->GetDict());
}
} // namespace translate
|
