Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(576)

Unified Diff: ios/web_view/internal/web_view_early_page_script_provider.mm

Issue 2764773002: Add CWVUserContentController which enables injecting JavaScripts. (Closed)
Patch Set: Apply new comments. Created 3 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: ios/web_view/internal/web_view_early_page_script_provider.mm
diff --git a/ios/web_view/internal/web_view_early_page_script_provider.mm b/ios/web_view/internal/web_view_early_page_script_provider.mm
new file mode 100644
index 0000000000000000000000000000000000000000..68dcd1d8170fabaa02de2459277d7c27a7ef9a9b
--- /dev/null
+++ b/ios/web_view/internal/web_view_early_page_script_provider.mm
@@ -0,0 +1,49 @@
+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#import "ios/web_view/internal/web_view_early_page_script_provider.h"
+
+#import <Foundation/Foundation.h>
+
+#include "ios/web/public/browser_state.h"
+#include "ios/web/public/web_thread.h"
+
+#if !defined(__has_feature) || !__has_feature(objc_arc)
+#error "This file requires ARC support."
+#endif
+
+namespace ios_web_view {
+
+namespace {
+// A key used to associate a WebViewEarlyPageScriptProvider with a BrowserState.
+const char kWebViewEarlyPageScriptProviderKeyName[] =
+ "web_view_early_page_script_provider";
+
+} // namespace
+
+// static
+WebViewEarlyPageScriptProvider&
+WebViewEarlyPageScriptProvider::FromBrowserState(
+ web::BrowserState* _Nonnull browser_state) {
+ DCHECK_CURRENTLY_ON(web::WebThread::UI);
+ DCHECK(browser_state);
+ if (!browser_state->GetUserData(kWebViewEarlyPageScriptProviderKeyName)) {
+ browser_state->SetUserData(kWebViewEarlyPageScriptProviderKeyName,
+ std::unique_ptr<WebViewEarlyPageScriptProvider>(
+ new WebViewEarlyPageScriptProvider()));
+ }
+ return *(static_cast<WebViewEarlyPageScriptProvider*>(
+ browser_state->GetUserData(kWebViewEarlyPageScriptProviderKeyName)));
+}
+
+WebViewEarlyPageScriptProvider::~WebViewEarlyPageScriptProvider() = default;
+
+void WebViewEarlyPageScriptProvider::SetScript(NSString* _Nonnull script) {
+ script_.reset((NSString * _Nonnull)[script copy]);
michaeldo 2017/03/22 23:03:33 Is this cast needed? I would expect this to be OK:
Hiroshi Ichikawa 2017/03/23 02:18:40 It worked, thanks. Done.
+}
+
+WebViewEarlyPageScriptProvider::WebViewEarlyPageScriptProvider()
+ : script_([[NSString alloc] init]) {}
+
+} // namespace ios_web_view

Powered by Google App Engine
This is Rietveld 408576698