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
|
// Copyright 2023 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/language/ios/browser/language_detection_java_script_feature.h"
#import "base/metrics/histogram_macros.h"
#import "components/language/ios/browser/ios_language_detection_tab_helper.h"
#import "components/prefs/pref_member.h"
#import "components/translate/core/browser/translate_pref_names.h"
#import "ios/web/public/browser_state.h"
#import "ios/web/public/js_messaging/script_message.h"
#import "ios/web/public/js_messaging/web_frame.h"
#import "ios/web/public/js_messaging/web_frames_manager.h"
namespace {
const char kScriptName[] = "language_detection";
const char kLanguageDetectionTextCapturedMessageHandlerName[] =
"LanguageDetectionTextCaptured";
} // namespace
namespace language {
// static
LanguageDetectionJavaScriptFeature*
LanguageDetectionJavaScriptFeature::GetInstance() {
static base::NoDestructor<LanguageDetectionJavaScriptFeature> instance;
return instance.get();
}
LanguageDetectionJavaScriptFeature::LanguageDetectionJavaScriptFeature()
: web::JavaScriptFeature(
web::ContentWorld::kIsolatedWorld,
{FeatureScript::CreateWithFilename(
kScriptName,
FeatureScript::InjectionTime::kDocumentStart,
FeatureScript::TargetFrames::kMainFrame,
FeatureScript::ReinjectionBehavior::kInjectOncePerWindow)}) {}
LanguageDetectionJavaScriptFeature::~LanguageDetectionJavaScriptFeature() =
default;
void LanguageDetectionJavaScriptFeature::StartLanguageDetection(
web::WebFrame* frame) {
CallJavaScriptFunction(frame, "languageDetection.detectLanguage", {});
}
void LanguageDetectionJavaScriptFeature::ScriptMessageReceived(
web::WebState* web_state,
const web::ScriptMessage& script_message) {
if (!script_message.is_main_frame()) {
// Translate is only supported on main frame.
return;
}
if (!script_message.body() || !script_message.body()->is_dict()) {
return;
}
base::Value::Dict& body_dict = script_message.body()->GetDict();
std::optional<bool> has_notranslate = body_dict.FindBool("hasNoTranslate");
const std::string* html_lang = body_dict.FindString("htmlLang");
const std::string* http_content_language =
body_dict.FindString("httpContentLanguage");
const std::string* frame_id = body_dict.FindString("frameId");
if (!has_notranslate.has_value() || !html_lang || !http_content_language ||
!frame_id) {
return;
}
web::WebFrame* sender_frame =
GetWebFramesManager(web_state)->GetFrameWithId(*frame_id);
if (!sender_frame) {
return;
}
if (!script_message.request_url()) {
return;
}
GURL url = script_message.request_url().value();
language::IOSLanguageDetectionTabHelper* tab_helper =
language::IOSLanguageDetectionTabHelper::FromWebState(web_state);
if (!tab_helper) {
return;
}
CallJavaScriptFunction(
sender_frame, "languageDetection.retrieveBufferedTextContent", {},
base::BindOnce(&IOSLanguageDetectionTabHelper::OnTextRetrieved,
tab_helper->GetWeakPtr(), *has_notranslate,
*http_content_language, *html_lang, url),
base::Milliseconds(web::kJavaScriptFunctionCallDefaultTimeout));
}
std::optional<std::string>
LanguageDetectionJavaScriptFeature::GetScriptMessageHandlerName() const {
return kLanguageDetectionTextCapturedMessageHandlerName;
}
} // namespace language
|
