diff --git a/src/main/java/org/jcnc/snow/compiler/cli/commands/CompileCommand.java b/src/main/java/org/jcnc/snow/compiler/cli/commands/CompileCommand.java index ff8f33d..36bc2b2 100644 --- a/src/main/java/org/jcnc/snow/compiler/cli/commands/CompileCommand.java +++ b/src/main/java/org/jcnc/snow/compiler/cli/commands/CompileCommand.java @@ -26,45 +26,36 @@ import java.util.*; /** *

- * 编译器命令实现：`snow compile` - *
+ * 编译器子命令：snow compile
* 将一个或多个 .snow 源文件编译为 VM 字节码文件（.water）。 *

+ * * - *

- * 用法：
- * snow compile [run] [-o <name>] [-d <srcDir>] [file1.snow file2.snow …] - *

+ * + * 
用法：
+ * snow compile [run] [-o <name>] [-d <srcDir>] [file1.snow file2.snow …]
+ *
*/ public final class CompileCommand implements CLICommand { - /** - * 获取命令名。 - * - * @return "compile" - */ + /* --------------------------------------------------------------------- */ + /* CLICommand 接口实现 */ + /* --------------------------------------------------------------------- */ + @Override public String name() { return "compile"; } - /** - * 获取命令描述。 - * - * @return 命令简介 - */ @Override public String description() { return "Compile .snow source files into VM byte-code (.water)."; } - /** - * 打印该命令的用法说明。 - */ @Override public void printUsage() { System.out.println("Usage:"); @@ -75,15 +66,13 @@ public final class CompileCommand implements CLICommand { System.out.println(" -d recursively compile all .snow files in directory"); } - /** - * 执行 compile 命令，编译 .snow 源文件。 - * - * @param args 剩余参数（不含命令名） - * @return 0 表示成功，1 表示参数错误或编译失败 - * @throws Exception 编译过程中可能抛出的异常 - */ + /* --------------------------------------------------------------------- */ + /* 核心：执行 compile 子命令 */ + /* --------------------------------------------------------------------- */ + @Override public int execute(String[] args) throws Exception { + /* ---------------- 解析命令行参数 ---------------- */ boolean runAfterCompile = false; String outputName = null; Path dir = null; @@ -92,28 +81,24 @@ public final class CompileCommand implements CLICommand { for (int i = 0; i < args.length; i++) { String arg = args[i]; switch (arg) { - case "run": - runAfterCompile = true; - break; - case "-o": - if (i + 1 < args.length) { - outputName = args[++i]; - } else { + case "run" -> runAfterCompile = true; + case "-o" -> { + if (i + 1 < args.length) outputName = args[++i]; + else { System.err.println("Missing argument for -o"); printUsage(); return 1; } - break; - case "-d": - if (i + 1 < args.length) { - dir = Path.of(args[++i]); - } else { + } + case "-d" -> { + if (i + 1 < args.length) dir = Path.of(args[++i]); + else { System.err.println("Missing argument for -d"); printUsage(); return 1; } - break; - default: + } + default -> { if (arg.endsWith(".snow")) { sources.add(Path.of(arg)); } else { @@ -121,10 +106,11 @@ public final class CompileCommand implements CLICommand { printUsage(); return 1; } + } } } - // 目录编译时收集所有 .snow 文件 + /* --------- 如果指定了目录则递归收集所有 *.snow --------- */ if (dir != null) { if (!Files.isDirectory(dir)) { System.err.println("Not a directory: " + dir); @@ -132,7 +118,7 @@ public final class CompileCommand implements CLICommand { } try (var stream = Files.walk(dir)) { stream.filter(p -> p.toString().endsWith(".snow")) - .sorted() + .sorted() // 确保稳定顺序 .forEach(sources::add); } } @@ -142,42 +128,64 @@ public final class CompileCommand implements CLICommand { return 1; } - // 多文件且未指定输出时错误 + /* 多文件但未指定 -o 且非目录编译 —— 提示必须指定输出名 */ if (sources.size() > 1 && outputName == null && dir == null) { System.err.println("Please specify output name using -o "); return 1; } - // 1. 词法和语法分析 + /* ----------------------------------------------------------------- */ + /* 1. 词法 + 语法分析；同时打印源代码 */ + /* ----------------------------------------------------------------- */ List allAst = new ArrayList<>(); + + System.out.println("## 编译器输出"); + System.out.println("### Snow 源代码"); // ========== 新增：二级标题 ========== + for (Path p : sources) { if (!Files.exists(p)) { System.err.println("File not found: " + p); return 1; } + String code = Files.readString(p, StandardCharsets.UTF_8); + + // ------- 打印每个文件的源码 ------- + System.out.println("#### " + p.getFileName()); + System.out.println(code); + // -------------------------------------------------------- + + /* 词法 + 语法 */ LexerEngine lexer = new LexerEngine(code, p.toString()); ParserContext ctx = new ParserContext(lexer.getAllTokens(), p.toString()); allAst.addAll(new ParserEngine(ctx).parse()); } - // 2. 语义分析 + /* ----------------------------------------------------------------- */ + /* 2. 语义分析 */ + /* ----------------------------------------------------------------- */ SemanticAnalyzerRunner.runSemanticAnalysis(allAst, false); - // 3. AST -> IR 并重排序入口 + /* ----------------------------------------------------------------- */ + /* 3. AST → IR，并把 main 函数调到首位 */ + /* ----------------------------------------------------------------- */ IRProgram program = new IRProgramBuilder().buildProgram(allAst); program = reorderForEntry(program); - System.out.println("## 编译器输出"); + /* ---------------- 打印 AST / IR ---------------- */ System.out.println("### AST"); ASTPrinter.printJson(allAst); + System.out.println("### IR"); System.out.println(program); - // 4. IR -> VM 指令 + /* ----------------------------------------------------------------- */ + /* 4. IR → VM 指令 */ + /* ----------------------------------------------------------------- */ VMProgramBuilder builder = new VMProgramBuilder(); List> generators = InstructionGeneratorProvider.defaultGenerators(); + for (IRFunction fn : program.functions()) { Map slotMap = new RegisterAllocator().allocate(fn); @@ -188,26 +196,36 @@ public final class CompileCommand implements CLICommand { System.out.println("### VM code"); finalCode.forEach(System.out::println); - // 5. 写出 .water 文件 + /* ----------------------------------------------------------------- */ + /* 5. 写出 .water 文件 */ + /* ----------------------------------------------------------------- */ Path outputFile = deriveOutputPath(sources, outputName, dir); Files.write(outputFile, finalCode, StandardCharsets.UTF_8); System.out.println("Written to " + outputFile.toAbsolutePath()); - // 6. 可选执行 + /* ----------------------------------------------------------------- */ + /* 6. 可选：立即运行 VM */ + /* ----------------------------------------------------------------- */ if (runAfterCompile) { System.out.println("\n=== Launching VM ==="); VMLauncher.main(new String[]{outputFile.toString()}); } + return 0; } + /* --------------------------------------------------------------------- */ + /* 辅助方法 */ + /* --------------------------------------------------------------------- */ + /** - * 推断输出 .water 文件的路径。 - * - * @param sources 源文件列表 - * @param outName 用户指定的输出基名 - * @param dir 目录编译时的源目录 - * @return 输出文件的完整路径 + * 根据输入情况推断 .water 输出文件名： + *
    + *
  • 若指定 -o，则直接使用
    • + *
  • 目录编译：取目录名
    • + *
  • 单文件编译：取文件名去掉 .snow
    • + *
  • 其他情况兜底为 "program"
    • + *
*/ private static Path deriveOutputPath(List sources, String outName, Path dir) { String base; @@ -216,7 +234,8 @@ public final class CompileCommand implements CLICommand { } else if (dir != null) { base = dir.getFileName().toString(); } else if (sources.size() == 1) { - base = sources.getFirst().getFileName().toString().replaceFirst("\\.snow$", ""); + base = sources.getFirst().getFileName().toString() + .replaceFirst("\\.snow$", ""); } else { base = "program"; } @@ -224,10 +243,7 @@ public final class CompileCommand implements CLICommand { } /** - * 保证 main 函数为程序入口。 - * - * @param in 输入的 IR 程序 - * @return 重新排序，使 main 函数位于首位的 IR 程序 + * 把 main 函数交换到程序函数列表首位，确保 PC=0 即入口。 */ private static IRProgram reorderForEntry(IRProgram in) { List ordered = new ArrayList<>(in.functions());