#include "ir/ir_function.hpp"
#include <iomanip>
#include <iostream>
#include <string>

// ─────────────────────────────────────────────────────────────────────────────
// Yardımcılar
// ─────────────────────────────────────────────────────────────────────────────

// Slot adını kısa göster: s0, s1, ...
static std::string slot(int s) {
    if (s == -1) return "?";
    return "s" + std::to_string(s);
}

// İkili op sembolü: ADD → "+"
static const char* opSymbol(Opcode op) {
    switch (op) {
        case Opcode::ADD:           return "+";
        case Opcode::SUB:           return "-";
        case Opcode::MUL:           return "*";
        case Opcode::DIV:           return "/";
        case Opcode::MOD:           return "%";
        case Opcode::LESS:          return "<";
        case Opcode::LESS_EQUAL:    return "<=";
        case Opcode::GREATER:       return ">";
        case Opcode::GREATER_EQUAL: return ">=";
        case Opcode::EQUAL_EQUAL:   return "==";
        case Opcode::NOT_EQUAL:     return "!=";
        default:                    return "?";
    }
}

static bool isBinaryOp(Opcode op) {
    switch (op) {
        case Opcode::ADD: case Opcode::SUB: case Opcode::MUL:
        case Opcode::DIV: case Opcode::MOD:
        case Opcode::LESS: case Opcode::LESS_EQUAL:
        case Opcode::GREATER: case Opcode::GREATER_EQUAL:
        case Opcode::EQUAL_EQUAL: case Opcode::NOT_EQUAL:
            return true;
        default: return false;
    }
}

// ─────────────────────────────────────────────────────────────────────────────
// IRFunction::dump
// ─────────────────────────────────────────────────────────────────────────────

void IRFunction::dump() const {
    // Başlık: fonksiyon adı + slot bilgisi
    std::string header = "  " + name + "()";
    if (paramCount > 0) {
        header = "  " + name + "(";
        for (int i = 0; i < paramCount; i++) {
            if (i) header += ", ";
            header += "s" + std::to_string(i);
        }
        header += ")";
    }
    header += "  [" + std::to_string(slotCount) + " slot]";

    // Başlık: NAME=fibonacci PARAMS=1 SLOTS=10
    std::cout << "NAME=" << name
              << " PARAMS=" << paramCount
              << " SLOTS=" << slotCount
              << "\n";

    // Talimatlar
    for (int i = 0; i < (int)instructions.size(); i++) {
        const Instruction& ins = instructions[i];

        // Satır numarası
        std::cout << "  " << std::setw(3) << std::right << i << "  ";

        // Opcode sütunu (12 karakter genişlik)
        std::cout << std::left << std::setw(12) << opcodeName(ins.opcode);

        // Operandlar — opcode'a göre farklı format
        if (ins.opcode == Opcode::LOAD_CONST) {
            std::cout << slot(ins.dest) << " = " << ins.intValue;

        } else if (ins.opcode == Opcode::LOAD_STRING) {
            std::cout << slot(ins.dest) << " = \"" << ins.stringValue << "\"";

        } else if (ins.opcode == Opcode::LOAD_SLOT) {
            std::cout << slot(ins.dest) << " = " << slot(ins.src);

        } else if (isBinaryOp(ins.opcode)) {
            std::cout << slot(ins.dest) << " = "
                      << slot(ins.left) << " " << opSymbol(ins.opcode)
                      << " " << slot(ins.right);

        } else if (ins.opcode == Opcode::JMP) {
            std::cout << "→ " << ins.jumpTarget;

        } else if (ins.opcode == Opcode::JIF_FALSE) {
            std::cout << "!" << slot(ins.cond) << " → " << ins.jumpTarget;

        } else if (ins.opcode == Opcode::CALL) {
            std::cout << slot(ins.dest) << " = " << ins.functionName << "(";
            for (int j = 0; j < (int)ins.argSlots.size(); j++) {
                if (j) std::cout << ", ";
                std::cout << slot(ins.argSlots[j]);
            }
            std::cout << ")";

        } else if (ins.opcode == Opcode::CALLHOST) {
            std::cout << ins.functionName << "(";
            for (int j = 0; j < (int)ins.argSlots.size(); j++) {
                if (j) std::cout << ", ";
                std::cout << slot(ins.argSlots[j]);
            }
            std::cout << ")";

        } else if (ins.opcode == Opcode::RETURN) {
            std::cout << slot(ins.src);
        }

        std::cout << "\n";
    }
    std::cout << "\n";
}