feat(debarrel): sharding

codemods/debarrel/scripts/codemod.ts

@@ -1,4 +1,4 @@
-import type { Transform, Edit, SgNode } from "codemod:ast-grep";
+import type { Transform, Edit, SgNode, GetSelector } from "codemod:ast-grep";
 import type TSX from "codemod:ast-grep/langs/tsx";
 import type TypeScript from "codemod:ast-grep/langs/typescript";
 import type JavaScript from "codemod:ast-grep/langs/javascript";
@@ -432,4 +432,12 @@ const transform: Transform<Language> = async (root) => {
   return rootNode.commitEdits(edits);
 };
 
+export const getSelector: GetSelector<Language> = () => {
+  return {
+    rule: {
+      any: [{ kind: "import_statement" }, { kind: "export_statement" }],
+    },
+  };
+};
+
 export default transform;

codemods/debarrel/tsconfig.json

@@ -5,13 +5,14 @@
     "types": ["@codemod.com/jssg-types"],
     "allowImportingTsExtensions": true,
     "noEmit": true,
-    "verbatimModuleSyntax": true,
-    "erasableSyntaxOnly": true,
     "strict": true,
     "strictNullChecks": true,
     "noImplicitReturns": true,
     "noFallthroughCasesInSwitch": true,
-    "noUncheckedIndexedAccess": true
+    "noUncheckedIndexedAccess": true,
+    // Keep this until the LLRT ambient declarations in @codemod.com/jssg-types
+    // stop conflicting with TypeScript's standard library declarations.
+    "skipLibCheck": true
   },
   "exclude": ["tests"]
 }

codemods/debarrel/workflow.yaml

@@ -2,19 +2,102 @@
 
 version: "1"
 
+state:
+  schema:
+    shards:
+      type: array
+      items:
+        type: object
+        properties:
+          name:
+            type: string
+          team:
+            type: string
+          directory:
+            type: string
+          _meta_shard:
+            type: number
+          _meta_files:
+            type: array
+            items:
+              type: string
+
+params:
+  schema:
+    shardingMethod:
+      name: "Sharding Method"
+      description: "The method to use for sharding"
+      type: string
+      default: "codeowner"
+      oneOf:
+        - type: string
+          enum:
+            - "codeowner"
+            - "directory"
+    shardingDirectoryTarget:
+      name: "Sharding Directory Target"
+      description: "When sharding by directory, the directory to target for the codemod"
+      type: string
+      default: "./src"
+
 nodes:
+  - id: evaluate-shards
+    name: Evaluate Shards
+    trigger:
+      type: automatic
+    steps:
+      - name: "Evaluate shards by directory"
+        if: params.shardingMethod == "directory"
+        shard:
+          method:
+            type: directory
+            max_files_per_shard: 30
+          target: ${{ params.shardingDirectoryTarget }}
+          output_state: shards
+          js-ast-grep:
+            js_file: scripts/codemod.ts
+            language: "typescript"
+      - name: "Evaluate shards by codeowner"
+        if: params.shardingMethod == "codeowner"
+        shard:
+          method:
+            type: codeowner
+            max_files_per_shard: 30
+          target: "."
+          output_state: shards
+          js-ast-grep:
+            js_file: scripts/codemod.ts
+            language: "typescript"
+
   - id: apply-transforms
     name: Apply AST Transformations
-    type: automatic
+    trigger:
+      type: manual
+    depends_on:
+      - evaluate-shards
+    strategy:
+      type: matrix
+      from_state: shards
+    pull_request:
+      title: "[DRAFT] refactor: Run debarrel shard ${{ matrix.name }}"
+      body: "This pull request runs debarrel for shard ${{ matrix.name }}."
+      draft: true
     steps:
       - id: debarrel-js-ast-grep
         name: "Debarrel: rewrite imports and clean up barrels"
         js-ast-grep:
           js_file: scripts/codemod.ts
           language: "typescript"
           semantic_analysis: workspace
-      - id: install-package-skill
-        name: Install package skill
+
+  - id: install-skill
+    name: Install Skill
+    trigger:
+      type: manual
+    depends_on:
+      - apply-transforms
+    steps:
+      - name: "Install debarrel skill"
         install-skill:
           package: "debarrel"
           path: "./agents/skill/SKILL.md"