luau2ts 0.1.0

package/dist/compile/class-shape.js

@@ -0,0 +1,291 @@
+import ts from 'typescript';
+const { factory } = ts;
+/** Walk a flat list of statements and detect class patterns. Returns the
+ *  detected classes plus the union of consumed indexes (so callers can
+ *  skip them in the regular pipeline). */
+export function detectClasses(stmts) {
+    const out = [];
+    for (let i = 0; i < stmts.length; i++) {
+        const candidate = matchClassDeclaration(stmts, i);
+        if (!candidate)
+            continue;
+        // Walk forward, collecting __index, .new, :constructor, :method
+        // statements that belong to this class.
+        const pattern = {
+            name: candidate.name,
+            superclass: candidate.superclass,
+            consumed: new Set([i]),
+            ctorFactory: null,
+            constructor: null,
+            methods: [],
+        };
+        for (let j = i + 1; j < stmts.length; j++) {
+            const s = stmts[j];
+            if (matchIndexAssignment(s, pattern.name)) {
+                pattern.consumed.add(j);
+                continue;
+            }
+            const fnMatch = matchClassFunction(s, pattern.name);
+            if (fnMatch) {
+                pattern.consumed.add(j);
+                if (fnMatch.kind === 'static' && fnMatch.methodName === 'new') {
+                    pattern.ctorFactory = s;
+                }
+                else if (fnMatch.kind === 'method' && fnMatch.methodName === 'constructor') {
+                    pattern.constructor = s;
+                }
+                else if (fnMatch.kind === 'method') {
+                    pattern.methods.push(s);
+                }
+                else {
+                    // Static non-`new` method — drop into instance methods for now.
+                    pattern.methods.push(s);
+                }
+                continue;
+            }
+            // Any non-matching statement breaks the class block.
+            break;
+        }
+        // Require at least a `.new` factory or a `:constructor` to be confident
+        // we matched a real class — bare `setmetatable({}, …)` blocks aren't
+        // necessarily classes.
+        if (!pattern.ctorFactory && !pattern.constructor && pattern.methods.length === 0)
+            continue;
+        out.push(pattern);
+    }
+    return out;
+}
+/** Check whether a statement is `local <Name> = setmetatable({}, <table>)`.
+ *  Returns the class name + (optional) superclass identifier read from the
+ *  metatable's `__index` field. */
+function matchClassDeclaration(stmts, i) {
+    const stat = stmts[i];
+    if (!stat || stat.type !== 'Local')
+        return null;
+    const ls = stat;
+    if (ls.vars.length !== 1 || ls.values.length !== 1)
+        return null;
+    const name = ls.vars[0].name;
+    const value = ls.values[0];
+    if (value.type !== 'Call')
+        return null;
+    const call = value;
+    if (call.func.type !== 'Global' || call.func.name !== 'setmetatable')
+        return null;
+    if (call.args.length < 1 || call.args[0].type !== 'Table')
+        return null;
+    // Optional second argument: a table literal whose `__index` field gives
+    // the superclass.
+    let superclass = null;
+    if (call.args.length >= 2 && call.args[1].type === 'Table') {
+        const meta = call.args[1];
+        for (const item of meta.items) {
+            if (item.kind !== 'Record')
+                continue;
+            const key = item.key;
+            if (!key || key.type !== 'ConstantString')
+                continue;
+            if (key.value !== '__index')
+                continue;
+            if (item.value.type === 'Global')
+                superclass = item.value.name;
+            else if (item.value.type === 'Local')
+                superclass = item.value.name;
+        }
+    }
+    return { name, superclass };
+}
+/** Match `<Class>.__index = <Class>` (or `= <Superclass>`). */
+function matchIndexAssignment(stat, className) {
+    if (stat.type !== 'Assign')
+        return false;
+    const a = stat;
+    if (a.vars.length !== 1)
+        return false;
+    const v = a.vars[0];
+    if (v.type !== 'IndexName')
+        return false;
+    if (v.index !== '__index')
+        return false;
+    if (v.expr.type !== 'Global' && v.expr.type !== 'Local')
+        return false;
+    return v.expr.name === className;
+}
+/** Match `function <Class>.method(...)` or `function <Class>:method(...)`. */
+function matchClassFunction(stat, className) {
+    if (stat.type !== 'Function')
+        return null;
+    const fs = stat;
+    // The name expr is an IndexName with `op` = '.' or ':'. We want the
+    // root to be `<className>` and the index to be the method name.
+    const nameExpr = fs.name;
+    if (nameExpr.type !== 'IndexName')
+        return null;
+    if (nameExpr.expr.type !== 'Global' && nameExpr.expr.type !== 'Local')
+        return null;
+    if (nameExpr.expr.name !== className)
+        return null;
+    const op = nameExpr.op;
+    return {
+        kind: op === ':' ? 'method' : 'static',
+        methodName: nameExpr.index,
+    };
+}
+/** Compile a detected class pattern into a TS `class` declaration. The
+ *  returned node replaces the consumed statements in the output. */
+export function compileClassPattern(pattern, ctx, compileBlockBody, compileExpr) {
+    const members = [];
+    // Constructor — prefer :constructor; fall back to the .new factory body.
+    const ctorFn = pattern.constructor ?? pattern.ctorFactory;
+    if (ctorFn) {
+        const fnExpr = ctorFn.func;
+        const params = (fnExpr.args ?? []).map((a) => factory.createParameterDeclaration(undefined, undefined, a.name));
+        const body = ctorBody(fnExpr.body, pattern, ctx, compileBlockBody, compileExpr);
+        members.push(factory.createConstructorDeclaration(undefined, params, factory.createBlock(body, true)));
+    }
+    // Methods.
+    for (const method of pattern.methods) {
+        const nameExpr = method.name;
+        if (nameExpr.type !== 'IndexName')
+            continue;
+        const methodName = nameExpr.index;
+        const fnExpr = method.func;
+        const params = (fnExpr.args ?? []).map((a) => factory.createParameterDeclaration(undefined, undefined, a.name));
+        const isStatic = nameExpr.op === '.';
+        const modifiers = isStatic
+            ? [factory.createModifier(ts.SyntaxKind.StaticKeyword)]
+            : undefined;
+        // Method bodies have the same `self` → `this` requirement as the
+        // constructor. Rewrite once after compileBlockBody so the emitted
+        // method reads `return this.x + arg` rather than `return self.x + arg`.
+        const rawBody = compileBlockBody(fnExpr.body, ctx);
+        const bodyStmts = rawBody.map((s) => rewriteSelfToThis(s));
+        members.push(factory.createMethodDeclaration(modifiers, undefined, methodName, undefined, undefined, params, undefined, factory.createBlock(bodyStmts, true)));
+    }
+    // Inheritance.
+    const heritage = pattern.superclass
+        ? [
+            factory.createHeritageClause(ts.SyntaxKind.ExtendsKeyword, [
+                factory.createExpressionWithTypeArguments(factory.createIdentifier(pattern.superclass), undefined),
+            ]),
+        ]
+        : undefined;
+    return factory.createClassDeclaration(undefined, pattern.name, undefined, heritage, members);
+}
+/** Compile the body of a `:constructor` or `.new` factory. The factory
+ *  body typically contains `local self = setmetatable({}, Class)` and
+ *  `self:constructor(...)`; we skip those bookkeeping lines and emit the
+ *  rest verbatim. The constructor body itself becomes the TS constructor. */
+function ctorBody(body, pattern, ctx, compileBlockBody, compileExpr) {
+    // If `:constructor` exists, use its body. Otherwise, the factory body
+    // typically wraps a `self:constructor(...)` call — we inline the
+    // factory body instead, keeping the user's actual init statements.
+    const stmts = body.type === 'Block' ? body.body : [body];
+    const filtered = stmts.filter((s) => !isClassPlumbing(s, pattern));
+    let compiled = filtered.flatMap((s) => compileBlockBody(s, ctx));
+    // Rewrite `self` → `this` and `Superclass.constructor(this, ...)` →
+    // `super(...)` so the synthesized class body uses TS-idiomatic names.
+    compiled = compiled.map((s) => rewriteSelfToThis(s));
+    if (pattern.superclass) {
+        compiled = compiled.map((s) => rewriteSuperCall(s, pattern.superclass));
+    }
+    return compiled;
+    // `compileExpr` is unused here today but kept in the signature for the
+    // future where we'll inline literal field assignments as TS class fields
+    // (`x: number = 0` syntax) instead of `this.x = …` constructor stmts.
+    void compileExpr;
+}
+/** Rewrite every `Identifier('self')` reference in a TS statement to
+ *  `this`. The class-shape detector copies Luau bodies that bind `self`
+ *  via `self = setmetatable({}, Class)`; once we elevate the body into a
+ *  TS class, every `self.X` is the synthesized `this.X`. */
+function rewriteSelfToThis(stat) {
+    function visit(node) {
+        if (ts.isIdentifier(node) && node.text === 'self') {
+            // Don't rewrite when `self` is the property name of a member
+            // access — `obj.self` should stay as `obj.self`.
+            // ts.isIdentifier covers both Identifier and the property-name
+            // position; we use the parent check at the call site below.
+            return factory.createThis();
+        }
+        return ts.visitEachChild(node, visit, undefined);
+    }
+    // Wrap so we don't rewrite property *names* (e.g. `obj.self` stays).
+    function topLevel(node) {
+        if (ts.isPropertyAccessExpression(node)) {
+            // Recurse only into the `expression` part; leave `name` alone.
+            const expr = topLevel(node.expression);
+            return factory.createPropertyAccessExpression(expr, node.name);
+        }
+        if (ts.isIdentifier(node) && node.text === 'self') {
+            return factory.createThis();
+        }
+        return ts.visitEachChild(node, topLevel, undefined);
+    }
+    void visit;
+    return topLevel(stat);
+}
+/** Statements that are pure metatable plumbing — `local self = setmetatable(...)`,
+ *  `self:constructor(...)`, `return self`. Kept out of the TS constructor body. */
+function isClassPlumbing(stat, pattern) {
+    // `local self = setmetatable({}, Class)` — first line of .new factory.
+    if (stat.type === 'Local') {
+        const ls = stat;
+        if (ls.vars.length === 1 && ls.vars[0].name === 'self' && ls.values.length === 1) {
+            const v = ls.values[0];
+            if (v.type === 'Call' && v.func.type === 'Global' && v.func.name === 'setmetatable') {
+                return true;
+            }
+        }
+    }
+    // `self:constructor(...)` call — second line of .new factory.
+    if (stat.type === 'Expr') {
+        const e = stat.expr;
+        if (e.type === 'Call' && e.func.type === 'IndexName') {
+            const f = e.func;
+            if (f.index === 'constructor'
+                && f.expr.type === 'Local'
+                && f.expr.name === 'self')
+                return true;
+        }
+    }
+    // `return self` — last line of .new factory.
+    if (stat.type === 'Return') {
+        const r = stat;
+        if (r.values.length === 1) {
+            const v = r.values[0];
+            if (v.type === 'Local' && v.name === 'self')
+                return true;
+        }
+    }
+    void pattern;
+    return false;
+}
+/** Best-effort: rewrite any `Superclass.constructor(this, ...)` call in a
+ *  TS statement to `super(...)`. */
+function rewriteSuperCall(stat, superclass) {
+    function visit(node) {
+        if (ts.isCallExpression(node)
+            && ts.isPropertyAccessExpression(node.expression)
+            && ts.isIdentifier(node.expression.expression)
+            && node.expression.expression.text === superclass
+            && ts.isIdentifier(node.expression.name)
+            && node.expression.name.text === 'constructor') {
+            // Drop the first arg if it's `this` / `self` — both forms appear
+            // depending on whether the self→this rewrite ran first (which
+            // produces a ts.ThisExpression node, not an identifier).
+            const args = node.arguments.slice();
+            if (args.length > 0) {
+                const first = args[0];
+                const isThisLike = first.kind === ts.SyntaxKind.ThisKeyword
+                    || (ts.isIdentifier(first) && (first.text === 'this' || first.text === 'self'));
+                if (isThisLike)
+                    args.shift();
+            }
+            return factory.createCallExpression(factory.createSuper(), undefined, args);
+        }
+        return ts.visitEachChild(node, visit, undefined);
+    }
+    return visit(stat);
+}
+//# sourceMappingURL=class-shape.js.map

package/dist/compile/class-shape.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"class-shape.js","sourceRoot":"","sources":["../../src/compile/class-shape.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,YAAY,CAAC;AAU5B,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;AAuBvB;;0CAE0C;AAC1C,MAAM,UAAU,aAAa,CAAC,KAAa;IACzC,MAAM,GAAG,GAAmB,EAAE,CAAC;IAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAClD,IAAI,CAAC,SAAS;YAAE,SAAS;QAEzB,gEAAgE;QAChE,wCAAwC;QACxC,MAAM,OAAO,GAAiB;YAC5B,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,UAAU,EAAE,SAAS,CAAC,UAAU;YAChC,QAAQ,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACtB,WAAW,EAAE,IAAI;YACjB,WAAW,EAAE,IAAI;YACjB,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;YACpB,IAAI,oBAAoB,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC1C,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACxB,SAAS;YACX,CAAC;YACD,MAAM,OAAO,GAAG,kBAAkB,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YACpD,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACxB,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,UAAU,KAAK,KAAK,EAAE,CAAC;oBAC9D,OAAO,CAAC,WAAW,GAAG,CAAiB,CAAC;gBAC1C,CAAC;qBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,UAAU,KAAK,aAAa,EAAE,CAAC;oBAC7E,OAAO,CAAC,WAAW,GAAG,CAAiB,CAAC;gBAC1C,CAAC;qBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACrC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAiB,CAAC,CAAC;gBAC1C,CAAC;qBAAM,CAAC;oBACN,gEAAgE;oBAChE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAiB,CAAC,CAAC;gBAC1C,CAAC;gBACD,SAAS;YACX,CAAC;YACD,qDAAqD;YACrD,MAAM;QACR,CAAC;QAED,wEAAwE;QACxE,qEAAqE;QACrE,uBAAuB;QACvB,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAE3F,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;mCAEmC;AACnC,SAAS,qBAAqB,CAAC,KAAa,EAAE,CAAS;IACrD,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACtB,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,IAAI,CAAC;IAChD,MAAM,EAAE,GAAG,IAAiB,CAAC;IAC7B,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAChE,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,IAAI,CAAC;IAC9B,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC;IAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;QAAE,OAAO,IAAI,CAAC;IACvC,MAAM,IAAI,GAAG,KAAwC,CAAC;IACtD,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc;QAAE,OAAO,IAAI,CAAC;IAClF,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,IAAI,CAAC;IAExE,wEAAwE;IACxE,kBAAkB;IAClB,IAAI,UAAU,GAAkB,IAAI,CAAC;IACrC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAqC,CAAC;QAC9D,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ;gBAAE,SAAS;YACrC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;YACrB,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,gBAAgB;gBAAE,SAAS;YACpD,IAAK,GAAyB,CAAC,KAAK,KAAK,SAAS;gBAAE,SAAS;YAC7D,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ;gBAAE,UAAU,GAAI,IAAI,CAAC,KAA0B,CAAC,IAAI,CAAC;iBAChF,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO;gBAAE,UAAU,GAAI,IAAI,CAAC,KAA0B,CAAC,IAAI,CAAC;QAC3F,CAAC;IACH,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;AAC9B,CAAC;AAED,+DAA+D;AAC/D,SAAS,oBAAoB,CAAC,IAAU,EAAE,SAAiB;IACzD,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACzC,MAAM,CAAC,GAAG,IAAkB,CAAC;IAC7B,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACtC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC;IACrB,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW;QAAE,OAAO,KAAK,CAAC;IACzC,IAAI,CAAC,CAAC,KAAK,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IACxC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,KAAK,CAAC;IACtE,OAAQ,CAAC,CAAC,IAAyB,CAAC,IAAI,KAAK,SAAS,CAAC;AACzD,CAAC;AAED,8EAA8E;AAC9E,SAAS,kBAAkB,CACzB,IAAU,EACV,SAAiB;IAEjB,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU;QAAE,OAAO,IAAI,CAAC;IAC1C,MAAM,EAAE,GAAG,IAAoB,CAAC;IAChC,oEAAoE;IACpE,gEAAgE;IAChE,MAAM,QAAQ,GAAG,EAAE,CAAC,IAAI,CAAC;IACzB,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW;QAAE,OAAO,IAAI,CAAC;IAC/C,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,IAAI,CAAC;IACnF,IAAK,QAAQ,CAAC,IAAyB,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACxE,MAAM,EAAE,GAAI,QAA4B,CAAC,EAAE,CAAC;IAC5C,OAAO;QACL,IAAI,EAAE,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;QACtC,UAAU,EAAE,QAAQ,CAAC,KAAK;KAC3B,CAAC;AACJ,CAAC;AAED;oEACoE;AACpE,MAAM,UAAU,mBAAmB,CACjC,OAAqB,EACrB,GAAmB,EACnB,gBAAqE,EACrE,WAA+D;IAE/D,MAAM,OAAO,GAAsB,EAAE,CAAC;IAEtC,yEAAyE;IACzE,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,CAAC;IAC1D,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC;QAC3B,MAAM,MAAM,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC3C,OAAO,CAAC,0BAA0B,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CACjE,CAAC;QACF,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAChF,OAAO,CAAC,IAAI,CACV,OAAO,CAAC,4BAA4B,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CACzF,CAAC;IACJ,CAAC;IAED,WAAW;IACX,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC;QAC7B,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW;YAAE,SAAS;QAC5C,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC;QAClC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC;QAC3B,MAAM,MAAM,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC3C,OAAO,CAAC,0BAA0B,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CACjE,CAAC;QACF,MAAM,QAAQ,GAAI,QAA4B,CAAC,EAAE,KAAK,GAAG,CAAC;QAC1D,MAAM,SAAS,GAAG,QAAQ;YACxB,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;YACvD,CAAC,CAAC,SAAS,CAAC;QACd,iEAAiE;QACjE,kEAAkE;QAClE,wEAAwE;QACxE,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,IAAY,EAAE,GAAG,CAAC,CAAC;QAC3D,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,OAAO,CAAC,IAAI,CACV,OAAO,CAAC,uBAAuB,CAC7B,SAAS,EACT,SAAS,EACT,UAAU,EACV,SAAS,EACT,SAAS,EACT,MAAM,EACN,SAAS,EACT,OAAO,CAAC,WAAW,CAAC,SAAS,EAAE,IAAI,CAAC,CACrC,CACF,CAAC;IACJ,CAAC;IAED,eAAe;IACf,MAAM,QAAQ,GAAG,OAAO,CAAC,UAAU;QACjC,CAAC,CAAC;YACE,OAAO,CAAC,oBAAoB,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,EAAE;gBACzD,OAAO,CAAC,iCAAiC,CACvC,OAAO,CAAC,gBAAgB,CAAC,OAAO,CAAC,UAAU,CAAC,EAC5C,SAAS,CACV;aACF,CAAC;SACH;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO,OAAO,CAAC,sBAAsB,CACnC,SAAS,EACT,OAAO,CAAC,IAAI,EACZ,SAAS,EACT,QAAQ,EACR,OAAO,CACR,CAAC;AACJ,CAAC;AAED;;;6EAG6E;AAC7E,SAAS,QAAQ,CACf,IAAU,EACV,OAAqB,EACrB,GAAmB,EACnB,gBAAqE,EACrE,WAA+D;IAE/D,sEAAsE;IACtE,iEAAiE;IACjE,mEAAmE;IACnE,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACzD,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;IACnE,IAAI,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IAEjE,oEAAoE;IACpE,sEAAsE;IACtE,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;QACvB,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,CAAC,EAAE,OAAO,CAAC,UAAW,CAAC,CAAC,CAAC;IAC3E,CAAC;IACD,OAAO,QAAQ,CAAC;IAChB,uEAAuE;IACvE,yEAAyE;IACzE,sEAAsE;IACtE,KAAK,WAAW,CAAC;AACnB,CAAC;AAED;;;4DAG4D;AAC5D,SAAS,iBAAiB,CAAC,IAAkB;IAC3C,SAAS,KAAK,CAAC,IAAa;QAC1B,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAClD,6DAA6D;YAC7D,iDAAiD;YACjD,+DAA+D;YAC/D,4DAA4D;YAC5D,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IACnD,CAAC;IACD,qEAAqE;IACrE,SAAS,QAAQ,CAAC,IAAa;QAC7B,IAAI,EAAE,CAAC,0BAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,+DAA+D;YAC/D,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAkB,CAAC;YACxD,OAAO,OAAO,CAAC,8BAA8B,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAClD,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;IACtD,CAAC;IACD,KAAK,KAAK,CAAC;IACX,OAAO,QAAQ,CAAC,IAAI,CAAiB,CAAC;AACxC,CAAC;AAED;mFACmF;AACnF,SAAS,eAAe,CAAC,IAAU,EAAE,OAAqB;IACxD,uEAAuE;IACvE,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC1B,MAAM,EAAE,GAAG,IAAiB,CAAC;QAC7B,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,IAAI,KAAK,MAAM,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClF,MAAM,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC;YACxB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gBACpF,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IACD,8DAA8D;IAC9D,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACzB,MAAM,CAAC,GAAI,IAAuB,CAAC,IAAI,CAAC;QACxC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACrD,MAAM,CAAC,GAAG,CAAC,CAAC,IAAqC,CAAC;YAClD,IACE,CAAC,CAAC,KAAK,KAAK,aAAa;mBACtB,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO;mBACtB,CAAC,CAAC,IAAyB,CAAC,IAAI,KAAK,MAAM;gBAC/C,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IACD,6CAA6C;IAC7C,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,IAA0B,CAAC;QACrC,IAAI,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC;YACvB,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,IAAK,CAAsB,CAAC,IAAI,KAAK,MAAM;gBAAE,OAAO,IAAI,CAAC;QACjF,CAAC;IACH,CAAC;IACD,KAAK,OAAO,CAAC;IACb,OAAO,KAAK,CAAC;AACf,CAAC;AAED;oCACoC;AACpC,SAAS,gBAAgB,CAAC,IAAkB,EAAE,UAAkB;IAC9D,SAAS,KAAK,CAAC,IAAa;QAC1B,IACE,EAAE,CAAC,gBAAgB,CAAC,IAAI,CAAC;eACtB,EAAE,CAAC,0BAA0B,CAAC,IAAI,CAAC,UAAU,CAAC;eAC9C,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;eAC3C,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,KAAK,UAAU;eAC9C,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;eACrC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,EAC9C,CAAC;YACD,iEAAiE;YACjE,8DAA8D;YAC9D,yDAAyD;YACzD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YACpC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;gBACvB,MAAM,UAAU,GACd,KAAK,CAAC,IAAI,KAAK,EAAE,CAAC,UAAU,CAAC,WAAW;uBACrC,CAAC,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC;gBAClF,IAAI,UAAU;oBAAE,IAAI,CAAC,KAAK,EAAE,CAAC;YAC/B,CAAC;YACD,OAAO,OAAO,CAAC,oBAAoB,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QAC9E,CAAC;QACD,OAAO,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IACnD,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAiB,CAAC;AACrC,CAAC"}

package/dist/compile/context.d.ts

@@ -0,0 +1,86 @@
+export declare const RUNTIME_MODULE = "luau2ts/runtime";
+/** What we know about a value at compile time.
+ *
+ *  Primitives use plain string tags. Roblox datatypes (Vector3, CFrame,
+ *  Color3, …) use the `'datatype:<Name>'` form so the compiler can fast-
+ *  path arithmetic operators to the underlying instance methods. Anything
+ *  we can't narrow falls back to `'unknown'`. */
+export type StaticValueType = 'number' | 'boolean' | 'string' | 'nil' | 'unknown' | `datatype:${string}`;
+/** Roblox datatypes that expose `__add`/`__sub`/`__mul`/`__div`/`__unm`
+ *  metamethods we can fast-path. The compiler narrows constructor calls
+ *  and annotated locals to `'datatype:<Name>'`; arithmetic between two
+ *  values of the same datatype emits `a.add(b)` instead of `luaAdd(a, b)`. */
+export declare const ARITH_DATATYPES: Set<string>;
+/** Compatibility mode for emitted TypeScript.
+ *
+ *  - `native`: emit TS that imports stdlib helpers from `luau2ts/runtime`.
+ *    `Vector3.new(...)`, `game:GetService(...)`, 1-indexed array semantics.
+ *  - `rbxts`:  emit TS that mirrors what roblox-ts would accept as input.
+ *    `new Vector3(...)`, `import { Workspace } from "@rbxts/services"`,
+ *    `new ClassName()` for `Instance.new("ClassName")`, optional 0-indexed
+ *    arrays for statically-array-typed expressions.
+ */
+export type CompatMode = 'native' | 'rbxts';
+export declare class CompileContext {
+    readonly compatMode: CompatMode;
+    private readonly imports;
+    private readonly scopes;
+    /** Per-scope Luau-name → JS-name override map. Used when a `local X = expr`
+     *  whose `expr` captures the outer `X` forces us to rename the new local
+     *  to a fresh name (so the inner reference still binds to the outer). */
+    private readonly jsNameOverrides;
+    /** Names of user-defined functions in this file whose bodies transitively
+     *  contain a yielding call. Calls to these names are themselves yielding
+     *  and the compiler must wrap each call site in `await`. Populated by a
+     *  pre-pass over the AST before code emission. */
+    readonly yieldingFunctions: Set<string>;
+    private tempCounter;
+    /** Reserved-import bookkeeping for module paths other than RUNTIME_MODULE.
+     *  Used by the macro registry to collect e.g. `@rbxts/services` symbols so
+     *  the emitter can prepend the right imports alongside the runtime helpers. */
+    private readonly extraImports;
+    /** Names of classes the class-shape detector inferred from
+     *  metatable-OOP patterns. Recorded so subsequent `<Class>.new(...)`
+     *  calls in the same file are lowered to `new <Class>(...)` rather than
+     *  staying as static-method references. */
+    private readonly detectedClasses;
+    /** Names locally bound to imported module identifiers — used to suppress
+     *  redundant `let X = X` declarations when a macro rewrote the RHS to
+     *  the same name as the LHS variable (e.g. `local Workspace =
+     *  game:GetService('Workspace')` after the macro fires). The key is the
+     *  Luau-level local name; the value is the imported module name (used
+     *  for diagnostics). */
+    private readonly suppressedLocals;
+    constructor(compatMode?: CompatMode);
+    /** Record a named import from `module`. The emitter will write
+     *  `import { ...names } from "<module>"` once per module. */
+    useImport(module: string, name: string): string;
+    /** Returns the recorded extra imports as `{ module, names[] }` tuples,
+     *  sorted by module path then by name for deterministic output. */
+    extraImportEntries(): {
+        module: string;
+        names: string[];
+    }[];
+    /** Record a name as a TS class detected by the class-shape pass. Lookup
+     *  happens at every `<Name>.new(...)` call site so we can lower it to
+     *  `new <Name>(...)` instead of leaving it as a static-method call. */
+    recordDetectedClass(name: string): void;
+    isDetectedClass(name: string): boolean;
+    /** Mark a local identifier as already bound to an import of the same
+     *  name. Subsequent `local <name> = <name>` declarations are dropped to
+     *  avoid the redundant shadowing. */
+    suppressLocal(name: string): void;
+    isSuppressedLocal(name: string): boolean;
+    use(helper: string): string;
+    importedHelpers(): string[];
+    freshIdentifier(prefix: string): string;
+    withScope<T>(fn: () => T): T;
+    setLocalJsName(name: string, jsName: string): void;
+    getLocalJsName(name: string): string | undefined;
+    defineLocal(name: string, type: StaticValueType): void;
+    hasLocalInCurrentScope(name: string): boolean;
+    hasLocalInOuterScope(name: string): boolean;
+    assignLocal(name: string, type: StaticValueType): void;
+    lookupLocal(name: string): StaticValueType;
+}
+//# sourceMappingURL=context.d.ts.map

package/dist/compile/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../src/compile/context.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,cAAc,oBAAoB,CAAC;AAEhD;;;;;iDAKiD;AACjD,MAAM,MAAM,eAAe,GACvB,QAAQ,GACR,SAAS,GACT,QAAQ,GACR,KAAK,GACL,SAAS,GACT,YAAY,MAAM,EAAE,CAAC;AAEzB;;;8EAG8E;AAC9E,eAAO,MAAM,eAAe,aAE1B,CAAC;AAEH;;;;;;;;GAQG;AACH,MAAM,MAAM,UAAU,GAAG,QAAQ,GAAG,OAAO,CAAC;AAE5C,qBAAa,cAAc;aAiCG,UAAU,EAAE,UAAU;IAhClD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;IAC7C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA+C;IACtE;;6EAEyE;IACzE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAsC;IACtE;;;sDAGkD;IAClD,QAAQ,CAAC,iBAAiB,cAAqB;IAC/C,OAAO,CAAC,WAAW,CAAK;IAExB;;mFAE+E;IAC/E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAkC;IAE/D;;;+CAG2C;IAC3C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAqB;IAErD;;;;;4BAKwB;IACxB,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAqB;gBAE1B,UAAU,GAAE,UAAqB;IAE7D;iEAC6D;IAC7D,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM;IAU/C;uEACmE;IACnE,kBAAkB,IAAI;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,EAAE,CAAA;KAAE,EAAE;IAM3D;;2EAEuE;IACvE,mBAAmB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAIvC,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAItC;;yCAEqC;IACrC,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAIjC,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIxC,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAK3B,eAAe,IAAI,MAAM,EAAE;IAI3B,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAMvC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC;IAW5B,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAIlD,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAQhD,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,GAAG,IAAI;IAItD,sBAAsB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI7C,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAO3C,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,GAAG,IAAI;IAWtD,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe;CAO3C"}

package/dist/compile/context.js

@@ -0,0 +1,144 @@
+export const RUNTIME_MODULE = 'luau2ts/runtime';
+/** Roblox datatypes that expose `__add`/`__sub`/`__mul`/`__div`/`__unm`
+ *  metamethods we can fast-path. The compiler narrows constructor calls
+ *  and annotated locals to `'datatype:<Name>'`; arithmetic between two
+ *  values of the same datatype emits `a.add(b)` instead of `luaAdd(a, b)`. */
+export const ARITH_DATATYPES = new Set([
+    'Vector3', 'Vector2', 'Vector3int16', 'Vector2int16', 'CFrame',
+]);
+export class CompileContext {
+    compatMode;
+    imports = new Set();
+    scopes = [new Map()];
+    /** Per-scope Luau-name → JS-name override map. Used when a `local X = expr`
+     *  whose `expr` captures the outer `X` forces us to rename the new local
+     *  to a fresh name (so the inner reference still binds to the outer). */
+    jsNameOverrides = [new Map()];
+    /** Names of user-defined functions in this file whose bodies transitively
+     *  contain a yielding call. Calls to these names are themselves yielding
+     *  and the compiler must wrap each call site in `await`. Populated by a
+     *  pre-pass over the AST before code emission. */
+    yieldingFunctions = new Set();
+    tempCounter = 0;
+    /** Reserved-import bookkeeping for module paths other than RUNTIME_MODULE.
+     *  Used by the macro registry to collect e.g. `@rbxts/services` symbols so
+     *  the emitter can prepend the right imports alongside the runtime helpers. */
+    extraImports = new Map();
+    /** Names of classes the class-shape detector inferred from
+     *  metatable-OOP patterns. Recorded so subsequent `<Class>.new(...)`
+     *  calls in the same file are lowered to `new <Class>(...)` rather than
+     *  staying as static-method references. */
+    detectedClasses = new Set();
+    /** Names locally bound to imported module identifiers — used to suppress
+     *  redundant `let X = X` declarations when a macro rewrote the RHS to
+     *  the same name as the LHS variable (e.g. `local Workspace =
+     *  game:GetService('Workspace')` after the macro fires). The key is the
+     *  Luau-level local name; the value is the imported module name (used
+     *  for diagnostics). */
+    suppressedLocals = new Set();
+    constructor(compatMode = 'native') {
+        this.compatMode = compatMode;
+    }
+    /** Record a named import from `module`. The emitter will write
+     *  `import { ...names } from "<module>"` once per module. */
+    useImport(module, name) {
+        let names = this.extraImports.get(module);
+        if (!names) {
+            names = new Set();
+            this.extraImports.set(module, names);
+        }
+        names.add(name);
+        return name;
+    }
+    /** Returns the recorded extra imports as `{ module, names[] }` tuples,
+     *  sorted by module path then by name for deterministic output. */
+    extraImportEntries() {
+        return [...this.extraImports.entries()]
+            .sort((a, b) => a[0].localeCompare(b[0]))
+            .map(([module, names]) => ({ module, names: [...names].sort() }));
+    }
+    /** Record a name as a TS class detected by the class-shape pass. Lookup
+     *  happens at every `<Name>.new(...)` call site so we can lower it to
+     *  `new <Name>(...)` instead of leaving it as a static-method call. */
+    recordDetectedClass(name) {
+        this.detectedClasses.add(name);
+    }
+    isDetectedClass(name) {
+        return this.detectedClasses.has(name);
+    }
+    /** Mark a local identifier as already bound to an import of the same
+     *  name. Subsequent `local <name> = <name>` declarations are dropped to
+     *  avoid the redundant shadowing. */
+    suppressLocal(name) {
+        this.suppressedLocals.add(name);
+    }
+    isSuppressedLocal(name) {
+        return this.suppressedLocals.has(name);
+    }
+    use(helper) {
+        this.imports.add(helper);
+        return helper;
+    }
+    importedHelpers() {
+        return [...this.imports].sort();
+    }
+    freshIdentifier(prefix) {
+        const id = this.tempCounter;
+        this.tempCounter += 1;
+        return `${prefix}_${id}`;
+    }
+    withScope(fn) {
+        this.scopes.push(new Map());
+        this.jsNameOverrides.push(new Map());
+        try {
+            return fn();
+        }
+        finally {
+            this.scopes.pop();
+            this.jsNameOverrides.pop();
+        }
+    }
+    setLocalJsName(name, jsName) {
+        this.jsNameOverrides[this.jsNameOverrides.length - 1].set(name, jsName);
+    }
+    getLocalJsName(name) {
+        for (let i = this.jsNameOverrides.length - 1; i >= 0; i -= 1) {
+            const v = this.jsNameOverrides[i].get(name);
+            if (v !== undefined)
+                return v;
+        }
+        return undefined;
+    }
+    defineLocal(name, type) {
+        this.scopes[this.scopes.length - 1].set(name, type);
+    }
+    hasLocalInCurrentScope(name) {
+        return this.scopes[this.scopes.length - 1].has(name);
+    }
+    hasLocalInOuterScope(name) {
+        for (let i = this.scopes.length - 2; i >= 0; i -= 1) {
+            if (this.scopes[i].has(name))
+                return true;
+        }
+        return false;
+    }
+    assignLocal(name, type) {
+        for (let i = this.scopes.length - 1; i >= 0; i -= 1) {
+            const scope = this.scopes[i];
+            if (scope.has(name)) {
+                scope.set(name, type);
+                return;
+            }
+        }
+        this.defineLocal(name, type);
+    }
+    lookupLocal(name) {
+        for (let i = this.scopes.length - 1; i >= 0; i -= 1) {
+            const type = this.scopes[i].get(name);
+            if (type)
+                return type;
+        }
+        return 'unknown';
+    }
+}
+//# sourceMappingURL=context.js.map

package/dist/compile/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../../src/compile/context.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,cAAc,GAAG,iBAAiB,CAAC;AAgBhD;;;8EAG8E;AAC9E,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC;IACrC,SAAS,EAAE,SAAS,EAAE,cAAc,EAAE,cAAc,EAAE,QAAQ;CAC/D,CAAC,CAAC;AAaH,MAAM,OAAO,cAAc;IAiCG;IAhCX,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5B,MAAM,GAAmC,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;IACtE;;6EAEyE;IACxD,eAAe,GAA0B,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;IACtE;;;sDAGkD;IACzC,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IACvC,WAAW,GAAG,CAAC,CAAC;IAExB;;mFAE+E;IAC9D,YAAY,GAAG,IAAI,GAAG,EAAuB,CAAC;IAE/D;;;+CAG2C;IAC1B,eAAe,GAAG,IAAI,GAAG,EAAU,CAAC;IAErD;;;;;4BAKwB;IACP,gBAAgB,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtD,YAA4B,aAAyB,QAAQ;QAAjC,eAAU,GAAV,UAAU,CAAuB;IAAG,CAAC;IAEjE;iEAC6D;IAC7D,SAAS,CAAC,MAAc,EAAE,IAAY;QACpC,IAAI,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACvC,CAAC;QACD,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChB,OAAO,IAAI,CAAC;IACd,CAAC;IAED;uEACmE;IACnE,kBAAkB;QAChB,OAAO,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;aACpC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;aACxC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;IACtE,CAAC;IAED;;2EAEuE;IACvE,mBAAmB,CAAC,IAAY;QAC9B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,eAAe,CAAC,IAAY;QAC1B,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAED;;yCAEqC;IACrC,aAAa,CAAC,IAAY;QACxB,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,iBAAiB,CAAC,IAAY;QAC5B,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,GAAG,CAAC,MAAc;QAChB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACzB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,eAAe;QACb,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;IAClC,CAAC;IAED,eAAe,CAAC,MAAc;QAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC;QAC5B,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC;QACtB,OAAO,GAAG,MAAM,IAAI,EAAE,EAAE,CAAC;IAC3B,CAAC;IAED,SAAS,CAAI,EAAW;QACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;QAC5B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC;YACH,OAAO,EAAE,EAAE,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,cAAc,CAAC,IAAY,EAAE,MAAc;QACzC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC3E,CAAC;IAED,cAAc,CAAC,IAAY;QACzB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7D,MAAM,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC7C,IAAI,CAAC,KAAK,SAAS;gBAAE,OAAO,CAAC,CAAC;QAChC,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,WAAW,CAAC,IAAY,EAAE,IAAqB;QAC7C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACvD,CAAC;IAED,sBAAsB,CAAC,IAAY;QACjC,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC;IAED,oBAAoB,CAAC,IAAY;QAC/B,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACpD,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,GAAG,CAAC,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;QAC7C,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,WAAW,CAAC,IAAY,EAAE,IAAqB;QAC7C,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACpD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC;YAC9B,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpB,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACtB,OAAO;YACT,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED,WAAW,CAAC,IAAY;QACtB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,IAAI;gBAAE,OAAO,IAAI,CAAC;QACxB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF"}

package/dist/compile/index.d.ts

@@ -0,0 +1,43 @@
+import { type ParseResult } from '../parser/index.js';
+import { type CompatMode } from './context.js';
+import './macros/datatypes.js';
+import './macros/instance.js';
+import './macros/stdlib.js';
+import './rbxts-runtime.js';
+import { type SourceMap } from './sourcemap.js';
+export interface CompileOptions {
+    /** Path of the source file. Used in the source map's `sources` field. */
+    sourceFile?: string;
+    /** Generate a source map mapping each TS statement back to its .luau line. */
+    sourceMap?: boolean;
+    /** Inline the source map as a base64 data URL appended to the output. */
+    inlineSourceMap?: boolean;
+    /** Preserve `--` line comments and `--[[ ]]` block comments from the source. */
+    preserveComments?: boolean;
+    /** Emit-shape compatibility mode.
+     *
+     *  - `'native'` (default): emit TS that imports stdlib helpers from
+     *    `luau2ts/runtime` and pairs with any Roblox runtime that mirrors
+     *    Roblox's Luau API surface: `Vector3.new(...)`, `game:GetService(...)`, etc.
+     *  - `'rbxts'`: emit TS that mirrors what roblox-ts accepts as input:
+     *    `new Vector3(...)`, `import { Workspace } from "@rbxts/services"`,
+     *    `new ClassName()` for `Instance.new("ClassName")`, etc.
+     */
+    compatMode?: CompatMode;
+    /** Run Prettier on the emitted TypeScript. Defaults to `true`. Disable
+     *  if you want raw TypeScript-printer output (faster, but the output is
+     *  4-space indented without canonical spacing around blocks). */
+    pretty?: boolean;
+}
+export interface CompileResult {
+    /** Full TypeScript source of the compiled output. */
+    source: string;
+    /** Helper names the output imports from luau2ts/runtime. */
+    helpers: string[];
+    /** Parser errors, if any (compiler returns parser-error output even so). */
+    errors: ParseResult['errors'];
+    /** Source map JSON when sourceMap was requested. */
+    sourceMap?: SourceMap;
+}
+export declare function compile(source: string, options?: CompileOptions): Promise<CompileResult>;
+//# sourceMappingURL=index.d.ts.map

package/dist/compile/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/compile/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAgBL,KAAK,WAAW,EAKjB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EAAmD,KAAK,UAAU,EAAwB,MAAM,cAAc,CAAC;AAItH,OAAO,uBAAuB,CAAC;AAC/B,OAAO,sBAAsB,CAAC;AAC9B,OAAO,oBAAoB,CAAC;AAC5B,OAAO,oBAAoB,CAAC;AAG5B,OAAO,EAGL,KAAK,SAAS,EAEf,MAAM,gBAAgB,CAAC;AAisExB,MAAM,WAAW,cAAc;IAC7B,yEAAyE;IACzE,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,8EAA8E;IAC9E,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,yEAAyE;IACzE,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,gFAAgF;IAChF,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B;;;;;;;;OAQG;IACH,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB;;qEAEiE;IACjE,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,aAAa;IAC5B,qDAAqD;IACrD,MAAM,EAAE,MAAM,CAAC;IACf,4DAA4D;IAC5D,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,4EAA4E;IAC5E,MAAM,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC;IAC9B,oDAAoD;IACpD,SAAS,CAAC,EAAE,SAAS,CAAC;CACvB;AAED,wBAAsB,OAAO,CAC3B,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,cAAmB,GAC3B,OAAO,CAAC,aAAa,CAAC,CAgLxB"}