#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>
#include "quickjs.h"
#include "quickjs-libc.h"

#ifdef _WIN32
#include <Windows.h>
#include <direct.h>
static JSValue js_os_exec_win(JSContext *ctx, JSValue this_val, int argc, JSValue *argv) {
	JSValue args = argv[0];
	JSValue val = JS_GetPropertyStr(ctx, args, "length");

	uint32_t exec_argc;
	JS_ToUint32(ctx, &exec_argc, val);

	char **exec_argv = js_mallocz(ctx, sizeof(exec_argv[0]) * (exec_argc + 1));
	for(int i = 0; i < exec_argc; i++) {
		val = JS_GetPropertyUint32(ctx, args, i);
		exec_argv[i] = JS_ToCString(ctx, val);
		JS_FreeValue(ctx, val);
	}
	exec_argv[exec_argc] = NULL;

	if (argc >= 2) {
		JSValue options = argv[1];
		val = JS_GetPropertyStr(ctx, options, "cwd");
		if (!JS_IsUndefined(val)) {
			char *cwd = JS_ToCString(ctx, val);
			JS_FreeValue(ctx, val);
			_chdir(cwd);
		}
	}

	char cmd[1024];
	cmd[0] = 0;
	for (int i = 0; i < exec_argc; ++i) {
		strcat(cmd, exec_argv[i]);
		strcat(cmd, " ");
	}

	HANDLE hReadPipe, hWritePipe;
	SECURITY_ATTRIBUTES saAttr;
	saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
	saAttr.bInheritHandle = TRUE;
	saAttr.lpSecurityDescriptor = NULL;
	CreatePipe(&hReadPipe, &hWritePipe, &saAttr, 0);
	SetHandleInformation(hReadPipe, HANDLE_FLAG_INHERIT, 0);

	STARTUPINFO si;
	PROCESS_INFORMATION pi;
	ZeroMemory(&si, sizeof(si));
	si.cb = sizeof(si);
	si.dwFlags = STARTF_USESTDHANDLES;
	si.hStdOutput = hWritePipe;
	si.hStdError = hWritePipe;
	ZeroMemory(&pi, sizeof(pi));
	CreateProcessA(NULL, cmd, NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi);
	CloseHandle(hWritePipe);

	char buf[4096];
	int buf_len = 0;
	int bytes_read;
	while (ReadFile(hReadPipe, buf + buf_len, 4096 - buf_len - 1, &bytes_read, NULL) && bytes_read > 0) {
		buf_len += bytes_read;
		if (buf_len >= 4096 - 1) {
			break;
		}
	}
	buf[buf_len] = '\0';

	CloseHandle(hReadPipe);
	WaitForSingleObject(pi.hProcess, INFINITE);
	DWORD exit_code;
	GetExitCodeProcess(pi.hProcess, &exit_code);
	CloseHandle(pi.hProcess);
	CloseHandle(pi.hThread);

	printf("%s", buf);

	JSValue result = JS_NewObject(ctx);
	JS_SetPropertyStr(ctx, result, "stdout", JS_NewString(ctx, buf));
	JS_SetPropertyStr(ctx, result, "status", JS_NewInt32(ctx, (int32_t)exit_code));
	return result;
}

#endif

void alang(char *source, char *output);
static JSValue js_alang(JSContext *ctx, JSValue this_val, int argc, JSValue *argv) {
	const char *source = JS_ToCString(ctx, argv[0]);
	const char *output = JS_ToCString(ctx, argv[1]);
	alang(source, output);
	return JS_UNDEFINED;
}

int ashader(char *shader_lang, char *from, char *to);
static JSValue js_ashader(JSContext *ctx, JSValue this_val, int argc, JSValue *argv) {
	const char *shader_lang = JS_ToCString(ctx, argv[0]);
	const char *from = JS_ToCString(ctx, argv[1]);
	const char *to = JS_ToCString(ctx, argv[2]);
	ashader(shader_lang, from, to);
	return JS_UNDEFINED;
}

void export_k(const char *from, const char *to);
JSValue js_export_k(JSContext *ctx, JSValue this_val, int argc, JSValue *argv) {
	const char *from = JS_ToCString(ctx, argv[0]);
	const char *to = JS_ToCString(ctx, argv[1]);
	export_k(from, to);
	return JS_UNDEFINED;
}

void export_ico(const char *from, const char *to);
JSValue js_export_ico(JSContext *ctx, JSValue this_val, int argc, JSValue *argv) {
	const char *from = JS_ToCString(ctx, argv[0]);
	const char *to = JS_ToCString(ctx, argv[1]);
	export_ico(from, to);
	return JS_UNDEFINED;
}

void export_png(const char *from, const char *to, int width, int height);
JSValue js_export_png(JSContext *ctx, JSValue this_val, int argc, JSValue *argv) {
	const char *from = JS_ToCString(ctx, argv[0]);
	const char *to = JS_ToCString(ctx, argv[1]);
	int32_t width;
	JS_ToInt32(ctx, &width, argv[2]);
	int32_t height;
	JS_ToInt32(ctx, &height, argv[3]);
	export_png(from, to, width, height);
	return JS_UNDEFINED;
}

int main(int argc, char **argv) {
	FILE *fp = fopen(argv[1], "rb");
	fseek(fp , 0, SEEK_END);
	int size = ftell(fp);
	rewind(fp);
	char *buffer = malloc(size + 1);
	buffer[size] = 0;
	fread(buffer, size, 1, fp);
	fclose(fp);

	JSRuntime *runtime = JS_NewRuntime();
	JSContext *ctx = JS_NewContext(runtime);

	js_std_init_handlers(runtime);
	js_std_add_helpers(ctx, argc, argv);
	js_init_module_std(ctx, "std");
	js_init_module_os(ctx, "os");

	JSValue global_obj = JS_GetGlobalObject(ctx);
	JSValue amake = JS_NewObject(ctx);
	JS_SetPropertyStr(ctx, amake, "export_k", JS_NewCFunction(ctx, js_export_k, "export_k", 2));
	JS_SetPropertyStr(ctx, amake, "export_ico", JS_NewCFunction(ctx, js_export_ico, "export_ico", 2));
	JS_SetPropertyStr(ctx, amake, "export_png", JS_NewCFunction(ctx, js_export_png, "export_png", 4));
	#ifdef _WIN32
	JS_SetPropertyStr(ctx, amake, "os_exec_win", JS_NewCFunction(ctx, js_os_exec_win, "os_exec_win", 1));
	#endif
	JS_SetPropertyStr(ctx, amake, "alang", JS_NewCFunction(ctx, js_alang, "alang", 2));
	JS_SetPropertyStr(ctx, amake, "ashader", JS_NewCFunction(ctx, js_ashader, "ashader", 3));

	JS_SetPropertyStr(ctx, global_obj, "amake", amake);
	JS_FreeValue(ctx, global_obj);

	JSValue ret = JS_Eval(ctx, buffer, size, "make.js", JS_EVAL_TYPE_MODULE);

	if (JS_IsException(ret)) {
		js_std_dump_error(ctx);
		JS_ResetUncatchableError(ctx);
	}

	JS_RunGC(runtime);
	free(buffer);
	return 0;
}