@tilua/compiler 6.1.0 → 7.0.0

package/dist/{chunk-574KDV5W.js → chunk-NM3EVX2J.js}

@@ -27,7 +27,7 @@ var index = (object2, key) => ({ type: "IndexExpression", object: object2, index
 var call = (callee, args) => ({ type: "CallExpression", callee, arguments: args, ...span() });
 var methodCall = (object2, method, args) => ({ type: "MethodCallExpression", object: object2, method: identifier(method), arguments: args, ...span() });
 var functionExpression = (func) => ({ type: "FunctionExpression", func, ...span() });
-var binary = (operator, left, right) => ({ type: "BinaryExpression", operator, left, right, ...span() });
+var binary = (operator, left, right) => ({ type: "BinaryExpression", operator, left: operator === "^" && left.type === "UnaryExpression" ? parenthesized(left) : left, right, ...span() });
 var unary = (operator, argument) => ({ type: "UnaryExpression", operator, argument, ...span() });
 var parenthesized = (expression) => ({ type: "ParenthesizedExpression", expression, ...span() });
 var table = (fields) => ({ type: "TableExpression", fields, ...span() });
@@ -86,19 +86,19 @@ var Names = class _Names {
    *  a name shared across modules (the module table) collides with none. */
   static from(...programs) {
     const taken = /* @__PURE__ */ new Set(["select", "pairs", "table", "tostring", "setmetatable", "error"]);
-    const walk = (value) => {
+    const walk2 = (value) => {
       if (!value || typeof value !== "object") return;
       if (Array.isArray(value)) {
-        for (const item of value) walk(item);
+        for (const item of value) walk2(item);
         return;
       }
       const node = value;
       if (typeof node.name === "string" && node.name) taken.add(node.name);
       for (const [key, child] of Object.entries(value)) {
-        if (key !== "line" && key !== "column") walk(child);
+        if (key !== "line" && key !== "column") walk2(child);
       }
     };
-    for (const program2 of programs) walk(program2);
+    for (const program2 of programs) walk2(program2);
     return new _Names(taken);
   }
   /** A copy that goes on to take names independently — one per module, each
@@ -106,6 +106,10 @@ var Names = class _Names {
   fork() {
     return new _Names(new Set(this.taken));
   }
+  /** `name`, chosen elsewhere, is not to be chosen again here. */
+  take(name) {
+    this.taken.add(name);
+  }
   /** `base`, or `base2`, `base3`, ... — whichever is free first. */
   fresh(base) {
     let name = base;
@@ -123,6 +127,7 @@ var implemented = (source) => new Set([...source.matchAll(/function __NAME__\.(\
 var array = read("array.luau");
 var string2 = read("string.luau");
 var object = read("object.luau");
+var json = read("json.luau");
 var arrayMethods = implemented(array);
 var stringMethods = implemented(string2);
 var objectMethods = implemented(object);
@@ -155,7 +160,7 @@ function objectKeys(type) {
   return { known, open: String(open) };
 }
 var plugin = {
-  runtime: { array, string: string2, object },
+  runtime: { array, string: string2, object, json },
   methodCall({ method, receiver, use }) {
     if (isString(receiver)) {
       return stringMethods.has(method) ? { callee: `${use("string")}.${method}` } : void 0;
@@ -244,6 +249,141 @@ var plugin2 = {
 var standardLowering = { plugin: plugin2, from: "tilua (standard library)" };
 
 // src/lower.ts
+function syntaxOf(lowerings) {
+  return new Set((lowerings ?? []).flatMap((l) => l.plugin.syntax ?? []));
+}
+var SharedRuntime = class {
+  constructor(names) {
+    this.names = names;
+  }
+  names;
+  byKey = /* @__PURE__ */ new Map();
+  /** Each group of names and the statements that declare them as locals,
+   *  in the order they were first defined: each after what it uses. */
+  definitions = [];
+  defined = /* @__PURE__ */ new Set();
+  /** The name the helper `key` goes by, in every module. */
+  name(key, base) {
+    let name = this.byKey.get(key);
+    if (!name) {
+      name = this.names.fresh(base);
+      this.byKey.set(key, name);
+    }
+    return name;
+  }
+  /** `statements` declare `names`. Only the first module to say so is heard:
+   *  the others' are the same. */
+  define(names, statements) {
+    if (names.every((name) => this.defined.has(name))) return;
+    for (const name of names) this.defined.add(name);
+    this.definitions.push({ names, statements });
+  }
+  /** Tree shaking: leave out of each runtime table the members nothing
+   *  reads. `code` is everything else the output holds — the modules —
+   *  and a member is read when it, or a member read already, names it:
+   *  `tilua_array.filter(xs, f)` keeps `filter`, and whatever `filter`
+   *  itself calls. A table used as anything but `table.member` — handed
+   *  on, walked, indexed by what only the running program knows — keeps
+   *  every member. */
+  shake(code) {
+    const names = new Set(this.definitions.flatMap((group) => group.names));
+    const read3 = /* @__PURE__ */ new Map();
+    const use = (name, key) => {
+      const now = read3.get(name);
+      if (now === "all") return;
+      if (key === void 0) {
+        read3.set(name, "all");
+        return;
+      }
+      read3.set(name, (now ?? /* @__PURE__ */ new Set()).add(key));
+    };
+    const isRead = (name, key) => {
+      const now = read3.get(name);
+      return now === "all" || (now?.has(key) ?? false);
+    };
+    const scan = (value) => {
+      if (!value || typeof value !== "object") return;
+      if (Array.isArray(value)) return void value.forEach(scan);
+      const node = value;
+      const object2 = node.object;
+      if (object2?.type === "Identifier" && names.has(object2.name)) {
+        if (node.type === "MemberExpression") return use(object2.name, node.property.name);
+        if (node.type === "MethodCallExpression") {
+          use(object2.name, node.method.name);
+          return scan(node.arguments);
+        }
+        if (node.type === "IndexExpression") {
+          const index2 = node.index;
+          if (index2.type === "StringLiteral") return use(object2.name, String(index2.value));
+        }
+      }
+      if (node.type === "Identifier" && names.has(node.name)) return use(node.name);
+      if (node.type === "LocalFunctionStatement") return scan(node.func);
+      for (const [key, child] of Object.entries(node)) {
+        if (key !== "line" && key !== "column") scan(child);
+      }
+    };
+    const defines = (statement, own) => {
+      if (statement.type === "FunctionDeclarationStatement") {
+        const target = statement.target;
+        if (own.includes(target.base.name) && target.path.length === 1 && !target.method) {
+          return { name: target.base.name, key: target.path[0].name, value: statement.func };
+        }
+      }
+      if (statement.type === "AssignmentStatement" && statement.targets.length === 1 && statement.values.length === 1) {
+        const target = statement.targets[0];
+        if (target.object?.type === "Identifier" && own.includes(target.object.name)) {
+          if (target.type === "MemberExpression") return { name: target.object.name, key: target.property.name, value: statement.values[0] };
+          if (target.type === "IndexExpression" && target.index?.type === "StringLiteral") {
+            return { name: target.object.name, key: String(target.index.value), value: statement.values[0] };
+          }
+        }
+      }
+      return void 0;
+    };
+    scan(code);
+    const scanned = /* @__PURE__ */ new Set();
+    let more = true;
+    while (more) {
+      more = false;
+      for (const group of this.definitions) {
+        for (const statement of group.statements) {
+          if (scanned.has(statement)) continue;
+          const definition = defines(statement, group.names);
+          if (definition && !isRead(definition.name, definition.key)) continue;
+          scanned.add(statement);
+          more = true;
+          scan(definition ? definition.value : statement);
+        }
+      }
+    }
+    for (const group of this.definitions) {
+      group.statements = group.statements.filter((statement) => {
+        const definition = defines(statement, group.names);
+        return !definition || isRead(definition.name, definition.key);
+      });
+    }
+  }
+  /** Luau setting up the helpers `used` for a chunk whose code names them:
+   *  `declare` goes before that code, which reads each as an upvalue, and
+   *  `define` after it — before any of it runs, and after what a helper
+   *  reads as it is set up (the line map). Each definition runs in a
+   *  function of its own, so what it keeps to itself stays there. */
+  chunk(used) {
+    const groups = this.definitions.filter((group) => group.names.some((name) => used.has(name)));
+    const all = groups.flatMap((group) => group.names);
+    const declare = [];
+    for (let i = 0; i < all.length; i += 100) declare.push(local(all.slice(i, i + 100), []));
+    const define = groups.map((group) => assign(
+      group.names.map((name) => identifier(name)),
+      [call(parenthesized(functionExpression(functionBody([], [
+        ...group.statements,
+        returns(group.names.map((name) => identifier(name)))
+      ]))), [])]
+    ));
+    return { declare, define };
+  }
+};
 function lower(program2, scopes, options = {}) {
   const result2 = new Lowerer(program2, scopes, options).run();
   dropUnreachable(result2.statements);
@@ -274,7 +414,7 @@ var Lowerer = class {
     this.source = source;
     this.scopes = scopes;
     this.options = options;
-    this.lua51 = options.target === "lua51";
+    this.syntax = syntaxOf(options.lowerings);
     this.names = options.names ? options.names.fork() : Names.from(source);
     this.exportsName = this.names.fresh("exports");
     for (const binding of scopes.bindings.values()) {
@@ -299,6 +439,11 @@ var Lowerer = class {
   names;
   diagnostics = [];
   helpers = /* @__PURE__ */ new Map();
+  /** The shared helpers this file names. */
+  sharedUsed = /* @__PURE__ */ new Set();
+  /** While a shared helper is written: it is defined outside every module,
+   *  where none of a module's own names shadows a global. */
+  rawGlobals = false;
   /** The binding each declaration node creates. */
   bindingByDeclaration = /* @__PURE__ */ new Map();
   /** tilua names that are Luau keywords, and what they are called instead. */
@@ -314,8 +459,8 @@ var Lowerer = class {
   tryFrames = [];
   /** The loops being lowered in the current function, innermost last. */
   loops = [];
-  /** Whether the output has to be plain Lua 5.1 rather than Luau. */
-  lua51;
+  /** What the output's Lua can write beyond Lua 5.1 (`syntaxOf`). */
+  syntax;
   /** The source line each emitted statement came from. */
   origins = /* @__PURE__ */ new WeakMap();
   /** The name each class written as a value is known by. */
@@ -323,15 +468,19 @@ var Lowerer = class {
   run() {
     const body = this.options.module ? this.module(this.options.module) : this.topLevel(this.source.body.statements);
     const scriptArgs = this.usesScriptArgs && !this.options.module ? [local(["scriptArgs"], [this.array(table([{ type: "TableFieldPositional", value: vararg() }]))])] : [];
+    this.rawGlobals = this.options.shared !== void 0;
     const helpers = this.helperDefinitions();
+    const classes = this.classRuntimeStatements();
+    this.rawGlobals = false;
     const captured = [...this.builtins].map(([global, local2]) => local([local2], [identifier(global)]));
     return {
-      statements: [...captured, ...helpers, ...scriptArgs, ...this.classRuntimeStatements(), ...body],
+      statements: [...captured, ...helpers, ...scriptArgs, ...classes, ...body],
       exportsName: this.exportsName,
       module: this.info,
       diagnostics: this.diagnostics,
       origins: this.origins,
-      usesScriptArgs: this.usesScriptArgs
+      usesScriptArgs: this.usesScriptArgs,
+      shared: [...this.sharedUsed]
     };
   }
   /** Whether the file reads `scriptArgs`, the language's own global. */
@@ -364,6 +513,7 @@ var Lowerer = class {
   /** A global the generated code calls — `pairs`, `table`, ... — under a name
    *  the source cannot have shadowed. */
   builtin(global) {
+    if (this.rawGlobals) return identifier(global);
     let local2 = this.builtins.get(global);
     if (!local2) {
       const shadowed = [...this.scopes.bindings.values()].some((b) => b.name === global && b.kind !== "global");
@@ -375,16 +525,31 @@ var Lowerer = class {
   }
   /** `table.move`. Lua 5.1 has none, and gets a function that does the same. */
   move() {
-    return this.lua51 ? this.helper("move") : member(this.builtin("table"), "move");
+    return this.helper("move");
   }
   helper(kind) {
     let name = this.helpers.get(kind);
     if (!name) {
-      name = this.names.fresh(`tilua_${kind === "array" ? "asArray" : kind}`);
+      name = this.sharedName(`helper:${kind}`, `tilua_${kind === "array" ? "asArray" : kind}`);
       this.helpers.set(kind, name);
     }
     return identifier(name);
   }
+  /** A fresh name for a helper — in a bundle, the one every module calls
+   *  that helper by. */
+  sharedName(key, base) {
+    const shared = this.options.shared;
+    if (!shared) return this.names.fresh(base);
+    const name = shared.name(key, base);
+    this.names.take(name);
+    this.sharedUsed.add(name);
+    return name;
+  }
+  /** `statements` declare `names`: in this file, or once for the bundle. */
+  define(names, statements, out) {
+    if (this.options.shared) this.options.shared.define(names, statements);
+    else out.push(...statements);
+  }
   // `typeof v` tells an array from an object, which Luau cannot: both are
   // tables. So every array the program makes — `[a, b]`, a rest parameter,
   // what `:map` answers — is written down as one, in a table that holds its
@@ -417,7 +582,8 @@ var Lowerer = class {
     const key = `${index2}:${runtime2}`;
     let name = this.loweringNames.get(key);
     if (!name) {
-      name = this.names.fresh(`tilua_${runtime2.replace(/[^A-Za-z0-9_]/g, "_")}`);
+      const lines = this.lowerings[index2].plugin.runtime?.[runtime2]?.includes("__LINES__") ? this.options.module?.lines : "";
+      name = this.sharedName(`lowering:${key}:${lines}`, `tilua_${runtime2.replace(/[^A-Za-z0-9_]/g, "_")}`);
       this.loweringNames.set(key, name);
       this.loweringUsed.push({ index: index2, runtime: runtime2, name });
     }
@@ -617,19 +783,25 @@ var Lowerer = class {
     const runtimes = this.runtimeDefinitions();
     if (this.helpers.has("concat")) this.helper("array");
     if (this.helpers.has("concat")) this.move();
+    const unpack = this.helpers.get("unpack");
+    if (unpack) {
+      this.define([unpack], parseLuau(
+        `local ${unpack} = ${this.builtin("table").name}.unpack or ${this.builtin("unpack").name}`
+      ).body.statements, out);
+    }
     const move = this.helpers.get("move");
     if (move) {
-      out.push(...parseLuau(`
-local function ${move}(from, first, last, at, to)
+      this.define([move], parseLuau(`
+local ${move} = ${this.builtin("table").name}.move or function(from, first, last, at, to)
 	for i = first, last do
 		to[at + i - first] = from[i]
 	end
 	return to
-end`).body.statements);
+end`).body.statements, out);
     }
     const each = this.helpers.get("each");
     if (each) {
-      out.push(...parseLuau(`
+      this.define([each], parseLuau(`
 local function ${each}(source, ...)
 	if ${this.builtin("type").name}(source) == "function" then
 		return source, ...
@@ -639,7 +811,7 @@ local function ${each}(source, ...)
 		return meta.__iter(source)
 	end
 	return ${this.builtin("next").name}, source
-end`).body.statements);
+end`).body.statements, out);
     }
     const marker = this.helpers.get("array");
     const typeOf = this.helpers.get("typeof");
@@ -647,19 +819,19 @@ end`).body.statements);
       const arrays = this.arrays();
       const own = this.helpers.get("arrays");
       if (own) {
-        out.push(...parseLuau(`local ${own} = ${this.builtin("setmetatable").name}({}, { __mode = "k" })`).body.statements);
+        this.define([own], parseLuau(`local ${own} = ${this.builtin("setmetatable").name}({}, { __mode = "k" })`).body.statements, out);
       }
       if (marker) {
-        out.push(...parseLuau(`
+        this.define([marker], parseLuau(`
 local function ${marker}(t)
 	${arrays}[t] = true
 	return t
-end`).body.statements);
+end`).body.statements, out);
       }
       if (typeOf) {
-        out.push(...parseLuau(`
+        this.define([typeOf], parseLuau(`
 local function ${typeOf}(value, known)
-	local name = ${this.builtin(this.lua51 ? "type" : "typeof").name}(value)
+	local name = (${this.builtin("typeof").name} or ${this.builtin("type").name})(value)
 	if name ~= "table" then
 		return name
 	end
@@ -676,22 +848,22 @@ local function ${typeOf}(value, known)
 		return "array"
 	end
 	return "object"
-end`).body.statements);
+end`).body.statements, out);
       }
     }
     const add = this.helpers.get("add");
     if (add) {
-      out.push(...parseLuau(`
+      this.define([add], parseLuau(`
 local function ${add}(a, b)
 	if ${this.builtin("type").name}(a) == "string" or ${this.builtin("type").name}(b) == "string" then
 		return ${this.builtin("tostring").name}(a) .. ${this.builtin("tostring").name}(b)
 	end
 	return a + b
-end`).body.statements);
+end`).body.statements, out);
     }
     const instanceOf = this.helpers.get("instanceof");
     if (instanceOf) {
-      out.push(...parseLuau(`
+      this.define([instanceOf], parseLuau(`
 local function ${instanceOf}(value, class)
 	if ${this.builtin("type").name}(value) ~= "table" then
 		return false
@@ -704,11 +876,11 @@ local function ${instanceOf}(value, class)
 		current = current.ParentClass
 	end
 	return false
-end`).body.statements);
+end`).body.statements, out);
     }
     const accessors = this.helpers.get("accessors");
     if (accessors) {
-      out.push(...parseLuau(`
+      this.define([accessors], parseLuau(`
 local function ${accessors}(target, getters, setters)
 	return ${this.builtin("setmetatable").name}(target, {
 		-- Read by a spread, which copies what the getters give.
@@ -732,11 +904,11 @@ local function ${accessors}(target, getters, setters)
 			${this.builtin("rawset").name}(self, key, value)
 		end,
 	})
-end`).body.statements);
+end`).body.statements, out);
     }
     const assign2 = this.helpers.get("assign");
     if (assign2) {
-      out.push(...parseLuau(`
+      this.define([assign2], parseLuau(`
 local function ${assign2}(target, ...)
 	for i = 1, ${this.builtin("select").name}("#", ...) do
 		local source = (${this.builtin("select").name}(i, ...))
@@ -754,12 +926,12 @@ local function ${assign2}(target, ...)
 		end
 	end
 	return target
-end`).body.statements);
+end`).body.statements, out);
     }
-    out.push(...runtimes);
+    for (const { names, statements } of runtimes) this.define(names, statements, out);
     const concat = this.helpers.get("concat");
     if (concat) {
-      out.push(localFunction(concat, functionBody([], [
+      this.define([concat], [localFunction(concat, functionBody([], [
         local(["result"], [this.array(table([]))]),
         numericFor("i", number(1), selectCount(this.builtin("select")), [
           local(["part"], [call(this.builtin("select"), [identifier("i"), vararg()])]),
@@ -772,7 +944,7 @@ end`).body.statements);
           ]))
         ]),
         returns([identifier("result")])
-      ], true)));
+      ], true))], out);
     }
     return out;
   }
@@ -792,7 +964,12 @@ end`).body.statements);
       }
       const lines = this.options.module?.lines ?? "nil";
       const fail = this.options.module?.fail ?? "nil";
-      out.push(...parseLuau(source.replace(/__NAME__/g, name).replace(/__LINES__/g, lines).replace(/__FAIL__/g, fail).replace(/__ARRAY__/g, () => this.helper("array").name)).body.statements);
+      const group = {
+        names: [name],
+        statements: parseLuau(source.replace(/__NAME__/g, name).replace(/__LINES__/g, lines).replace(/__FAIL__/g, fail).replace(/__ARRAYS__/g, () => this.arrays()).replace(/__ARRAY__/g, () => this.helper("array").name).replace(/__JSON__/g, () => this.loweringRuntime(0, "json"))).body.statements
+      };
+      if (index2 === 0 && runtime2 === "json") out.unshift(group);
+      else out.push(group);
     }
     return out;
   }
@@ -807,7 +984,8 @@ end`).body.statements);
    *    util = require("util")    -- the imports
    *    ...                       -- the rest, in source order */
   module(context) {
-    const statements = this.source.body.statements;
+    const shaken = this.options.shaken;
+    const statements = shaken ? this.source.body.statements.filter((s) => !shaken.removed.has(s)) : this.source.body.statements;
     const exportsTable = () => identifier(this.exportsName);
     const { names, links, stars } = this.info;
     const locals = [];
@@ -846,13 +1024,24 @@ end`).body.statements);
     };
     for (const statement of statements) {
       if (statement.type !== "ImportStatement" || statement.isTypeOnly) continue;
+      if (!statement.defaultImport && !statement.namespaceImport && !statement.specifiers.length) {
+        const key2 = resolve3(statement.source);
+        if (key2 !== void 0) load(key2);
+        continue;
+      }
       const specifiers = [
         ...statement.defaultImport ? [{ local: statement.defaultImport, imported: "default" }] : [],
         ...statement.namespaceImport ? [{ local: statement.namespaceImport, imported: void 0 }] : [],
         ...statement.specifiers.map((s) => ({ local: s.local, imported: s.imported.name }))
       ];
-      const used = specifiers.filter((s) => (this.bindingByDeclaration.get(s.local)?.references.length ?? 0) > 0);
-      if (!used.length) continue;
+      const used = specifiers.filter((s) => (this.bindingByDeclaration.get(s.local)?.references ?? []).some((reference) => !shaken?.dead.has(reference)));
+      if (!used.length) {
+        if (shaken?.runs(statement.source.value)) {
+          const key2 = resolve3(statement.source);
+          if (key2 !== void 0) load(key2);
+        }
+        continue;
+      }
       const key = resolve3(statement.source);
       if (key === void 0) continue;
       const local2 = moduleLocal(key);
@@ -894,6 +1083,7 @@ end`).body.statements);
           break;
         }
         case "ExportAllStatement": {
+          if (shaken && !shaken.needed(statement.source.value)) break;
           const key = resolve3(statement.source);
           if (key === void 0) break;
           load(key);
@@ -1168,7 +1358,7 @@ end`).body.statements);
       case "VariableDeclaration":
         return this.variableDeclaration(node, "declare");
       case "FunctionDeclaration":
-        return [{ ...localFunction(this.name(node.name.name), this.functionBody(node.func)), attributes: this.lua51 ? void 0 : node.attributes }];
+        return [{ ...localFunction(this.name(node.name.name), this.functionBody(node.func)), attributes: this.syntax.has("attributes") ? node.attributes : void 0 }];
       case "FunctionDeclarationStatement":
         return [this.functionDeclarationStatement(node)];
       case "ClassDeclaration":
@@ -1180,7 +1370,7 @@ end`).body.statements);
         const join = node.operator === "+=" ? this.options.types?.joins.get(node) : void 0;
         if (join === "dynamic") return this.dynamicAddition(node);
         const operator = join === "concat" ? "..=" : node.operator;
-        if (this.lua51) return this.compoundAssignment51(node, operator);
+        if (!this.syntax.has("compoundAssignment")) return this.compoundAssignment51(node, operator);
         return [{
           type: "CompoundAssignmentStatement",
           operator,
@@ -1218,10 +1408,10 @@ end`).body.statements);
       case "RepeatStatement": {
         const body = this.block(node.body);
         const condition = this.expression(node.condition);
-        if (this.lua51 && hasOwnJump(body.statements, "ContinueStatement")) {
+        if (!this.syntax.has("continue") && hasOwnJump(body.statements, "ContinueStatement")) {
           const hidden = declaredNames(body.statements).filter((name) => mentions(condition, name));
           if (hidden.length) {
-            this.report(node, `'continue' in a 'repeat' whose 'until' reads ${hidden.map((n) => `'${n}'`).join(", ")} cannot be compiled for Lua 5.1, which has no 'continue' to lower it to. Move what the condition reads out of the loop body, or use a 'while' loop.`);
+            this.report(node, `'continue' in a 'repeat' whose 'until' reads ${hidden.map((n) => `'${n}'`).join(", ")} cannot be compiled to Lua, which has no 'continue' to lower it to. Move what the condition reads out of the loop body, or use a 'while' loop \u2014 or load a library whose Lua has 'continue' ("luau").`);
           }
         }
         return [{ type: "RepeatStatement", body: this.loopBody51(body), condition, ...spanOf(node) }];
@@ -1262,8 +1452,8 @@ end`).body.statements);
         let source = this.expression(node.iterator);
         if (form.viaIter) source = methodCall(source, "__iter", []);
         const variables = form.walks === "values" ? [this.names.fresh("_"), item] : [item];
-        let iterators = form.walks === "iteration" ? [call(this.builtin("unpack"), [source, number(1), number(3)])] : [source];
-        if (this.lua51 && form.walks === "values") {
+        let iterators = form.walks === "iteration" ? [call(this.helper("unpack"), [source, number(1), number(3)])] : [source];
+        if (!this.syntax.has("iteration") && form.walks === "values") {
           const kind = form.viaIter ? void 0 : this.options.types?.typeOf.get(node.iterator)?.kind;
           iterators = [call(kind === "array" || kind === "tuple" ? this.builtin("ipairs") : this.helper("each"), [source])];
         }
@@ -1419,8 +1609,8 @@ end`).body.statements);
   classHelpers;
   classRuntime() {
     return this.classHelpers ??= {
-      build: this.names.fresh("tilua_class"),
-      accessors: this.names.fresh("tilua_accessors")
+      build: this.sharedName("class:build", "tilua_class"),
+      accessors: this.sharedName("class:accessors", "tilua_accessors")
     };
   }
   /** The helpers, as Luau — emitted only in a file that declares a class.
@@ -1438,7 +1628,8 @@ end`).body.statements);
     if (!this.classHelpers) return [];
     const { build, accessors } = this.classHelpers;
     const metamethods = `{ ${CLASS_METAMETHODS.map((name) => `"${name}"`).join(", ")} }`;
-    return parseLuau(`
+    const out = [];
+    this.define([build, accessors], parseLuau(`
 local function ${build}(base, name)
     local class = { __getters = {}, __setters = {} }
     class.__index = class
@@ -1476,7 +1667,8 @@ local function ${accessors}(class)
         rawset(this, key, value)
     end
 end
-`).body.statements;
+`).body.statements, out);
+    return out;
   }
   /** The class being lowered, so `super` knows what to name. */
   classContext;
@@ -1620,7 +1812,7 @@ end
       isMethod,
       func,
       // `@native` and the like are Luau's.
-      attributes: this.lua51 ? void 0 : node.attributes,
+      attributes: this.syntax.has("attributes") ? node.attributes : void 0,
       ...spanOf(node)
     };
   }
@@ -1860,7 +2052,7 @@ end
           if (join === "concat") return binary("..", left, right);
           if (join === "dynamic") return call(this.helper("add"), [this.keptToOne(left), this.keptToOne(right)]);
         }
-        if (this.lua51 && node.operator === "//") return this.floorDivide(left, right);
+        if (!this.syntax.has("floorDivision") && node.operator === "//") return this.floorDivide(left, right);
         return binary(node.operator, left, right);
       }
       case "UnaryExpression":
@@ -1894,7 +2086,7 @@ end
       case "AsConstExpression":
         return this.expression(node.expression);
       case "IfElseExpression":
-        if (this.lua51) return this.ifElse51(node);
+        if (!this.syntax.has("ifExpression")) return this.ifElse51(node);
         return {
           type: "IfElseExpression",
           clauses: node.clauses.map((c) => ({ condition: this.expression(c.condition), body: this.expression(c.body) })),
@@ -1941,7 +2133,7 @@ end
       const last2 = list.length - 1;
       return lowered.length && !many.has(list[last2]) ? [...lowered.slice(0, -1), this.keptToOne(lowered[last2])] : lowered;
     }
-    const unpack = (value) => call(this.lua51 ? this.builtin("unpack") : member(this.builtin("table"), "unpack"), [value]);
+    const unpack = (value) => call(this.helper("unpack"), [value]);
     const last = list[list.length - 1];
     if (spreads.length === 1 && last.type === "SpreadElement") {
       return [...list.slice(0, -1).map((a) => this.expression(a)), unpack(this.expression(last.argument))];
@@ -2055,7 +2247,7 @@ end
     const segments = this.segments(chain);
     const base = this.expression(chain.base);
     const last = chain.links[chain.links.length - 1];
-    if (!this.lua51 && base.type === "Identifier" && segments.length === 1 && last.type !== "CallExpression" && last.type !== "MethodCallExpression") {
+    if (this.syntax.has("ifExpression") && base.type === "Identifier" && segments.length === 1 && last.type !== "CallExpression" && last.type !== "MethodCallExpression") {
       return {
         type: "IfElseExpression",
         clauses: [{ condition: binary("==", base, nil()), body: nil() }],
@@ -2082,7 +2274,7 @@ end
   nullish(node) {
     const left = this.keptToOne(this.expression(node.left));
     const right = this.keptToOne(this.expression(node.right));
-    if (!this.lua51 && left.type === "Identifier") {
+    if (this.syntax.has("ifExpression") && left.type === "Identifier") {
       return {
         type: "IfElseExpression",
         clauses: [{ condition: binary("~=", left, nil()), body: left }],
@@ -2169,14 +2361,19 @@ end
     let target;
     if (node.target.type === "MemberExpression" || node.target.type === "IndexExpression") {
       const object2 = this.expression(node.target.object);
-      const objectName = this.names.fresh("target");
-      before.push(local([objectName], [object2]));
-      if (node.target.type === "MemberExpression") {
-        target = member(identifier(objectName), node.target.property.name);
+      const key = node.target.type === "IndexExpression" ? node.target.index : void 0;
+      if (object2.type === "Identifier" && (key === void 0 || key.type === "StringLiteral" || key.type === "NumberLiteral")) {
+        target = key === void 0 ? member(object2, node.target.property.name) : index(object2, this.expression(key));
       } else {
-        const keyName = this.names.fresh("key");
-        before.push(local([keyName], [this.expression(node.target.index)]));
-        target = index(identifier(objectName), identifier(keyName));
+        const objectName = this.names.fresh("target");
+        before.push(local([objectName], [object2]));
+        if (node.target.type === "MemberExpression") {
+          target = member(identifier(objectName), node.target.property.name);
+        } else {
+          const keyName = this.names.fresh("key");
+          before.push(local([keyName], [this.expression(node.target.index)]));
+          target = index(identifier(objectName), identifier(keyName));
+        }
       }
     } else {
       target = this.expression(node.target);
@@ -2197,7 +2394,7 @@ end
    *  done, and breaks for real there. A body with no `continue` is left
    *  exactly as it was. */
   loopBody51(body) {
-    if (!this.lua51 || !hasOwnJump(body.statements, "ContinueStatement")) return body;
+    if (this.syntax.has("continue") || !hasOwnJump(body.statements, "ContinueStatement")) return body;
     const flag = hasOwnJump(body.statements, "BreakStatement") ? this.names.fresh("broke") : void 0;
     const once = repeatUntil(rewriteOwnJumps(body.statements, flag), boolean(true));
     if (!flag) return block([once]);
@@ -2249,7 +2446,7 @@ end
    *  underscores and `0b1010` becomes the decimal it means. A hex literal
    *  keeps its base — 5.1 reads those. */
   numberRaw(node) {
-    if (!this.lua51) return node.raw;
+    if (this.syntax.has("numberLiterals")) return node.raw;
     if (/^0[bB]/.test(node.raw)) return String(node.value);
     return node.raw.replace(/_/g, "");
   }
@@ -2502,10 +2699,10 @@ function declaredNames(statements) {
 }
 function mentions(expression, name) {
   let found = false;
-  const walk = (value) => {
+  const walk2 = (value) => {
     if (found || !value || typeof value !== "object") return;
     if (Array.isArray(value)) {
-      for (const item of value) walk(item);
+      for (const item of value) walk2(item);
       return;
     }
     const node = value;
@@ -2514,10 +2711,10 @@ function mentions(expression, name) {
       return;
     }
     for (const [key, child] of Object.entries(value)) {
-      if (key !== "line" && key !== "column") walk(child);
+      if (key !== "line" && key !== "column") walk2(child);
     }
   };
-  walk(expression);
+  walk2(expression);
   return found;
 }
 function allNodes(root) {
@@ -2600,6 +2797,269 @@ import {
 } from "@tilua/parser";
 import { parse as parseLuau2, parseExpressionFromSource as parseExpressionFromSource2, print } from "luau-parser";
 
+// src/treeshake.ts
+function walk(node, visit, parent) {
+  if (!node || typeof node !== "object") return;
+  if (Array.isArray(node)) {
+    for (const item of node) walk(item, visit, parent);
+    return;
+  }
+  const n = node;
+  if (typeof n.type === "string" && visit(n, parent) === false) return;
+  for (const [key, child] of Object.entries(n)) {
+    if (key === "line" || key === "column") continue;
+    walk(child, visit, typeof n.type === "string" ? n : parent);
+  }
+}
+function pure(expression) {
+  let ok = true;
+  walk(expression, (node) => {
+    if (!ok) return false;
+    switch (node.type) {
+      case "CallExpression":
+      case "MethodCallExpression":
+      case "ImportExpression":
+      case "SuperExpression":
+        ok = false;
+        return false;
+      case "FunctionExpression":
+      case "FunctionBody":
+      case "ClassMethod":
+      case "ClassAccessor":
+      case "ClassConstructor":
+        return false;
+      case "ClassField":
+        if (!node.isStatic) return false;
+    }
+  });
+  return ok;
+}
+function declarationOf(statement) {
+  return statement.type === "ExportStatement" ? statement.declaration : statement;
+}
+function removable(statement) {
+  const declaration = declarationOf(statement);
+  switch (declaration.type) {
+    case "FunctionDeclaration":
+      return true;
+    case "ClassDeclaration":
+      return pure(declaration);
+    case "VariableDeclaration":
+      return declaration.init === void 0 || pure(declaration.init);
+    case "ExportDefaultStatement":
+      return pure(declaration.declaration);
+    default:
+      return false;
+  }
+}
+function structural(statement) {
+  switch (statement.type) {
+    case "TypeAliasStatement":
+    case "ExportTypeAliasStatement":
+    case "DeclareStatement":
+    case "DeclareClassStatement":
+    case "DeclareMetatableStatement":
+    case "ImportStatement":
+    case "ExportNamedStatement":
+    case "ExportAllStatement":
+      return true;
+    default:
+      return false;
+  }
+}
+function exportedNames(statement) {
+  switch (statement.type) {
+    case "ExportStatement": {
+      const declaration = statement.declaration;
+      if (declaration.type === "FunctionDeclaration" || declaration.type === "ClassDeclaration") return [declaration.name.name];
+      const names = [];
+      walk(declaration.name, (node) => {
+        if (node.type === "IdentifierPattern") names.push(node.name);
+      });
+      return names;
+    }
+    case "ExportDefaultStatement":
+      return ["default"];
+    default:
+      return [];
+  }
+}
+function factsOf(module) {
+  const tops = module.program.body.statements;
+  const byNode = /* @__PURE__ */ new Map();
+  for (const binding of module.scopes.bindings.values()) {
+    if (binding.declarationNode) byNode.set(binding.declarationNode, binding.id);
+  }
+  const declaredBy = /* @__PURE__ */ new Map();
+  const imported = /* @__PURE__ */ new Map();
+  for (const statement of tops) {
+    const declaration = declarationOf(statement);
+    const declare = (node) => {
+      const id = node && byNode.get(node);
+      if (id !== void 0) declaredBy.set(id, statement);
+    };
+    if (declaration.type === "FunctionDeclaration" || declaration.type === "ClassDeclaration") declare(declaration.name);
+    else if (declaration.type === "VariableDeclaration") {
+      walk(declaration.name, (node) => {
+        if (node.type === "IdentifierPattern") declare(node);
+      });
+    } else if (declaration.type === "ImportStatement" && !declaration.isTypeOnly) {
+      const target = module.resolve(declaration.source.value);
+      if (target === void 0) continue;
+      const add = (node, name) => {
+        const id = node && byNode.get(node);
+        if (id !== void 0) imported.set(id, { target, name });
+      };
+      if (declaration.defaultImport) add(declaration.defaultImport, "default");
+      if (declaration.namespaceImport) add(declaration.namespaceImport, "*");
+      for (const s of declaration.specifiers) add(s.local, s.imported.name);
+    }
+  }
+  const parents = /* @__PURE__ */ new WeakMap();
+  walk(module.program.body, (node, parent) => {
+    if (parent) parents.set(node, parent);
+  });
+  const sideEffects = tops.some((s) => !structural(s) && !removable(s));
+  return { module, tops, declaredBy, imported, parents, sideEffects };
+}
+function treeShake(modules, entry) {
+  const facts = new Map(modules.map((m) => [m.file, factsOf(m)]));
+  const used = /* @__PURE__ */ new Map();
+  const live = /* @__PURE__ */ new Map();
+  const required = /* @__PURE__ */ new Set();
+  let changed = true;
+  const require2 = (file) => {
+    if (required.has(file) || !facts.has(file)) return;
+    required.add(file);
+    changed = true;
+  };
+  const read3 = (file, name) => {
+    if (!facts.has(file)) return;
+    require2(file);
+    const now = used.get(file);
+    if (now === "all") return;
+    if (name === "*") {
+      used.set(file, "all");
+      changed = true;
+      return;
+    }
+    const names = now ?? /* @__PURE__ */ new Set();
+    if (!names.has(name)) {
+      names.add(name);
+      used.set(file, names);
+      changed = true;
+    }
+  };
+  const isRead = (file, name) => {
+    const now = used.get(file);
+    return now === "all" || (now?.has(name) ?? false);
+  };
+  require2(entry);
+  used.set(entry, "all");
+  while (changed) {
+    changed = false;
+    for (const file of [...required]) {
+      const f = facts.get(file);
+      const kept = live.get(file) ?? /* @__PURE__ */ new Set();
+      live.set(file, kept);
+      const queue = [];
+      const keep = (statement) => {
+        if (!statement || kept.has(statement)) return;
+        kept.add(statement);
+        queue.push(statement);
+        changed = true;
+      };
+      for (const statement of f.tops) {
+        if (!structural(statement) && !removable(statement)) keep(statement);
+        if (exportedNames(statement).some((name) => isRead(file, name))) keep(statement);
+        const declaration = declarationOf(statement);
+        if (declaration.type === "ImportStatement" && !declaration.isTypeOnly && !declaration.defaultImport && !declaration.namespaceImport && !declaration.specifiers.length) {
+          const target = f.module.resolve(declaration.source.value);
+          if (target !== void 0) require2(target);
+        }
+        if (declaration.type === "ExportNamedStatement" && !declaration.typeOnly) {
+          const target = declaration.source ? f.module.resolve(declaration.source.value) : void 0;
+          for (const s of declaration.specifiers) {
+            if (!isRead(file, s.exported.name)) continue;
+            if (declaration.source) {
+              if (target !== void 0) read3(target, s.local.name);
+              continue;
+            }
+            const id = f.module.scopes.bindingOf.get(s.local);
+            const own = [...f.declaredBy].find(([bid, st]) => (id !== void 0 ? bid === id : false) || exportedLocal(st, s.local.name));
+            if (own) keep(own[1]);
+            const from = id !== void 0 ? f.imported.get(id) : void 0;
+            if (from) read3(from.target, from.name);
+          }
+        }
+        if (declaration.type === "ExportAllStatement") {
+          const target = f.module.resolve(declaration.source.value);
+          if (target === void 0) continue;
+          const mine = used.get(file);
+          if (mine === "all") read3(target, "*");
+          else for (const name of mine ?? []) if (!ownExport(f, name)) read3(target, name);
+        }
+      }
+      for (const statement of [...kept]) queue.push(statement);
+      while (queue.length) {
+        const statement = queue.pop();
+        walk(statement, (node) => {
+          if (node.type === "ImportExpression") {
+            const source = node.source;
+            const target = source?.type === "StringLiteral" ? f.module.resolve(source.value) : void 0;
+            if (target !== void 0) read3(target, "*");
+            return;
+          }
+          if (node.type !== "Identifier") return;
+          const id = f.module.scopes.bindingOf.get(node);
+          if (id === void 0) return;
+          const declaring = f.declaredBy.get(id);
+          if (declaring && declaring !== statement) keep(declaring);
+          const from = f.imported.get(id);
+          if (!from) return;
+          if (from.name !== "*") return read3(from.target, from.name);
+          const parent = f.parents.get(node);
+          if (parent?.type === "MemberExpression" && parent.object === node) read3(from.target, parent.property.name);
+          else read3(from.target, "*");
+        });
+      }
+      for (const statement of f.tops) {
+        if (statement.type !== "ImportStatement" || statement.isTypeOnly) continue;
+        const target = f.module.resolve(statement.source.value);
+        if (target !== void 0 && facts.get(target)?.sideEffects) require2(target);
+      }
+    }
+  }
+  const shaken = /* @__PURE__ */ new Map();
+  for (const [file, f] of facts) {
+    const kept = live.get(file) ?? /* @__PURE__ */ new Set();
+    const removed = new Set(f.tops.filter((s) => !kept.has(s) && removable(s)));
+    const dead = /* @__PURE__ */ new Set();
+    for (const statement of removed) walk(statement, (node) => {
+      if (node.type === "Identifier") dead.add(node);
+    });
+    shaken.set(file, { removed, dead });
+  }
+  return {
+    modules: shaken,
+    sideEffects: new Set([...facts].filter(([, f]) => f.sideEffects).map(([file]) => file)),
+    required
+  };
+}
+function exportedLocal(statement, name) {
+  const declaration = declarationOf(statement);
+  if (declaration.type === "FunctionDeclaration" || declaration.type === "ClassDeclaration") return declaration.name.name === name;
+  if (declaration.type !== "VariableDeclaration") return false;
+  let found = false;
+  walk(declaration.name, (node) => {
+    if (node.type === "IdentifierPattern" && node.name === name) found = true;
+  });
+  return found;
+}
+function ownExport(f, name) {
+  return f.tops.some((statement) => exportedNames(statement).includes(name) || statement.type === "ExportNamedStatement" && statement.specifiers.some((s) => s.exported.name === name));
+}
+
 // src/linemap.ts
 import { luauString as luauString3 } from "@tilua/parser";
 function statementsOf(node) {
@@ -2663,8 +3123,8 @@ async function bundle(options) {
     sources.set(file, { file, name: nameOf(file), text, program: program3, scopes });
     const dynamic = importExpressions(program3).flatMap((node) => node.source.type === "StringLiteral" ? [node.source.value] : []);
     for (const specifier of [...importedSpecifiers(program3), ...dynamic]) {
-      const target2 = resolveModulePath(file, specifier, config);
-      if (target2 && !target2.endsWith(".d.tilua")) queue.push(target2);
+      const target = resolveModulePath(file, specifier, config);
+      if (target && !target.endsWith(".d.tilua")) queue.push(target);
     }
   }
   const analysis = analyzeModules([...sources.values()], config);
@@ -2678,9 +3138,16 @@ async function bundle(options) {
   const requireExpression = member(identifier(G), "require");
   const importExpression = member(identifier(G), "import");
   const mode = options.output ?? config?.output ?? "single";
-  const target = options.target ?? config?.target ?? "luau";
+  const byPath = syntaxOf(lowerings).has("requireByPath");
   const modules = /* @__PURE__ */ new Map();
   const origins = /* @__PURE__ */ new WeakMap();
+  const shared = new SharedRuntime(names);
+  const shaking = options.treeShaking ?? config?.treeShaking ?? false;
+  const resolveIn = (file) => (specifier) => {
+    const found = resolveModulePath(file, specifier, config);
+    return found !== void 0 && sources.has(found) ? found : void 0;
+  };
+  const shaken = shaking ? treeShake([...sources.values()].map((s) => ({ file: s.file, program: s.program, scopes: s.scopes, resolve: resolveIn(s.file) })), entry) : void 0;
   const pending = [entry];
   while (pending.length) {
     const file = pending.shift();
@@ -2693,10 +3160,21 @@ async function bundle(options) {
       names,
       types: analysis.types.get(file),
       lowerings,
-      target,
       noLib: config?.noLib,
       source: source.text,
       file: shown,
+      shared,
+      shaken: shaken && {
+        ...shaken.modules.get(file),
+        runs: (specifier) => {
+          const target = resolveIn(file)(specifier);
+          return target !== void 0 && shaken.sideEffects.has(target);
+        },
+        needed: (specifier) => {
+          const target = resolveIn(file)(specifier);
+          return target !== void 0 && shaken.required.has(target);
+        }
+      },
       module: {
         name: source.name,
         require: requireExpression,
@@ -2727,6 +3205,7 @@ async function bundle(options) {
       exportsName: lowered2.exportsName,
       info: lowered2.module,
       usesScriptArgs: lowered2.usesScriptArgs,
+      shared: lowered2.shared,
       imports,
       dynamicImports
     });
@@ -2752,16 +3231,17 @@ async function bundle(options) {
   if (diagnostics.some((d) => d.category === "syntax" || d.category === "scope" || d.category === "module")) {
     return { files: [], modules: moduleNames, diagnostics };
   }
+  if (shaking) shared.shake(lowered.flatMap((m) => m.statements));
   const several = mode !== "single";
   const layout = layoutOf(mode, modules.get(entry), lowered, options.outFile);
   const usesScriptArgs = lowered.some((m) => m.usesScriptArgs);
   const elsewhere = /* @__PURE__ */ new Map();
   for (const file of layout.slice(1)) {
-    for (const module of file.modules) elsewhere.set(module.name, loadPath(layout[0].path, file.path, target));
+    for (const module of file.modules) elsewhere.set(module.name, loadPath(layout[0].path, file.path, byPath));
   }
   const program2 = parseLuau2(runtime(G, {
     several,
-    target,
+    byPath,
     files: elsewhere,
     imports: lowered.some((m) => m.dynamicImports.size > 0)
   }));
@@ -2776,6 +3256,8 @@ async function bundle(options) {
   for (const module of layout[0].modules) {
     modulesTable.fields.push({ type: "TableFieldComputed", key: string(module.name), value: moduleTable(module) });
   }
+  const helpers = shared.chunk(new Set(layout[0].modules.flatMap((m) => m.shared)));
+  program2.body.statements.unshift(...helpers.declare);
   const ok = names.fresh("ok");
   const result2 = names.fresh("result");
   const entryStatements = parseLuau2(`
@@ -2797,9 +3279,11 @@ ${modules.get(entry).info.exports ? `return ${result2}` : ""}
     if (several) mapStatements.push(...mapRegistration(G, layout[0], `${G}.lines`));
     program2.body.statements.splice(program2.body.statements.length - entryStatements.length, 0, ...mapStatements);
   }
+  program2.body.statements.splice(program2.body.statements.length - entryStatements.length, 0, ...helpers.define);
   const files = [{ path: layout[0].path, code: print(program2) + "\n", modules: layout[0].modules.map((m) => m.name) }];
   for (const file of layout.slice(1)) {
-    const body = [];
+    const fileHelpers = shared.chunk(new Set(file.modules.flatMap((m) => m.shared)));
+    const body = [...fileHelpers.declare];
     if (file.modules.some((m) => m.usesScriptArgs)) body.push(...parseLuau2(`local scriptArgs = ${G}.scriptArgs`).body.statements);
     for (const module of file.modules) {
       body.push(assign(
@@ -2809,7 +3293,7 @@ ${modules.get(entry).info.exports ? `return ${result2}` : ""}
     }
     const chunk = program([returns([functionExpression(functionBody([G], body))])]);
     const lines = buildLineMap(chunk, parseLuau2(print(chunk)), origins);
-    body.push(...mapRegistration(G, file, lineMapSource(lines)));
+    body.push(...mapRegistration(G, file, lineMapSource(lines)), ...fileHelpers.define);
     files.push({ path: file.path, code: print(chunk) + "\n", modules: file.modules.map((m) => m.name) });
   }
   return { code: files[0].code, files, modules: moduleNames, diagnostics };
@@ -2871,9 +3355,9 @@ function layoutOf(mode, entry, modules, outFile) {
   }
   return files;
 }
-function loadPath(entryPath, file, target) {
+function loadPath(entryPath, file, byPath) {
   const path = posix.relative(posix.dirname(entryPath), file);
-  if (target === "lua51") return path;
+  if (!byPath) return path;
   const bare = path.slice(0, path.length - posix.extname(path).length);
   return bare.startsWith("../") ? bare : `./${bare}`;
 }
@@ -2886,8 +3370,8 @@ function mapRegistration(G, file, map) {
     ...file.modules.map((module) => `${G}.maps[${luauString4(module.name)}] = ${G}.sources[#${G}.sources][2]`)
   ].join("\n")).body.statements;
 }
-function loader(G, target) {
-  if (target === "lua51") {
+function loader(G, byPath) {
+  if (!byPath) {
     return `
         -- From beside this file, wherever that is: its own path says where.
         local folder = ""
@@ -3095,7 +3579,7 @@ ${several ? `
         if ${G}.loaded[file] then
             return
         end
-        ${G}.loaded[file] = true${loader(G, options.target)}
+        ${G}.loaded[file] = true${loader(G, options.byPath)}
     end,` : ""}${options.imports ? `
     -- \`import("./x")\`: the module's exports, asked for where the code runs
     -- rather than before its module does. \`require\` does the rest \u2014 and