#pragma once
#include <JuceHeader.h>
#include <regex>
#include <numbers>
#include "../shape/Shape.h"

class ErrorListener {
public:
	virtual void onError(int lineNumber, juce::String error) = 0;
	virtual juce::String getId() = 0;
};

const int NUM_SLIDERS = 26;

const char SLIDER_NAMES[NUM_SLIDERS][9] = {
	"slider_a",
	"slider_b",
	"slider_c",
	"slider_d",
	"slider_e",
	"slider_f",
	"slider_g",
	"slider_h",
	"slider_i",
	"slider_j",
	"slider_k",
	"slider_l",
	"slider_m",
	"slider_n",
	"slider_o",
	"slider_p",
	"slider_q",
	"slider_r",
	"slider_s",
	"slider_t",
	"slider_u",
	"slider_v",
	"slider_w",
	"slider_x",
	"slider_y",
	"slider_z",
};

struct LuaVariables {

	double sliders[NUM_SLIDERS] = { 0.0 };

	double step = 1;
	double phase = 0;
	double sampleRate = 0;
	double frequency = 0;

	// x, y, z are only used for effects
	bool isEffect = false;

	double x = 0;
	double y = 0;
	double z = 0;
};

struct lua_State;
struct lua_Debug;
class LuaParser {
public:
	LuaParser(juce::String fileName, juce::String script, std::function<void(int, juce::String, juce::String)> errorCallback, juce::String fallbackScript = "return { 0.0, 0.0 }");

	std::vector<float> run(lua_State*& L, LuaVariables& vars);
	bool isFunctionValid();
	juce::String getScript();
	void resetErrors();
	void close(lua_State*& L);

	static std::function<void(const std::string&)> onPrint;
	static std::function<void()> onClear;

private:
	static void maximumInstructionsReached(lua_State* L, lua_Debug* D);
	
	void reset(lua_State*& L, juce::String script);
	void reportError(const char* error);
	void parse(lua_State*& L);
	void setGlobalVariable(lua_State*& L, const char* name, double value);
	void setGlobalVariable(lua_State*& L, const char* name, int value);
	void setGlobalVariables(lua_State*& L, LuaVariables& vars);
	void incrementVars(LuaVariables& vars);
	void clearStack(lua_State*& L);
	void revertToFallback(lua_State*& L);
	void readTable(lua_State*& L, std::vector<float>& values);
	void setMaximumInstructions(lua_State*& L, int count);
	void resetMaximumInstructions(lua_State*& L);

	int functionRef = -1;
	bool usingFallbackScript = false;
	juce::String script;
	juce::String fallbackScript;
	std::function<void(int, juce::String, juce::String)> errorCallback;
	juce::String fileName;
	std::vector<lua_State*> seenStates;
};