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

int xyw_debug = 0;

// Instruction mnemonics definition
const char xyw_instructions[][4] = {
    "BRK", "PSH", "POP", "CLR",
    "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"};

int flag(const char *arg, const char *str)
{
    if (!str || !arg || !*str)
        return 0;
    if (strncmp(arg, "--", 2) == 0 && strcmp(arg + 2, str) == 0)
        return 1;
    if (arg[0] == '-' && arg[1] != '\0' && arg[2] == '\0' && arg[1] == str[0])
        return 1;
    return 0;
}

int main(int argc, char *argv[])
{
    (void)argc;

    for (int i = 1; i < argc; i++)
    {
        if (flag(argv[i], "debug"))
        {
            xyw_debug = 1;
        }
    }

    for (int i = 1; i < argc; i++)
    {
        if (flag(argv[i], "debug"))
        {
            continue; // already handled
        }

        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;
        }
    }
}