#include "ScriptEngine.hpp"
#include <duktape.h>


struct DuktapeEngine : ScriptEngine {
	duk_context* ctx = NULL;

	~DuktapeEngine() {
		if (ctx)
			duk_destroy_heap(ctx);
	}

	std::string getEngineName() override {
		return "JavaScript";
	}

	int run(const std::string& path, const std::string& script) override {
		assert(!ctx);
		ProcessBlock* block = getProcessBlock();

		// Create duktape context
		ctx = duk_create_heap_default();
		if (!ctx) {
			display("Could not create duktape context");
			return -1;
		}

		// Initialize globals
		// user pointer
		duk_push_pointer(ctx, this);
		duk_put_global_string(ctx, DUK_HIDDEN_SYMBOL("engine"));

		// console
		duk_idx_t consoleIdx = duk_push_object(ctx);
		{
			// log
			duk_push_c_function(ctx, native_console_log, 1);
			duk_put_prop_string(ctx, consoleIdx, "log");
		}
		duk_put_global_string(ctx, "console");

		// display
		duk_push_c_function(ctx, native_display, 1);
		duk_put_global_string(ctx, "display");

		// config: Set defaults
		duk_idx_t configIdx = duk_push_object(ctx);
		{
			// frameDivider
			duk_push_int(ctx, 32);
			duk_put_prop_string(ctx, configIdx, "frameDivider");
			// bufferSize
			duk_push_int(ctx, 1);
			duk_put_prop_string(ctx, configIdx, "bufferSize");
		}
		duk_put_global_string(ctx, "config");

		// Compile string
		duk_push_string(ctx, path.c_str());
		if (duk_pcompile_lstring_filename(ctx, 0, script.c_str(), script.size()) != 0) {
			const char* s = duk_safe_to_string(ctx, -1);
			WARN("duktape: %s", s);
			display(s);
			duk_pop(ctx);
			return -1;
		}
		// Execute function
		if (duk_pcall(ctx, 0)) {
			const char* s = duk_safe_to_string(ctx, -1);
			WARN("duktape: %s", s);
			display(s);
			duk_pop(ctx);
			return -1;
		}
		// Ignore return value
		duk_pop(ctx);

		// config: Read values
		duk_get_global_string(ctx, "config");
		{
			// frameDivider
			duk_get_prop_string(ctx, -1, "frameDivider");
			setFrameDivider(duk_get_int(ctx, -1));
			duk_pop(ctx);
			// bufferSize
			duk_get_prop_string(ctx, -1, "bufferSize");
			setBufferSize(duk_get_int(ctx, -1));
			duk_pop(ctx);
		}
		duk_pop(ctx);

		// Keep process function on stack for faster calling
		duk_get_global_string(ctx, "process");
		if (!duk_is_function(ctx, -1)) {
			display("No process() function");
			return -1;
		}

		// block (keep on stack)
		duk_idx_t blockIdx = duk_push_object(ctx);
		{
			// inputs
			duk_idx_t inputsIdx = duk_push_array(ctx);
			for (int i = 0; i < NUM_ROWS; i++) {
				duk_push_external_buffer(ctx);
				duk_config_buffer(ctx, -1, block->inputs[i], sizeof(float) * block->bufferSize);
				duk_push_buffer_object(ctx, -1, 0, sizeof(float) * block->bufferSize, DUK_BUFOBJ_FLOAT32ARRAY);
				duk_put_prop_index(ctx, inputsIdx, i);
				duk_pop(ctx);
			}
			duk_put_prop_string(ctx, blockIdx, "inputs");

			// outputs
			duk_idx_t outputsIdx = duk_push_array(ctx);
			for (int i = 0; i < NUM_ROWS; i++) {
				duk_push_external_buffer(ctx);
				duk_config_buffer(ctx, -1, block->outputs[i], sizeof(float) * block->bufferSize);
				duk_push_buffer_object(ctx, -1, 0, sizeof(float) * block->bufferSize, DUK_BUFOBJ_FLOAT32ARRAY);
				duk_put_prop_index(ctx, outputsIdx, i);
				duk_pop(ctx);
			}
			duk_put_prop_string(ctx, blockIdx, "outputs");

			// knobs
			duk_push_external_buffer(ctx);
			duk_config_buffer(ctx, -1, block->knobs, sizeof(float) * NUM_ROWS);
			duk_push_buffer_object(ctx, -1, 0, sizeof(float) * NUM_ROWS, DUK_BUFOBJ_FLOAT32ARRAY);
			duk_put_prop_string(ctx, blockIdx, "knobs");
			duk_pop(ctx);

			// switches
			duk_push_external_buffer(ctx);
			duk_config_buffer(ctx, -1, block->switches, sizeof(bool) * NUM_ROWS);
			duk_push_buffer_object(ctx, -1, 0, sizeof(bool) * NUM_ROWS, DUK_BUFOBJ_UINT8ARRAY);
			duk_put_prop_string(ctx, blockIdx, "switches");
			duk_pop(ctx);

			// lights
			duk_idx_t lightsIdx = duk_push_array(ctx);
			for (int i = 0; i < NUM_ROWS; i++) {
				duk_push_external_buffer(ctx);
				duk_config_buffer(ctx, -1, block->lights[i], sizeof(float) * 3);
				duk_push_buffer_object(ctx, -1, 0, sizeof(float) * 3, DUK_BUFOBJ_FLOAT32ARRAY);
				duk_put_prop_index(ctx, lightsIdx, i);
				duk_pop(ctx);
			}
			duk_put_prop_string(ctx, blockIdx, "lights");

			// switchLights
			duk_idx_t switchLightsIdx = duk_push_array(ctx);
			for (int i = 0; i < NUM_ROWS; i++) {
				duk_push_external_buffer(ctx);
				duk_config_buffer(ctx, -1, block->switchLights[i], sizeof(float) * 3);
				duk_push_buffer_object(ctx, -1, 0, sizeof(float) * 3, DUK_BUFOBJ_FLOAT32ARRAY);
				duk_put_prop_index(ctx, switchLightsIdx, i);
				duk_pop(ctx);
			}
			duk_put_prop_string(ctx, blockIdx, "switchLights");
		}

		return 0;
	}

	int process() override {
		// block
		ProcessBlock* block = getProcessBlock();
		duk_idx_t blockIdx = duk_get_top(ctx) - 1;
		{
			// sampleRate
			duk_push_number(ctx, block->sampleRate);
			duk_put_prop_string(ctx, blockIdx, "sampleRate");

			// sampleTime
			duk_push_number(ctx, block->sampleTime);
			duk_put_prop_string(ctx, blockIdx, "sampleTime");

			// bufferSize
			duk_push_int(ctx, block->bufferSize);
			duk_put_prop_string(ctx, blockIdx, "bufferSize");
		}

		// Duplicate process function
		duk_dup(ctx, -2);
		// Duplicate block object
		duk_dup(ctx, -2);
		// Call process function
		if (duk_pcall(ctx, 1)) {
			const char* s = duk_safe_to_string(ctx, -1);
			WARN("duktape: %s", s);
			display(s);
			duk_pop(ctx);
			return -1;
		}
		// return value
		duk_pop(ctx);

		return 0;
	}

	static DuktapeEngine* getDuktapeEngine(duk_context* ctx) {
		duk_get_global_string(ctx, DUK_HIDDEN_SYMBOL("engine"));
		DuktapeEngine* engine = (DuktapeEngine*) duk_get_pointer(ctx, -1);
		duk_pop(ctx);
		return engine;
	}

	static duk_ret_t native_console_log(duk_context* ctx) {
		const char* s = duk_safe_to_string(ctx, -1);
		INFO("Duktape: %s", s);
		return 0;
	}
	static duk_ret_t native_display(duk_context* ctx) {
		const char* s = duk_safe_to_string(ctx, -1);
		getDuktapeEngine(ctx)->display(s);
		return 0;
	}
};


__attribute__((constructor(1000)))
static void constructor() {
	addScriptEngine<DuktapeEngine>("js");
}