rbxts-transform-luau 1.0.0 → 1.0.1

package/out/index.js

@@ -28,6 +28,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = default_1;
 const typescript_1 = __importDefault(require("typescript"));
+const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 const format_1 = require("./passes/format");
 const LUAU_TYPE = {
@@ -170,23 +171,60 @@ function commonRoot(files) {
     return root.join(path.sep) || undefined;
 }
 const pending = new Map();
-let finalizeRegistered = false;
+const writingFiles = new Set();
+// dir path → watcher, shared across files in the same dir
+const dirWatchers = new Map();
+function processFile(meta) {
+    try {
+        (0, format_1.formatFile)(meta.outPath, meta.strict, meta.optimizeLevel, meta.sidecar, meta.annotate ? meta.types : new Map());
+    }
+    catch {
+        // silently skip files that fail — they stay as-is
+    }
+}
+function watchDir(dir) {
+    if (dirWatchers.has(dir))
+        return;
+    try {
+        const watcher = fs.watch(dir, { persistent: false }, (_event, filename) => {
+            if (!filename)
+                return;
+            const outPath = path.join(dir, filename);
+            const meta = pending.get(outPath);
+            if (!meta)
+                return;
+            if (writingFiles.has(outPath))
+                return;
+            pending.delete(outPath);
+            writingFiles.add(outPath);
+            try {
+                processFile(meta);
+            }
+            finally {
+                setTimeout(() => writingFiles.delete(outPath), 50).unref();
+            }
+            // clean up watcher if no more pending files in this dir
+            const stillPending = [...pending.keys()].some(p => path.dirname(p) === dir);
+            if (!stillPending) {
+                watcher.close();
+                dirWatchers.delete(dir);
+            }
+        });
+        dirWatchers.set(dir, watcher);
+    }
+    catch {
+        // dir may not exist yet — fall back to exit handler
+    }
+}
 function flushPending() {
     for (const [, meta] of pending) {
-        try {
-            (0, format_1.formatFile)(meta.outPath, meta.strict, meta.optimizeLevel, meta.sidecar, meta.annotate ? meta.types : new Map());
-        }
-        catch {
-            // silently skip files that fail — they stay as-is
-        }
+        processFile(meta);
     }
     pending.clear();
-}
-function registerFinalizer() {
-    if (finalizeRegistered)
-        return;
-    finalizeRegistered = true;
-    process.on("exit", flushPending);
+    for (const [dir, watcher] of dirWatchers) {
+        watcher.close();
+        dirWatchers.delete(dir);
+    }
 }
 function jsDocText(comment) {
     if (!comment)
@@ -236,9 +274,9 @@ function collectJsDoc(ts, sourceFile) {
 function default_1(program, config = {}) {
     const { strict = true, optimize = false, annotate = true, verbose = false } = config;
     const optimizeLevel = optimize === false ? false : [0, 1, 2].includes(optimize) ? optimize : 2;
-    // Watch mode: flush the previous run's pending files before starting this one.
-    flushPending();
-    registerFinalizer();
+    // Safety net: flush any files that weren't caught by the dir watcher.
+    // Also handles the last cycle when the process exits.
+    process.on("exit", flushPending);
     const outDir = program.getCompilerOptions().outDir;
     const checker = annotate ? program.getTypeChecker() : null;
     return (_ctx) => (sourceFile) => {
@@ -247,6 +285,7 @@ function default_1(program, config = {}) {
             const sidecar = collectJsDoc(typescript_1.default, sourceFile);
             const types = checker ? collectTypes(checker, sourceFile) : new Map();
             pending.set(outPath, { outPath, strict, optimizeLevel, annotate, verbose, sidecar, types });
+            watchDir(path.dirname(outPath));
             if (verbose) {
                 const rel = outDir ? path.relative(outDir, outPath) : outPath;
                 console.log(`luau: ${rel}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "rbxts-transform-luau",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "roblox-ts transformer: idiomatic Luau output — preamble organization, comment cleanup, const promotion, and directive injection",
   "main": "out/index.js",
   "types": "out/index.d.ts",