Allow the user to cancel the capture region action, and show its active status in the tray.

ash/common/system/chromeos/palette/tools/capture_region_mode.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/chromeos/palette/tools/capture_region_action.h"
+#include "ash/common/system/chromeos/palette/tools/capture_region_mode.h"
 
 #include "ash/common/accelerators/accelerator_controller.h"
 #include "ash/common/palette_delegate.h"
 #include "ash/common/system/chromeos/palette/palette_ids.h"
 #include "ash/common/system/toast/toast_data.h"
 #include "ash/common/system/toast/toast_manager.h"
 #include "ash/common/wm_shell.h"
 #include "grit/ash_strings.h"
 #include "ui/base/l10n/l10n_util.h"
 
 namespace ash {
 
 namespace {
 
 const char kToastId[] = "palette_capture_region";
 const int kToastDurationMs = 2500;
 
 }  // namespace
 
-CaptureRegionAction::CaptureRegionAction(Delegate* delegate)
-    : CommonPaletteTool(delegate) {}
+CaptureRegionMode::CaptureRegionMode(Delegate* delegate)
+    : CommonPaletteTool(delegate), weak_factory_(this) {}
 
-CaptureRegionAction::~CaptureRegionAction() {}
+CaptureRegionMode::~CaptureRegionMode() {}
 
-PaletteGroup CaptureRegionAction::GetGroup() const {
-  return PaletteGroup::ACTION;
+PaletteGroup CaptureRegionMode::GetGroup() const {
+  return PaletteGroup::MODE;
 }
 
-PaletteToolId CaptureRegionAction::GetToolId() const {
+PaletteToolId CaptureRegionMode::GetToolId() const {
   return PaletteToolId::CAPTURE_REGION;
 }
 
-void CaptureRegionAction::OnEnable() {
+gfx::VectorIconId CaptureRegionMode::GetActiveTrayIcon() {
+  // TODO(jdufault): Update to real icon once new tray icons are ready.
+  return gfx::VectorIconId::PALETTE_TRAY_ICON_MAGNIFY;
+}
+
+void CaptureRegionMode::OnEnable() {
   CommonPaletteTool::OnEnable();
 
   ToastData toast(kToastId, l10n_util::GetStringUTF16(
                                 IDS_ASH_STYLUS_TOOLS_CAPTURE_REGION_TOAST),
                   kToastDurationMs, base::string16());
   ash::WmShell::Get()->toast_manager()->Show(toast);
 
-  WmShell::Get()->palette_delegate()->TakePartialScreenshot();
-  delegate()->DisableTool(GetToolId());
+  WmShell::Get()->palette_delegate()->TakePartialScreenshot(base::Bind(
+      &CaptureRegionMode::OnScreenshotDone, weak_factory_.GetWeakPtr()));
   delegate()->HidePalette();
 }
 
-views::View* CaptureRegionAction::CreateView() {
+void CaptureRegionMode::OnDisable() {
+  CommonPaletteTool::OnDisable();
+
+  // If the user manually cancelled the action we need to make sure to cancel
+  // the screenshot session as well.
+  WmShell::Get()->palette_delegate()->CancelPartialScreenshot();
+}
+
+views::View* CaptureRegionMode::CreateView() {
   return CreateDefaultView(
       l10n_util::GetStringUTF16(IDS_ASH_STYLUS_TOOLS_CAPTURE_REGION_ACTION));
 }
 
-gfx::VectorIconId CaptureRegionAction::GetPaletteIconId() {
+gfx::VectorIconId CaptureRegionMode::GetPaletteIconId() {
   return gfx::VectorIconId::PALETTE_ACTION_CAPTURE_REGION;
 }
 
+void CaptureRegionMode::OnScreenshotDone() {
+  // The screenshot finished, so disable the tool.
+  delegate()->DisableTool(GetToolId());
+}
+
 }  // namespace ash