#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <stdlib.h>
#include "parser.h"
#include "translator.h"

char* verfname(char* fname) {
	int len = strlen(fname);
	if(len > 3) {
		int extind = len - 3;
		char* extstr = fname + (sizeof(char)*extind);
		if(strcmp(extstr, ".vm") == 0) {
			int startind = 0;
			for(int i = extind-1; i >= 0; i--) {
				if(fname[i] == '/') {
					startind = i+1;
					break;
				}
			}
			int size = sizeof(char)*(extind - startind + 1);
			char* startstr = fname + (sizeof(char)*startind);
			char* retstr = (char*)malloc(size);
			snprintf(retstr, size, "%s", startstr);
			return retstr;
		}
	}
	fprintf(stderr, "Name format must be Xxx.vm\n");
	exit(1);
}

char* getoutname(char* fname) {
	int len = strlen(fname);
	int sz = sizeof(char)*(len+2);
	char* outname = (char*)malloc(sz);
	strcpy(outname, fname);
	int ind = len-2;
	char* ext = sizeof(char)*ind+outname;
	snprintf(ext, 4, "asm");
	return outname;
}

int main(int argc, char* argv[]) {
	if(argc < 2) {
		fprintf(stderr, "Usage: %s {file}\n", argv[0]);
		return 1;
	}
	// file name validating
	char* fname = verfname(argv[1]);
	
	FILE* input = fopen(argv[1], "r");

	if(input == NULL) {
		fprintf(stderr, "%s\n", strerror(errno));
		return errno;
	}

	// parsing
	struct Parser* p = mkparser(input);
	parse(p);

	// translating
	struct Translator* t = mktranslator(p->lns, fname);
	translate(t);
	freeparser(p);
	free(fname);
	
	// output
	char* outname = getoutname(argv[1]);
	
	FILE* output = fopen(outname, "w");
	printasmlns(t, output);
	fclose(output);

	// freeing rest
	free(outname);
	freetranslator(t);

	return 0;
}