Index: extensions/components/javascript_dialog_extensions_client/javascript_dialog_extension_client_impl.cc |
diff --git a/extensions/components/javascript_dialog_extensions_client/javascript_dialog_extension_client_impl.cc b/extensions/components/javascript_dialog_extensions_client/javascript_dialog_extension_client_impl.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..ec0f1f2c79bb96b0aa1e4541e1424a1b485861c5 |
--- /dev/null |
+++ b/extensions/components/javascript_dialog_extensions_client/javascript_dialog_extension_client_impl.cc |
@@ -0,0 +1,82 @@ |
+// Copyright 2014 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. |
+ |
+#include "extensions/components/javascript_dialog_extensions_client/javascript_dialog_extension_client_impl.h" |
+ |
+#include "components/app_modal_dialogs/javascript_dialog_extensions_client.h" |
+#include "components/app_modal_dialogs/javascript_dialog_manager.h" |
+#include "content/public/browser/web_contents.h" |
+#include "extensions/browser/process_manager.h" |
+#include "extensions/common/extension.h" |
+#include "ui/gfx/native_widget_types.h" |
+ |
+namespace { |
+ |
+using extensions::Extension; |
+ |
+// Returns the ProcessManager for the browser context from |web_contents|. |
+extensions::ProcessManager* GetExtensionsProcessManager( |
Yoyo Zhou
2014/11/06 01:05:03
nit: I'd drop the Extensions from this function na
|
+ content::WebContents* web_contents) { |
+ return extensions::ProcessManager::Get(web_contents->GetBrowserContext()); |
+} |
+ |
+// Returns the extension associated with |web_contents| or NULL if there is no |
+// associated extension (or extensions are not supported). |
+const Extension* GetExtensionForWebContents( |
+ content::WebContents* web_contents) { |
+ extensions::ProcessManager* pm = GetExtensionsProcessManager(web_contents); |
+ return pm->GetExtensionForRenderViewHost(web_contents->GetRenderViewHost()); |
+} |
+ |
+class JavaScriptDialogExtensionsClientImpl |
+ : public JavaScriptDialogExtensionsClient { |
+ public: |
+ JavaScriptDialogExtensionsClientImpl() {} |
+ ~JavaScriptDialogExtensionsClientImpl() override {} |
+ |
+ // JavaScriptDialogExtensionsClient: |
+ void IncrementLazyKeepaliveCount( |
+ content::WebContents* web_contents) override { |
+ const Extension* extension = GetExtensionForWebContents(web_contents); |
+ if (extension == nullptr) |
+ return; |
+ |
+ DCHECK(web_contents); |
+ extensions::ProcessManager* pm = GetExtensionsProcessManager(web_contents); |
+ if (pm) |
+ pm->IncrementLazyKeepaliveCount(extension); |
+ } |
+ void DecrementLazyKeepaliveCount( |
+ content::WebContents* web_contents) override { |
+ const Extension* extension = GetExtensionForWebContents(web_contents); |
+ if (extension == nullptr) |
+ return; |
+ |
+ DCHECK(web_contents); |
+ extensions::ProcessManager* pm = GetExtensionsProcessManager(web_contents); |
+ if (pm) |
+ pm->DecrementLazyKeepaliveCount(extension); |
+ } |
+ bool GetExtensionName(content::WebContents* web_contents, |
+ const GURL& origin_url, |
+ std::string* name_out) override { |
+ const Extension* extension = GetExtensionForWebContents(web_contents); |
+ if (extension && |
+ web_contents->GetLastCommittedURL().GetOrigin() == origin_url) { |
+ *name_out = extension->name(); |
+ return true; |
+ } |
+ return false; |
+ } |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(JavaScriptDialogExtensionsClientImpl); |
+}; |
+ |
+} // namespace |
+ |
+void InstallJavaScriptDialogExtensionsClient() { |
+ SetJavaScriptDialogExtensionsClient( |
+ make_scoped_ptr(new JavaScriptDialogExtensionsClientImpl)); |
+} |