rbxts-transform-boost 0.1.0 → 1.1.0

package/bench/src/shared/fns-bare.ts

@@ -1,127 +0,0 @@
-// ── Vector3 / physics ────────────────────────────────────────────────────────
-
-export function integrate(pos: Vector3, vel: Vector3, acc: Vector3, dt: number): [Vector3, Vector3] {
-    const newVel = vel.add(acc.mul(dt));
-    const newPos = pos.add(newVel.mul(dt));
-    return [newPos, newVel];
-}
-
-export function dot(a: Vector3, b: Vector3): number {
-    return a.X * b.X + a.Y * b.Y + a.Z * b.Z;
-}
-
-export function cross(a: Vector3, b: Vector3): Vector3 {
-    return new Vector3(
-        a.Y * b.Z - a.Z * b.Y,
-        a.Z * b.X - a.X * b.Z,
-        a.X * b.Y - a.Y * b.X,
-    );
-}
-
-export function lerpVec3(a: Vector3, b: Vector3, t: number): Vector3 {
-    return new Vector3(
-        a.X + (b.X - a.X) * t,
-        a.Y + (b.Y - a.Y) * t,
-        a.Z + (b.Z - a.Z) * t,
-    );
-}
-
-// ── Buffer / encoding ────────────────────────────────────────────────────────
-
-export function encodeFixed(buf: buffer, offset: number, value: number, scale: number): number {
-    const fixed = math.floor(value * scale);
-    const clamped = math.clamp(fixed, -32768, 32767);
-    buffer.writei16(buf, offset, clamped);
-    return offset + 2;
-}
-
-export function encodePacket(buf: buffer, x: number, y: number, z: number, scale: number): void {
-    let off = 0;
-    off = encodeFixed(buf, off, x, scale);
-    off = encodeFixed(buf, off, y, scale);
-    encodeFixed(buf, off, z, scale);
-}
-
-// ── Math / arithmetic ────────────────────────────────────────────────────────
-
-export function sumWeighted(values: Array<number>, weights: Array<number>): number {
-    let total = 0;
-    for (let i = 0; i < values.size(); i++) {
-        total += values[i] * weights[i];
-    }
-    return total;
-}
-
-export function dotProduct(a: Array<number>, b: Array<number>): number {
-    let sum = 0;
-    for (let i = 0; i < a.size(); i++) {
-        sum += a[i] * b[i];
-    }
-    return sum;
-}
-
-export function norm(values: Array<number>): number {
-    let sq = 0;
-    for (let i = 0; i < values.size(); i++) {
-        sq += values[i] * values[i];
-    }
-    return math.sqrt(sq);
-}
-
-export function mathHeavy(x: number, y: number): number {
-    return math.sin(x) * math.cos(y) + math.sqrt(x * x + y * y) + math.atan2(y, x);
-}
-
-export function fib(n: number): number {
-    if (n <= 1) return n;
-    let a = 0;
-    let b = 1;
-    for (let i = 2; i <= n; i++) {
-        const tmp = a + b;
-        a = b;
-        b = tmp;
-    }
-    return b;
-}
-
-// ── CFrame ───────────────────────────────────────────────────────────────────
-
-export function cfLookAt(eye: Vector3, target: Vector3): CFrame {
-    return CFrame.lookAt(eye, target);
-}
-
-export function cfChain(cf: CFrame, dt: number): CFrame {
-    return cf.mul(CFrame.Angles(0, dt, 0));
-}
-
-// ── Services / instance ──────────────────────────────────────────────────────
-
-export function serviceWork(): string {
-    const count = game.GetService("Players").GetPlayers().size();
-    const running = game.GetService("RunService").IsRunning();
-    return `${count}-${running}`;
-}
-
-export function multiService(): boolean {
-    const players = game.GetService("Players");
-    const rs = game.GetService("RunService");
-    const ws = game.GetService("Workspace");
-    return rs.IsRunning() && players.MaxPlayers > 0 && ws.Gravity > 0;
-}
-
-export function cameraWork(camera: Camera): number {
-    const pos = camera.CFrame.Position;
-    const look = camera.CFrame.LookVector;
-    const fov = camera.FieldOfView;
-    return pos.Magnitude + look.X + fov;
-}
-
-// ── String ───────────────────────────────────────────────────────────────────
-
-export function formatStats(label: string, value: number, unit: string): string {
-    return `${label}: ${string.format("%.4f", value)} ${unit}`;
-}
-
-export function buildKey(prefix: string, id: number, suffix: string): string {
-    return `${prefix}_${id}_${suffix}`;
-}

package/bench/src/shared/fns.ts

@@ -1,127 +0,0 @@
-// ── Vector3 / physics ────────────────────────────────────────────────────────
-
-export function integrate(pos: Vector3, vel: Vector3, acc: Vector3, dt: number): [Vector3, Vector3] {
-    const newVel = vel.add(acc.mul(dt));
-    const newPos = pos.add(newVel.mul(dt));
-    return [newPos, newVel];
-}
-
-export function dot(a: Vector3, b: Vector3): number {
-    return a.X * b.X + a.Y * b.Y + a.Z * b.Z;
-}
-
-export function cross(a: Vector3, b: Vector3): Vector3 {
-    return new Vector3(
-        a.Y * b.Z - a.Z * b.Y,
-        a.Z * b.X - a.X * b.Z,
-        a.X * b.Y - a.Y * b.X,
-    );
-}
-
-export function lerpVec3(a: Vector3, b: Vector3, t: number): Vector3 {
-    return new Vector3(
-        a.X + (b.X - a.X) * t,
-        a.Y + (b.Y - a.Y) * t,
-        a.Z + (b.Z - a.Z) * t,
-    );
-}
-
-// ── Buffer / encoding ────────────────────────────────────────────────────────
-
-export function encodeFixed(buf: buffer, offset: number, value: number, scale: number): number {
-    const fixed = math.floor(value * scale);
-    const clamped = math.clamp(fixed, -32768, 32767);
-    buffer.writei16(buf, offset, clamped);
-    return offset + 2;
-}
-
-export function encodePacket(buf: buffer, x: number, y: number, z: number, scale: number): void {
-    let off = 0;
-    off = encodeFixed(buf, off, x, scale);
-    off = encodeFixed(buf, off, y, scale);
-    encodeFixed(buf, off, z, scale);
-}
-
-// ── Math / arithmetic ────────────────────────────────────────────────────────
-
-export function sumWeighted(values: Array<number>, weights: Array<number>): number {
-    let total = 0;
-    for (let i = 0; i < values.size(); i++) {
-        total += values[i] * weights[i];
-    }
-    return total;
-}
-
-export function dotProduct(a: Array<number>, b: Array<number>): number {
-    let sum = 0;
-    for (let i = 0; i < a.size(); i++) {
-        sum += a[i] * b[i];
-    }
-    return sum;
-}
-
-export function norm(values: Array<number>): number {
-    let sq = 0;
-    for (let i = 0; i < values.size(); i++) {
-        sq += values[i] * values[i];
-    }
-    return math.sqrt(sq);
-}
-
-export function mathHeavy(x: number, y: number): number {
-    return math.sin(x) * math.cos(y) + math.sqrt(x * x + y * y) + math.atan2(y, x);
-}
-
-export function fib(n: number): number {
-    if (n <= 1) return n;
-    let a = 0;
-    let b = 1;
-    for (let i = 2; i <= n; i++) {
-        const tmp = a + b;
-        a = b;
-        b = tmp;
-    }
-    return b;
-}
-
-// ── CFrame ───────────────────────────────────────────────────────────────────
-
-export function cfLookAt(eye: Vector3, target: Vector3): CFrame {
-    return CFrame.lookAt(eye, target);
-}
-
-export function cfChain(cf: CFrame, dt: number): CFrame {
-    return cf.mul(CFrame.Angles(0, dt, 0));
-}
-
-// ── Services / instance ──────────────────────────────────────────────────────
-
-export function serviceWork(): string {
-    const count = game.GetService("Players").GetPlayers().size();
-    const running = game.GetService("RunService").IsRunning();
-    return `${count}-${running}`;
-}
-
-export function multiService(): boolean {
-    const players = game.GetService("Players");
-    const rs = game.GetService("RunService");
-    const ws = game.GetService("Workspace");
-    return rs.IsRunning() && players.MaxPlayers > 0 && ws.Gravity > 0;
-}
-
-export function cameraWork(camera: Camera): number {
-    const pos = camera.CFrame.Position;
-    const look = camera.CFrame.LookVector;
-    const fov = camera.FieldOfView;
-    return pos.Magnitude + look.X + fov;
-}
-
-// ── String ───────────────────────────────────────────────────────────────────
-
-export function formatStats(label: string, value: number, unit: string): string {
-    return `${label}: ${string.format("%.4f", value)} ${unit}`;
-}
-
-export function buildKey(prefix: string, id: number, suffix: string): string {
-    return `${prefix}_${id}_${suffix}`;
-}

package/bench/tsconfig.json

@@ -1,32 +0,0 @@
-{
-    "compilerOptions": {
-        "target": "ESNext",
-        "module": "commonjs",
-        "moduleDetection": "force",
-        "moduleResolution": "Node",
-        "strict": true,
-        "noLib": true,
-        "esModuleInterop": true,
-        "allowSyntheticDefaultImports": true,
-        "forceConsistentCasingInFileNames": true,
-        "rootDir": "./src",
-        "outDir": "./out/src",
-        "typeRoots": [
-            "./node_modules/@rbxts"
-        ],
-        "plugins": [
-            {
-                "transform": "rbxts-transform-boost",
-                "optimize": true,
-                "hoist": true
-            }
-        ]
-    },
-    "include": [
-        "./src/server",
-        "./src/shared/fns.ts"
-    ],
-    "rbxts": {
-        "includePath": "out/include"
-    }
-}

package/bench/tsconfig.notransform.json

@@ -1,24 +0,0 @@
-{
-    "compilerOptions": {
-        "target": "ESNext",
-        "module": "commonjs",
-        "moduleDetection": "force",
-        "moduleResolution": "Node",
-        "strict": true,
-        "noLib": true,
-        "esModuleInterop": true,
-        "allowSyntheticDefaultImports": true,
-        "forceConsistentCasingInFileNames": true,
-        "rootDir": "./src",
-        "outDir": "./out/src",
-        "typeRoots": [
-            "./node_modules/@rbxts"
-        ],
-    },
-    "include": [
-        "./src/shared/fns-bare.ts"
-    ],
-    "rbxts": {
-        "includePath": "out/include"
-    }
-}

package/rokit.toml

@@ -1,3 +0,0 @@
-[tools]
-rojo = "rojo-rbx/rojo@7.6.1"
-rotor = "uproot/rotor@2.2.0"

package/scripts/postprocess.js

@@ -1,29 +0,0 @@
-#!/usr/bin/env node
-// Replaces "-- !fn-native\n" markers (emitted by rbxts-transform-perf's native pass)
-// with bare "@native\n" Luau attributes in compiled output files.
-// Run after `rotor build` to make per-function @native actually take effect.
-
-const fs = require("fs");
-const path = require("path");
-
-const MARKER = "--!fn-native\n";
-const ATTRIBUTE = "@native\n";
-
-function processDir(dir) {
-    if (!fs.existsSync(dir)) return;
-    for (const entry of fs.readdirSync(dir, { withFileTypes: true })) {
-        const full = path.join(dir, entry.name);
-        if (entry.isDirectory()) {
-            processDir(full);
-        } else if (entry.name.endsWith(".luau") || entry.name.endsWith(".lua")) {
-            const original = fs.readFileSync(full, "utf8");
-            if (!original.includes(MARKER)) continue;
-            const patched = original.replaceAll(MARKER, ATTRIBUTE);
-            fs.writeFileSync(full, patched, "utf8");
-            console.log(`@native patched: ${full}`);
-        }
-    }
-}
-
-const outDir = process.argv[2] ?? path.join(__dirname, "..", "test", "out");
-processDir(outDir);

package/src/config.ts

@@ -1,10 +0,0 @@
-export interface PluginConfig {
-    // Prepend --!optimize 2 to every file that doesn't already have it.
-    // Default: true
-    optimize?: boolean;
-
-    // Hoist repeated game.GetService() calls to module-level locals,
-    // and hoist repeated property access chains within functions to locals.
-    // Default: true
-    hoist?: boolean;
-}

package/src/index.ts

@@ -1,25 +0,0 @@
-import type ts from "typescript";
-import type { PluginConfig } from "./config";
-import { nativePass } from "./passes/native";
-import { cachePass } from "./passes/cache";
-import { loopsPass } from "./passes/loops";
-import { annotatePass } from "./passes/annotate";
-
-export type { PluginConfig };
-
-export default function(
-    program: ts.Program,
-    config: PluginConfig = {},
-    { ts }: { ts: typeof import("typescript") },
-): ts.TransformerFactory<ts.SourceFile> {
-    const { optimize = true, hoist = true } = config;
-
-    return (ctx: ts.TransformationContext) => (sourceFile: ts.SourceFile): ts.SourceFile => {
-        annotatePass(ts, program, sourceFile);
-        let result = sourceFile;
-        if (hoist) result = cachePass(ts, program, ctx, result);
-        result = loopsPass(ts, program, ctx, result);
-        if (optimize) result = nativePass(ts, ctx, result);
-        return result;
-    };
-}

package/src/passes/annotate.ts

@@ -1,207 +0,0 @@
-import type ts from "typescript";
-import * as fs from "fs";
-import * as path from "path";
-
-// Luau type names for TypeScript type strings rotor knows about
-const LUAU_TYPE: Record<string, string> = {
-    number: "number",
-    string: "string",
-    boolean: "boolean",
-    Vector3: "Vector3",
-    Vector2: "Vector2",
-    Vector2int16: "Vector2int16",
-    Vector3int16: "Vector3int16",
-    CFrame: "CFrame",
-    UDim: "UDim",
-    UDim2: "UDim2",
-    Color3: "Color3",
-    BrickColor: "BrickColor",
-    TweenInfo: "TweenInfo",
-    NumberRange: "NumberRange",
-    NumberSequence: "NumberSequence",
-    ColorSequence: "ColorSequence",
-    Rect: "Rect",
-    Region3: "Region3",
-    Ray: "Ray",
-    buffer: "buffer",
-    // Roblox service types
-    Instance: "Instance",
-    BasePart: "BasePart",
-    Part: "Part",
-    Model: "Model",
-    Player: "Player",
-    Camera: "Camera",
-    Workspace: "Workspace",
-    RunService: "RunService",
-    Players: "Players",
-    // Luau numeric arrays — kept as {number} in Luau
-};
-
-type FnAnnotation = {
-    params: Array<string | null>; // null = unknown/skip
-};
-
-// Global sidecar: outLuauPath → list of function annotations for that file
-const sidecar = new Map<string, Map<string, FnAnnotation>>();
-let hooked = false;
-
-function luauTypeForTsType(
-    ts: typeof import("typescript"),
-    checker: ts.TypeChecker,
-    node: ts.ParameterDeclaration,
-): string | null {
-    if (node.type) {
-        const mapped = mapTypeNode(ts, node.type);
-        if (mapped) return mapped;
-    }
-    const type = checker.getTypeAtLocation(node);
-    const name = checker.typeToString(type);
-    return LUAU_TYPE[name] ?? null;
-}
-
-function mapTypeNode(ts: typeof import("typescript"), typeNode: ts.TypeNode): string | null {
-    if (ts.isTypeReferenceNode(typeNode)) {
-        const name = ts.isIdentifier(typeNode.typeName) ? typeNode.typeName.text : null;
-        if (!name) return null;
-        if (LUAU_TYPE[name]) return LUAU_TYPE[name];
-        // Array<T> → {T}
-        if ((name === "Array" || name === "ReadonlyArray") && typeNode.typeArguments?.length === 1) {
-            const inner = mapTypeNode(ts, typeNode.typeArguments[0]);
-            return inner ? `{${inner}}` : "{any}";
-        }
-        return null;
-    }
-    if (ts.isArrayTypeNode(typeNode)) {
-        const inner = mapTypeNode(ts, typeNode.elementType);
-        return inner ? `{${inner}}` : "{any}";
-    }
-    const kw: Partial<Record<number, string>> = {
-        [ts.SyntaxKind.NumberKeyword]: "number",
-        [ts.SyntaxKind.StringKeyword]: "string",
-        [ts.SyntaxKind.BooleanKeyword]: "boolean",
-    };
-    if (typeNode.kind in kw) return kw[typeNode.kind]!;
-    return null;
-}
-
-function outPathForSource(
-    sourceFile: ts.SourceFile,
-    program: ts.Program,
-): string | null {
-    const options = program.getCompilerOptions();
-    const outDir = options.outDir;
-    if (!outDir) return null;
-
-    // Compute rootDir: explicit option or the common root of all source files
-    const rootDir = options.rootDir
-        ?? commonRoot(program.getRootFileNames());
-    if (!rootDir) return null;
-
-    const rel = path.relative(rootDir, sourceFile.fileName);
-    if (rel.startsWith("..")) return null;
-
-    // Change .ts / .tsx extension to .luau
-    const luauRel = rel.replace(/\.tsx?$/, ".luau");
-    return path.join(outDir, luauRel);
-}
-
-function commonRoot(files: readonly string[]): string | undefined {
-    if (files.length === 0) return undefined;
-    const parts = files[0].split(path.sep);
-    let root = parts.slice(0, parts.length - 1);
-    for (const f of files.slice(1)) {
-        const fp = f.split(path.sep);
-        let i = 0;
-        while (i < root.length && i < fp.length - 1 && root[i] === fp[i]) i++;
-        root = root.slice(0, i);
-    }
-    return root.join(path.sep) || undefined;
-}
-
-function collectAnnotations(
-    ts: typeof import("typescript"),
-    checker: ts.TypeChecker,
-    sourceFile: ts.SourceFile,
-    outPath: string,
-): void {
-    const fileMap = sidecar.get(outPath) ?? new Map<string, FnAnnotation>();
-    sidecar.set(outPath, fileMap);
-
-    function visit(node: ts.Node): void {
-        if (ts.isFunctionDeclaration(node) && node.name) {
-            const fnName = node.name.text;
-            const params = node.parameters.map(p => luauTypeForTsType(ts, checker, p));
-            if (params.some(p => p !== null)) {
-                fileMap.set(fnName, { params });
-            }
-        }
-        ts.forEachChild(node, visit);
-    }
-    visit(sourceFile);
-}
-
-function injectAnnotations(luauPath: string, fileMap: Map<string, FnAnnotation>): void {
-    if (!fs.existsSync(luauPath)) return;
-    let src = fs.readFileSync(luauPath, "utf8");
-    let changed = false;
-
-    for (const [fnName, ann] of fileMap) {
-        if (ann.params.every(p => p === null)) continue;
-
-        // Match: local function fnName(a, b, c)
-        // Captures the param list so we can replace individual names
-        const re = new RegExp(
-            `(local function ${escapeRegex(fnName)}\\()([^)]*)(\\.\\.\\.\\))?\\)`,
-        );
-        src = src.replace(re, (_match: string, open: string, rawParams: string, vararg: string | undefined) => {
-            const names = rawParams.split(",").map((s: string) => s.trim()).filter(Boolean);
-            const annotated = names.map((name: string, i: number) => {
-                // strip any existing annotation
-                const bare = name.split(":")[0].trim();
-                const typ = ann.params[i];
-                return typ ? `${bare}: ${typ}` : bare;
-            });
-            if (vararg) annotated.push("...");
-            changed = true;
-            return `${open}${annotated.join(", ")})`;
-        });
-    }
-
-    if (changed) fs.writeFileSync(luauPath, src, "utf8");
-}
-
-function escapeRegex(s: string): string {
-    return s.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-}
-
-function installWatcher(outDir: string): void {
-    if (hooked) return;
-    hooked = true;
-    const seen = new Set<string>();
-    const watcher = fs.watch(outDir, { recursive: true }, (_event, filename) => {
-        if (!filename || !filename.endsWith(".luau")) return;
-        const full = path.join(outDir, filename);
-        if (seen.has(full)) return;
-        const fileMap = sidecar.get(full);
-        if (!fileMap) return;
-        seen.add(full);
-        if (fileMap.size > 0) {
-            try { injectAnnotations(full, fileMap); } catch { /* ignore */ }
-        }
-    });
-    watcher.unref();
-}
-
-export function annotatePass(
-    ts: typeof import("typescript"),
-    program: ts.Program,
-    sourceFile: ts.SourceFile,
-): void {
-    const outPath = outPathForSource(sourceFile, program);
-    if (!outPath) return;
-
-    const checker = program.getTypeChecker();
-    collectAnnotations(ts, checker, sourceFile, outPath);
-    const outDir = program.getCompilerOptions().outDir!;
-    installWatcher(outDir);
-}