#pragma once

#include <Geode/DefaultInclude.hpp>
#include <cocos2d.h>

namespace geode {
    enum WrappingMode {
        NO_WRAP,
        WORD_WRAP,
        CUTOFF_WRAP
    };

    /**
     * A class which provides a textarea with proper alignment and some extra features like:
     * 
     * - Max lines
     * - Changing all aspects after creation
     * - Custom text alignment
     * - Configurable and automatic word wrapping
     * - Line padding
     * 
     * Contact me on Discord (\@smjs) if you have any questions, suggestions or bugs.
    */
    class GEODE_DLL SimpleTextArea : public cocos2d::CCNode {
        static SimpleTextArea* create(const std::string& text, const std::string& font = "chatFont.fnt", const float scale = 1);
        static SimpleTextArea* create(const std::string& text, const std::string& font, const float scale, const float width);

        cocos2d::CCMenu* m_container;
        std::string m_font;
        std::string m_text;
        std::vector<cocos2d::CCLabelBMFont*> m_lines;
        cocos2d::ccColor4B m_color;
        cocos2d::CCTextAlignment m_alignment;
        WrappingMode m_wrappingMode;
        size_t m_maxLines;
        float m_scale;
        float m_lineHeight;
        float m_linePadding;

        void setFont(const std::string& font);
        std::string getFont();
        void setColor(const cocos2d::ccColor4B& color);
        cocos2d::ccColor4B getColor();
        void setAlignment(const cocos2d::CCTextAlignment alignment);
        cocos2d::CCTextAlignment getAlignment();
        void setWrappingMode(const WrappingMode mode);
        WrappingMode getWrappingMode();
        void setText(const std::string& text);
        std::string getText();
        void setMaxLines(const size_t maxLines);
        size_t getMaxLines();
        void setWidth(const float width);
        float getWidth();
        void setScale(const float scale) override;
        float getScale() override;
        void setLinePadding(const float padding);
        float getLinePadding();
        std::vector<cocos2d::CCLabelBMFont*> getLines();
        float getHeight();
        float getLineHeight();
    private:
        static SimpleTextArea* create(const std::string& font, const std::string& text, const float scale, const float width, const bool artificialWidth);

        bool m_shouldUpdate;
        bool m_artificialWidth;

        SimpleTextArea(const std::string& font, const std::string& text, const float scale, const float width, const bool artificialWidth);
        cocos2d::CCLabelBMFont* createLabel(const std::string& text, const float top);
        float calculateOffset(cocos2d::CCLabelBMFont* label);
        void charIteration(const std::function<cocos2d::CCLabelBMFont*(cocos2d::CCLabelBMFont* line, const char c, const float top)>& overflowHandling);
        void updateLinesNoWrap();
        void updateLinesWordWrap();
        void updateLinesCutoffWrap();
        void updateContainer();
        virtual void draw() override;
    };
}