#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include "threads.h"
#include "parser.h"
#include "compiler.h"
#include "io.h"
#include "assembler.h"

int main(int argc, char* argv[]) {
	if(argc < 2) {
		eprintf("Usage: %s {input file(s)}\n", argv[0]);
		return 1;
	}

	FILELIST* files = getfiles(argv[1]);
	FILELIST* curr = files->next;

	COMPILEUNIT* head = (COMPILEUNIT*)malloc(sizeof(COMPILEUNIT));

	head->file = files;
	head->parser = mkparser(tokenize(files->fullname), files->name);

	COMPILEUNIT* currunit = head;
	while(curr != NULL) {
		COMPILEUNIT* newunit = (COMPILEUNIT*)malloc(sizeof(COMPILEUNIT));
		newunit->file = curr;
		newunit->parser = mkparser(tokenize(curr->fullname), curr->name);
		currunit->next = newunit;
		currunit = newunit;
		curr = curr->next;
	}
	currunit->next = NULL;

	actonunits(head, parseunit);

	CLASS* headclass = head->parsed;
	CLASS* currclass = headclass;
	currunit = head->next;
	while(currunit != NULL) {
		currclass->next = currunit->parsed;
		currclass = currunit->parsed;
		currunit = currunit->next;
	}
	currclass->next = NULL;
	COMPILER* compiler = mkcompiler(headclass);

	currunit = head;
	while(currunit != NULL) {
		currunit->compiler = compiler;
		currunit = currunit->next;
	}

	actonunits(head, compileunit);
	actonunits(head, vmtranslateunit);

	ASMBLK* asmlns = head->asmlns;
	currunit = head->next;
	while(currunit != NULL) {
		mergeasmblks(asmlns, currunit->asmlns);
		currunit = currunit->next;
	}

	ASSEMBLER* assembler = mkassembler(asmlns->head);
	preprocess(assembler);
	assemble(assembler);


	char* outname = getouthack(argv[1]);
	FILE* output = fopen(outname, "w");
	if(output == NULL) {
		eprintf("%s", strerror(errno));
		exit(1);
	}

	printstrlist(asmlns->head, output);
	free(asmlns);

	fclose(output);
	free(outname);

	currunit = head;
	while(currunit != NULL) {
		COMPILEUNIT* next = currunit->next;
		freeunit(currunit);
		currunit = next;
	}

	freecompiler(compiler);
	freeassembler(assembler);
	freetree(headclass);
	freefilelist(files);
	return 0;
}