constrained_web_dialog_ui.cc revision 5d1f7b1de12d16ceb2c938c56701a3e8bfa558f7
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) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::RenderViewHost; 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::WebContents; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::WebUIMessageHandler; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kConstrainedWebDialogDelegateUserDataKey[] = 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "ConstrainedWebDialogDelegateUserData"; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ConstrainedWebDialogDelegateUserData 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : public base::SupportsUserData::Data { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit ConstrainedWebDialogDelegateUserData( 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate) : delegate_(delegate) {} 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~ConstrainedWebDialogDelegateUserData() {} 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate() { return delegate_; } 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate_; // unowned 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ConstrainedWebDialogDelegateUserData); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ConstrainedWebDialogUI::ConstrainedWebDialogUI(content::WebUI* web_ui) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : WebUIController(web_ui) { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ConstrainedWebDialogUI::~ConstrainedWebDialogUI() { 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ConstrainedWebDialogUI::RenderViewCreated( 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RenderViewHost* render_view_host) { 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate = GetConstrainedDelegate(); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!delegate) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::WebDialogDelegate* dialog_delegate = delegate->GetWebDialogDelegate(); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<WebUIMessageHandler*> handlers; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dialog_delegate->GetWebUIMessageHandlers(&handlers); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) render_view_host->SetWebUIProperty("dialogArguments", 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dialog_delegate->GetDialogArgs()); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (std::vector<WebUIMessageHandler*>::iterator it = handlers.begin(); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != handlers.end(); ++it) { 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->AddMessageHandler(*it); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Add a "DialogClose" callback which matches WebDialogUI behavior. 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->RegisterMessageCallback("DialogClose", 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&ConstrainedWebDialogUI::OnDialogCloseMessage, 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) dialog_delegate->OnDialogShown(web_ui(), render_view_host); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void ConstrainedWebDialogUI::OnDialogCloseMessage(const base::ListValue* args) { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate = GetConstrainedDelegate(); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!delegate) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string json_retval; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!args->empty() && !args->GetString(0, &json_retval)) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "Could not read JSON argument"; 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->GetWebDialogDelegate()->OnDialogClosed(json_retval); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->OnDialogCloseFromWebUI(); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ConstrainedWebDialogUI::SetConstrainedDelegate( 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::WebContents* web_contents, 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegate* delegate) { 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_contents->SetUserData(&kConstrainedWebDialogDelegateUserDataKey, 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new ConstrainedWebDialogDelegateUserData(delegate)); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ConstrainedWebDialogDelegate* ConstrainedWebDialogUI::GetConstrainedDelegate() { 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ConstrainedWebDialogDelegateUserData* user_data = 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static_cast<ConstrainedWebDialogDelegateUserData*>( 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->GetWebContents()-> 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetUserData(&kConstrainedWebDialogDelegateUserDataKey)); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return user_data ? user_data->delegate() : NULL; 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 106