/*
 * Copyright © 2024 Matthew Wozniak <me@woz.blue>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED “AS IS” AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
 * OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
 * PERFORMANCE OF THIS SOFTWARE.
 */

#include "api.h"
#include "hook.h"
#include "log.h"
#include "render.h"
#include "os.h"

#include "ms/getopt.h"

#include <stddef.h>
#define WIN32_LEAN_AND_MEAN
#include <Windows.h>

struct {
	int width;
	int height;
	char *game;
	int fps;
	int quality;
	int bitrate;
	char *out;
	char **demo;
	bool combine;
} args = {0};

void (*orig_cbuf_addtext)(char *);
void hook_cbuf_addtext(char *str) {
	orig_cbuf_addtext(str);
	// this is the last thing that happens when the game is opened
	if (!strcmp(str, "exec modsettings.cfg mod\n")) {
		bool use_bitrate = !!args.bitrate;
		if (!render_init(args.width, args.height, args.fps, use_bitrate,
					use_bitrate ? args.bitrate : args.quality, args.out)) 
			die("couldn't init render");
		// play the demo
		demoplayer->vt->start_playback(demoplayer, args.demo[0], false);
	}
}

char cmdline[128];
char WINAPI *hook_GetCommandLineA(void) {
	return cmdline;
}

void *(WINAPI *orig_LoadLibraryExA)(const char *, void *, int);
void WINAPI *hook_LoadLibraryExA(const char *filename, void *hfile, int flags) {
	// if the dll is already loaded, don't run our code again
	if (os_dlhandle(filename))
		return orig_LoadLibraryExA(filename, hfile, flags);
	void *ret = orig_LoadLibraryExA(filename, hfile, flags);
	if (!ret) return ret;
	// cut down to basename for display
	const char *basename = filename;
	for (const char *p = filename; *p; p++)
		if (*p == '\\') basename = p + 1;

	if (!strcmp(basename, "engine.dll")) {
		if (!api_init()) die("couldn't get apis");
		orig_cbuf_addtext = (void (*)(char *))
			hook_inline((void *)cbuf_addtext, (void *)hook_cbuf_addtext);
	}
	return ret;
}

typedef int (*LauncherMain_t)(void *instance, void *prev_inst, char *cmdline,
		int cmd_show);

int main(int argc, char **argv) {
	SetDllDirectoryA("bin/");
	int c;
	char *strend;
	while ((c = getopt(argc, argv, "w:h:g:r:q:b:")) != -1) {
		switch (c) {
		case 'w':
			args.width = strtol(optarg, &strend, 10);
			if (strend == optarg) die("width must be a number!");
			break;
		case 'h':
			args.height = strtol(optarg, &strend, 10);
			if (strend == optarg) die("height must be a number");
			break;
		case 'r':
			args.fps = strtol(optarg, &strend, 10);
			if (strend == optarg) die("must pass a number to -w!");
			break;
		case 'q':
			args.quality = strtol(optarg, &strend, 10);
			if (strend == optarg) die("must pass a number to -w!");
			break;
		case 'b':
			args.bitrate = strtol(optarg, &strend, 10);
			if (strend == optarg) die("must pass a number to -w!");
			break;
		case 'g':
			args.game = optarg;
		case '1':
			args.combine = true;
		case '?':
			break;
		}
	}

	if (!args.width) args.width = 1280;
	if (!args.height) args.height = 720;
	if (!args.game) args.game = "hl2";
	if (!args.fps) args.fps = 30;
	if (!args.quality) args.quality = 75;

	if (argc - optind < 2) {
		printf(
			"usage:\n"
			"   rt -w <width> -h <height> -g <game> -r <fps>  [-q <quality>] [-b <bitrate>]\n"
			"   path/to/video.mp4 path/to/demo1.dem...\n");
		exit(1);
	}
	args.out = argv[optind++];
	args.demo = argv + optind;

	sprintf(cmdline, "hl2.exe -game %s -w %d -h %d -window -console",
			args.game, args.width, args.height);

	info("cmdline = %s", cmdline);

	hook_init();
	orig_LoadLibraryExA = (typeof(orig_LoadLibraryExA))hook_dllapi("kernel32",
			"LoadLibraryExA", (void *)hook_LoadLibraryExA);
	hook_dllapi("kernel32", "GetCommandLineA", (void *)hook_GetCommandLineA);

	void *launcher_dll = os_dlopen("launcher");
	LauncherMain_t launcher_main =
		(LauncherMain_t)os_dlsym(launcher_dll, "LauncherMain");

	if (!launcher_main) die("couldn't open launcher");
	launcher_main(NULL, NULL, cmdline, 0);
}

// vi: sw=4 ts=4 noet tw=80 cc=80