blob: f50b4e98560c45d32ffdba959c4b46a4b7f2621a [file] [log] [blame]
// 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/javascript_dialog_extensions_client.h"
#include "components/app_modal/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 javascript_dialog_extensions_client {
namespace {
using extensions::Extension;
// Returns the ProcessManager for the browser context from |web_contents|.
extensions::ProcessManager* GetProcessManager(
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 = GetProcessManager(web_contents);
return pm->GetExtensionForRenderViewHost(web_contents->GetRenderViewHost());
}
class JavaScriptDialogExtensionsClientImpl
: public app_modal::JavaScriptDialogExtensionsClient {
public:
JavaScriptDialogExtensionsClientImpl() {}
~JavaScriptDialogExtensionsClientImpl() override {}
// JavaScriptDialogExtensionsClient:
void OnDialogOpened(content::WebContents* web_contents) override {
const Extension* extension = GetExtensionForWebContents(web_contents);
if (extension == nullptr)
return;
DCHECK(web_contents);
extensions::ProcessManager* pm = GetProcessManager(web_contents);
if (pm)
pm->IncrementLazyKeepaliveCount(extension);
}
void OnDialogClosed(content::WebContents* web_contents) override {
const Extension* extension = GetExtensionForWebContents(web_contents);
if (extension == nullptr)
return;
DCHECK(web_contents);
extensions::ProcessManager* pm = GetProcessManager(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 InstallClient() {
app_modal::JavaScriptDialogManager::GetInstance()->
SetExtensionsClient(
make_scoped_ptr(new JavaScriptDialogExtensionsClientImpl));
}
} // namespace javascript_dialog_extensions_client