chrome/browser/ui/views/extensions/device_permissions_dialog_view.cc

   1 // Copyright 2014 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 "chrome/browser/ui/views/extensions/device_permissions_dialog_view.h"
   6
   7 #include "base/bind.h"
   8 #include "base/location.h"
   9 #include "base/single_thread_task_runner.h"
  10 #include "base/strings/utf_string_conversions.h"
  11 #include "base/thread_task_runner_handle.h"
  12 #include "chrome/browser/extensions/api/chrome_device_permissions_prompt.h"
  13 #include "chrome/grit/generated_resources.h"
  14 #include "components/constrained_window/constrained_window_views.h"
  15 #include "components/web_modal/web_contents_modal_dialog_manager.h"
  16 #include "content/public/browser/browser_thread.h"
  17 #include "device/usb/usb_device.h"
  18 #include "extensions/common/extension.h"
  19 #include "ui/base/l10n/l10n_util.h"
  20 #include "ui/base/models/table_model.h"
  21 #include "ui/base/models/table_model_observer.h"
  22 #include "ui/views/controls/label.h"
  23 #include "ui/views/controls/table/table_view.h"
  24 #include "ui/views/layout/box_layout.h"
  25 #include "ui/views/layout/layout_constants.h"
  26
  27 using device::UsbDevice;
  28 using extensions::DevicePermissionsPrompt;
  29
  30 class DevicePermissionsTableModel
  31     : public ui::TableModel,
  32       public DevicePermissionsPrompt::Prompt::Observer {
  33  public:
  34   explicit DevicePermissionsTableModel(
  35       scoped_refptr<DevicePermissionsPrompt::Prompt> prompt)
  36       : prompt_(prompt) {
  37     prompt_->SetObserver(this);
  38   }
  39
  40   ~DevicePermissionsTableModel() override { prompt_->SetObserver(nullptr); }
  41
  42   // ui::TableModel
  43   int RowCount() override;
  44   base::string16 GetText(int row, int column) override;
  45   void SetObserver(ui::TableModelObserver* observer) override;
  46
  47   // extensions::DevicePermissionsPrompt::Prompt::Observer
  48   void OnDevicesChanged() override;
  49
  50  private:
  51   scoped_refptr<DevicePermissionsPrompt::Prompt> prompt_;
  52   ui::TableModelObserver* observer_ = nullptr;
  53 };
  54
  55 int DevicePermissionsTableModel::RowCount() {
  56   return prompt_->GetDeviceCount();
  57 }
  58
  59 base::string16 DevicePermissionsTableModel::GetText(int row, int col_id) {
  60   switch (col_id) {
  61     case IDS_DEVICE_PERMISSIONS_DIALOG_DEVICE_NAME_COLUMN:
  62       return prompt_->GetDeviceName(row);
  63     case IDS_DEVICE_PERMISSIONS_DIALOG_SERIAL_NUMBER_COLUMN:
  64       return prompt_->GetDeviceSerialNumber(row);
  65     default:
  66       NOTREACHED();
  67       return base::string16();
  68   }
  69 }
  70
  71 void DevicePermissionsTableModel::SetObserver(
  72     ui::TableModelObserver* observer) {
  73   observer_ = observer;
  74 }
  75
  76 void DevicePermissionsTableModel::OnDevicesChanged() {
  77   if (observer_) {
  78     observer_->OnModelChanged();
  79   }
  80 }
  81
  82 DevicePermissionsDialogView::DevicePermissionsDialogView(
  83     scoped_refptr<DevicePermissionsPrompt::Prompt> prompt)
  84     : prompt_(prompt) {
  85   views::BoxLayout* layout =
  86       new views::BoxLayout(views::BoxLayout::kVertical,
  87                            views::kButtonHEdgeMarginNew,
  88                            0,
  89                            views::kRelatedControlVerticalSpacing);
  90   SetLayoutManager(layout);
  91
  92   views::Label* label = new views::Label(prompt_->GetPromptMessage());
  93   label->SetHorizontalAlignment(gfx::ALIGN_LEFT);
  94   label->SetMultiLine(true);
  95   AddChildView(label);
  96
  97   std::vector<ui::TableColumn> table_columns;
  98   table_columns.push_back(
  99       ui::TableColumn(IDS_DEVICE_PERMISSIONS_DIALOG_DEVICE_NAME_COLUMN,
 100                       ui::TableColumn::LEFT,
 101                       -1,
 102                       0.8f));
 103   table_columns.back().title = l10n_util::GetStringUTF16(
 104       IDS_DEVICE_PERMISSIONS_DIALOG_DEVICE_NAME_COLUMN);
 105   table_columns.push_back(
 106       ui::TableColumn(IDS_DEVICE_PERMISSIONS_DIALOG_SERIAL_NUMBER_COLUMN,
 107                       ui::TableColumn::LEFT,
 108                       -1,
 109                       0.2f));
 110   table_columns.back().title = l10n_util::GetStringUTF16(
 111       IDS_DEVICE_PERMISSIONS_DIALOG_SERIAL_NUMBER_COLUMN);
 112
 113   table_model_.reset(new DevicePermissionsTableModel(prompt_));
 114   table_view_ = new views::TableView(table_model_.get(),
 115                                      table_columns,
 116                                      views::TEXT_ONLY,
 117                                      !prompt_->multiple());
 118
 119   views::View* table_parent = table_view_->CreateParentIfNecessary();
 120   AddChildView(table_parent);
 121   layout->SetFlexForView(table_parent, 1);
 122 }
 123
 124 DevicePermissionsDialogView::~DevicePermissionsDialogView() {
 125   RemoveAllChildViews(true);
 126 }
 127
 128 void DevicePermissionsDialogView::DeleteDelegate() {
 129   // Calling prompt_->Dismissed() here ensures it will be called regardless of
 130   // how the view is closed, including shutdown of the entire view hierarchy.
 131   prompt_->Dismissed();
 132   delete this;
 133 }
 134
 135 bool DevicePermissionsDialogView::Accept() {
 136   for (int index : table_view_->selection_model().selected_indices()) {
 137     prompt_->GrantDevicePermission(index);
 138   }
 139   return true;
 140 }
 141
 142 base::string16 DevicePermissionsDialogView::GetDialogButtonLabel(
 143     ui::DialogButton button) const {
 144   if (button == ui::DIALOG_BUTTON_OK) {
 145     return l10n_util::GetStringUTF16(IDS_DEVICE_PERMISSIONS_DIALOG_SELECT);
 146   }
 147   return views::DialogDelegateView::GetDialogButtonLabel(button);
 148 }
 149
 150 ui::ModalType DevicePermissionsDialogView::GetModalType() const {
 151   return ui::MODAL_TYPE_CHILD;
 152 }
 153
 154 base::string16 DevicePermissionsDialogView::GetWindowTitle() const {
 155   return prompt_->GetHeading();
 156 }
 157
 158 gfx::Size DevicePermissionsDialogView::GetPreferredSize() const {
 159   return gfx::Size(500, 250);
 160 }
 161
 162 void ChromeDevicePermissionsPrompt::ShowDialog() {
 163   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 164
 165   web_modal::WebContentsModalDialogManager* manager =
 166       web_modal::WebContentsModalDialogManager::FromWebContents(web_contents());
 167   if (manager) {
 168     constrained_window::ShowWebModalDialogViews(
 169         new DevicePermissionsDialogView(prompt()), web_contents());
 170   } else {
 171     base::ThreadTaskRunnerHandle::Get()->PostTask(
 172         FROM_HERE,
 173         base::Bind(&DevicePermissionsPrompt::Prompt::Dismissed, prompt()));
 174   }
 175 }