#pragma once
#include <JuceHeader.h>
#include "../lua/LuaParser.h"
#include "../PluginProcessor.h"

class ErrorCodeEditorComponent : public juce::CodeEditorComponent, public ErrorListener {
 public:
    ErrorCodeEditorComponent(juce::CodeDocument& document, juce::CodeTokeniser* codeTokeniser, OscirenderAudioProcessor& p, juce::String fileName) : juce::CodeEditorComponent(document, codeTokeniser), audioProcessor(p), document(document), fileName(fileName) {
        audioProcessor.addErrorListener(this);
    }

    ~ErrorCodeEditorComponent() override {
        audioProcessor.removeErrorListener(this);
    }

    void paint(juce::Graphics& g) override {
        juce::CodeEditorComponent::paint(g);
        if (errorLine != -1) {
            int firstVisibleLine = getFirstLineOnScreen();
            int lastVisibleLine = firstVisibleLine + getNumLinesOnScreen();
            if (errorLine < firstVisibleLine || errorLine > lastVisibleLine) {
                return;
            }
            
            auto lineBounds = getErrorLineBounds();

            // draw error line
            g.setColour(juce::Colours::red);
            // draw red squiggly line
            juce::Path path;

            double xOffset = 0;
            double yOffset = 2;
            
            double lineIncrement = 2.5;
            double squiggleHeight = 3;

            juce::String line = document.getLine(errorLine - 1);
            // get number of leading whitespace characters
            int leadingWhitespace = line.length() - line.trimStart().length();
            double start = getCharWidth() * leadingWhitespace;
            double width = getCharWidth() * line.length() - getCharWidth();

            double squiggleMax = lineBounds.getY() + lineBounds.getHeight() - squiggleHeight + yOffset;
            double squiggleMin = lineBounds.getY() + lineBounds.getHeight() + yOffset;

            path.startNewSubPath(lineBounds.getX() + start + xOffset, squiggleMax);

            for (double i = start; i < width - xOffset; i += 2 * lineIncrement) {
                path.lineTo(lineBounds.getX() + i + xOffset, squiggleMax);
                path.lineTo(lineBounds.getX() + i + lineIncrement + xOffset, squiggleMin);
            }
            g.strokePath(path, juce::PathStrokeType(1.0f));


            if (errorLineHovered) {
                // draw error text on line below the one being hovered
                auto bounds = juce::Rectangle<int>(lineBounds.getX(), lineBounds.getY() + lineBounds.getHeight(), lineBounds.getWidth(), lineBounds.getHeight());

                // draw over existing line
                g.setColour(juce::Colours::red);
                g.fillRect(bounds);

                // draw text
                g.setColour(juce::Colours::white);
                g.setFont(12);
                g.drawText(errorText, bounds, juce::Justification::centred);
            }
        }
    }

    void mouseMove(const juce::MouseEvent& event) override {
        juce::CodeEditorComponent::mouseMove(event);
        if (errorLine != -1) {
            errorLineHovered = getErrorLineBounds().contains(event.getPosition());
            repaint(getErrorRepaintBounds());
        }
    }

    juce::Rectangle<int> getErrorLineBounds() {
        int firstVisibleLine = getFirstLineOnScreen();
        double gutterWidth = 35;

        return juce::Rectangle<int>(gutterWidth, getLineHeight() * (errorLine - 1 - firstVisibleLine), getWidth() - gutterWidth, getLineHeight());
    }

    juce::Rectangle<int> getErrorRepaintBounds() {
        auto lineBounds = getErrorLineBounds();
        return juce::Rectangle<int>(lineBounds.getX(), lineBounds.getY(), lineBounds.getWidth(), lineBounds.getHeight() * 2);
    }

    bool keyPressed(const juce::KeyPress& key) override {
        repaint();
        return juce::CodeEditorComponent::keyPressed(key);
    }

private:
    void onError(int lineNumber, juce::String error) override {
        errorLine = lineNumber;
        errorText = error;
    }

    juce::String getFileName() override {
        return fileName;
    }

    juce::CodeDocument& document;
    juce::String fileName;
    int errorLine = -1;
    juce::String errorText;
    bool errorLineHovered = false;
    OscirenderAudioProcessor& audioProcessor;
};