#ifndef RAGNAROK_SOURCE_UI_EDIT_UIEDITBOXEX_H
#define RAGNAROK_SOURCE_UI_EDIT_UIEDITBOXEX_H

#include "UI/Edit/UIEditCtrl.h"
#include "UI/Window/UIControl.h"

#include <chrono>
#include <cstddef>
#include <cstdint>
#include <functional>
#include <string>
#include <vector>

struct EditLine {
  /* 0000 */ std::string text;
  /* 0018 */ unsigned char unknown_0018;
  /* 001C */ int selectionBegin;
  /* 0020 */ int selectionEnd;
};

class CUIEditCtrlEx : public UIEditCtrl3 {
public:
  CUIEditCtrlEx(UIWindowContext& windows, UIEditContext& edit);
  void Paint() override;
  void Invalidate() override;
  void SelectAll() override;
  void UpdateScrollBar() override;
  int SendMsg(UIWindow* sender, int message, std::intptr_t a, std::intptr_t b, std::intptr_t c, std::intptr_t d) override;

  void RefreshPlaceholder();
  void HandleKeyDown(int key, std::intptr_t lparam);
  void ResetDragState();
  void DrawCaret(int x, int y);
  void PaintTextLines();
  UIEditContext& EditContext() {
    return *this->m_edit;
  }
  int LeadingBytesFitting(const std::string& text, int width);
  int TrailingOffsetFitting(const std::string& text, int width);

  /* 0188 */ std::uint32_t normalColor;
  /* 018C */ int unknown_018C;
  /* 0190 */ unsigned char unknown_0190;
  /* 0194 */ std::string placeholder;
  /* 01AC */ std::uint32_t placeholderColor;
  /* 01B0 */ int alignment;
  /* 01B4 */ int unknown_01B4;
  /* 01B8 */ int verticalPadding;
  /* 01BC */ std::vector<EditLine> lines;
  /* 01C8 */ unsigned char selectedAll;
  /* 01CC */ int column;
  /* 01D0 */ int line;
  /* 01D4 */ int unknown_01D4;
  /* 01D8 */ int unknown_01D8;
  /* 01DC */ unsigned char unknown_01DC;
  /* 01E0 */ std::chrono::system_clock::time_point caretStart;
  /* 01E8 */ int scrollMaximum;
  /* 01EC */ int visibleLines;
  /* 01F0 */ int unknown_01F0;
  /* 01F4 */ int scrollPosition;
  /* 01F8 */ int unknown_01F8;
};

class CUIEditBoxEx : public CUIControl {
public:
  explicit CUIEditBoxEx(UIWindowContext& context);
  void Resize(int width, int height) override;
  bool Slot05() override;
  void Paint() override;
  int SendMsg(UIWindow* sender, int message, std::intptr_t a, std::intptr_t b, std::intptr_t c, std::intptr_t d) override;

  void Create(int width, int height, bool withScrollBar);
  void ResetSelection();
  void SetCursor(std::size_t position);
  int Length();
  void SetAlignment(int value);
  void SetBackgroundColor(std::uint32_t color);
  void SetTextColor(std::uint32_t color);
  void SetHighlight(int value);
  void SetBorderColor(std::uint32_t color);
  void SetBorderWidth(int width);
  void SetMaximum(int value);
  void SetEditBounds(int left, int top, int right, int bottom);
  void SetMask(char value);
  void SetText(const std::string& text);
  void SetFocus();
  void SetOnSubmit(std::function<void()> handler);

  /* 0080 */ std::uint32_t backgroundColor;
  /* 0084 */ std::uint32_t borderColor;
  /* 0088 */ int borderWidth;
  /* 008C */ int editLeft;
  /* 0090 */ int editTop;
  /* 0094 */ int editRight;
  /* 0098 */ int editBottom;
  /* 009C */ CUIEditCtrlEx* edit;
  /* 00A0 */ std::function<void()> onSubmit;
};

#endif