#include <stdio.h>
#include <string.h>
#include "xyw.h"

int xyw_debug = 0;

// Instruction mnemonics definition
const char xyw_instructions[][4] = {
    "BRK", "PSH", "POP", "???",
    "LDB", "LDW", "STB", "STW",
    "INC", "DEC", "SHL", "SHR",
    "ADD", "SUB", "MUL", "DIV",
    "EQU", "NEQ", "GTH", "LTH",
    "AND", "IOR", "XOR", "NOT",
    "SWP", "DUP", "OVR", "ROT",
    "JMP", "JCN", "JSR", "RTS"};

//// Main Function
int main(int argc, char *argv[])
{
    (void)argc;
    for (int i = 1; i < argc; i++)
    {
        if (strcmp(argv[i], "-d") == 0 || strcmp(argv[i], "--debug") == 0)
        {
            xyw_debug = 1;
        }
        else
        {
            const char *input_file = argv[i];
            char output_file[256];
            snprintf(output_file, sizeof(output_file), "%.*s.xim", (int)(strlen(input_file) - 4), input_file);
            const char *ext = strrchr(input_file, '.');
            if (ext && strcmp(ext, ".xyw") == 0)
            {
                if (xyw_assemble(input_file, output_file) != 0)
                {
                    fprintf(stderr, "Assembly failed for file: %s\n", input_file);
                    return -1;
                }
                else
                {
                    if (xyw_debug)
                    {
                        printf("Assembled %s to %s\n", input_file, output_file);
                    }
                }
            }
            else if (ext && strcmp(ext, ".xim") == 0)
            {
                if (xyw_run(input_file) != 0)
                {
                    fprintf(stderr, "Execution failed for file: %s\n", input_file);
                    return -1;
                }
            }
            else
            {
                fprintf(stderr, "Unsupported file type: %s\n", input_file);
                return -1;
            }
        }
    }
}