// Copyright 2017 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.

#ifndef ASH_HIGHLIGHTER_HIGHLIGHTER_CONTROLLER_TEST_API_H_
#define ASH_HIGHLIGHTER_HIGHLIGHTER_CONTROLLER_TEST_API_H_

#include "ash/highlighter/highlighter_selection_observer.h"
#include "base/callback.h"
#include "base/macros.h"
#include "ui/gfx/geometry/rect.h"

namespace ash {

class FastInkPoints;
class HighlighterController;

// An api for testing the HighlighterController class.
// Inheriting from HighlighterSelectionObserver to provide the tests
// with access to gesture recognition results.
class HighlighterControllerTestApi : public HighlighterSelectionObserver {
 public:
  explicit HighlighterControllerTestApi(HighlighterController* instance);
  ~HighlighterControllerTestApi() override;

  void SetMetalayerDone(base::OnceClosure done) {
    metalayer_done_ = std::move(done);
  }
  void CallMetalayerDone() {
    if (!metalayer_done_.is_null())
      std::move(metalayer_done_).Run();
  }
  void SetViaButton(bool via_button) { via_button_ = via_button; }
  void SetEnabled(bool enabled);
  void DestroyPointerView();
  void SimulateInterruptedStrokeTimeout();
  bool IsShowingHighlighter() const;
  bool IsFadingAway() const;
  bool IsWaitingToResumeStroke() const;
  bool IsShowingSelectionResult() const;
  const FastInkPoints& points() const;
  const FastInkPoints& predicted_points() const;

  void ResetSelection() {
    handle_selection_called_ = false;
    handle_failed_selection_called_ = false;
  }
  bool handle_selection_called() const { return handle_selection_called_; }
  bool handle_failed_selection_called() const {
    return handle_failed_selection_called_;
  }
  const gfx::Rect& selection() const { return selection_; }

 private:
  // HighlighterSelectionObserver:
  void HandleSelection(const gfx::Rect& rect) override;
  void HandleFailedSelection() override;

  HighlighterController* instance_;

  bool handle_selection_called_ = false;
  bool handle_failed_selection_called_ = false;
  gfx::Rect selection_;

  base::OnceClosure metalayer_done_;
  bool via_button_ = false;

  DISALLOW_COPY_AND_ASSIGN(HighlighterControllerTestApi);
};

}  // namespace ash

#endif  // ASH_HIGHLIGHTER_HIGHLIGHTER_CONTROLLER_TEST_API_H_