#ifndef MO_TEXTEDITOR_H
#define MO_TEXTEDITOR_H

#include <QPlainTextEdit>

class TextEditor;

class TextEditorToolbar
{
public:
  TextEditorToolbar(TextEditor& editor);

  QWidget* widget();

private:
  TextEditor& m_editor;
  QWidget* m_widget;
  QAction* m_save;
  QAction* m_wordWrap;

  void onTextModified(bool b);
  void onWordWrap(bool b);
};


class TextEditorLineNumbers : public QWidget
{
public:
  TextEditorLineNumbers(TextEditor& editor);
  QSize sizeHint() const override;
  int areaWidth() const;

protected:
  void paintEvent(QPaintEvent *event) override;

private:
  TextEditor& m_editor;

  void updateAreaWidth();
  void updateArea(const QRect &rect, int dy);
};


class TextEditor : public QPlainTextEdit
{
  Q_OBJECT
  friend class TextEditorLineNumbers;

public:
  TextEditor(QWidget* parent=nullptr);

  void setupToolbar();

  bool load(const QString& filename);
  bool save();

  const QString& filename() const;

  void wordWrap(bool b);
  void toggleWordWrap();
  bool wordWrap() const;

  bool dirty() const;

signals:
  void modified(bool b);
  void wordWrapChanged(bool b);

protected:
  void resizeEvent(QResizeEvent* e) override;

private:
  TextEditorToolbar m_toolbar;
  TextEditorLineNumbers* m_lineNumbers;
  QString m_filename;
  QString m_encoding;
  bool m_dirty;

  void onModified(bool b);
  void dirty(bool b);

  QWidget* wrapEditWidget();

  void paintLineNumbers(QPaintEvent* e);
};

#endif // MO_TEXTEDITOR_H