#include "opcode.h"
#include <ctype.h>
#include <string.h>

#define debug 1

#define OPNUM 85
#define SETOP(num, _mne, _IMM, _ZM, _ZMX, _ZMY, _ABS, _IMPL) \
{opcodes[num].mnemonic[3] = '\0'; strncpy(opcodes[num].mnemonic, _mne, 3); \
opcodes[num].imm = _IMM; \
opcodes[num].zm = _ZM; opcodes[num].zmx = _ZMX; opcodes[num].zmy = _ZMY; \
opcodes[num].abs = _ABS; opcodes[num].impl = _IMPL;}


struct fixup {
	struct fixup *nxt;
	struct label *l;
	uint64_t adr;
};

struct label {
	struct label* nxt;
	uint64_t adr;
	uint8_t def;
	char name[1];
};
struct label *labels = 0;
struct fixup *fixups = 0;
uint8_t defined = 0;
struct label *mklabel(const char *name, uint64_t adr, uint8_t def) {
	struct label *l;
	for (l = labels; l; l = l->nxt) {
		if (!strcasecmp(name, l->name)) {
			if (def) {
				if (l->def) {
					printf("oof, you cannot redefine the label: %s\n", name);
					defined = 1;
				} else {
					defined = 0;
				}
				l->def = def;
				l->adr = adr;
			}
			return l;
		}
	}
	l = malloc(sizeof(*l) + strlen(name));
	l->def = def;
	l->adr = adr;
	strcpy(l->name, name);
	l->nxt = labels;
	labels = l;
	defined = 0;
	return l;
}

void use_label(const char *name, uint64_t *adr) {
	struct label *l = mklabel(name, 0, 0);
	if (l->def) {
		*adr = l->adr;
	} else {
		struct fixup *f = malloc(sizeof(*f));
		f->nxt = fixups;
		f->adr = *adr;
		f->l = l;
		fixups = f;
	}
}

int asmmon() {
	opent opcodes[OPNUM];
	/*	  mne	 IMM	ZM   ZMX   ZMY	 ABS  IMPL*/
	SETOP(0, "CPS", 0x00, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(1, "ADC", 0x01, 0x05, 0x00, 0x00, 0x03, 0x00);
	SETOP(2, "PHP", 0x08, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(3, "PHA", 0x09, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(4, "PHY", 0x0A, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(5, "TAY", 0x00, 0x00, 0x00, 0x00, 0x00, 0x0B);
	SETOP(6, "PHX", 0x0C, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(7, "TAX", 0x00, 0x00, 0x00, 0x00, 0x00, 0x0D);
	SETOP(8, "TYX", 0x00, 0x00, 0x00, 0x00, 0x00, 0x0E);
	SETOP(9, "JMP", 0x00, 0xD0, 0x00, 0x00, 0x10, 0x00);
	SETOP(10, "SBC", 0x11, 0x15, 0x00, 0x00, 0x13, 0x00);
	SETOP(11, "PLP", 0x18, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(12, "PLA", 0x19, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(13, "PLY", 0x1A, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(14, "TYA", 0x00, 0x00, 0x00, 0x00, 0x00, 0x1B);
	SETOP(15, "PLX", 0x1C, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(16, "TXA", 0x00, 0x00, 0x00, 0x00, 0x00, 0x1D);
	SETOP(17, "TXY", 0x00, 0x00, 0x00, 0x00, 0x00, 0x1E);
	SETOP(18, "JSR", 0x00, 0x20, 0x00, 0x00, 0x00, 0x00);
	SETOP(19, "AND", 0x21, 0x2B, 0x00, 0x00, 0x29, 0x00);
	SETOP(20, "ANY", 0x22, 0x82, 0x00, 0x00, 0x52, 0x00);
	SETOP(21, "AAY", 0x00, 0x00, 0x00, 0x00, 0x00, 0x23);
	SETOP(22, "ANX", 0x24, 0x84, 0x00, 0x00, 0x54, 0x00);
	SETOP(23, "AAX", 0x00, 0x00, 0x00, 0x00, 0x00, 0x25);
	SETOP(24, "STT", 0x28, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(25, "BPO", 0x00, 0x00, 0x00, 0x00, 0x30, 0x00);
	SETOP(26, "ORA", 0x31, 0x3B, 0x00, 0x00, 0x39, 0x00);
	SETOP(27, "ORY", 0x32, 0x00, 0x00, 0x00, 0x62, 0x00);
	SETOP(28, "OAY", 0x00, 0x00, 0x00, 0x00, 0x00, 0x33);
	SETOP(29, "ORX", 0x34, 0x94, 0x00, 0x00, 0x64, 0x00);
	SETOP(30, "OAX", 0x00, 0x00, 0x00, 0x00, 0x00, 0x35);
	SETOP(31, "SEI", 0x00, 0x00, 0x00, 0x00, 0x00, 0x38);
	SETOP(32, "BNG", 0x00, 0x00, 0x00, 0x00, 0x40, 0x00);
	SETOP(33, "XOR", 0x41, 0x4B, 0x00, 0x00, 0x49, 0x00);
	SETOP(34, "XRY", 0x42, 0xA2, 0x00, 0x00, 0x72, 0x00);
	SETOP(35, "XAY", 0x00, 0x00, 0x00, 0x00, 0x00, 0x43);
	SETOP(36, "XRX", 0x44, 0xA4, 0x00, 0x00, 0x74, 0x00);
	SETOP(37, "XAX", 0x00, 0x00, 0x00, 0x00, 0x00, 0x45);
	SETOP(38, "CLI", 0x00, 0x00, 0x00, 0x00, 0x00, 0x48);
	SETOP(39, "BCS", 0x00, 0x00, 0x00, 0x00, 0x50, 0x00);
	SETOP(40, "LSL", 0x51, 0x55, 0x00, 0x00, 0x53, 0x00);
	SETOP(41, "SEC", 0x00, 0x00, 0x00, 0x00, 0x00, 0x58);
	SETOP(42, "STA", 0x00, 0x7B, 0x8B, 0x9B, 0x5B, 0x00);
	SETOP(43, "STY", 0x00, 0x7D, 0x8D, 0x00, 0x5D, 0x00);
	SETOP(44, "STX", 0x00, 0x7E, 0x00, 0x9E, 0x5E, 0x00);
	SETOP(45, "BCC", 0x00, 0x00, 0x00, 0x00, 0x60, 0x00);
	SETOP(46, "LSR", 0x61, 0x65, 0x00, 0x00, 0x63, 0x00);
	SETOP(47, "CLC", 0x00, 0x00, 0x00, 0x00, 0x00, 0x68);
	SETOP(48, "LDA", 0x69, 0x79, 0x89, 0x99, 0x59, 0x00);
	SETOP(49, "LDY", 0x6A, 0x7A, 0x8A, 0x00, 0x5A, 0x00);
	SETOP(50, "LDX", 0x6C, 0x7C, 0x00, 0x9C, 0x5C, 0x00);
	SETOP(51, "BEQ", 0x00, 0x00, 0x00, 0x00, 0x70, 0x00);
	SETOP(52, "ROL", 0x71, 0x75, 0x00, 0x00, 0x73, 0x00);
	SETOP(53, "SSP", 0x00, 0x00, 0x00, 0x00, 0x00, 0x78);
	SETOP(54, "BNE", 0x00, 0x00, 0x00, 0x00, 0x80, 0x00);
	SETOP(55, "ROR", 0x81, 0x85, 0x00, 0x00, 0x83, 0x00);
	SETOP(56, "CSP", 0x00, 0x00, 0x00, 0x00, 0x00, 0x88);
	SETOP(57, "BVS", 0x00, 0x00, 0x00, 0x00, 0x90, 0x00);
	SETOP(58, "MUL", 0x91, 0x95, 0x00, 0x00, 0x93, 0x00);
	SETOP(59, "SEV", 0x00, 0x00, 0x00, 0x00, 0x00, 0x98);
	SETOP(60, "BVC", 0x00, 0x00, 0x00, 0x00, 0xA0, 0x00);
	SETOP(61, "DIV", 0xA1, 0xA5, 0x00, 0x00, 0xA3, 0x00);
	SETOP(62, "CLV", 0x00, 0x00, 0x00, 0x00, 0x00, 0xA8);
	SETOP(63, "RTS", 0x00, 0x00, 0x00, 0x00, 0x00, 0xB0);
	SETOP(64, "CMP", 0xB1, 0xF5, 0x00, 0x00, 0xE5, 0x00);
	SETOP(65, "CPY", 0xB2, 0xF2, 0x00, 0x00, 0xE2, 0x00);
	SETOP(66, "CAY", 0x00, 0x00, 0x00, 0x00, 0x00, 0xB3);
	SETOP(67, "CPX", 0xB4, 0xF4, 0x00, 0x00, 0xE4, 0x00);
	SETOP(68, "CAX", 0x00, 0x00, 0x00, 0x00, 0x00, 0xB5);
	SETOP(69, "ENT", 0xB8, 0x00, 0x00, 0x00, 0x00, 0x00);
	SETOP(70, "RTI", 0x00, 0x00, 0x00, 0x00, 0x00, 0xC0);
	SETOP(71, "INC", 0x00, 0xE3, 0x00, 0x00, 0xE1, 0xC1);
	SETOP(72, "INY", 0x00, 0x00, 0x00, 0x00, 0x00, 0xC2);
	SETOP(73, "IAY", 0x00, 0x00, 0x00, 0x00, 0x00, 0xC3);
	SETOP(74, "INX", 0x00, 0x00, 0x00, 0x00, 0x00, 0xC4);
	SETOP(75, "IAX", 0x00, 0x00, 0x00, 0x00, 0x00, 0xC5);
	SETOP(76, "DEC", 0x00, 0x00, 0x00, 0x00, 0x00, 0xD1);
	SETOP(77, "DEY", 0x00, 0x00, 0x00, 0x00, 0x00, 0xD2);
	SETOP(78, "DAY", 0x00, 0xF3, 0x00, 0x00, 0xF1, 0xD3);
	SETOP(79, "DEX", 0x00, 0x00, 0x00, 0x00, 0x00, 0xD4);
	SETOP(80, "DAX", 0x00, 0x00, 0x00, 0x00, 0x00, 0xD5);
	SETOP(81, "JSL", 0x00, 0x00, 0x00, 0x00, 0xE0, 0x00);
	SETOP(82, "NOP", 0x00, 0x00, 0x00, 0x00, 0x00, 0xE8);
	SETOP(83, "RTL", 0x00, 0x00, 0x00, 0x00, 0x00, 0xF0);
	SETOP(84, "BRK", 0x00, 0x00, 0x00, 0x00, 0x00, 0xF8);
	uint8_t done = 0;
	uint64_t address = 0x0000;
	while (!(done & 1)) {
		char *buf = NULL;
		char *ins;
		char *postfix;
		char mode[3];
		opent op;
		uint8_t addrmode = 0;
		uint64_t value;
		char *oprand;
		char *cmd;
		char *tmp = malloc(sizeof(char *)*128);
		size_t size;
		done &= ~0x1F;
		getline(&buf, &size, stdin);
		cmd = strtok_r(buf, "\n", &tmp);
		if (cmd != NULL) {
			if (strcasecmp(cmd, "done") == 0) {
				done |= 1;
			} else {
				ins = strtok(buf, "\t\n ");
				if (ins != NULL) {
					oprand = strtok(NULL, "\t\n ");
					strtok_r(ins, ".", &postfix);
				}
				if (strcasecmp(cmd, "quit") == 0 || strcasecmp(cmd, "q") == 0)
					return 2;
				if (strcasecmp(cmd, "viewmem") == 0) {
					done |= 4;
					printf("\t\t\t");
					for (int ind = 0; ind < 0x10; ind++) {
						printf("%02x", ind);
						if (ind < 0x0F)
							printf(" ");
					}
					printf("\n\n");
					for (int hi = 0; hi < 0x10; hi++) {
						printf("%016llx:\t", (address & ~0xF)+(hi*0x10));
						for (int lo = 0; lo < 0x10; lo++) {
							printf("%02x", addr[(address & ~0xF)+lo+(hi*0x10)]);
							if (lo < 0x0F)
								printf(" ");
						}
						printf("\n");
					}
				}
				if (oprand == NULL && ins == NULL && postfix == NULL) {
					done |= 2;
				}
				if (ins != NULL) {
					for (int i = 0; i < strlen(ins); i++) {
						if (i && ins[i] == ':') {
							ins[i] = '\0';
							mklabel(ins, address, 1);
							#if debug
							printf("Created label with the name %s, at address: $%llx\n", ins, address);
							#endif
							done |= 6;
							break;
						}
						if (ins[i] == ';') {
							if (i && (ins[i-1] == ' ' || ins[i-1] == '\t'))
								ins[i] = '\0';
							else
								done |=6;
							break;
						}
					}
					if (strcasecmp(ins, ".org") == 0) {
						done |= 6;
						oprand = strtok(oprand, "$");
						address = strtoull(oprand, NULL, 16);
						#if debug
						printf("Origin for program code is now at address $%llx.\n", address);
						#endif
					}
					if (strcasecmp(ins, ".byte") == 0 || strcasecmp(ins, ".word") == 0 || strcasecmp(ins, ".dword") == 0 || strcasecmp(ins, ".qword") == 0) {
						done |= 6;
						for (int i = 0; i < strlen(oprand); i++) {
							if (oprand[i] == '$') {
								oprand = strtok(oprand, "$");
								value = strtoull(oprand, NULL, 16);
								break;
							}
							if (oprand[i] == ';') {
								done |= 16;
								break;
							}
							if (isalnum(oprand[i]) || oprand[i] == '_') {
								use_label(oprand, &value);
								sprintf(oprand, "%llx", value);
								break;
							}
						}
						if (strcasecmp(ins, ".byte") == 0)
							addr[address++] = value & 0xFF;
						if (strcasecmp(ins, ".word") == 0) {
							addr[address] = value & 0xFF;
							addr[address+1] = value >> 8;
							address+=2;
						}
						if (strcasecmp(ins, ".dword") == 0) {
							addr[address] = value & 0xFF;
							addr[address+1] = value >> 8;
							addr[address+2] = value >> 16;
							addr[address+3] = value >> 24;
							address+=4;
						}
						if (strcasecmp(ins, ".qword") == 0) {
							addr[address] = value & 0xFF;
                	        	                addr[address+1] = value >> 8;
                	        	                addr[address+2] = value >> 16;
                	        	                addr[address+3] = value >> 24;
                	        	                addr[address+4] = value >> 32;
                	        	                addr[address+5] = value >> 40;
                	        	                addr[address+6] = value >> 48;
                	        	                addr[address+7] = value >> 56;
							address+=8;
						}
						#if debug
						printf("The value $%llx was placed at address $%llx.\n", value, address);
						#endif
					}
				}

				if (!(done & 2) && oprand != NULL) {
					if (oprand[0] == '#' || oprand[0] == '$') {
						if(oprand[0] == '#' && (oprand[1] == '$' || oprand[1] == '%')) {
							mode[1] = oprand[1];
							mode[2] = '\0';
						} else {
							mode[1] = '\0';
						}
						mode[0] = oprand[0];
						oprand = strtok(oprand, "#$%");
						if (mode[0] == '#') {
							addrmode = 1;
							if (mode[1] == '$')
								value = strtoull(oprand, NULL, 16);
							if (mode[1] == '%')
								value = strtoull(oprand, NULL, 2);
						}
						if (mode[0] == '$') {
							value = strtoull(oprand, NULL, 16);
							if (value == 0)
								addrmode = 2;
							if (value & 0xFFFFFFFF) {
								char *stf[] = {"BPO", "BNG", "BCS", "BCC", "BEQ", "BNE", "BVS", "BVC"};
								for (int i = 0; i < 8; i++) {
									if (strcasecmp(ins, stf[i]) == 0) {
										addrmode = 5;
										break;
									} else {
										addrmode = 2;
									}
								}
							} else if (value & 0xFFFFFFFF00000000) {
								addrmode = 5;
							}
						}
					} else {
						for (int i = 0; i < strlen(oprand); i++) {
							if (oprand[i] == ';') {
								done |= 16;
								break;
							}
							if (isalnum(oprand[i]) || oprand[i] == '_') {
								use_label(oprand, &value);
								if (value == 0)
									addrmode = 2;
								if (value & 0xFFFFFFFF) {
									char *stf[] = {"BPO", "BNG", "BCS", "BCC", "BEQ", "BNE", "BVS", "BVC"};
									for (int i = 0; i < 8; i++) {
										if (strcasecmp(ins, stf[i]) == 0) {
											addrmode = 5;
											break;
										} else {
											addrmode = 2;
										}
									}
								} else if (value & 0xFFFFFFFF00000000) {
									addrmode = 5;
								}
								sprintf(oprand, "%llx", value);
								break;
							}
						}
					}
				}
				if (ins != NULL && !(done & 6)) {
					uint8_t i;
					for (i = 0; i < OPNUM; i++) {
						if (strcasecmp(opcodes[i].mnemonic, ins) == 0) {
							if (addrmode == 0 && (opcodes[i].impl || opcodes[i].impl == CPS)) {
								done |= 8;
							} else if (addrmode == 1) {
								switch (opcodes[i].imm) {
									case PHP:
									case PHA:
									case PHY:
									case PHX:
									case PLP:
									case PLA:
									case PLY:
									case PLX:
									case STT:
									case LSL:
									case LSR:
									case ROL:
									case ROR:
									case ENT:
										done |= 8;
										break;
								}
							} else {
								if (strcasecmp(ins, "JMP") == 0)
									done |=8;
								if (strcasecmp(ins, "JSR") == 0)
									done |=8;
								if (strcasecmp(ins, "JSL") == 0)
									done |=8;
								if (strcasecmp(ins, "INC") == 0)
									done |=8;
								if (strcasecmp(ins, "BPO") == 0)
									done |=8;
								if (strcasecmp(ins, "BNG") == 0)
									done |=8;
								if (strcasecmp(ins, "BCS") == 0)
									done |=8;
								if (strcasecmp(ins, "BCC") == 0)
									done |=8;
								if (strcasecmp(ins, "BEQ") == 0)
									done |=8;
								if (strcasecmp(ins, "BNE") == 0)
									done |=8;
								if (strcasecmp(ins, "BVS") == 0)
									done |=8;
								if (strcasecmp(ins, "BVC") == 0)
									done |=8;
							}
							op = opcodes[i];
							break;
						}
					}
					if (postfix != NULL && !(done & 8)) {
						if (strcasecmp(postfix, "w") == 0) {
							addr[address++] = 0x17;
						} else if (strcasecmp(postfix, "d") == 0) {
							addr[address++] = 0x27;
						} else if (strcasecmp(postfix, "q") == 0) {
							addr[address++] = 0x37;
						} else {
							done |=8;
						}
					} else if (postfix == NULL && !(done & 8)) {
							done |=8;
					}
					uint8_t r;
					if (!(done & 8))
						r = addr[address-1];
					else
						r = 0;
					switch (addrmode) {
						case 0:
							if (op.impl || op.impl == CPS) {
								addr[address++] = op.impl;
								break;
							} else {
								fprintf(stderr, "oof, %s requires an operand.\n", op.mnemonic);
							}
							break;
						case 1:
							if (op.imm) {
								addr[address++] = op.imm;
								switch (op.imm) {
									case PHP:
									case PHA:
									case PHY:
									case PHX:
									case PLP:
									case PLA:
									case PLY:
									case PLX:
									case STT:
									case LSL:
									case LSR:
									case ROL:
									case ROR:
									case ENT:
										addr[address++] = value & 0xFF;
										break;
									default:
										addr[address] = value & 0xFF;
										if (r & 0x10)
                	        	        					addr[address+1] = value >> 8;
										if (r & 0x20)
                	        	        					addr[address+2] = value >> 16;
                	        	        					addr[address+3] = value >> 24;
										if (r & 0x30)
                	        	        					addr[address+4] = value >> 32;
                	        	        					addr[address+5] = value >> 40;
                	        	        					addr[address+6] = value >> 48;
                	        	        					addr[address+7] = value >> 56;

										address+=(1 << ((r & 0x30) >> 4));
										break;

								}
								break;
							} else {
								fprintf(stderr, "oof, %s does not use Immediate data.\n", op.mnemonic);
							}
							break;
						case 2:
							if (op.zm) {
								addr[address++] = op.zm;
								addr[address] = value & 0xFF;
                	        	        		addr[address+1] = value >> 8;
                	        	        		addr[address+2] = value >> 16;
                	        	        		addr[address+3] = value >> 24;
								address+=4;
								break;
							} else {
								fprintf(stderr, "oof, %s does not use Zero Matrix.\n", op.mnemonic);
							}
							break;
						case 3:
							if (op.zmx) {
								addr[address++] = op.zmy;
								addr[address] = value & 0xFF;
                	        	        		addr[address+1] = value >> 8;
                	        	        		addr[address+2] = value >> 16;
                	        	        		addr[address+3] = value >> 24;
								address+=4;
								break;
							} else {
								fprintf(stderr, "oof, %s does not use Zero Matrix, indexed with x.\n", op.mnemonic);
							}
							break;
						case 4:
							if (op.zmy) {
								addr[address++] = op.zmy;
								addr[address] = value & 0xFF;
                	        	        		addr[address+1] = value >> 8;
                	        	        		addr[address+2] = value >> 16;
                	        	        		addr[address+3] = value >> 24;
								address+=4;
								break;
							} else {
								fprintf(stderr, "oof, %s does not use Zero Matrix, indexed with y.\n", op.mnemonic);
							}
							break;
						case 5:
							if (op.abs) {
								addr[address++] = op.abs;
								addr[address] = value & 0xFF;
                	        	        		addr[address+1] = value >> 8;
                	        	        		addr[address+2] = value >> 16;
                	        	        		addr[address+3] = value >> 24;
                	        	        		addr[address+4] = value >> 32;
                	        	        		addr[address+5] = value >> 40;
                	        	        		addr[address+6] = value >> 48;
                	        	        		addr[address+7] = value >> 56;
								address+=8;
								break;
							} else {
								fprintf(stderr, "oof, %s cannot be an absolute dictator.\n", op.mnemonic);
							}
							break;
					}
					#if debug
					if (!(done & 6)) {
						printf("instruction: %s, ", ins);
						#if (!__GLIBC__) || (__TINYC__)
						printf("Postfix: %s, ", (postfix != NULL) ? postfix : "none");
						#else
						printf("Postfix: %s, ", (postfix[0] != '\0') ? postfix : "none");
						#endif
						printf("Operand: %s, Address: $%llx\n", (oprand != NULL && !(done & 16)) ? oprand : "none", address);
					}
					#endif
				}
			}
		}
	}
	return 0;
}