diff --git a/src/main/java/org/jcnc/snow/compiler/semantic/core/SignatureRegistrar.java b/src/main/java/org/jcnc/snow/compiler/semantic/core/SignatureRegistrar.java
index e542646..5040493 100644
--- a/src/main/java/org/jcnc/snow/compiler/semantic/core/SignatureRegistrar.java
+++ b/src/main/java/org/jcnc/snow/compiler/semantic/core/SignatureRegistrar.java
@@ -1,12 +1,10 @@
package org.jcnc.snow.compiler.semantic.core;
-import org.jcnc.snow.compiler.parser.ast.FunctionNode;
-import org.jcnc.snow.compiler.parser.ast.ImportNode;
-import org.jcnc.snow.compiler.parser.ast.ModuleNode;
-import org.jcnc.snow.compiler.parser.ast.ParameterNode;
+import org.jcnc.snow.compiler.parser.ast.*;
import org.jcnc.snow.compiler.semantic.error.SemanticError;
import org.jcnc.snow.compiler.semantic.type.BuiltinType;
import org.jcnc.snow.compiler.semantic.type.FunctionType;
+import org.jcnc.snow.compiler.semantic.type.StructType;
import org.jcnc.snow.compiler.semantic.type.Type;
import java.util.ArrayList;
@@ -14,71 +12,96 @@ import java.util.List;
import java.util.Optional;
/**
- * {@code SignatureRegistrar} 负责函数签名登记与导入语义检查。
+ * {@code SignatureRegistrar}
*
- * 在语义分析初期阶段，它遍历每个模块，完成以下任务:
+ * 语义分析准备阶段：负责函数签名登记与 import 模块存在性校验。
*
- * - 验证所有 {@link ImportNode} 导入的模块是否存在于全局模块表 {@link Context#modules()} 中；
- * - 将每个 {@link FunctionNode} 的函数签名（参数类型和返回类型）注册到对应 {@link ModuleInfo} 中；
- * - 在参数或返回类型无法识别时，记录 {@link SemanticError}，并进行容错降级。
+ * - 验证每个模块声明的 import 模块在全局模块表 {@link Context#modules()} 是否存在。
+ * - 将每个函数、结构体方法、构造函数的类型签名登记到 {@link ModuleInfo}，便于后续类型推断。
+ * - 若参数或返回类型无法解析，则报错并降级为 int 或 void，保证语义分析流程健壮。
*
- * 本组件作为语义分析的准备阶段，为后续函数体检查提供函数类型上下文。
- *
- * @param ctx 全局语义分析上下文，提供模块、类型、错误管理等功能
+ * 作为语义分析前置流程，为后续函数体和表达式分析提供类型环境。
+ *
*/
public record SignatureRegistrar(Context ctx) {
/**
- * 构造函数签名注册器。
+ * 遍历所有模块，注册函数/方法签名，校验 import 合法性，填充 {@link ModuleInfo} 元信息。
*
- * @param ctx 当前语义分析上下文
+ * @param modules 需要分析的所有模块列表（AST 顶层节点）
*/
- public SignatureRegistrar {
- }
-
- /**
- * 遍历模块并注册函数签名，同时校验导入模块的合法性。
- *
- * @param mods 所有模块的语法树节点集合
- */
- public void register(Iterable mods) {
- for (ModuleNode mod : mods) {
+ public void register(Iterable modules) {
+ for (ModuleNode mod : modules) {
+ ctx.setCurrentModule(mod.name()); // 切换上下文到当前模块
ModuleInfo mi = ctx.modules().get(mod.name());
- // ---------- 1. 模块导入检查 ----------
+ // ========== 1) 校验 imports ==========
for (ImportNode imp : mod.imports()) {
if (!ctx.modules().containsKey(imp.moduleName())) {
- ctx.errors().add(new SemanticError(
- imp,
- "未知模块: " + imp.moduleName()
- ));
+ // 导入的模块在全局表中不存在，报错
+ ctx.errors().add(new SemanticError(imp, "未知模块: " + imp.moduleName()));
} else {
+ // 添加到本模块导入集合
mi.getImports().add(imp.moduleName());
}
}
- // ---------- 2. 函数签名注册 ----------
- for (FunctionNode fn : mod.functions()) {
- List params = new ArrayList<>();
+ // ========== 2) 结构体签名登记 ==========
+ for (StructNode stn : mod.structs()) {
+ // 构造结构体类型对象，唯一标识为 (模块名, 结构体名)
+ StructType st = new StructType(mod.name(), stn.name());
+ mi.getStructs().put(stn.name(), st);
- // 参数类型解析
- for (ParameterNode p : fn.parameters()) {
- Type t = Optional.ofNullable(ctx.parseType(p.type()))
- .orElseGet(() -> {
- ctx.errors().add(new SemanticError(
- p,
- "未知类型: " + p.type()
- ));
- return BuiltinType.INT; // 容错降级
- });
- params.add(t);
+ // --- 2.1 构造函数 init ---
+ if (stn.init() != null) {
+ List ptypes = new ArrayList<>();
+ for (ParameterNode p : stn.init().parameters()) {
+ // 解析参数类型，不存在则报错降级为 int
+ Type t = ctx.parseType(p.type());
+ if (t == null) {
+ ctx.errors().add(new SemanticError(p, "未知类型: " + p.type()));
+ t = BuiltinType.INT;
+ }
+ ptypes.add(t);
+ }
+ // 构造函数返回类型固定为 void
+ st.setConstructor(new FunctionType(ptypes, BuiltinType.VOID));
}
- // 返回类型解析（默认降级为 void）
+ // --- 2.2 结构体方法签名 ---
+ for (FunctionNode fn : stn.methods()) {
+ List ptypes = new ArrayList<>();
+ for (ParameterNode p : fn.parameters()) {
+ // 解析参数类型，不存在则报错降级为 int
+ Type t = ctx.parseType(p.type());
+ if (t == null) {
+ ctx.errors().add(new SemanticError(p, "未知类型: " + p.type()));
+ t = BuiltinType.INT;
+ }
+ ptypes.add(t);
+ }
+ // 返回类型未指定时降级为 void
+ Type ret = Optional.ofNullable(ctx.parseType(fn.returnType()))
+ .orElse(BuiltinType.VOID);
+ st.getMethods().put(fn.name(), new FunctionType(ptypes, ret));
+ }
+ }
+
+ // ========== 3) 模块级函数签名登记 ==========
+ for (FunctionNode fn : mod.functions()) {
+ List params = new ArrayList<>();
+ for (ParameterNode p : fn.parameters()) {
+ // 解析参数类型，不存在则报错降级为 int
+ Type t = ctx.parseType(p.type());
+ if (t == null) {
+ ctx.errors().add(new SemanticError(p, "未知类型: " + p.type()));
+ t = BuiltinType.INT;
+ }
+ params.add(t);
+ }
+ // 返回类型未指定时降级为 void
Type ret = Optional.ofNullable(ctx.parseType(fn.returnType()))
.orElse(BuiltinType.VOID);
-
- // 注册函数签名
mi.getFunctions().put(fn.name(), new FunctionType(params, ret));
}
}