1 files changed, 103 insertions, 0 deletions
diff --git a/chrome/browser/views/user_data_dir_dialog.cc b/chrome/browser/views/user_data_dir_dialog.cc
new file mode 100644
index 0000000..90460fb
--- /dev/null
+++ b/chrome/browser/views/user_data_dir_dialog.cc
@@ -0,0 +1,103 @@
+// Copyright (c) 2006-2008 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 "base/logging.h"
+#include "chrome/browser/views/user_data_dir_dialog.h"
+#include "chrome/common/l10n_util.h"
+#include "chrome/views/message_box_view.h"
+#include "chrome/views/window.h"
+
+#include "chromium_strings.h"
+#include "generated_resources.h"
+
+// static
+std::wstring UserDataDirDialog::RunUserDataDirDialog(
+    const std::wstring& user_data_dir) {
+  // When the window closes, it will delete itself.
+  UserDataDirDialog* dlg = new UserDataDirDialog(user_data_dir);
+  MessageLoopForUI::current()->Run(dlg);
+  return dlg->user_data_dir();
+}
+
+UserDataDirDialog::UserDataDirDialog(const std::wstring& user_data_dir)
+    : select_file_dialog_(SelectFileDialog::Create(this)),
+      is_blocking_(true) {
+  std::wstring message_text = l10n_util::GetStringF(
+      IDS_CANT_WRITE_USER_DIRECTORY_SUMMARY, user_data_dir);
+  const int kDialogWidth = 400;
+  message_box_view_ = new MessageBoxView(MessageBoxView::kIsConfirmMessageBox,
+      message_text.c_str(), std::wstring(), kDialogWidth);
+
+  ChromeViews::Window::CreateChromeWindow(NULL, gfx::Rect(), this)->Show();
+}
+
+UserDataDirDialog::~UserDataDirDialog() {
+  select_file_dialog_->ListenerDestroyed();
+}
+
+int UserDataDirDialog::GetDialogButtons() const {
+  return DIALOGBUTTON_OK | DIALOGBUTTON_CANCEL;
+
+}
+
+std::wstring UserDataDirDialog::GetDialogButtonLabel(
+    DialogButton button) const {
+
+  switch (button) {
+    case DIALOGBUTTON_OK:
+      return l10n_util::GetString(
+          IDS_CANT_WRITE_USER_DIRECTORY_CHOOSE_DIRECTORY_BUTTON);
+    case DIALOGBUTTON_CANCEL:
+      return l10n_util::GetString(IDS_CANT_WRITE_USER_DIRECTORY_EXIT_BUTTON);
+    default:
+      NOTREACHED();
+  }
+
+  return std::wstring();
+}
+
+std::wstring UserDataDirDialog::GetWindowTitle() const {
+  return l10n_util::GetString(IDS_CANT_WRITE_USER_DIRECTORY_TITLE);
+}
+
+void UserDataDirDialog::WindowClosing() {
+  delete this;
+}
+
+bool UserDataDirDialog::Accept() {
+  // Directory picker
+  std::wstring dialog_title = l10n_util::GetString(
+      IDS_CANT_WRITE_USER_DIRECTORY_CHOOSE_DIRECTORY_BUTTON);
+  HWND owning_hwnd =
+      GetAncestor(message_box_view_->GetViewContainer()->GetHWND(), GA_ROOT);
+  select_file_dialog_->SelectFile(SelectFileDialog::SELECT_FOLDER,
+                                  dialog_title, std::wstring(), owning_hwnd,
+                                  NULL);
+  return false;
+}
+
+bool UserDataDirDialog::Cancel() {
+  is_blocking_ = false;
+  return true;
+}
+
+ChromeViews::View* UserDataDirDialog::GetContentsView() {
+  return message_box_view_;
+}
+
+bool UserDataDirDialog::Dispatch(const MSG& msg) {
+  TranslateMessage(&msg);
+  DispatchMessage(&msg);
+  return is_blocking_;
+}
+
+void UserDataDirDialog::FileSelected(const std::wstring& path, void* params) {
+  user_data_dir_ = path;
+  is_blocking_ = false;
+  window()->Close();
+}
+
+void UserDataDirDialog::FileSelectionCanceled(void* params) {
+}
+