Hide dismiss button on toast when doing partial screenshot using stylus tools.

ash/common/system/toast/toast_overlay.cc

 // 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/toast/toast_overlay.h"
 
 #include "ash/common/shelf/wm_shelf.h"
 #include "ash/common/wm_lookup.h"
 #include "ash/common/wm_root_window_controller.h"
 #include "ash/common/wm_shell.h"
@@ skipping 111 matching lines @@
   int verticalSpacing = kToastVerticalSpacing -
                         (GetPreferredSize().height() - label()->GetBaseline());
   SetBorder(views::CreateEmptyBorder(verticalSpacing, kToastHorizontalSpacing,
                                      verticalSpacing, kToastHorizontalSpacing));
 }
 
 ///////////////////////////////////////////////////////////////////////////////
 //  ToastOverlayView
 class ToastOverlayView : public views::View, public views::ButtonListener {
  public:
-  // This object is not owned by the views hiearchy or by the widget.
+  // This object is not owned by the views hierarchy or by the widget.
   ToastOverlayView(ToastOverlay* overlay,
                    const base::string16& text,
-                   const base::string16& dismiss_text);
+                   const base::Optional<base::string16>& dismiss_text);
   ~ToastOverlayView() override;
 
   // views::View overrides:
   void OnPaint(gfx::Canvas* canvas) override;
 
   ToastOverlayButton* button() { return button_; }
 
  private:
-  ToastOverlay* overlay_;       // weak
-  ToastOverlayButton* button_;  // weak
-
+  // views::View overrides:
   gfx::Size GetMaximumSize() const override;
   gfx::Size GetMinimumSize() const override;
 
+  // views::ButtonListener overrides:
   void ButtonPressed(views::Button* sender, const ui::Event& event) override;
 
+  ToastOverlay* overlay_ = nullptr;       // weak
+  ToastOverlayButton* button_ = nullptr;  // weak
+
   DISALLOW_COPY_AND_ASSIGN(ToastOverlayView);
 };
 
-ToastOverlayView::ToastOverlayView(ToastOverlay* overlay,
-                                   const base::string16& text,
-                                   const base::string16& dismiss_text)
-    : overlay_(overlay),
-      button_(new ToastOverlayButton(
-          this,
-          dismiss_text.empty()
-              ? l10n_util::GetStringUTF16(IDS_ASH_TOAST_DISMISS_BUTTON)
-              : dismiss_text)) {
-  ToastOverlayLabel* label = new ToastOverlayLabel(text);
-  label->SetMaximumWidth(
-      GetMaximumSize().width() - button_->GetPreferredSize().width() -
-      kToastHorizontalSpacing * 2 - kToastHorizontalSpacing * 2);
-  AddChildView(label);
-
-  AddChildView(button_);
-
+ToastOverlayView::ToastOverlayView(
+    ToastOverlay* overlay,
+    const base::string16& text,
+    const base::Optional<base::string16>& dismiss_text)
+    : overlay_(overlay) {
   auto* layout = new views::BoxLayout(views::BoxLayout::kHorizontal, 0, 0, 0);
   SetLayoutManager(layout);
+
+  if (dismiss_text.has_value()) {
+    button_ = new ToastOverlayButton(
+        this, dismiss_text.value().empty()
+                  ? l10n_util::GetStringUTF16(IDS_ASH_TOAST_DISMISS_BUTTON)
+                  : dismiss_text.value());
+  }
+
+  ToastOverlayLabel* label = new ToastOverlayLabel(text);
+  AddChildView(label);
   layout->SetFlexForView(label, 1);
-  layout->SetFlexForView(button_, 0);
+
+  if (button_) {
+    label->SetMaximumWidth(
+        GetMaximumSize().width() - button_->GetPreferredSize().width() -
+        kToastHorizontalSpacing * 2 - kToastHorizontalSpacing * 2);
+    AddChildView(button_);
+  }
 }
 
 ToastOverlayView::~ToastOverlayView() {}
 
 void ToastOverlayView::OnPaint(gfx::Canvas* canvas) {
   SkPaint paint;
   paint.setStyle(SkPaint::kFill_Style);
   paint.setColor(kButtonBackgroundColor);
   canvas->DrawRoundRect(GetLocalBounds(), 2, paint);
   views::View::OnPaint(canvas);
 }
 
 gfx::Size ToastOverlayView::GetMinimumSize() const {
   return gfx::Size(kToastMinimumWidth, 0);
 }
 
 gfx::Size ToastOverlayView::GetMaximumSize() const {
   gfx::Rect work_area_bounds = GetUserWorkAreaBounds();
   return gfx::Size(kToastMaximumWidth, work_area_bounds.height() - kOffset * 2);
 }
 
 void ToastOverlayView::ButtonPressed(views::Button* sender,
                                      const ui::Event& event) {
+  DCHECK_EQ(button_, sender);
   overlay_->Show(false);
 }
 
 ///////////////////////////////////////////////////////////////////////////////
 //  ToastOverlay
 ToastOverlay::ToastOverlay(Delegate* delegate,
                            const base::string16& text,
-                           const base::string16& dismiss_text)
+                           base::Optional<base::string16> dismiss_text)
     : delegate_(delegate),
       text_(text),
       dismiss_text_(dismiss_text),
       overlay_widget_(new views::Widget),
       overlay_view_(new ToastOverlayView(this, text, dismiss_text)),
       widget_size_(overlay_view_->GetPreferredSize()) {
   views::Widget::InitParams params;
   params.type = views::Widget::InitParams::TYPE_POPUP;
   params.name = "ToastOverlay";
   params.opacity = views::Widget::InitParams::TRANSLUCENT_WINDOW;
@@ skipping 69 matching lines @@
 
 void ToastOverlay::OnImplicitAnimationsCompleted() {
   if (!overlay_widget_->GetLayer()->GetTargetVisibility())
     delegate_->OnClosed();
 }
 
 views::Widget* ToastOverlay::widget_for_testing() {
   return overlay_widget_.get();
 }
 
+ToastOverlayButton* ToastOverlay::dismiss_button_for_testing() {
+  return overlay_view_->button();
+}
+
 void ToastOverlay::ClickDismissButtonForTesting(const ui::Event& event) {
+  DCHECK(overlay_view_->button());
   overlay_view_->button()->NotifyClick(event);
 }
 
 }  // namespace ash