#pragma once
#include "Helpers/ConvertVec.hpp"
#include <include/core/SkCanvas.h>
#include "../../TimePoint.hpp"
#include <Eigen/Dense>
#include <include/core/SkFont.h>
#include <include/core/SkFontMgr.h>
#include <include/core/SkTypeface.h>
#include "../../FontData.hpp"
#include <modules/skparagraph/include/FontCollection.h>
#include <modules/svg/include/SkSVGDOM.h>
#include <Helpers/SCollision.hpp>
#include <Helpers/Serializers.hpp>
#include <nlohmann/json.hpp>
#include "../../RichText/TextBox.hpp"
#include "../../InputManager.hpp"

using namespace Eigen;

namespace GUIStuff {

// Taken from https://github.com/TimothyHoytBSME/ClayMan (rewritten to be a separate struct, and use templates to change size)
template <size_t S> class StringArena {
    public:
        void reset() {
            stringArenaIndex = 0;
        }
        const char* insert_string_into_arena(std::string_view str) {
            size_t strSize = str.size();
            if(stringArenaIndex + strSize > S)
                throw std::overflow_error("[StringArena::insert_string_into_arena] Not enough space to insert string");
            char* startPtr = &stringArena[stringArenaIndex];
            std::copy(str.begin(), str.end(), startPtr);
            stringArenaIndex += strSize;
            return startPtr;
        }
        Clay_String std_str_to_clay_str(std::string_view str) {
            return Clay_String{.length = static_cast<int32_t>(str.size()), .chars = insert_string_into_arena(str)};
        }
    private:
        std::array<char, S> stringArena;
        size_t stringArenaIndex;
};

typedef StringArena<4000000> DefaultStringArena;

SkFont get_setup_skfont();

struct Theme {
    SkColor4f fillColor1 = {0.8f, 0.785f, 1.0f, 1.0f};
    SkColor4f fillColor2 = {0.3f, 0.3f, 0.477f, 1.0f};
    SkColor4f backColor1 = {0.0f, 0.0f, 0.0f, 1.0f};
    SkColor4f backColor2 = {1.0f, 1.0f, 1.0f, 1.0f};
    SkColor4f frontColor1 = {1.0f, 1.0f, 1.0f, 1.0f};
    SkColor4f frontColor2 = {0.9f, 0.9f, 0.9f, 1.0f};

    SkColor4f errorColor = {1.0f, 0.0f, 0.0f, 1.0f};
    SkColor4f warningColor = {1.0f, 1.0f, 0.0f, 1.0f};

    float hoverExpandTime = 0.1f;
    uint16_t childGap1 = 10;
    uint16_t padding1 = 8;
    float windowCorners1 = 8;

    NLOHMANN_DEFINE_TYPE_INTRUSIVE_WITH_DEFAULT(Theme, fillColor1, fillColor2, backColor1, backColor2, frontColor1, frontColor2, errorColor, warningColor, hoverExpandTime, childGap1, padding1, windowCorners1)
};

std::shared_ptr<Theme> get_default_dark_mode();

struct UpdateInputData {
    struct {
        int leftClick = 0;
        int rightClick = 0;
        bool leftDoubleClick = false;
        bool leftHeld = false;
        bool rightHeld = false;
        Vector2f globalPos{0, 0};
        Vector2f pos{0, 0}; // Position relative to window 
        Vector2f scroll;
    } mouse;

    struct {
        bool left = false;
        bool right = false;
        bool up = false;
        bool down = false;
        bool leftShift = false;
        bool leftCtrl = false;
        bool enter = false;
        bool escape = false;
    } key;

    struct {
        std::function<std::string()> textInFunc;
        std::optional<std::string> textOut;
    } clipboard;

    std::shared_ptr<RichText::TextBox> previousRichTextBoxToEdit;
    std::shared_ptr<RichText::TextBox> richTextBoxToEdit;
    std::shared_ptr<RichText::TextBox::Cursor> richTextBoxToEditCursor;
    InputManager::TextInputProperties richTextInputProperties;
    std::shared_ptr<SCollision::AABB<float>> richTextBoxToEditRectangle;

    SkFont get_font(float fSize) const;

    uint16_t fontSize = 16;

    std::string textInput;
    bool hoverObstructed = false;
    std::vector<SCollision::AABB<float>> hoverObstructingAABBs;
    std::vector<SCollision::Circle<float>> hoverObstructingCircles;
    bool acceptingTextInput = false;
    float deltaTime = 0.0f;
    bool isTouch = true;

    float guiScaleMultiplier = 1.0f;
    Vector2f windowPos = {0.0f, 0.0f};

    std::shared_ptr<FontData> fonts;
    sk_sp<SkTypeface> textTypeface;

    DefaultStringArena* strArena;
    std::shared_ptr<Theme> theme;
    std::unordered_map<std::string, sk_sp<SkSVGDOM>> svgData;
};

class SelectionHelper {
    public:
        void update(bool isHovering, bool isLeftClick, bool isLeftHeld, const Vector2f& cursorPos);
        bool held = false;
        bool hovered = false;
        bool selected = false;
        bool clicked = false;
        bool justUnselected = false;
        bool tapped = false;
        Vector2f clickPos;
        std::chrono::steady_clock::time_point timeClicked;
};

class Element {
    public:
        virtual void clay_draw(SkCanvas* canvas, UpdateInputData& io, Clay_RenderCommand* command, bool skiaAA) = 0;
        virtual ~Element() = default;

    protected:
        static SCollision::AABB<float> get_bb(Clay_RenderCommand* command);
};

}