content/shell/browser/shell_javascript_dialog_win.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/shell/browser/shell_javascript_dialog.h"
   6
   7 #include "base/strings/string_util.h"
   8 #include "content/shell/app/resource.h"
   9 #include "content/shell/browser/shell.h"
  10 #include "content/shell/browser/shell_javascript_dialog_manager.h"
  11
  12 namespace content {
  13
  14 class ShellJavaScriptDialog;
  15
  16 INT_PTR CALLBACK ShellJavaScriptDialog::DialogProc(HWND dialog,
  17                                                    UINT message,
  18                                                    WPARAM wparam,
  19                                                    LPARAM lparam) {
  20   switch (message) {
  21     case WM_INITDIALOG: {
  22       SetWindowLongPtr(dialog, DWLP_USER, static_cast<LONG_PTR>(lparam));
  23       ShellJavaScriptDialog* owner =
  24           reinterpret_cast<ShellJavaScriptDialog*>(lparam);
  25       owner->dialog_win_ = dialog;
  26       SetDlgItemText(dialog, IDC_DIALOGTEXT, owner->message_text_.c_str());
  27       if (owner->message_type_ == JAVASCRIPT_MESSAGE_TYPE_PROMPT)
  28         SetDlgItemText(dialog, IDC_PROMPTEDIT,
  29                        owner->default_prompt_text_.c_str());
  30       break;
  31     }
  32     case WM_DESTROY: {
  33       ShellJavaScriptDialog* owner = reinterpret_cast<ShellJavaScriptDialog*>(
  34           GetWindowLongPtr(dialog, DWLP_USER));
  35       if (owner->dialog_win_) {
  36         owner->dialog_win_ = 0;
  37         owner->callback_.Run(false, base::string16());
  38         owner->manager_->DialogClosed(owner);
  39       }
  40       break;
  41     }
  42     case WM_COMMAND: {
  43       ShellJavaScriptDialog* owner = reinterpret_cast<ShellJavaScriptDialog*>(
  44           GetWindowLongPtr(dialog, DWLP_USER));
  45       base::string16 user_input;
  46       bool finish = false;
  47       bool result;
  48       switch (LOWORD(wparam)) {
  49         case IDOK:
  50           finish = true;
  51           result = true;
  52           if (owner->message_type_ == JAVASCRIPT_MESSAGE_TYPE_PROMPT) {
  53             int length =
  54                 GetWindowTextLength(GetDlgItem(dialog, IDC_PROMPTEDIT)) + 1;
  55             GetDlgItemText(dialog, IDC_PROMPTEDIT,
  56                            base::WriteInto(&user_input, length), length);
  57           }
  58           break;
  59         case IDCANCEL:
  60           finish = true;
  61           result = false;
  62           break;
  63       }
  64       if (finish) {
  65         owner->dialog_win_ = 0;
  66         owner->callback_.Run(result, user_input);
  67         DestroyWindow(dialog);
  68         owner->manager_->DialogClosed(owner);
  69       }
  70       break;
  71     }
  72     default:
  73       return DefWindowProc(dialog, message, wparam, lparam);
  74   }
  75   return 0;
  76 }
  77
  78 ShellJavaScriptDialog::ShellJavaScriptDialog(
  79     ShellJavaScriptDialogManager* manager,
  80     gfx::NativeWindow parent_window,
  81     JavaScriptMessageType message_type,
  82     const base::string16& message_text,
  83     const base::string16& default_prompt_text,
  84     const JavaScriptDialogManager::DialogClosedCallback& callback)
  85     : manager_(manager),
  86       callback_(callback),
  87       message_type_(message_type),
  88       message_text_(message_text),
  89       default_prompt_text_(default_prompt_text) {
  90   int dialog_type;
  91   if (message_type == JAVASCRIPT_MESSAGE_TYPE_ALERT)
  92     dialog_type = IDD_ALERT;
  93   else if (message_type == JAVASCRIPT_MESSAGE_TYPE_CONFIRM)
  94     dialog_type = IDD_CONFIRM;
  95   else // JAVASCRIPT_MESSAGE_TYPE_PROMPT
  96     dialog_type = IDD_PROMPT;
  97
  98   dialog_win_ = CreateDialogParam(GetModuleHandle(0),
  99                                   MAKEINTRESOURCE(dialog_type), 0, DialogProc,
 100                                   reinterpret_cast<LPARAM>(this));
 101   ShowWindow(dialog_win_, SW_SHOWNORMAL);
 102 }
 103
 104 ShellJavaScriptDialog::~ShellJavaScriptDialog() {
 105   Cancel();
 106 }
 107
 108 void ShellJavaScriptDialog::Cancel() {
 109   if (dialog_win_)
 110     DestroyWindow(dialog_win_);
 111 }
 112
 113 }  // namespace content