flake-monster 0.3.0 → 0.3.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "flake-monster",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "Source-to-source test hardener that injects async delays to surface flaky tests",
   "type": "module",
   "bin": {

package/src/adapters/javascript/index.js

@@ -1,7 +1,7 @@
 import { fileURLToPath } from 'node:url';
 import { dirname, join, posix } from 'node:path';
 import { parseSource } from './parser.js';
-import { computeInjections, computeRuntimeImportInsertion, applyInsertions } from './injector.js';
+import { computeInjections, computeRuntimeImportInsertion, applyInsertions, MARKER_PREFIX } from './injector.js';
 import { recoverDelays, scanForRecovery } from './remover.js';
 
 const __dirname = dirname(fileURLToPath(import.meta.url));
@@ -43,18 +43,28 @@ export function createJavaScriptAdapter() {
     },
 
     inject(source, options) {
-      const { ast } = parseSource(source);
-      const { insertions, points } = computeInjections(ast, source, options);
-      const runtimeNeeded = points.length > 0;
-
-      if (runtimeNeeded) {
-        const importPath = computeRuntimeImportPath(options.filePath);
-        const imp = computeRuntimeImportInsertion(ast, source, importPath);
-        if (imp) insertions.push(imp);
+      if (source.includes(MARKER_PREFIX)) {
+        console.warn(`  Skipping ${options.filePath}: already injected (restore first)`);
+        return { source, points: [], runtimeNeeded: false };
       }
 
-      const output = applyInsertions(source, insertions);
-      return { source: output, points, runtimeNeeded };
+      try {
+        const { ast } = parseSource(source);
+        const { insertions, points } = computeInjections(ast, source, options);
+        const runtimeNeeded = points.length > 0;
+
+        if (runtimeNeeded) {
+          const importPath = computeRuntimeImportPath(options.filePath);
+          const imp = computeRuntimeImportInsertion(ast, source, importPath);
+          if (imp) insertions.push(imp);
+        }
+
+        const output = applyInsertions(source, insertions);
+        return { source: output, points, runtimeNeeded };
+      } catch (err) {
+        console.warn(`  Skipping ${options.filePath}: ${err.message}`);
+        return { source, points: [], runtimeNeeded: false };
+      }
     },
 
     remove(source) {

package/src/adapters/javascript/parser.js

@@ -1,5 +1,15 @@
 import * as acorn from 'acorn';
-import { attachComments } from 'astravel';
+import { defaultTraveler, attachComments } from 'astravel';
+
+// Patch astravel bug: PropertyDefinition reuses MethodDefinition handler which
+// calls this.go(node.value) without a null guard. Uninitialized class fields
+// (e.g. `bar;`) have value: null, crashing the traversal.
+defaultTraveler.PropertyDefinition = function (node, state) {
+  this.go(node.key, state);
+  if (node.value != null) {
+    this.go(node.value, state);
+  }
+};
 
 /**
  * Parse JS source to ESTree AST with comments attached to nodes.

package/src/cli/index.js

@@ -9,7 +9,7 @@ export function createCli() {
   program
     .name('flake-monster')
     .description('Source-to-source test hardener, injects async delays to surface flaky tests')
-    .version('0.1.0');
+    .version('0.3.1');
 
   registerInjectCommand(program);
   registerRestoreCommand(program);