#pragma once

#include "stfwd.h"

#include <QWidget>

class CaptureAreaSelector : public QWidget
{
  Q_OBJECT

public:
  CaptureAreaSelector(Capturer &capturer, const Settings &settings,
                      const CommonModels &models, const QPixmap &pixmap);
  ~CaptureAreaSelector();

  void activate();
  void setScreenRects(const std::vector<QRect> &screens);
  void updateSettings();

protected:
  void showEvent(QShowEvent *event) override;
  void hideEvent(QHideEvent *event) override;
  void keyPressEvent(QKeyEvent *event) override;
  void mousePressEvent(QMouseEvent *event) override;
  void mouseMoveEvent(QMouseEvent *event) override;
  void mouseReleaseEvent(QMouseEvent *event) override;
  void paintEvent(QPaintEvent *event) override;

private:
  struct HelpRect {
    QRect current;
    std::vector<QRect> possible;
  };

  bool updateCurrentHelpRects();
  void drawHelpRects(QPainter &painter, const HelpRect &rect) const;

  void customize(const CaptureArea &area);
  void applyEditor();
  void drawCaptureArea(QPainter &painter, const CaptureArea &area) const;

  Capturer &capturer_;
  const Settings &settings_;
  const QPixmap &pixmap_;
  QPoint startSelectPos_;
  QPoint currentSelectPos_;
  QString help_;
  std::vector<HelpRect> helpRects_;
  std::unique_ptr<CaptureArea> area_;
  std::unique_ptr<CaptureAreaEditor> editor_;
};