Add palette tray to ash.

ash/common/system/chromeos/palette/palette_tray.cc

Index: ash/common/system/chromeos/palette/palette_tray.cc
diff --git a/ash/common/system/chromeos/palette/palette_tray.cc b/ash/common/system/chromeos/palette/palette_tray.cc
new file mode 100644
index 0000000000000000000000000000000000000000..121c7dbb74d04e7fe59a992149569096e03cc64a
--- /dev/null
+++ b/ash/common/system/chromeos/palette/palette_tray.cc
@@ -0,0 +1,321 @@
+// Copyright 2016 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 "ash/common/system/chromeos/palette/palette_tray.h"
+
+#include "ash/common/material_design/material_design_controller.h"
+#include "ash/common/shelf/shelf_constants.h"
+#include "ash/common/shelf/wm_shelf.h"
+#include "ash/common/shelf/wm_shelf_util.h"
+#include "ash/common/shell_window_ids.h"
+#include "ash/common/system/chromeos/palette/palette_tool.h"
+#include "ash/common/system/chromeos/palette/palette_tool_manager.h"
+#include "ash/common/system/tray/system_tray_delegate.h"
+#include "ash/common/system/tray/tray_bubble_wrapper.h"
+#include "ash/common/system/tray/tray_constants.h"
+#include "ash/common/system/tray/tray_popup_header_button.h"
+#include "ash/common/wm_lookup.h"
+#include "ash/common/wm_root_window_controller.h"
+#include "ash/common/wm_shell.h"
+#include "ash/common/wm_window.h"
+#include "base/sys_info.h"
+#include "grit/ash_resources.h"
+#include "grit/ash_strings.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "ui/base/resource/resource_bundle.h"
+#include "ui/gfx/paint_vector_icon.h"
+#include "ui/gfx/vector_icons_public.h"
+#include "ui/views/controls/image_view.h"
+#include "ui/views/controls/label.h"
+#include "ui/views/controls/separator.h"
+#include "ui/views/layout/box_layout.h"
+
+#include "ui/views/layout/fill_layout.h"
+
+namespace {
+
+// Predefined padding for the icon used in this tray. These are to be set to the
+// border of the icon, depending on the current shelf_alignment()
+const int kHorizontalShelfHorizontalPadding = 8;
+const int kHorizontalShelfVerticalPadding = 4;
+const int kVerticalShelfHorizontalPadding = 2;
+const int kVerticalShelfVerticalPadding = 5;
+
+// Width of the palette itself.
+const int kPaletteWidth = 360;
+
+const int kSeparatorInset = 10;
+
+const int kBubbleViewTopMargin = 7;
+
+// Returns the icon that should be used in the status tray for the given tool.
+gfx::VectorIconId GetTrayIconForTool(ash::PaletteToolId tool_id) {
+  switch (tool_id) {
+    case ash::PaletteToolId::PRESENTATION:
+      return gfx::VectorIconId::PALETTE_PRESENTATION_BEHAVIOR;
+    case ash::PaletteToolId::MAGNIFY:
+      return gfx::VectorIconId::PALETTE_MAGNIFY_BEHAVIOR;
+    default:
+      return gfx::VectorIconId::PALETTE_DEFAULT_BEHAVIOR;
+  }
+}
+
+// Creates a separator.
+views::Separator* CreateSeparator(views::Separator::Orientation orientation) {
+  views::Separator* separator =
+      new views::Separator(views::Separator::HORIZONTAL);
+  separator->SetColor(ash::kBorderDarkColor);
+  separator->SetBorder(
+      views::Border::CreateEmptyBorder(kSeparatorInset, 0, kSeparatorInset, 0));
+  return separator;
+}
+
+class TitleBarView : public views::View {
+ public:
+  TitleBarView() {
+    auto& rb = ui::ResourceBundle::GetSharedInstance();
+
+    auto* box_layout =
+        new views::BoxLayout(views::BoxLayout::kHorizontal, 0, 0, 0);
+    SetLayoutManager(box_layout);
+
+    views::Label* text_label =
+        new views::Label(l10n_util::GetStringUTF16(IDS_ASH_PALETTE_TITLE));
+    text_label->SetBorder(views::Border::CreateEmptyBorder(
+        0, ash::kTrayPopupPaddingHorizontal, 0, 0));
+    text_label->SetHorizontalAlignment(gfx::ALIGN_LEFT);
+    text_label->SetFontList(rb.GetFontList(ui::ResourceBundle::BoldFont));
+    AddChildView(text_label);
+    box_layout->SetFlexForView(text_label, 1);
+
+    // TODO(jdufault): Use proper icons.
+    ash::TrayPopupHeaderButton* help_button = new ash::TrayPopupHeaderButton(
+        nullptr, IDR_AURA_UBER_TRAY_SHUTDOWN, IDR_AURA_UBER_TRAY_SHUTDOWN,
+        IDR_AURA_UBER_TRAY_SHUTDOWN_HOVER, IDR_AURA_UBER_TRAY_SHUTDOWN_HOVER,
+        IDS_ASH_STATUS_TRAY_SHUTDOWN);
+    help_button->SetTooltipText(
+        l10n_util::GetStringUTF16(IDS_ASH_STATUS_TRAY_SHUTDOWN));
+    AddChildView(help_button);
+
+    AddChildView(CreateSeparator(views::Separator::VERTICAL));
+
+    // TODO(jdufault): Use proper icons.
+    ash::TrayPopupHeaderButton* settings_button =
+        new ash::TrayPopupHeaderButton(
+            nullptr, IDR_AURA_UBER_TRAY_SHUTDOWN, IDR_AURA_UBER_TRAY_SHUTDOWN,
+            IDR_AURA_UBER_TRAY_SHUTDOWN_HOVER,
+            IDR_AURA_UBER_TRAY_SHUTDOWN_HOVER, IDS_ASH_STATUS_TRAY_SHUTDOWN);
+    settings_button->SetTooltipText(
+        l10n_util::GetStringUTF16(IDS_ASH_STATUS_TRAY_SHUTDOWN));
+    AddChildView(settings_button);
+  }
+
+  ~TitleBarView() override {}
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(TitleBarView);
+};
+
+}  // namespace
+
+namespace ash {
+
+PaletteTray::PaletteTray(WmShelf* wm_shelf)
+    : TrayBackgroundView(wm_shelf),
+      palette_tool_manager_(new PaletteToolManager(
+          wm_shelf->GetWindow(),
+          base::Bind(&PaletteTray::UpdateTrayIcon, base::Unretained(this)))) {
+  PaletteTool::RegisterToolInstances(palette_tool_manager_.get());
+
+  SetContentsBackground();
+
+  SetLayoutManager(new views::FillLayout());
+  icon_ = new views::ImageView();
+  UpdateTrayIcon();
+
+  SetIconBorderForShelfAlignment();
+  tray_container()->AddChildView(icon_);
+
+  WmShell::Get()->AddShellObserver(this);
+  WmShell::Get()->GetSessionStateDelegate()->AddSessionStateObserver(this);
+
+  UpdateIconVisibility();
+}
+
+PaletteTray::~PaletteTray() {
+  if (bubble_)
+    bubble_->bubble_view()->reset_delegate();
+
+  WmShell::Get()->RemoveShellObserver(this);
+  WmShell::Get()->GetSessionStateDelegate()->RemoveSessionStateObserver(this);
+}
+
+bool PaletteTray::PerformAction(const ui::Event& event) {
+  if (bubble_) {
+    bubble_.reset();
+    return true;
+  }
+
+  return OpenBubble();
+}
+
+bool PaletteTray::OpenBubble() {
+  views::TrayBubbleView::InitParams init_params(
+      views::TrayBubbleView::ANCHOR_TYPE_TRAY, GetAnchorAlignment(),
+      kPaletteWidth, kPaletteWidth);
+  init_params.first_item_has_no_margin = true;
+  init_params.can_activate = true;
+  init_params.close_on_deactivate = true;
+
+  views::View* anchor = this->tray_container();

[oshima, 2016/07/13 21:14:23]

try_container()? 

[jdufault, 2016/07/18 20:44:02]

Done.

+  DCHECK(anchor);

[oshima, 2016/07/13 21:14:23]

or just inline tray_container()

[jdufault, 2016/07/18 20:44:02]

Done.

+
+  // Create view, customize it.
+  views::TrayBubbleView* bubble_view =
+      views::TrayBubbleView::Create(anchor, this, &init_params);
+  bubble_view->set_margins(gfx::Insets(kBubbleViewTopMargin, 0, 0, 0));
+  bubble_view->SetArrowPaintType(views::BubbleBorder::PAINT_NONE);
+
+  // Add child views.
+  bubble_view->AddChildView(new TitleBarView());
+  bubble_view->AddChildView(CreateSeparator(views::Separator::HORIZONTAL));
+  AddToolsToView(bubble_view);
+
+  // Show the bubble.
+  bubble_.reset(new ash::TrayBubbleWrapper(this, bubble_view));
+
+  SetDrawBackgroundAsActive(true);
+
+  return true;
+}
+
+void PaletteTray::AddToolsToView(views::View* host) {
+  std::vector<PaletteToolView> views = palette_tool_manager_->CreateViews();
+  if (!views.size())
+    return;

[oshima, 2016/07/13 21:14:23]

can you add comment? 

[jdufault, 2016/07/18 20:44:02]

Done.

+
+  PaletteGroup group = views[0].group;
+  for (const PaletteToolView& view : views) {
+    // If the group changes, add a separator.
+    if (group != view.group) {
+      group = view.group;
+      host->AddChildView(CreateSeparator(views::Separator::HORIZONTAL));
+    }
+
+    host->AddChildView(view.view);
+  }
+}
+
+void PaletteTray::SessionStateChanged(
+    SessionStateDelegate::SessionState state) {
+  UpdateIconVisibility();
+}
+
+void PaletteTray::ClickedOutsideBubble() {
+  bubble_.reset();
+}
+
+base::string16 PaletteTray::GetAccessibleNameForTray() {
+  return l10n_util::GetStringUTF16(IDS_ASH_PALETTE_TITLE);
+}
+
+void PaletteTray::HideBubbleWithView(const views::TrayBubbleView* bubble_view) {
+  if (bubble_->bubble_view() == bubble_view)
+    bubble_.reset();
+}
+
+void PaletteTray::BubbleViewDestroyed() {
+  palette_tool_manager_->ViewsDestroyed();
+  SetDrawBackgroundAsActive(false);
+}
+
+void PaletteTray::OnMouseEnteredView() {}
+
+void PaletteTray::OnMouseExitedView() {}
+
+base::string16 PaletteTray::GetAccessibleNameForBubble() {
+  return GetAccessibleNameForTray();
+}
+
+gfx::Rect PaletteTray::GetAnchorRect(
+    views::Widget* anchor_widget,
+    views::TrayBubbleView::AnchorType anchor_type,
+    views::TrayBubbleView::AnchorAlignment anchor_alignment) const {
+  gfx::Rect r =
+      GetBubbleAnchorRect(anchor_widget, anchor_type, anchor_alignment);
+
+  // Move the palette to the left so the right edge of the palette aligns with
+  // the right edge of the tray button.
+  if (IsHorizontalAlignment(shelf_alignment()))
+    r.Offset(-r.width() + tray_container()->width(), 0);
+  else
+    r.Offset(0, -r.height() + tray_container()->height());
+
+  return r;
+}
+
+void PaletteTray::OnBeforeBubbleWidgetInit(
+    views::Widget* anchor_widget,
+    views::Widget* bubble_widget,
+    views::Widget::InitParams* params) const {
+  // Place the bubble in the same root window as |anchor_widget|.
+  WmLookup::Get()
+      ->GetWindowForWidget(anchor_widget)
+      ->GetRootWindowController()
+      ->ConfigureWidgetInitParamsForContainer(
+          bubble_widget, kShellWindowId_SettingBubbleContainer, params);
+}
+
+void PaletteTray::HideBubble(const views::TrayBubbleView* bubble_view) {
+  HideBubbleWithView(bubble_view);
+}
+
+void PaletteTray::SetShelfAlignment(ShelfAlignment alignment) {
+  if (alignment == shelf_alignment())
+    return;
+
+  TrayBackgroundView::SetShelfAlignment(alignment);
+  SetIconBorderForShelfAlignment();
+}
+
+void PaletteTray::AnchorUpdated() {
+  if (bubble_)
+    bubble_->bubble_view()->UpdateBubble();
+}
+
+void PaletteTray::SetIconBorderForShelfAlignment() {
+  if (IsHorizontalAlignment(shelf_alignment())) {
+    icon_->SetBorder(views::Border::CreateEmptyBorder(
+        kHorizontalShelfVerticalPadding, kHorizontalShelfHorizontalPadding,

[oshima, 2016/07/13 21:14:23]

nit: Insets(h, v) may be shorter.

[jdufault, 2016/07/18 20:44:02]

Easier to read, at least.

+        kHorizontalShelfVerticalPadding, kHorizontalShelfHorizontalPadding));
+  } else {
+    icon_->SetBorder(views::Border::CreateEmptyBorder(
+        kVerticalShelfVerticalPadding, kVerticalShelfHorizontalPadding,
+        kVerticalShelfVerticalPadding, kVerticalShelfHorizontalPadding));
+  }
+}
+
+void PaletteTray::UpdateTrayIcon() {
+  gfx::VectorIconId icon = GetTrayIconForTool(
+      palette_tool_manager_->GetActiveTool(PaletteGroup::MODE));
+
+  gfx::ImageSkia image_md = CreateVectorIcon(icon, kShelfIconColor);
+  icon_->SetImage(image_md);
+}
+
+void PaletteTray::UpdateIconVisibility() {
+  // TODO(jdufault): Properly plumb if the UI should be shown.
+  return;

[oshima, 2016/07/13 21:14:23]

debug code?

[jdufault, 2016/07/18 20:44:02]

There's no command line flag added yet that this will check. I've pulled this
out into a separate function so we're closer to the final logic.

+
+  SessionStateDelegate* session_state_delegate =
+      WmShell::Get()->GetSessionStateDelegate();
+
+  SetVisible(!session_state_delegate->IsScreenLocked() &&
+             session_state_delegate->GetSessionState() ==
+                 SessionStateDelegate::SESSION_STATE_ACTIVE &&
+             WmShell::Get()->system_tray_delegate()->GetUserLoginStatus() !=
+                 LoginStatus::KIOSK_APP);
+}
+
+}  // namespace ash