@astryxdesign/cli 0.6.3 → 0.6.4-canary.10dd683

package/assets/codemods/integration-discovery.test.mjs

@@ -163,6 +163,21 @@ describe('integration codemod discovery', () => {
     ).rejects.toThrow(/is invalid/i);
   });
 
+  it('names a broken codemod by its path inside the package', async () => {
+    scaffold({
+      '1.1.0/no-default.mjs': `export const nope = 1;\n`,
+    });
+    const project = await Project.load(tmpDir);
+    const failure = await discoverIntegrationCodemods(
+      project.loadedIntegrations,
+    ).then(
+      () => null,
+      error => error,
+    );
+    expect(failure?.message).toContain('(codemods/1.1.0/no-default.mjs)');
+    expect(failure?.message).not.toContain(tmpDir);
+  });
+
   it('fails discovery when default export is not a codemod envelope', async () => {
     scaffold({
       '0.2.0/bad.mjs': `export default { title: 'x', transform: () => null };\n`,
@@ -221,3 +236,61 @@ describe('integration codemod discovery', () => {
     ).rejects.toThrow(/across versions/i);
   });
 });
+
+describe('test files beside a codemod', () => {
+  // The incident this closes: every .ts/.mjs/.js under a version folder was
+  // loaded AND VALIDATED as a codemod, so a test file colocated with its
+  // transform failed validation — and because a definition error is a hard
+  // error, it took every codemod in that version with it. `astryx upgrade`
+  // then applied nothing and reported success. Core is immune because its own
+  // codemods are enumerated in a registry rather than discovered by walking a
+  // directory, so its colocated tests are simply never visited. Only
+  // integrations carry the landmine.
+  const TRANSFORM = `
+    export default {
+      type: 'code',
+      title: 'Drop foo',
+      transform: (file) => file.source.replace(/foo/g, 'bar'),
+    };
+  `;
+  // Not a codemod: no default export of the right shape. Loading it throws.
+  const A_TEST = `
+    import {describe, it, expect} from 'vitest';
+    describe('drop-foo', () => {
+      it('drops foo', () => expect(1).toBe(1));
+    });
+  `;
+
+  it.each([
+    ['a .test. sibling', '0.2.0/drop-foo.test.mjs'],
+    ['a .spec. sibling', '0.2.0/drop-foo.spec.mjs'],
+    ['a fixture sibling', '0.2.0/drop-foo.fixture.mjs'],
+    ['a __tests__ directory', '0.2.0/__tests__/drop-foo.mjs'],
+    ['a __fixtures__ directory', '0.2.0/__fixtures__/input.mjs'],
+  ])('ignores %s and still discovers the codemod', async (_label, testPath) => {
+    scaffold({'0.2.0/drop-foo.mjs': TRANSFORM, [testPath]: A_TEST});
+
+    const project = await Project.load(tmpDir);
+    const byVersion = await discoverIntegrationCodemods(
+      project.loadedIntegrations,
+    );
+
+    expect([...byVersion.keys()]).toEqual(['0.2.0']);
+    expect(byVersion.get('0.2.0').map(entry => entry.id)).toEqual(['drop-foo']);
+  });
+
+  it('a nested helper directory is still walked', async () => {
+    // Only test and fixture names are skipped. A package that organises its
+    // transforms into subdirectories keeps working.
+    scaffold({'0.2.0/imports/drop-foo.mjs': TRANSFORM});
+
+    const project = await Project.load(tmpDir);
+    const byVersion = await discoverIntegrationCodemods(
+      project.loadedIntegrations,
+    );
+
+    expect(byVersion.get('0.2.0').map(entry => entry.id)).toEqual([
+      'imports/drop-foo',
+    ]);
+  });
+});

package/assets/codemods/integration-runner.mjs

@@ -24,6 +24,7 @@ import {
   runCodeCodemod,
   runConfigCodemod,
 } from './run-codemod.mjs';
+import {createFileProtectionResolver} from '../../foundation/fs/file-protection.mjs';
 
 /**
  * Run file-based integration codemods, version-ordered. Config codemods run
@@ -40,18 +41,40 @@ import {
  * @param {Set<string>} [options.skipCodemods] codemod ids to exclude
  * @param {import('../../authoring/codemod/type').JscodeshiftFactory} options.jscodeshift
  * @param {boolean} [options.silent]
- * @returns {{totalFilesChanged: number, totalTransformsApplied: number, writtenFiles: string[], errors: Array<{file: string, codemod: string, error: string}>, skippedOptional: Array<import('../../authoring/codemod/type').CodemodEntry>}}
+ * @param {string} [options.root]
+ * @param {Map<string, string>} [options.contents] staged bytes from earlier dry-run phases
+ * @param {{root: string, classify: (file: string) => import('../../foundation/fs/file-protection.mjs').FileProtection[]}} [options.protection]
+ * @returns {{totalFilesChanged: number, totalTransformsApplied: number, changedFiles: string[], writtenFiles: string[], stagedContents: Map<string, string>, protectedFiles: import('../../authoring/codemod/type').CodemodRunResult['protectedFiles'], errors: Array<{file: string, codemod: string, error: string}>, skippedOptional: Array<import('../../authoring/codemod/type').CodemodEntry>}}
  */
 export function runIntegrationCodemods(
   versionGroups,
-  {apply, path: srcPath, codemod, skipCodemods, jscodeshift, silent = false},
+  {
+    apply,
+    path: srcPath,
+    codemod,
+    skipCodemods,
+    jscodeshift,
+    silent = false,
+    root = process.cwd(),
+    protection: providedProtection,
+    contents: providedContents,
+  },
 ) {
   const log = makeLog(silent);
+  let protection =
+    providedProtection ?? createFileProtectionResolver(path.resolve(root));
+  let protectionWriteCount = 0;
+  /** In-memory pipeline state keeps ordered dry-runs equivalent to apply. */
+  const virtualContents = new Map(providedContents ?? []);
 
   let totalFilesChanged = 0;
   let totalTransformsApplied = 0;
   /** @type {string[]} */
+  const changedFiles = [];
+  /** @type {string[]} */
   const writtenFiles = [];
+  /** @type {import('../../authoring/codemod/type').CodemodRunResult['protectedFiles']} */
+  const protectedFiles = [];
   /** @type {Array<{file: string, codemod: string, error: string}>} */
   const errors = [];
   /** @type {Array<import('../../authoring/codemod/type').CodemodEntry>} */
@@ -79,11 +102,24 @@ export function runIntegrationCodemods(
 
   // Config codemods first.
   for (const entry of configEntries) {
+    if (apply && writtenFiles.length !== protectionWriteCount) {
+      protection = createFileProtectionResolver(path.resolve(root));
+      protectionWriteCount = writtenFiles.length;
+    }
     log.info(`  ${entry.codemod.title} (v${entry.version}, ${entry.package})`);
-    const r = runConfigCodemod(entry, {apply, log, jscodeshift});
+    const r = runConfigCodemod(entry, {
+      apply,
+      log,
+      jscodeshift,
+      root,
+      protection,
+      contents: virtualContents,
+    });
     totalFilesChanged += r.filesChanged;
     totalTransformsApplied += r.filesChanged;
+    changedFiles.push(...r.changedFiles);
     writtenFiles.push(...r.writtenFiles);
+    protectedFiles.push(...r.protectedFiles);
     errors.push(...r.errors);
   }
 
@@ -94,13 +130,26 @@ export function runIntegrationCodemods(
       ? findSourceFiles(resolvedPath)
       : [];
     for (const entry of codeEntries) {
+      if (apply && writtenFiles.length !== protectionWriteCount) {
+        protection = createFileProtectionResolver(path.resolve(root));
+        protectionWriteCount = writtenFiles.length;
+      }
       log.info(
         `  ${entry.codemod.title} (v${entry.version}, ${entry.package})`,
       );
-      const r = runCodeCodemod(entry, files, {apply, log, jscodeshift});
+      const r = runCodeCodemod(entry, files, {
+        apply,
+        log,
+        jscodeshift,
+        root,
+        protection,
+        contents: virtualContents,
+      });
       totalFilesChanged += r.filesChanged;
       totalTransformsApplied += r.filesChanged;
+      changedFiles.push(...r.changedFiles);
       writtenFiles.push(...r.writtenFiles);
+      protectedFiles.push(...r.protectedFiles);
       errors.push(...r.errors);
     }
   }
@@ -108,7 +157,10 @@ export function runIntegrationCodemods(
   return {
     totalFilesChanged,
     totalTransformsApplied,
+    changedFiles,
     writtenFiles,
+    stagedContents: virtualContents,
+    protectedFiles,
     errors,
     skippedOptional,
   };

package/assets/codemods/integration-runner.protection.test.mjs

@@ -0,0 +1,153 @@
+// Copyright (c) Meta Platforms, Inc. and affiliates.
+
+import {afterEach, beforeEach, describe, expect, it} from 'vitest';
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+import * as path from 'node:path';
+import {runIntegrationCodemods} from './integration-runner.mjs';
+
+let root;
+let originalCwd;
+
+beforeEach(() => {
+  root = fs.mkdtempSync(
+    path.join(os.tmpdir(), 'astryx-integration-protection-'),
+  );
+  originalCwd = process.cwd();
+  process.chdir(root);
+});
+
+afterEach(() => {
+  process.chdir(originalCwd);
+  fs.rmSync(root, {recursive: true, force: true});
+});
+
+function write(relative, contents) {
+  const file = path.join(root, relative);
+  fs.mkdirSync(path.dirname(file), {recursive: true});
+  fs.writeFileSync(file, contents);
+  return file;
+}
+
+describe('integration codemod file protection', () => {
+  it('uses the same resolver for integration config and code codemods', async () => {
+    write(
+      '.gitattributes',
+      [
+        'astryx.config.mjs linguist-generated=true',
+        'src/generated.js linguist-vendored=true',
+        '',
+      ].join('\n'),
+    );
+    write('astryx.config.mjs', "export default {value: 'old'};\n");
+    write('src/generated.js', "export const value = 'old';\n");
+
+    const groups = [
+      {
+        version: '1.0.0',
+        codemods: [
+          {
+            id: 'config',
+            type: 'config',
+            package: '@acme/integration',
+            codemod: {
+              title: 'config',
+              transform: file => file.source.replace('old', 'new'),
+            },
+          },
+          {
+            id: 'code',
+            type: 'code',
+            package: '@acme/integration',
+            codemod: {
+              title: 'code',
+              fileExtensions: ['.js'],
+              transform: file => file.source.replace('old', 'new'),
+            },
+          },
+        ],
+      },
+    ];
+    const jscodeshift = (await import('jscodeshift')).default;
+    const result = runIntegrationCodemods(groups, {
+      apply: true,
+      path: path.join(root, 'src'),
+      root,
+      jscodeshift,
+      silent: true,
+    });
+
+    expect(result.totalFilesChanged).toBe(0);
+    expect(result.protectedFiles.map(item => item.file).sort()).toEqual([
+      'astryx.config.mjs',
+      'src/generated.js',
+    ]);
+    expect(
+      fs.readFileSync(path.join(root, 'astryx.config.mjs'), 'utf8'),
+    ).toContain('old');
+    expect(
+      fs.readFileSync(path.join(root, 'src/generated.js'), 'utf8'),
+    ).toContain('old');
+  });
+
+  it('lets a staged declaration protect another file in the same entry', async () => {
+    write('.gitattributes', '# authored\n');
+    write('src/x.js', "export const value = 'old';\n");
+    const groups = [
+      {
+        version: '1.0.0',
+        codemods: [
+          {
+            id: 'declare-and-change',
+            type: 'code',
+            package: '@acme/integration',
+            codemod: {
+              title: 'declare and change',
+              fileExtensions: ['', '.js'],
+              transform: file =>
+                path.basename(file.path) === '.gitattributes'
+                  ? 'src/x.js linguist-generated\n'
+                  : file.source.replace('old', 'new'),
+            },
+          },
+        ],
+      },
+    ];
+    const jscodeshift = (await import('jscodeshift')).default;
+
+    const preview = runIntegrationCodemods(groups, {
+      apply: false,
+      path: root,
+      root,
+      jscodeshift,
+      silent: true,
+    });
+    expect(preview.protectedFiles).toEqual([
+      expect.objectContaining({file: 'src/x.js', reason: 'generated'}),
+    ]);
+    expect(fs.readFileSync(path.join(root, '.gitattributes'), 'utf8')).toBe(
+      '# authored\n',
+    );
+
+    const result = runIntegrationCodemods(groups, {
+      apply: true,
+      path: root,
+      root,
+      jscodeshift,
+      silent: true,
+    });
+
+    expect(fs.readFileSync(path.join(root, '.gitattributes'), 'utf8')).toBe(
+      'src/x.js linguist-generated\n',
+    );
+    expect(fs.readFileSync(path.join(root, 'src/x.js'), 'utf8')).toContain(
+      'old',
+    );
+    expect(result.protectedFiles).toEqual([
+      expect.objectContaining({
+        file: 'src/x.js',
+        reason: 'generated',
+      }),
+    ]);
+  });
+});

package/assets/codemods/registry.mjs

@@ -29,6 +29,7 @@ const registry = new Map([
   ['0.4.0', () => import('./transforms/v0.4.0/index.mjs')],
   ['0.5.0', () => import('./transforms/v0.5.0/index.mjs')],
   ['0.6.0', () => import('./transforms/v0.6.0/index.mjs')],
+  ['0.6.4', () => import('./transforms/v0.6.4/index.mjs')],
 ]);
 
 // Re-export from the shared utility so registry callers and other consumers