#include "codegen.h"
#include <string>
#include <vector>
#include <iostream>

std::string generateAssembly(const CompilerState& state) {
    std::string result;

    // --- NAGŁÓWEK ---
    result += "global main\n";
    result += "extern printf\n";
    result += "extern GetAsyncKeyState\n\n";

    result += "section .data\n";
    for (const auto& v : state.variables) {
        result += "    " + v.first + " dd " + std::to_string(v.second) + "\n";
    }
    result += "    fmt db '%d', 10, 0\n";
    result += "    pause_msg db 'Nacisnij ESC aby zamknac...', 10, 0\n\n";

    result += "section .text\n";

    // --- GENEROWANIE FUNKCJI ---
    for (const auto& func : state.functions) {
        result += func.first + ":\n";
        result += "    sub rsp, 40\n";

        // Iterujemy przez LISTĘ ROZKAZÓW danej funkcji
        // Rozkazy mogą być typu "PRINT:x" albo "IF:index"
        for (const std::string& cmd : func.second) {

            // 1. ZWYKŁY PRINT (format: "PRINT:nazwa_zmiennej")
            if (cmd.find("PRINT:") == 0) {
                std::string varName = cmd.substr(6); // utnij "PRINT:"
                if (state.variables.count(varName)) {
                    result += "    mov edx, [" + varName + "]\n";
                    result += "    lea rcx, [rel fmt]\n";
                    result += "    call printf\n";
                }
            }
            // 2. BLOK IF (format: "IF:index_w_tablicy")
            else if (cmd.find("IF:") == 0) {
                int ifIndex = std::stoi(cmd.substr(3)); // pobierz numer IFa

                if (ifIndex >= 0 && ifIndex < state.ifBlocks.size()) {
                    const auto& ifBlock = state.ifBlocks[ifIndex];

                    if (state.variables.count(ifBlock.conditionVar)) {
                        std::string labelSkip = "skip_if_" + std::to_string(ifIndex);

                        result += "    ; --- IF START [" + ifBlock.conditionVar + "] ---\n";
                        result += "    mov eax, [" + ifBlock.conditionVar + "]\n";
                        result += "    cmp eax, 1\n";
                        result += "    jne " + labelSkip + "\n"; // Skocz jeśli FAŁSZ (0)

                        // Generuj zawartość środka IFa
                        for (const std::string& innerCmd : ifBlock.prints) {
                            if (state.variables.count(innerCmd)) {
                                result += "    mov edx, [" + innerCmd + "]\n";
                                result += "    lea rcx, [rel fmt]\n";
                                result += "    call printf\n";
                            }
                        }

                        result += labelSkip + ":\n";
                        result += "    ; --- IF END ---\n";
                    }
                }
            }
        }

        result += "    add rsp, 40\n";
        result += "    ret\n\n";
    }

    // --- MAIN ---
    result += "main:\n";
    result += "    sub rsp, 40\n";

    // Globalne printy (bez zmian)
    for (const std::string& var : state.globalPrints) {
        if (state.variables.count(var)) {
            result += "    mov edx, [" + var + "]\n";
            result += "    lea rcx, [rel fmt]\n";
            result += "    call printf\n";
        }
    }

    // Wywołania funkcji
    for (const std::string& call : state.printCalls) {
        if (call.find("CALL_") == 0) {
            std::string funcName = call.substr(5);
            if (state.functions.count(funcName)) {
                result += "    call " + funcName + "\n";
            }
        }
    }

    // Pauza
    result += "    lea rcx, [rel pause_msg]\n";
    result += "    call printf\n";
    result += "pause_loop:\n";
    result += "    mov ecx, 27\n";
    result += "    call GetAsyncKeyState\n";
    result += "    test ax, 8000h\n";
    result += "    jz pause_loop\n";
    result += "    add rsp, 40\n";
    result += "    ret\n";

    return result;
}