#include "driver/driver.h"
#include <camellia/flags.h>
#include <camellia/syscalls.h>
#include <stdint.h>
#include <stdio.h>
#include <unistd.h>
#include <user/lib/fs/misc.h>

#define die(fmt, ...) do { fprintf(stderr, "init: " fmt, __VA_ARGS__); exit(1); } while (0)

void redirect(const char *exe, const char *out, const char *in) {
	if (!fork()) {
		if (!freopen(out, "a+", stderr)) {
			fprintf(stdout, "couldn't open %s\n", out);
			exit(1);
		}
		if (!freopen(out, "a+", stdout))
			die("couldn't open %s\n", out);
		if (!freopen(in, "r", stdin))
			die(" couldn't open %s\n", in);

		for (;;) {
			if (!fork()) {
				termcook();
				execv(exe, NULL);
				fprintf(stderr, "init: couldn't start %s\n", exe);
				_syscall_sleep(5000);
				exit(1);
			}
			_syscall_await();
		}
	}
}

int main(void) {
	handle_t killswitch_pipe[2];

	freopen("/kdev/com1", "a+", stdout);
	freopen("/kdev/com1", "a+", stderr);
	printf("in init (stage 2), main at 0x%x\n", &main);

	MOUNT_AT("/keyboard") { ps2_drv(); }
	MOUNT_AT("/bin/") {
		fs_union((const char*[]){
			"/init/bin/amd64/",
			"/init/bin/sh/",
			"/init/usr/bin/",
			NULL
		});
	}
	MOUNT_AT("/Users/") {
		MOUNT_AT("/tmp/") {
			const char *argv[] = {"/bin/tmpfs", NULL};
			execv(argv[0], (void*)argv);
		}
		// TODO a simple union isn't enough here
		fs_union((const char*[]){
			"/tmp/",
			"/init/Users/",
			NULL
		});
	}
	MOUNT_AT("/tmp/") {
		const char *argv[] = {"/bin/tmpfs", NULL};
		execv(argv[0], (void*)argv);
	}
	MOUNT_AT("/vtty") {
		const char *argv[] = {"/bin/vterm", NULL};
		execv(argv[0], (void*)argv);
	}
	MOUNT_AT("/net/") {
		const char *argv[] = {"/bin/netstack", "/kdev/eth", "192.168.0.11", "192.168.0.2", NULL};
		execv(argv[0], (void*)argv);
	}

	if (_syscall_pipe(killswitch_pipe, 0) < 0) {
		printf("couldn't create the killswitch pipe, quitting...\n");
		return 1;
	}
	MOUNT_AT("/initctl") {
		close(killswitch_pipe[0]);
		initctl_drv(killswitch_pipe[1]);
	}
	close(killswitch_pipe[1]);

	if (!fork()) {
		// TODO close on exec
		close(killswitch_pipe[0]);
		redirect("/bin/shell", "/kdev/com1", "/kdev/com1");
		redirect("/bin/shell", "/vtty", "/keyboard");
		// TODO busy loop without no children
		for (;;) _syscall_await();
		exit(1);
	}

	_syscall_read(killswitch_pipe[0], NULL, 0, 0);
	return 0;
}