redscript-mc 1.2.30 → 2.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (269) hide show
  1. package/.claude/commands/build-test.md +10 -0
  2. package/.claude/commands/deploy-demo.md +12 -0
  3. package/.claude/commands/stage-status.md +13 -0
  4. package/.claude/settings.json +12 -0
  5. package/.github/workflows/ci.yml +1 -0
  6. package/CLAUDE.md +231 -0
  7. package/demo.gif +0 -0
  8. package/dist/cli.js +2 -554
  9. package/dist/compile.js +2 -266
  10. package/dist/index.js +2 -159
  11. package/dist/lowering/index.js +5 -3
  12. package/dist/src/__tests__/cli.test.d.ts +1 -0
  13. package/dist/src/__tests__/cli.test.js +104 -0
  14. package/dist/src/__tests__/codegen.test.d.ts +1 -0
  15. package/dist/src/__tests__/codegen.test.js +152 -0
  16. package/dist/src/__tests__/compile-all.test.d.ts +10 -0
  17. package/dist/src/__tests__/compile-all.test.js +108 -0
  18. package/dist/src/__tests__/dce.test.d.ts +1 -0
  19. package/dist/src/__tests__/dce.test.js +102 -0
  20. package/dist/src/__tests__/diagnostics.test.d.ts +4 -0
  21. package/dist/src/__tests__/diagnostics.test.js +177 -0
  22. package/dist/src/__tests__/e2e.test.d.ts +6 -0
  23. package/dist/src/__tests__/e2e.test.js +1789 -0
  24. package/dist/src/__tests__/entity-types.test.d.ts +1 -0
  25. package/dist/src/__tests__/entity-types.test.js +203 -0
  26. package/dist/src/__tests__/formatter.test.d.ts +1 -0
  27. package/dist/src/__tests__/formatter.test.js +40 -0
  28. package/dist/src/__tests__/lexer.test.d.ts +1 -0
  29. package/dist/src/__tests__/lexer.test.js +343 -0
  30. package/dist/src/__tests__/lowering.test.d.ts +1 -0
  31. package/dist/src/__tests__/lowering.test.js +1015 -0
  32. package/dist/src/__tests__/macro.test.d.ts +8 -0
  33. package/dist/src/__tests__/macro.test.js +306 -0
  34. package/dist/src/__tests__/mc-integration.test.d.ts +12 -0
  35. package/dist/src/__tests__/mc-integration.test.js +817 -0
  36. package/dist/src/__tests__/mc-syntax.test.d.ts +1 -0
  37. package/dist/src/__tests__/mc-syntax.test.js +124 -0
  38. package/dist/src/__tests__/nbt.test.d.ts +1 -0
  39. package/dist/src/__tests__/nbt.test.js +82 -0
  40. package/dist/src/__tests__/optimizer-advanced.test.d.ts +1 -0
  41. package/dist/src/__tests__/optimizer-advanced.test.js +124 -0
  42. package/dist/src/__tests__/optimizer.test.d.ts +1 -0
  43. package/dist/src/__tests__/optimizer.test.js +149 -0
  44. package/dist/src/__tests__/parser.test.d.ts +1 -0
  45. package/dist/src/__tests__/parser.test.js +807 -0
  46. package/dist/src/__tests__/repl.test.d.ts +1 -0
  47. package/dist/src/__tests__/repl.test.js +27 -0
  48. package/dist/src/__tests__/runtime.test.d.ts +1 -0
  49. package/dist/src/__tests__/runtime.test.js +289 -0
  50. package/dist/src/__tests__/stdlib-advanced.test.d.ts +4 -0
  51. package/dist/src/__tests__/stdlib-advanced.test.js +374 -0
  52. package/dist/src/__tests__/stdlib-bigint.test.d.ts +7 -0
  53. package/dist/src/__tests__/stdlib-bigint.test.js +426 -0
  54. package/dist/src/__tests__/stdlib-math.test.d.ts +7 -0
  55. package/dist/src/__tests__/stdlib-math.test.js +351 -0
  56. package/dist/src/__tests__/stdlib-vec.test.d.ts +4 -0
  57. package/dist/src/__tests__/stdlib-vec.test.js +263 -0
  58. package/dist/src/__tests__/structure-optimizer.test.d.ts +1 -0
  59. package/dist/src/__tests__/structure-optimizer.test.js +33 -0
  60. package/dist/src/__tests__/typechecker.test.d.ts +1 -0
  61. package/dist/src/__tests__/typechecker.test.js +552 -0
  62. package/dist/src/__tests__/var-allocator.test.d.ts +1 -0
  63. package/dist/src/__tests__/var-allocator.test.js +69 -0
  64. package/dist/src/ast/types.d.ts +515 -0
  65. package/dist/src/ast/types.js +9 -0
  66. package/dist/src/builtins/metadata.d.ts +36 -0
  67. package/dist/src/builtins/metadata.js +1014 -0
  68. package/dist/src/cli.d.ts +11 -0
  69. package/dist/src/cli.js +443 -0
  70. package/dist/src/codegen/cmdblock/index.d.ts +26 -0
  71. package/dist/src/codegen/cmdblock/index.js +45 -0
  72. package/dist/src/codegen/mcfunction/index.d.ts +40 -0
  73. package/dist/src/codegen/mcfunction/index.js +606 -0
  74. package/dist/src/codegen/structure/index.d.ts +24 -0
  75. package/dist/src/codegen/structure/index.js +279 -0
  76. package/dist/src/codegen/var-allocator.d.ts +45 -0
  77. package/dist/src/codegen/var-allocator.js +104 -0
  78. package/dist/src/compile.d.ts +37 -0
  79. package/dist/src/compile.js +165 -0
  80. package/dist/src/diagnostics/index.d.ts +44 -0
  81. package/dist/src/diagnostics/index.js +140 -0
  82. package/dist/src/events/types.d.ts +35 -0
  83. package/dist/src/events/types.js +59 -0
  84. package/dist/src/formatter/index.d.ts +1 -0
  85. package/dist/src/formatter/index.js +26 -0
  86. package/dist/src/index.d.ts +22 -0
  87. package/dist/src/index.js +45 -0
  88. package/dist/src/ir/builder.d.ts +33 -0
  89. package/dist/src/ir/builder.js +99 -0
  90. package/dist/src/ir/types.d.ts +132 -0
  91. package/dist/src/ir/types.js +15 -0
  92. package/dist/src/lexer/index.d.ts +37 -0
  93. package/dist/src/lexer/index.js +569 -0
  94. package/dist/src/lowering/index.d.ts +188 -0
  95. package/dist/src/lowering/index.js +3405 -0
  96. package/dist/src/mc-test/client.d.ts +128 -0
  97. package/dist/src/mc-test/client.js +174 -0
  98. package/dist/src/mc-test/runner.d.ts +28 -0
  99. package/dist/src/mc-test/runner.js +151 -0
  100. package/dist/src/mc-test/setup.d.ts +11 -0
  101. package/dist/src/mc-test/setup.js +98 -0
  102. package/dist/src/mc-validator/index.d.ts +17 -0
  103. package/dist/src/mc-validator/index.js +322 -0
  104. package/dist/src/nbt/index.d.ts +86 -0
  105. package/dist/src/nbt/index.js +250 -0
  106. package/dist/src/optimizer/commands.d.ts +38 -0
  107. package/dist/src/optimizer/commands.js +451 -0
  108. package/dist/src/optimizer/dce.d.ts +34 -0
  109. package/dist/src/optimizer/dce.js +639 -0
  110. package/dist/src/optimizer/passes.d.ts +34 -0
  111. package/dist/src/optimizer/passes.js +243 -0
  112. package/dist/src/optimizer/structure.d.ts +9 -0
  113. package/dist/src/optimizer/structure.js +356 -0
  114. package/dist/src/parser/index.d.ts +93 -0
  115. package/dist/src/parser/index.js +1687 -0
  116. package/dist/src/repl.d.ts +16 -0
  117. package/dist/src/repl.js +165 -0
  118. package/dist/src/runtime/index.d.ts +107 -0
  119. package/dist/src/runtime/index.js +1409 -0
  120. package/dist/src/typechecker/index.d.ts +61 -0
  121. package/dist/src/typechecker/index.js +1034 -0
  122. package/dist/src/types/entity-hierarchy.d.ts +29 -0
  123. package/dist/src/types/entity-hierarchy.js +107 -0
  124. package/dist/src2/__tests__/e2e/basic.test.d.ts +8 -0
  125. package/dist/src2/__tests__/e2e/basic.test.js +140 -0
  126. package/dist/src2/__tests__/e2e/macros.test.d.ts +9 -0
  127. package/dist/src2/__tests__/e2e/macros.test.js +182 -0
  128. package/dist/src2/__tests__/e2e/migrate.test.d.ts +13 -0
  129. package/dist/src2/__tests__/e2e/migrate.test.js +2739 -0
  130. package/dist/src2/__tests__/hir/desugar.test.d.ts +1 -0
  131. package/dist/src2/__tests__/hir/desugar.test.js +234 -0
  132. package/dist/src2/__tests__/lir/lower.test.d.ts +1 -0
  133. package/dist/src2/__tests__/lir/lower.test.js +559 -0
  134. package/dist/src2/__tests__/lir/types.test.d.ts +1 -0
  135. package/dist/src2/__tests__/lir/types.test.js +185 -0
  136. package/dist/src2/__tests__/lir/verify.test.d.ts +1 -0
  137. package/dist/src2/__tests__/lir/verify.test.js +221 -0
  138. package/dist/src2/__tests__/mir/arithmetic.test.d.ts +1 -0
  139. package/dist/src2/__tests__/mir/arithmetic.test.js +130 -0
  140. package/dist/src2/__tests__/mir/control-flow.test.d.ts +1 -0
  141. package/dist/src2/__tests__/mir/control-flow.test.js +205 -0
  142. package/dist/src2/__tests__/mir/verify.test.d.ts +1 -0
  143. package/dist/src2/__tests__/mir/verify.test.js +223 -0
  144. package/dist/src2/__tests__/optimizer/block_merge.test.d.ts +1 -0
  145. package/dist/src2/__tests__/optimizer/block_merge.test.js +78 -0
  146. package/dist/src2/__tests__/optimizer/branch_simplify.test.d.ts +1 -0
  147. package/dist/src2/__tests__/optimizer/branch_simplify.test.js +58 -0
  148. package/dist/src2/__tests__/optimizer/constant_fold.test.d.ts +1 -0
  149. package/dist/src2/__tests__/optimizer/constant_fold.test.js +131 -0
  150. package/dist/src2/__tests__/optimizer/copy_prop.test.d.ts +1 -0
  151. package/dist/src2/__tests__/optimizer/copy_prop.test.js +91 -0
  152. package/dist/src2/__tests__/optimizer/dce.test.d.ts +1 -0
  153. package/dist/src2/__tests__/optimizer/dce.test.js +76 -0
  154. package/dist/src2/__tests__/optimizer/pipeline.test.d.ts +1 -0
  155. package/dist/src2/__tests__/optimizer/pipeline.test.js +102 -0
  156. package/dist/src2/emit/compile.d.ts +19 -0
  157. package/dist/src2/emit/compile.js +80 -0
  158. package/dist/src2/emit/index.d.ts +17 -0
  159. package/dist/src2/emit/index.js +172 -0
  160. package/dist/src2/hir/lower.d.ts +15 -0
  161. package/dist/src2/hir/lower.js +378 -0
  162. package/dist/src2/hir/types.d.ts +373 -0
  163. package/dist/src2/hir/types.js +16 -0
  164. package/dist/src2/lir/lower.d.ts +15 -0
  165. package/dist/src2/lir/lower.js +453 -0
  166. package/dist/src2/lir/types.d.ts +136 -0
  167. package/dist/src2/lir/types.js +11 -0
  168. package/dist/src2/lir/verify.d.ts +14 -0
  169. package/dist/src2/lir/verify.js +113 -0
  170. package/dist/src2/mir/lower.d.ts +9 -0
  171. package/dist/src2/mir/lower.js +1030 -0
  172. package/dist/src2/mir/macro.d.ts +22 -0
  173. package/dist/src2/mir/macro.js +168 -0
  174. package/dist/src2/mir/types.d.ts +183 -0
  175. package/dist/src2/mir/types.js +11 -0
  176. package/dist/src2/mir/verify.d.ts +16 -0
  177. package/dist/src2/mir/verify.js +216 -0
  178. package/dist/src2/optimizer/block_merge.d.ts +12 -0
  179. package/dist/src2/optimizer/block_merge.js +84 -0
  180. package/dist/src2/optimizer/branch_simplify.d.ts +9 -0
  181. package/dist/src2/optimizer/branch_simplify.js +28 -0
  182. package/dist/src2/optimizer/constant_fold.d.ts +10 -0
  183. package/dist/src2/optimizer/constant_fold.js +85 -0
  184. package/dist/src2/optimizer/copy_prop.d.ts +9 -0
  185. package/dist/src2/optimizer/copy_prop.js +113 -0
  186. package/dist/src2/optimizer/dce.d.ts +8 -0
  187. package/dist/src2/optimizer/dce.js +155 -0
  188. package/dist/src2/optimizer/pipeline.d.ts +10 -0
  189. package/dist/src2/optimizer/pipeline.js +42 -0
  190. package/dist/tsconfig.tsbuildinfo +1 -0
  191. package/docs/compiler-pipeline-redesign.md +2243 -0
  192. package/docs/optimization-ideas.md +1076 -0
  193. package/editors/vscode/package-lock.json +3 -3
  194. package/editors/vscode/package.json +1 -1
  195. package/jest.config.js +1 -1
  196. package/package.json +6 -5
  197. package/scripts/postbuild.js +15 -0
  198. package/src/__tests__/cli.test.ts +8 -220
  199. package/src/__tests__/dce.test.ts +11 -56
  200. package/src/__tests__/diagnostics.test.ts +59 -38
  201. package/src/__tests__/mc-integration.test.ts +1 -2
  202. package/src/ast/types.ts +6 -1
  203. package/src/cli.ts +29 -156
  204. package/src/compile.ts +6 -162
  205. package/src/index.ts +14 -178
  206. package/src/mc-test/runner.ts +4 -3
  207. package/src/parser/index.ts +1 -1
  208. package/src/repl.ts +1 -1
  209. package/src/runtime/index.ts +1 -1
  210. package/src2/__tests__/e2e/basic.test.ts +154 -0
  211. package/src2/__tests__/e2e/macros.test.ts +199 -0
  212. package/src2/__tests__/e2e/migrate.test.ts +3008 -0
  213. package/src2/__tests__/hir/desugar.test.ts +263 -0
  214. package/src2/__tests__/lir/lower.test.ts +619 -0
  215. package/src2/__tests__/lir/types.test.ts +207 -0
  216. package/src2/__tests__/lir/verify.test.ts +249 -0
  217. package/src2/__tests__/mir/arithmetic.test.ts +156 -0
  218. package/src2/__tests__/mir/control-flow.test.ts +242 -0
  219. package/src2/__tests__/mir/verify.test.ts +254 -0
  220. package/src2/__tests__/optimizer/block_merge.test.ts +84 -0
  221. package/src2/__tests__/optimizer/branch_simplify.test.ts +64 -0
  222. package/src2/__tests__/optimizer/constant_fold.test.ts +145 -0
  223. package/src2/__tests__/optimizer/copy_prop.test.ts +99 -0
  224. package/src2/__tests__/optimizer/dce.test.ts +83 -0
  225. package/src2/__tests__/optimizer/pipeline.test.ts +116 -0
  226. package/src2/emit/compile.ts +99 -0
  227. package/src2/emit/index.ts +222 -0
  228. package/src2/hir/lower.ts +428 -0
  229. package/src2/hir/types.ts +216 -0
  230. package/src2/lir/lower.ts +556 -0
  231. package/src2/lir/types.ts +109 -0
  232. package/src2/lir/verify.ts +129 -0
  233. package/src2/mir/lower.ts +1160 -0
  234. package/src2/mir/macro.ts +167 -0
  235. package/src2/mir/types.ts +106 -0
  236. package/src2/mir/verify.ts +218 -0
  237. package/src2/optimizer/block_merge.ts +93 -0
  238. package/src2/optimizer/branch_simplify.ts +27 -0
  239. package/src2/optimizer/constant_fold.ts +88 -0
  240. package/src2/optimizer/copy_prop.ts +106 -0
  241. package/src2/optimizer/dce.ts +133 -0
  242. package/src2/optimizer/pipeline.ts +44 -0
  243. package/tsconfig.json +2 -2
  244. package/src/__tests__/codegen.test.ts +0 -161
  245. package/src/__tests__/e2e.test.ts +0 -2039
  246. package/src/__tests__/entity-types.test.ts +0 -236
  247. package/src/__tests__/lowering.test.ts +0 -1185
  248. package/src/__tests__/macro.test.ts +0 -343
  249. package/src/__tests__/nbt.test.ts +0 -58
  250. package/src/__tests__/optimizer-advanced.test.ts +0 -144
  251. package/src/__tests__/optimizer.test.ts +0 -162
  252. package/src/__tests__/runtime.test.ts +0 -305
  253. package/src/__tests__/stdlib-advanced.test.ts +0 -379
  254. package/src/__tests__/stdlib-bigint.test.ts +0 -427
  255. package/src/__tests__/stdlib-math.test.ts +0 -374
  256. package/src/__tests__/stdlib-vec.test.ts +0 -259
  257. package/src/__tests__/structure-optimizer.test.ts +0 -38
  258. package/src/__tests__/var-allocator.test.ts +0 -75
  259. package/src/codegen/cmdblock/index.ts +0 -63
  260. package/src/codegen/mcfunction/index.ts +0 -662
  261. package/src/codegen/structure/index.ts +0 -346
  262. package/src/codegen/var-allocator.ts +0 -104
  263. package/src/ir/builder.ts +0 -116
  264. package/src/ir/types.ts +0 -134
  265. package/src/lowering/index.ts +0 -3876
  266. package/src/optimizer/commands.ts +0 -534
  267. package/src/optimizer/dce.ts +0 -679
  268. package/src/optimizer/passes.ts +0 -250
  269. package/src/optimizer/structure.ts +0 -450
@@ -0,0 +1,606 @@
1
+ "use strict";
2
+ /**
3
+ * Code generator: IR → mcfunction datapack
4
+ *
5
+ * Output structure:
6
+ * <namespace>/
7
+ * functions/
8
+ * <fn_name>.mcfunction
9
+ * <fn_name>/<block_label>.mcfunction (for control-flow continuations)
10
+ * load.mcfunction (objective setup)
11
+ *
12
+ * Variable mapping:
13
+ * scoreboard objective: "rs"
14
+ * fake player: "$<varname>"
15
+ * temporaries: "$_0", "$_1", ...
16
+ * return value: "$ret"
17
+ * parameters: "$p0", "$p1", ...
18
+ */
19
+ Object.defineProperty(exports, "__esModule", { value: true });
20
+ exports.countMcfunctionCommands = countMcfunctionCommands;
21
+ exports.generateDatapackWithStats = generateDatapackWithStats;
22
+ exports.generateDatapack = generateDatapack;
23
+ const commands_1 = require("../../optimizer/commands");
24
+ const types_1 = require("../../events/types");
25
+ const var_allocator_1 = require("../var-allocator");
26
+ // ---------------------------------------------------------------------------
27
+ // Utilities
28
+ // ---------------------------------------------------------------------------
29
+ // Default scoreboard objective — overridden per-compilation via DatapackGenerationOptions.scoreboardObjective
30
+ let OBJ = 'rs';
31
+ function operandToScore(op, alloc) {
32
+ if (op.kind === 'var')
33
+ return `${alloc.alloc(op.name)} ${OBJ}`;
34
+ if (op.kind === 'const')
35
+ return `${alloc.constant(op.value)} ${OBJ}`;
36
+ if (op.kind === 'param')
37
+ return `${alloc.internal(`p${op.index}`)} ${OBJ}`;
38
+ throw new Error(`Cannot convert storage operand to score: ${op.path}`);
39
+ }
40
+ // Collect all constants used in a function for pre-setup
41
+ function collectConsts(fn) {
42
+ const consts = new Set();
43
+ for (const block of fn.blocks) {
44
+ for (const instr of block.instrs) {
45
+ if (instr.op === 'assign' && instr.src.kind === 'const')
46
+ consts.add(instr.src.value);
47
+ if (instr.op === 'binop') {
48
+ if (instr.lhs.kind === 'const')
49
+ consts.add(instr.lhs.value);
50
+ if (instr.rhs.kind === 'const')
51
+ consts.add(instr.rhs.value);
52
+ }
53
+ if (instr.op === 'cmp') {
54
+ if (instr.lhs.kind === 'const')
55
+ consts.add(instr.lhs.value);
56
+ if (instr.rhs.kind === 'const')
57
+ consts.add(instr.rhs.value);
58
+ }
59
+ }
60
+ const t = block.term;
61
+ if (t.op === 'return' && t.value?.kind === 'const')
62
+ consts.add(t.value.value);
63
+ }
64
+ return consts;
65
+ }
66
+ // MC scoreboard operation suffix
67
+ const BOP_OP = {
68
+ '+': '+=', '-': '-=', '*': '*=', '/': '/=', '%': '%=',
69
+ };
70
+ // ---------------------------------------------------------------------------
71
+ // Instruction codegen
72
+ // ---------------------------------------------------------------------------
73
+ function emitInstr(instr, ns, alloc) {
74
+ const lines = [];
75
+ switch (instr.op) {
76
+ case 'assign': {
77
+ const dst = alloc.alloc(instr.dst);
78
+ const src = instr.src;
79
+ if (src.kind === 'const') {
80
+ lines.push(`scoreboard players set ${dst} ${OBJ} ${src.value}`);
81
+ }
82
+ else if (src.kind === 'var') {
83
+ lines.push(`scoreboard players operation ${dst} ${OBJ} = ${alloc.alloc(src.name)} ${OBJ}`);
84
+ }
85
+ else if (src.kind === 'param') {
86
+ lines.push(`scoreboard players operation ${dst} ${OBJ} = ${alloc.internal(`p${src.index}`)} ${OBJ}`);
87
+ }
88
+ else {
89
+ lines.push(`execute store result score ${dst} ${OBJ} run data get storage ${src.path}`);
90
+ }
91
+ break;
92
+ }
93
+ case 'binop': {
94
+ const dst = alloc.alloc(instr.dst);
95
+ const bop = BOP_OP[instr.bop] ?? '+=';
96
+ // Copy lhs → dst, then apply op with rhs
97
+ lines.push(...emitInstr({ op: 'assign', dst: instr.dst, src: instr.lhs }, ns, alloc));
98
+ lines.push(`scoreboard players operation ${dst} ${OBJ} ${bop} ${operandToScore(instr.rhs, alloc)}`);
99
+ break;
100
+ }
101
+ case 'cmp': {
102
+ // MC doesn't have a direct compare-to-register; use execute store
103
+ const dst = alloc.alloc(instr.dst);
104
+ const lhsScore = operandToScore(instr.lhs, alloc);
105
+ const rhsScore = operandToScore(instr.rhs, alloc);
106
+ lines.push(`scoreboard players set ${dst} ${OBJ} 0`);
107
+ switch (instr.cop) {
108
+ case '==':
109
+ lines.push(`execute if score ${lhsScore} = ${rhsScore} run scoreboard players set ${dst} ${OBJ} 1`);
110
+ break;
111
+ case '!=':
112
+ lines.push(`execute unless score ${lhsScore} = ${rhsScore} run scoreboard players set ${dst} ${OBJ} 1`);
113
+ break;
114
+ case '<':
115
+ lines.push(`execute if score ${lhsScore} < ${rhsScore} run scoreboard players set ${dst} ${OBJ} 1`);
116
+ break;
117
+ case '<=':
118
+ lines.push(`execute if score ${lhsScore} <= ${rhsScore} run scoreboard players set ${dst} ${OBJ} 1`);
119
+ break;
120
+ case '>':
121
+ lines.push(`execute if score ${lhsScore} > ${rhsScore} run scoreboard players set ${dst} ${OBJ} 1`);
122
+ break;
123
+ case '>=':
124
+ lines.push(`execute if score ${lhsScore} >= ${rhsScore} run scoreboard players set ${dst} ${OBJ} 1`);
125
+ break;
126
+ }
127
+ break;
128
+ }
129
+ case 'call': {
130
+ // Push args into the internal parameter slots ($p0, $p1, ...).
131
+ // We emit the copy commands directly (not via emitInstr/alloc.alloc) to
132
+ // ensure the destination resolves to alloc.internal('p{i}') rather than
133
+ // alloc.alloc('p{i}') which would create a *different* user-var slot.
134
+ for (let i = 0; i < instr.args.length; i++) {
135
+ const paramSlot = alloc.internal(`p${i}`);
136
+ const arg = instr.args[i];
137
+ if (arg.kind === 'const') {
138
+ lines.push(`scoreboard players set ${paramSlot} ${OBJ} ${arg.value}`);
139
+ }
140
+ else if (arg.kind === 'var') {
141
+ lines.push(`scoreboard players operation ${paramSlot} ${OBJ} = ${alloc.alloc(arg.name)} ${OBJ}`);
142
+ }
143
+ else if (arg.kind === 'param') {
144
+ lines.push(`scoreboard players operation ${paramSlot} ${OBJ} = ${alloc.internal(`p${arg.index}`)} ${OBJ}`);
145
+ }
146
+ // storage args are rare for call sites; fall through to no-op
147
+ }
148
+ lines.push(`function ${ns}:${instr.fn}`);
149
+ if (instr.dst) {
150
+ const retSlot = alloc.internal('ret');
151
+ lines.push(`scoreboard players operation ${alloc.alloc(instr.dst)} ${OBJ} = ${retSlot} ${OBJ}`);
152
+ }
153
+ break;
154
+ }
155
+ case 'raw': {
156
+ // resolveRaw rewrites $var tokens that are registered in the allocator
157
+ // so that mangle=true mode produces correct mangled names instead of
158
+ // the raw IR names embedded by the lowering phase.
159
+ // \x01 is a sentinel for the MC macro line-start '$' (used by
160
+ // storage_get_int sub-functions). Replace it last, after resolveRaw,
161
+ // so '$execute' is never treated as a variable reference.
162
+ const rawResolved = alloc.resolveRaw(instr.cmd).replace(/^\x01/, '$');
163
+ lines.push(rawResolved);
164
+ break;
165
+ }
166
+ }
167
+ return lines;
168
+ }
169
+ // ---------------------------------------------------------------------------
170
+ // Terminator codegen
171
+ // ---------------------------------------------------------------------------
172
+ function emitTerm(term, ns, fnName, alloc) {
173
+ const lines = [];
174
+ switch (term.op) {
175
+ case 'jump':
176
+ lines.push(`function ${ns}:${fnName}/${term.target}`);
177
+ break;
178
+ case 'jump_if':
179
+ lines.push(`execute if score ${alloc.alloc(term.cond)} ${OBJ} matches 1.. run function ${ns}:${fnName}/${term.then}`);
180
+ lines.push(`execute if score ${alloc.alloc(term.cond)} ${OBJ} matches ..0 run function ${ns}:${fnName}/${term.else_}`);
181
+ break;
182
+ case 'jump_unless':
183
+ lines.push(`execute if score ${alloc.alloc(term.cond)} ${OBJ} matches ..0 run function ${ns}:${fnName}/${term.then}`);
184
+ lines.push(`execute if score ${alloc.alloc(term.cond)} ${OBJ} matches 1.. run function ${ns}:${fnName}/${term.else_}`);
185
+ break;
186
+ case 'return': {
187
+ // Emit the copy to the shared return slot directly — do NOT go through
188
+ // emitInstr/alloc.alloc(retSlot) which would allocate a *user* var slot
189
+ // (different from the internal slot) and break mangle mode.
190
+ const retSlot = alloc.internal('ret');
191
+ if (term.value) {
192
+ if (term.value.kind === 'const') {
193
+ lines.push(`scoreboard players set ${retSlot} ${OBJ} ${term.value.value}`);
194
+ }
195
+ else if (term.value.kind === 'var') {
196
+ lines.push(`scoreboard players operation ${retSlot} ${OBJ} = ${alloc.alloc(term.value.name)} ${OBJ}`);
197
+ }
198
+ else if (term.value.kind === 'param') {
199
+ lines.push(`scoreboard players operation ${retSlot} ${OBJ} = ${alloc.internal(`p${term.value.index}`)} ${OBJ}`);
200
+ }
201
+ }
202
+ // MC 1.20+: use `return` to propagate the value back to the caller's
203
+ // `execute store result … run function …` without an extra scoreboard read.
204
+ if (term.value?.kind === 'const') {
205
+ lines.push(`return ${term.value.value}`);
206
+ }
207
+ else if (term.value?.kind === 'var') {
208
+ lines.push(`return run scoreboard players get ${alloc.alloc(term.value.name)} ${OBJ}`);
209
+ }
210
+ else if (term.value?.kind === 'param') {
211
+ lines.push(`return run scoreboard players get ${alloc.internal(`p${term.value.index}`)} ${OBJ}`);
212
+ }
213
+ break;
214
+ }
215
+ case 'tick_yield':
216
+ lines.push(`schedule function ${ns}:${fnName}/${term.continuation} 1t replace`);
217
+ break;
218
+ }
219
+ return lines;
220
+ }
221
+ function toFunctionName(file) {
222
+ const match = file.path.match(/^data\/[^/]+\/function\/(.+)\.mcfunction$/);
223
+ return match?.[1] ?? null;
224
+ }
225
+ function applyFunctionOptimization(files) {
226
+ const functionFiles = files
227
+ .map(file => {
228
+ const functionName = toFunctionName(file);
229
+ if (!functionName)
230
+ return null;
231
+ const commands = file.content
232
+ .split('\n')
233
+ .map(line => line.trim())
234
+ .filter(line => line !== '' && !line.startsWith('#'))
235
+ .map(cmd => ({ cmd }));
236
+ return { file, functionName, commands };
237
+ })
238
+ .filter((entry) => entry !== null);
239
+ const optimized = (0, commands_1.optimizeCommandFunctions)(functionFiles.map(entry => ({
240
+ name: entry.functionName,
241
+ commands: entry.commands,
242
+ })));
243
+ const commandMap = new Map(optimized.functions.map(fn => [fn.name, fn.commands]));
244
+ // Filter out files for functions that were removed (inlined trivial functions)
245
+ const optimizedNames = new Set(optimized.functions.map(fn => fn.name));
246
+ return {
247
+ files: files
248
+ .filter(file => {
249
+ const functionName = toFunctionName(file);
250
+ // Keep non-function files and functions that weren't removed
251
+ return !functionName || optimizedNames.has(functionName);
252
+ })
253
+ .map(file => {
254
+ const functionName = toFunctionName(file);
255
+ if (!functionName)
256
+ return file;
257
+ const commands = commandMap.get(functionName);
258
+ if (!commands)
259
+ return file;
260
+ const lines = file.content.split('\n');
261
+ const header = lines.filter(line => line.trim().startsWith('#'));
262
+ return {
263
+ ...file,
264
+ content: [...header, ...commands.map(command => command.cmd)].join('\n'),
265
+ };
266
+ }),
267
+ stats: optimized.stats,
268
+ };
269
+ }
270
+ function countMcfunctionCommands(files) {
271
+ return files.reduce((sum, file) => {
272
+ if (!toFunctionName(file)) {
273
+ return sum;
274
+ }
275
+ return sum + file.content
276
+ .split('\n')
277
+ .map(line => line.trim())
278
+ .filter(line => line !== '' && !line.startsWith('#'))
279
+ .length;
280
+ }, 0);
281
+ }
282
+ // ---------------------------------------------------------------------------
283
+ // Pre-allocation helpers for the two-pass mangle strategy
284
+ // ---------------------------------------------------------------------------
285
+ /** Register every variable referenced in an instruction with the allocator. */
286
+ function preAllocInstr(instr, alloc) {
287
+ switch (instr.op) {
288
+ case 'assign':
289
+ alloc.alloc(instr.dst);
290
+ if (instr.src.kind === 'var')
291
+ alloc.alloc(instr.src.name);
292
+ break;
293
+ case 'binop':
294
+ alloc.alloc(instr.dst);
295
+ if (instr.lhs.kind === 'var')
296
+ alloc.alloc(instr.lhs.name);
297
+ if (instr.rhs.kind === 'var')
298
+ alloc.alloc(instr.rhs.name);
299
+ break;
300
+ case 'cmp':
301
+ alloc.alloc(instr.dst);
302
+ if (instr.lhs.kind === 'var')
303
+ alloc.alloc(instr.lhs.name);
304
+ if (instr.rhs.kind === 'var')
305
+ alloc.alloc(instr.rhs.name);
306
+ break;
307
+ case 'call':
308
+ for (const arg of instr.args) {
309
+ if (arg.kind === 'var')
310
+ alloc.alloc(arg.name);
311
+ }
312
+ if (instr.dst)
313
+ alloc.alloc(instr.dst);
314
+ break;
315
+ case 'raw':
316
+ // Scan for $varname tokens and pre-register each one
317
+ ;
318
+ instr.cmd.replace(/\$[A-Za-z_][A-Za-z0-9_]*/g, (tok) => {
319
+ alloc.alloc(tok);
320
+ return tok;
321
+ });
322
+ break;
323
+ }
324
+ }
325
+ /** Register every variable referenced in a terminator with the allocator. */
326
+ function preAllocTerm(term, alloc) {
327
+ switch (term.op) {
328
+ case 'jump_if':
329
+ case 'jump_unless':
330
+ alloc.alloc(term.cond);
331
+ break;
332
+ case 'return':
333
+ if (term.value?.kind === 'var')
334
+ alloc.alloc(term.value.name);
335
+ break;
336
+ }
337
+ }
338
+ function generateDatapackWithStats(module, options = {}) {
339
+ const { optimizeCommands = true, mangle = false, scoreboardObjective = 'rs' } = options;
340
+ // Set module-level OBJ so all helper functions in this module use the correct objective.
341
+ // This is safe because compilation is synchronous.
342
+ OBJ = scoreboardObjective;
343
+ (0, commands_1.setOptimizerObjective)(scoreboardObjective);
344
+ const alloc = new var_allocator_1.VarAllocator(mangle);
345
+ const files = [];
346
+ const advancements = [];
347
+ const ns = module.namespace;
348
+ // Collect all trigger handlers
349
+ const triggerHandlers = module.functions.filter(fn => fn.isTriggerHandler && fn.triggerName);
350
+ const triggerNames = new Set(triggerHandlers.map(fn => fn.triggerName));
351
+ const eventHandlers = module.functions.filter((fn) => !!fn.eventHandler && (0, types_1.isEventTypeName)(fn.eventHandler.eventType));
352
+ const eventTypes = new Set(eventHandlers.map(fn => fn.eventHandler.eventType));
353
+ // Collect all tick functions
354
+ const tickFunctionNames = [];
355
+ for (const fn of module.functions) {
356
+ if (fn.isTickLoop) {
357
+ tickFunctionNames.push(fn.name);
358
+ }
359
+ }
360
+ // pack.mcmeta
361
+ files.push({
362
+ path: 'pack.mcmeta',
363
+ content: JSON.stringify({
364
+ pack: { pack_format: 26, description: `${ns} datapack — compiled by redscript` }
365
+ }, null, 2),
366
+ });
367
+ // __load.mcfunction — create scoreboard objective + trigger registrations
368
+ const loadLines = [
369
+ `# RedScript runtime init`,
370
+ `scoreboard objectives add ${OBJ} dummy`,
371
+ ];
372
+ for (const g of module.globals) {
373
+ loadLines.push(`scoreboard players set ${alloc.alloc(g.name)} ${OBJ} ${g.init}`);
374
+ }
375
+ // Add trigger objectives
376
+ for (const triggerName of triggerNames) {
377
+ loadLines.push(`scoreboard objectives add ${triggerName} trigger`);
378
+ loadLines.push(`scoreboard players enable @a ${triggerName}`);
379
+ }
380
+ for (const eventType of eventTypes) {
381
+ const detection = types_1.EVENT_TYPES[eventType].detection;
382
+ if (eventType === 'PlayerDeath') {
383
+ loadLines.push(`scoreboard objectives add ${OBJ}.deaths deathCount`);
384
+ }
385
+ else if (eventType === 'EntityKill') {
386
+ loadLines.push(`scoreboard objectives add ${OBJ}.kills totalKillCount`);
387
+ }
388
+ else if (eventType === 'ItemUse') {
389
+ loadLines.push('# ItemUse detection requires a project-specific objective/tag setup');
390
+ }
391
+ else if (detection === 'tag' || detection === 'advancement') {
392
+ loadLines.push(`# ${eventType} detection expects tag ${types_1.EVENT_TYPES[eventType].tag} to be set externally`);
393
+ }
394
+ }
395
+ // Generate trigger dispatch functions
396
+ for (const triggerName of triggerNames) {
397
+ const handlers = triggerHandlers.filter(fn => fn.triggerName === triggerName);
398
+ // __trigger_{name}_dispatch.mcfunction
399
+ const dispatchLines = [
400
+ `# Trigger dispatch for ${triggerName}`,
401
+ ];
402
+ for (const handler of handlers) {
403
+ dispatchLines.push(`function ${ns}:${handler.name}`);
404
+ }
405
+ dispatchLines.push(`scoreboard players set @s ${triggerName} 0`);
406
+ dispatchLines.push(`scoreboard players enable @s ${triggerName}`);
407
+ files.push({
408
+ path: `data/${ns}/function/__trigger_${triggerName}_dispatch.mcfunction`,
409
+ content: dispatchLines.join('\n'),
410
+ });
411
+ }
412
+ // Collect all constants across all functions first (deduplicated)
413
+ const allConsts = new Set();
414
+ for (const fn of module.functions) {
415
+ for (const c of collectConsts(fn))
416
+ allConsts.add(c);
417
+ }
418
+ if (allConsts.size > 0) {
419
+ loadLines.push(...Array.from(allConsts).sort((a, b) => a - b).map(value => `scoreboard players set ${alloc.constant(value)} ${OBJ} ${value}`));
420
+ }
421
+ // ─────────────────────────────────────────────────────────────────────────
422
+ // Pre-allocation pass (mangle mode only)
423
+ //
424
+ // When mangle=true, the codegen assigns sequential names ($a, $b, …) the
425
+ // FIRST time alloc.alloc() is called for a given variable. Raw IR commands
426
+ // embed variable names (e.g. "$_0") as plain strings; resolveRaw() can only
427
+ // substitute them if the name was already registered in the allocator.
428
+ //
429
+ // Problem: a freshTemp ($\_0) used in a `raw` instruction and then in the
430
+ // immediately following `assign` gets registered by the `assign` AFTER the
431
+ // `raw` has already been emitted — so resolveRaw sees an unknown name and
432
+ // passes it through verbatim ($\_0), while the assign emits a different
433
+ // mangled slot ($e). The two slots never meet and the value is lost.
434
+ //
435
+ // Fix: walk every instruction (and terminator) of every function in order
436
+ // and call alloc.alloc() for each variable reference. This registers all
437
+ // names — with the same sequential order the main emit pass will encounter
438
+ // them — so that resolveRaw() can always find the correct mangled name.
439
+ // ─────────────────────────────────────────────────────────────────────────
440
+ if (mangle) {
441
+ for (const fn of module.functions) {
442
+ // Register internals used by the calling convention
443
+ for (let i = 0; i < fn.params.length; i++)
444
+ alloc.internal(`p${i}`);
445
+ alloc.internal('ret');
446
+ for (const block of fn.blocks) {
447
+ for (const instr of block.instrs) {
448
+ preAllocInstr(instr, alloc);
449
+ }
450
+ preAllocTerm(block.term, alloc);
451
+ }
452
+ }
453
+ }
454
+ // Generate each function
455
+ for (const fn of module.functions) {
456
+ // Entry block → <fn_name>.mcfunction
457
+ // Continuation blocks → <fn_name>/<label>.mcfunction
458
+ for (let i = 0; i < fn.blocks.length; i++) {
459
+ const block = fn.blocks[i];
460
+ const lines = [`# block: ${block.label}`];
461
+ // Param setup is now handled by the lowering IR itself via { kind: 'param' }
462
+ // operands, so we no longer need a separate codegen param-copy loop here.
463
+ // (Removing it prevents the double-assignment that caused mangle-mode collisions.)
464
+ for (const instr of block.instrs) {
465
+ lines.push(...emitInstr(instr, ns, alloc));
466
+ }
467
+ lines.push(...emitTerm(block.term, ns, fn.name, alloc));
468
+ const filePath = i === 0
469
+ ? `data/${ns}/function/${fn.name}.mcfunction`
470
+ : `data/${ns}/function/${fn.name}/${block.label}.mcfunction`;
471
+ // Skip empty continuation blocks (only contain the block comment, no real commands)
472
+ // Entry block (i === 0) is always emitted so the function file exists
473
+ const hasRealContent = lines.some(l => !l.startsWith('#') && l.trim() !== '');
474
+ if (i !== 0 && !hasRealContent)
475
+ continue;
476
+ files.push({ path: filePath, content: lines.join('\n') });
477
+ }
478
+ }
479
+ // Call @load functions and @requires-referenced load helpers from __load.
480
+ // We collect them in a set to deduplicate (multiple fns might @requires the same dep).
481
+ const loadCalls = new Set();
482
+ for (const fn of module.functions) {
483
+ if (fn.isLoadInit) {
484
+ loadCalls.add(fn.name);
485
+ }
486
+ // @requires: if this fn is compiled in, its required load-helpers must also run
487
+ for (const dep of fn.requiredLoads ?? []) {
488
+ loadCalls.add(dep);
489
+ }
490
+ }
491
+ for (const name of loadCalls) {
492
+ loadLines.push(`function ${ns}:${name}`);
493
+ }
494
+ // Write __load.mcfunction
495
+ files.push({
496
+ path: `data/${ns}/function/__load.mcfunction`,
497
+ content: loadLines.join('\n'),
498
+ });
499
+ // minecraft:load tag pointing to __load
500
+ files.push({
501
+ path: `data/minecraft/tags/function/load.json`,
502
+ content: JSON.stringify({ values: [`${ns}:__load`] }, null, 2),
503
+ });
504
+ // __tick.mcfunction — calls all @tick functions + trigger check
505
+ const tickLines = ['# RedScript tick dispatcher'];
506
+ // Call all @tick functions
507
+ for (const fnName of tickFunctionNames) {
508
+ tickLines.push(`function ${ns}:${fnName}`);
509
+ }
510
+ // Call trigger check if there are triggers
511
+ if (triggerNames.size > 0) {
512
+ tickLines.push(`# Trigger checks`);
513
+ for (const triggerName of triggerNames) {
514
+ tickLines.push(`execute as @a[scores={${triggerName}=1..}] run function ${ns}:__trigger_${triggerName}_dispatch`);
515
+ }
516
+ }
517
+ if (eventHandlers.length > 0) {
518
+ tickLines.push('# Event checks');
519
+ for (const eventType of eventTypes) {
520
+ const tag = types_1.EVENT_TYPES[eventType].tag;
521
+ const handlers = eventHandlers.filter(fn => fn.eventHandler?.eventType === eventType);
522
+ for (const handler of handlers) {
523
+ tickLines.push(`execute as @a[tag=${tag}] run function ${ns}:${handler.name}`);
524
+ }
525
+ tickLines.push(`tag @a[tag=${tag}] remove ${tag}`);
526
+ }
527
+ }
528
+ // Only generate __tick if there's something to run
529
+ if (tickFunctionNames.length > 0 || triggerNames.size > 0 || eventHandlers.length > 0) {
530
+ files.push({
531
+ path: `data/${ns}/function/__tick.mcfunction`,
532
+ content: tickLines.join('\n'),
533
+ });
534
+ // minecraft:tick tag pointing to __tick
535
+ files.push({
536
+ path: `data/minecraft/tags/function/tick.json`,
537
+ content: JSON.stringify({ values: [`${ns}:__tick`] }, null, 2),
538
+ });
539
+ }
540
+ for (const fn of module.functions) {
541
+ const eventTrigger = fn.eventTrigger;
542
+ if (!eventTrigger) {
543
+ continue;
544
+ }
545
+ let path = '';
546
+ let criteria = {};
547
+ switch (eventTrigger.kind) {
548
+ case 'advancement':
549
+ path = `data/${ns}/advancements/on_advancement_${fn.name}.json`;
550
+ criteria = {
551
+ trigger: {
552
+ trigger: `minecraft:${eventTrigger.value}`,
553
+ },
554
+ };
555
+ break;
556
+ case 'craft':
557
+ path = `data/${ns}/advancements/on_craft_${fn.name}.json`;
558
+ criteria = {
559
+ crafted: {
560
+ trigger: 'minecraft:inventory_changed',
561
+ conditions: {
562
+ items: [
563
+ {
564
+ items: [eventTrigger.value],
565
+ },
566
+ ],
567
+ },
568
+ },
569
+ };
570
+ break;
571
+ case 'death':
572
+ path = `data/${ns}/advancements/on_death_${fn.name}.json`;
573
+ criteria = {
574
+ death: {
575
+ trigger: 'minecraft:entity_killed_player',
576
+ },
577
+ };
578
+ break;
579
+ case 'login':
580
+ case 'join_team':
581
+ continue;
582
+ }
583
+ advancements.push({
584
+ path,
585
+ content: JSON.stringify({
586
+ criteria,
587
+ rewards: {
588
+ function: `${ns}:${fn.name}`,
589
+ },
590
+ }, null, 2),
591
+ });
592
+ }
593
+ const stats = (0, commands_1.createEmptyOptimizationStats)();
594
+ const sourceMap = mangle ? alloc.toSourceMap() : undefined;
595
+ if (!optimizeCommands) {
596
+ return { files, advancements, stats, sourceMap };
597
+ }
598
+ const optimized = applyFunctionOptimization(files);
599
+ (0, commands_1.mergeOptimizationStats)(stats, optimized.stats);
600
+ return { files: optimized.files, advancements, stats, sourceMap };
601
+ }
602
+ function generateDatapack(module) {
603
+ const generated = generateDatapackWithStats(module);
604
+ return [...generated.files, ...generated.advancements];
605
+ }
606
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1,24 @@
1
+ import { type OptimizationStats } from '../../optimizer/commands';
2
+ import type { IRModule } from '../../ir/types';
3
+ import type { DatapackFile } from '../mcfunction';
4
+ export interface StructureBlockInfo {
5
+ command: string;
6
+ conditional: boolean;
7
+ state: number;
8
+ functionName: string;
9
+ lineNumber: number;
10
+ }
11
+ export interface StructureCompileResult {
12
+ buffer: Buffer;
13
+ blockCount: number;
14
+ blocks: StructureBlockInfo[];
15
+ stats?: OptimizationStats;
16
+ }
17
+ export interface StructureCompileOptions {
18
+ dce?: boolean;
19
+ mangle?: boolean;
20
+ }
21
+ export declare function generateStructure(input: IRModule | DatapackFile[], options?: {
22
+ mangle?: boolean;
23
+ }): StructureCompileResult;
24
+ export declare function compileToStructure(source: string, namespace: string, filePath?: string, options?: StructureCompileOptions): StructureCompileResult;