constrained_web_dialog_ui.cc revision 1320f92c476a1ad9d19dba2a48c72b75566198e9
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/ui/webui/constrained_web_dialog_ui.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind_helpers.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/lazy_instance.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/values.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/notification_service.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/render_view_host.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/web_contents.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/web_ui.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/web_dialogs/web_dialog_delegate.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/web_dialogs/web_dialog_ui.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 211320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#if defined(ENABLE_EXTENSIONS) 221320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome/browser/extensions/tab_helper.h" 231320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#endif 241320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::RenderViewHost; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::WebContents; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::WebUIMessageHandler; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kConstrainedWebDialogDelegateUserDataKey[] = 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "ConstrainedWebDialogDelegateUserData"; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ConstrainedWebDialogDelegateUserData 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : public base::SupportsUserData::Data { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit ConstrainedWebDialogDelegateUserData( 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate) : delegate_(delegate) {} 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~ConstrainedWebDialogDelegateUserData() {} 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate() { return delegate_; } 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate_; // unowned 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ConstrainedWebDialogDelegateUserData); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ConstrainedWebDialogUI::ConstrainedWebDialogUI(content::WebUI* web_ui) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : WebUIController(web_ui) { 531320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#if defined(ENABLE_EXTENSIONS) 541320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci extensions::TabHelper::CreateForWebContents(web_ui->GetWebContents()); 551320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#endif 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ConstrainedWebDialogUI::~ConstrainedWebDialogUI() { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ConstrainedWebDialogUI::RenderViewCreated( 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RenderViewHost* render_view_host) { 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate = GetConstrainedDelegate(); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!delegate) 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::WebDialogDelegate* dialog_delegate = delegate->GetWebDialogDelegate(); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<WebUIMessageHandler*> handlers; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dialog_delegate->GetWebUIMessageHandlers(&handlers); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) render_view_host->SetWebUIProperty("dialogArguments", 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dialog_delegate->GetDialogArgs()); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (std::vector<WebUIMessageHandler*>::iterator it = handlers.begin(); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != handlers.end(); ++it) { 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->AddMessageHandler(*it); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 775c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu // Add a "dialogClose" callback which matches WebDialogUI behavior. 785c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu web_ui()->RegisterMessageCallback("dialogClose", 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&ConstrainedWebDialogUI::OnDialogCloseMessage, 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dialog_delegate->OnDialogShown(web_ui(), render_view_host); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void ConstrainedWebDialogUI::OnDialogCloseMessage(const base::ListValue* args) { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate = GetConstrainedDelegate(); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!delegate) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string json_retval; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!args->empty() && !args->GetString(0, &json_retval)) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "Could not read JSON argument"; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->GetWebDialogDelegate()->OnDialogClosed(json_retval); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->OnDialogCloseFromWebUI(); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ConstrainedWebDialogUI::SetConstrainedDelegate( 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::WebContents* web_contents, 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate) { 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_contents->SetUserData(&kConstrainedWebDialogDelegateUserDataKey, 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new ConstrainedWebDialogDelegateUserData(delegate)); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ConstrainedWebDialogDelegate* ConstrainedWebDialogUI::GetConstrainedDelegate() { 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegateUserData* user_data = 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static_cast<ConstrainedWebDialogDelegateUserData*>( 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->GetWebContents()-> 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetUserData(&kConstrainedWebDialogDelegateUserDataKey)); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return user_data ? user_data->delegate() : NULL; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 113