#include "irgen/IRGen.h"

#include <stdexcept>

#include "SysYParser.h"
#include "ir/IR.h"
#include "utils/Log.h"

// 表达式生成当前也只实现了很小的一个子集。
// 目前支持：
// - 整数字面量
// - 普通局部变量读取
// - 括号表达式
// - 二元加法
//
// 还未支持：
// - 减乘除与一元运算
// - 赋值表达式
// - 函数调用
// - 数组、指针、下标访问
// - 条件与比较表达式
// - ...
ir::Value* IRGenImpl::EvalExpr(SysYParser::ExpContext& expr) {
  return std::any_cast<ir::Value*>(expr.accept(this));
}


std::any IRGenImpl::visitParenExp(SysYParser::ParenExpContext* ctx) {
  if (!ctx || !ctx->exp()) {
    throw std::runtime_error(FormatError("irgen", "非法括号表达式"));
  }
  return EvalExpr(*ctx->exp());
}


std::any IRGenImpl::visitNumberExp(SysYParser::NumberExpContext* ctx) {
  if (!ctx || !ctx->number() || !ctx->number()->ILITERAL()) {
    throw std::runtime_error(FormatError("irgen", "当前仅支持整数字面量"));
  }
  return static_cast<ir::Value*>(
      builder_.CreateConstInt(std::stoi(ctx->number()->getText())));
}

// 变量使用的处理流程：
// 1. 先通过语义分析结果把变量使用绑定回声明；
// 2. 再通过 storage_map_ 找到该声明对应的栈槽位；
// 3. 最后生成 load，把内存中的值读出来。
//
// 因此当前 IRGen 自己不再做名字查找，而是直接消费 Sema 的绑定结果。
std::any IRGenImpl::visitVarExp(SysYParser::VarExpContext* ctx) {
  if (!ctx || !ctx->var() || !ctx->var()->ID()) {
    throw std::runtime_error(FormatError("irgen", "当前仅支持普通整型变量"));
  }
  auto* decl = sema_.ResolveVarUse(ctx->var());
  if (!decl) {
    throw std::runtime_error(
        FormatError("irgen",
                    "变量使用缺少语义绑定: " + ctx->var()->ID()->getText()));
  }
  auto it = storage_map_.find(decl);
  if (it == storage_map_.end()) {
    throw std::runtime_error(
        FormatError("irgen",
                    "变量声明缺少存储槽位: " + ctx->var()->ID()->getText()));
  }
  return static_cast<ir::Value*>(
      builder_.CreateLoad(it->second, module_.GetContext().NextTemp()));
}


std::any IRGenImpl::visitAdditiveExp(SysYParser::AdditiveExpContext* ctx) {
  if (!ctx || !ctx->exp(0) || !ctx->exp(1)) {
    throw std::runtime_error(FormatError("irgen", "非法加法表达式"));
  }
  ir::Value* lhs = EvalExpr(*ctx->exp(0));
  ir::Value* rhs = EvalExpr(*ctx->exp(1));
  return static_cast<ir::Value*>(
      builder_.CreateBinary(ir::Opcode::Add, lhs, rhs,
                            module_.GetContext().NextTemp()));
}
