#include <stdio.h>
#include "asm_rules.c"

void bin_to_hex(char *tryte, int *hex) {
	char *a = tryte;
	long num = 0;
	do {
		long b = *a=='1'?1:0;
		num = (num<<1)|b;
		a++;
	} while (*a);
	*hex = num;

}
void asm_to_hex(char *_asm, int *opcode, int *in0, int *in1, int *out){

	char tryte[108] = "\0";
	char t0[26] = "\0"; 
	char t1[26] = "\0"; 
	char t2[26] = "\0"; 
	char t3[26] = "\0"; 

	int size = sizeof(_asm)/sizeof(_asm[0]);
	//convert assembly to formated binary representation of ternary
	for (int i = 0; i < sizeof(_asm)/4; i++) {
		
		//printf("%c",_asm[i]);
		if(_asm[i]=='-') {
			strcat(tryte, "10");
		} else if (_asm[i]=='+') {
			strcat(tryte, "01");
		} else {
			strcat(tryte, "00");
		}
	}

	bin_to_hex(t0, opcode);
	bin_to_hex(t1, in0);
	bin_to_hex(t2, in1);
	bin_to_hex(t3, out);
}


int main(int argc, char **argv) {

	FILE *dest_asm;
	FILE *src_asm;
	if (argc > 2) {
		for (int i = 1 ; i < argc; i++) {
			if (strcmp("-i", argv[i]) == 0 || strcmp("--input", argv[i]) == 0) {
				src_asm = fopen(argv[i+1], "r");
				i=i+1;
			} else if (strcmp("-o", argv[i]) == 0 || strcmp("--output", argv[i]) == 0) {
				dest_asm = fopen(argv[i+1], "w");
				i=i+1;
			} else if (strcmp("-h", argv[i]) == 0 || strcmp("--help", argv[i]) == 0) {
				printf("Usage: tasm [options] file...\n");
				printf("Options:\n");
				printf("-h,--help		Prints this information\n");
				printf("-i,--input <arg> 	Path of input file\n");
				printf("-o,--output <arg>	Path of output file\n");
				return 0; 
			} else {
				printf("unknown argument: %s\n", argv[i]);
				return 1;
			}
		}
	} else {
		printf("tasm: no input file\n");
		return 1;
	}

	unsigned int size=0; //keeps the size of the assembled program
	char in_asm[64];
	char out_asm[27];

	int hex_op;
	int hex_in0;
	int hex_in1;
	int hex_out;

	while(fgets(in_asm, sizeof(in_asm), src_asm)) {
		hex_op = 0;
		hex_in0 = 0;
		hex_in1 = 0;
		hex_out = 0;

		if (in_asm[strlen(in_asm)-1] == '\n') {
			in_asm[strlen(in_asm)-1] = '\0'; //cleans newline from string
		}

		if (!(in_asm[0] == '\0')){ // if in_asm is an empty, skip
			asm_rules(in_asm, out_asm);
			asm_to_hex(out_asm, &hex_op, &hex_in0, &hex_in1, &hex_out);

			printf("%.5x %.5x %.5x %.5x\n", hex_op, hex_in0, hex_in1, hex_out);

			fprintf(dest_asm,"%.5x ", hex_op);
			fprintf(dest_asm,"%.5x ", hex_in0);
			fprintf(dest_asm,"%.5x ", hex_in1);
			fprintf(dest_asm,"%.5x \n", hex_out);
			// Write the hex to a file

			size = size + 3; //1 word = 3trytes
		}

	}

	printf("\n%u trytes\n\n", size);


	// Close the file
	fclose(dest_asm);
	fclose(src_asm);

	return 0;
}