#include "general.h"
#include "utils.h"

Napi::ThreadSafeFunction removalCB_function = nullptr;

Napi::Boolean wrap_vJoyEnabled(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	bool enabled = vJoyEnabled();

	return Napi::Boolean::New(env, enabled);
}

Napi::Number wrap_GetvJoyVersion(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	short version = GetvJoyVersion();

	return Napi::Number::New(env, (double)version);
}

Napi::String wrap_GetvJoyProductString(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	PVOID p_prodString = GetvJoyProductString();

	std::string prodString = utf8_encode((PWSTR)p_prodString);

	CoTaskMemFree(p_prodString);

	return Napi::String::New(env, prodString);
}

Napi::String wrap_GetvJoyManufacturerString(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	PVOID p_manuString = GetvJoyManufacturerString();

	std::string manuString = utf8_encode((PWSTR)p_manuString);

	CoTaskMemFree(p_manuString);

	return Napi::String::New(env, manuString);
}

Napi::String wrap_GetvJoySerialNumberString(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	PVOID p_serialString = GetvJoySerialNumberString();

	std::string serialString = utf8_encode((PWSTR)p_serialString);

	CoTaskMemFree(p_serialString);

	return Napi::String::New(env, serialString);
}

Napi::Object wrap_DriverMatch(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	WORD DllVersion;
	WORD DrvVersion;

	bool matches = DriverMatch(&DllVersion, &DrvVersion);

	Napi::Boolean var_matches = Napi::Boolean::New(env, matches);
	Napi::Number var_dllversion = Napi::Number::New(env, (double)DllVersion);
	Napi::Number var_drvversion = Napi::Number::New(env, (double)DrvVersion);

	Napi::Object res = Napi::Object::New(env);

	res.Set("matches", var_matches);
	res.Set("dll_version", var_dllversion);
	res.Set("drv_version", var_drvversion);

	return res;
}

Napi::Number wrap_GetvJoyMaxDevices(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	int maxDevices;

	GetvJoyMaxDevices(&maxDevices);

	return Napi::Number::New(env, (double)maxDevices);
}

Napi::Number wrap_GetNumberExistingVJD(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	int existingVJD;

	GetNumberExistingVJD(&existingVJD);

	return Napi::Number::New(env, (double)existingVJD);
}

void ConfChangedCB(BOOL Removed, BOOL First, PVOID _data) {
	RemovalCBData* data = new RemovalCBData{ Removed == TRUE, First == TRUE };

	removalCB_function.NonBlockingCall(data, [](Napi::Env env, Napi::Function fn, RemovalCBData* data) {
		Napi::Object evt = Napi::Object::New(env);
		evt.Set("removed", Napi::Boolean::New(env, data->removed));
		evt.Set("first", Napi::Boolean::New(env, data->first));

		// pass data to JS callback
		fn.Call( {evt} );

		delete data; // clean up
	});
}

// This function leaks memory if called more than once which is why it's not exposed to the users
void wrap_RegisterRemovalCB(const Napi::CallbackInfo& info) {
	Napi::Env env = info.Env();

	if (!info[0].IsFunction()) {
		Napi::Error::New(env, "The first parameter needs to be a function").ThrowAsJavaScriptException();
		return;
	}

	Napi::Function callback = info[0].As<Napi::Function>();
	removalCB_function = Napi::ThreadSafeFunction::New(env, callback, "removalCB", 0, 1);

	RegisterRemovalCB(&ConfChangedCB, nullptr);
}