blob: 1b88b6c1a49edae262d5305bc394ac605159b478 [file] [log] [blame]
// Copyright (c) 2012 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 "chrome/browser/ui/webui/constrained_web_dialog_ui.h"
#include <string>
#include <vector>
#include "base/bind.h"
#include "base/bind_helpers.h"
#include "base/lazy_instance.h"
#include "base/macros.h"
#include "base/values.h"
#include "content/public/browser/notification_service.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
#include "ui/web_dialogs/web_dialog_delegate.h"
#include "ui/web_dialogs/web_dialog_ui.h"
#if defined(ENABLE_EXTENSIONS)
#include "chrome/browser/extensions/tab_helper.h"
#endif
using content::RenderViewHost;
using content::WebContents;
using content::WebUIMessageHandler;
namespace {
const char kConstrainedWebDialogDelegateUserDataKey[] =
"ConstrainedWebDialogDelegateUserData";
class ConstrainedWebDialogDelegateUserData
: public base::SupportsUserData::Data {
public:
explicit ConstrainedWebDialogDelegateUserData(
ConstrainedWebDialogDelegate* delegate) : delegate_(delegate) {}
~ConstrainedWebDialogDelegateUserData() override {}
ConstrainedWebDialogDelegate* delegate() { return delegate_; }
private:
ConstrainedWebDialogDelegate* delegate_; // unowned
DISALLOW_COPY_AND_ASSIGN(ConstrainedWebDialogDelegateUserData);
};
} // namespace
ConstrainedWebDialogUI::ConstrainedWebDialogUI(content::WebUI* web_ui)
: WebUIController(web_ui) {
#if defined(ENABLE_EXTENSIONS)
extensions::TabHelper::CreateForWebContents(web_ui->GetWebContents());
#endif
}
ConstrainedWebDialogUI::~ConstrainedWebDialogUI() {
}
void ConstrainedWebDialogUI::RenderViewCreated(
RenderViewHost* render_view_host) {
// Add a "dialogClose" callback which matches WebDialogUI behavior.
web_ui()->RegisterMessageCallback("dialogClose",
base::Bind(&ConstrainedWebDialogUI::OnDialogCloseMessage,
base::Unretained(this)));
ConstrainedWebDialogDelegate* delegate = GetConstrainedDelegate();
if (!delegate)
return;
ui::WebDialogDelegate* dialog_delegate = delegate->GetWebDialogDelegate();
std::vector<WebUIMessageHandler*> handlers;
dialog_delegate->GetWebUIMessageHandlers(&handlers);
render_view_host->SetWebUIProperty("dialogArguments",
dialog_delegate->GetDialogArgs());
for (std::vector<WebUIMessageHandler*>::iterator it = handlers.begin();
it != handlers.end(); ++it) {
web_ui()->AddMessageHandler(*it);
}
dialog_delegate->OnDialogShown(web_ui(), render_view_host);
}
void ConstrainedWebDialogUI::OnDialogCloseMessage(const base::ListValue* args) {
ConstrainedWebDialogDelegate* delegate = GetConstrainedDelegate();
if (!delegate)
return;
std::string json_retval;
if (!args->empty() && !args->GetString(0, &json_retval))
NOTREACHED() << "Could not read JSON argument";
delegate->GetWebDialogDelegate()->OnDialogClosed(json_retval);
delegate->OnDialogCloseFromWebUI();
}
// static
void ConstrainedWebDialogUI::SetConstrainedDelegate(
content::WebContents* web_contents,
ConstrainedWebDialogDelegate* delegate) {
web_contents->SetUserData(&kConstrainedWebDialogDelegateUserDataKey,
new ConstrainedWebDialogDelegateUserData(delegate));
}
ConstrainedWebDialogDelegate* ConstrainedWebDialogUI::GetConstrainedDelegate() {
ConstrainedWebDialogDelegateUserData* user_data =
static_cast<ConstrainedWebDialogDelegateUserData*>(
web_ui()->GetWebContents()->
GetUserData(&kConstrainedWebDialogDelegateUserDataKey));
return user_data ? user_data->delegate() : NULL;
}