chrome/browser/dom_ui/constrained_html_dialog.cc


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

// Copyright (c) 2010 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/dom_ui/constrained_html_dialog.h"

#include "chrome/browser/dom_ui/dom_ui_util.h"
#include "chrome/browser/dom_ui/html_dialog_ui.h"
#include "chrome/browser/renderer_host/render_view_host.h"

ConstrainedHtmlDialog::ConstrainedHtmlDialog(Profile* profile,
                                             HtmlDialogUIDelegate* delegate)
    : DOMUI(NULL),
      profile_(profile),
      render_view_host_(NULL),
      html_dialog_ui_delegate_(delegate) {
}

ConstrainedHtmlDialog::~ConstrainedHtmlDialog() {
}

void ConstrainedHtmlDialog::InitializeDOMUI(RenderViewHost* render_view_host) {
  render_view_host_ = render_view_host;

  std::vector<DOMMessageHandler*> handlers;
  html_dialog_ui_delegate_->GetDOMMessageHandlers(&handlers);
  render_view_host->SetDOMUIProperty("dialogArguments",
      html_dialog_ui_delegate_->GetDialogArgs());
  for (std::vector<DOMMessageHandler*>::iterator it = handlers.begin();
       it != handlers.end(); ++it) {
    (*it)->Attach(this);
    AddMessageHandler(*it);
  }

  // Add a "DialogClosed" callback which matches HTMLDialogUI behavior.
  RegisterMessageCallback("DialogClose",
    NewCallback(this, &ConstrainedHtmlDialog::OnDialogClosed));
}

void ConstrainedHtmlDialog::OnDialogClosed(const ListValue* args) {
  html_dialog_ui_delegate()->OnDialogClosed(
    dom_ui_util::GetJsonResponseFromFirstArgumentInList(args));
}