#include "irgen/IRGen.h"

#include <stdexcept>

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

// 语句生成当前只实现了最小子集。
// 目前支持：
// - return <exp>;
//
// 还未支持：
// - 赋值语句
// - if / while 等控制流
// - 空语句、块语句嵌套分发之外的更多语句形态

std::any IRGenImpl::visitStmt(SysYParser::StmtContext* ctx) {
  if (!ctx) {
    throw std::runtime_error(FormatError("irgen", "缺少语句"));
  }
  
  // return 语句 - 通过 Return() 关键字判断
  if (ctx->Return()) {
    return HandleReturnStmt(ctx);
  }
  
  // 块语句
  if (ctx->block()) {
    return ctx->block()->accept(this);
  }
  
  // 空语句或表达式语句（先计算表达式）
  if (ctx->exp()) {
    EvalExpr(*ctx->exp());
    return BlockFlow::Continue;
  }
  
  throw std::runtime_error(FormatError("irgen", "暂不支持的语句类型"));
}

IRGenImpl::BlockFlow IRGenImpl::HandleReturnStmt(SysYParser::StmtContext* ctx) {
  if (!ctx) {
    throw std::runtime_error(FormatError("irgen", "缺少 return 语句"));
  }
  
  ir::Value* retValue = nullptr;
  if (ctx->exp()) {
    retValue = EvalExpr(*ctx->exp());
  }
  // 如果没有表达式，返回0（对于int main）
  if (!retValue) {
    retValue = builder_.CreateConstInt(0);
  }
  
  builder_.CreateRet(retValue);
  return BlockFlow::Terminated;
}

// if语句（待实现）
IRGenImpl::BlockFlow IRGenImpl::HandleIfStmt(SysYParser::StmtContext* ctx) {
  // TODO: 实现if语句
  throw std::runtime_error(FormatError("irgen", "if语句暂未实现"));
}

// while语句（待实现）
IRGenImpl::BlockFlow IRGenImpl::HandleWhileStmt(SysYParser::StmtContext* ctx) {
  // TODO: 实现while语句
  throw std::runtime_error(FormatError("irgen", "while语句暂未实现"));
}

// break语句（待实现）
IRGenImpl::BlockFlow IRGenImpl::HandleBreakStmt(SysYParser::StmtContext* ctx) {
  // TODO: 实现break
  throw std::runtime_error(FormatError("irgen", "break语句暂未实现"));
}

// continue语句（待实现）
IRGenImpl::BlockFlow IRGenImpl::HandleContinueStmt(SysYParser::StmtContext* ctx) {
  // TODO: 实现continue
  throw std::runtime_error(FormatError("irgen", "continue语句暂未实现"));
}

// 赋值语句（待实现）
IRGenImpl::BlockFlow IRGenImpl::HandleAssignStmt(SysYParser::StmtContext* ctx) {
  // TODO: 实现赋值
  throw std::runtime_error(FormatError("irgen", "赋值语句暂未实现"));
}