@mastra/codemod 0.0.0-fix-9244-clickhouse-metadata-20251105010900

package/CHANGELOG.md

@@ -0,0 +1,7 @@
+# @mastra/codemod
+
+## 0.0.0-fix-9244-clickhouse-metadata-20251105010900
+
+### Minor Changes
+
+- Initial release of `@mastra/codemod` package ([#9579](https://github.com/mastra-ai/mastra/pull/9579))

package/LICENSE.md

@@ -0,0 +1,15 @@
+# Apache License 2.0
+
+Copyright (c) 2025 Kepler Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

package/README.md

@@ -0,0 +1,99 @@
+# Mastra Codemods
+
+Mastra provides automated code transformations (codemods) to help upgrade your codebase when features are deprecated, removed, or changed between versions.
+
+Codemods are transformations that run on your codebase programmatically, allowing you to apply many changes without manually editing every file.
+
+## Quick Start
+
+### Run Individual Codemods
+
+To run a specific codemod:
+
+```sh
+npx @mastra/codemod <codemod-name> <path>
+```
+
+Examples:
+
+```sh
+# Transform a specific file
+npx @mastra/codemod v1/mastra-core-imports src/mastra.ts
+
+# Transform a directory
+npx @mastra/codemod v1/mastra-core-imports src/lib/
+
+# Transform entire project
+npx @mastra/codemod v1/mastra-core-imports .
+```
+
+## Available Codemods
+
+### v1 Codemods (v0 → v1 Migration)
+
+| Codemod                  | Description                                                                                                                                      |
+| ------------------------ | ------------------------------------------------------------------------------------------------------------------------------------------------ |
+| `v1/mastra-core-imports` | Updates all imports from `@mastra/core` to use the new subpath imports. For v1, all exports except `Mastra` and `Config` have moved to subpaths. |
+| `v1/runtime-context`     | Update `RuntimeContext` to `RequestContext` and rename all instances of parameter names.                                                         |
+
+## CLI Options
+
+### Commands
+
+```sh
+npx @mastra/codemod <command> [options]
+```
+
+**Available Commands:**
+
+- `<codemod-name> <path>` - Apply specific codemod
+
+### Global Options
+
+- `--dry` - Preview changes without applying them
+- `--print` - Print transformed code to stdout
+- `--verbose` - Show detailed transformation logs
+
+### Examples
+
+```sh
+# Show verbose output for specific codemod
+npx @mastra/codemod --verbose v1/mastra-core-imports src/
+
+# Print transformed code for specific codemod
+npx @mastra/codemod --print v1/mastra-core-imports src/mastra.ts
+```
+
+## Contributing
+
+### Adding New Codemods
+
+1. Create the codemod in `src/codemods/<version>`
+2. Add test fixtures in `src/test/__fixtures__/`
+3. Create tests in `src/test/`
+4. Use the scaffold script to generate boilerplate:
+
+   ```sh
+   pnpm scaffold
+   ```
+
+### Testing Codemods
+
+First, navigate to the codemod directory:
+
+```sh
+cd packages/codemod
+```
+
+Then run the tests:
+
+```sh
+# Run all tests
+pnpm test
+
+# Run specific codemod tests
+pnpm test mastra-core-imports
+
+# Test in development
+pnpm test:watch
+```

package/dist/codemods/lib/add-comment.js

@@ -0,0 +1,57 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/codemods/lib/add-comment.ts
+var add_comment_exports = {};
+__export(add_comment_exports, {
+  ERROR_PREFIX: () => ERROR_PREFIX,
+  insertCommentOnce: () => insertCommentOnce
+});
+module.exports = __toCommonJS(add_comment_exports);
+var ERROR_PREFIX = "FIXME(mastra): ";
+function existsComment(comments, comment) {
+  let hasComment = false;
+  if (comments) {
+    comments.forEach((commentNode) => {
+      const currentComment = commentNode.value.trim();
+      if (currentComment === comment) {
+        hasComment = true;
+      }
+    });
+    if (hasComment) {
+      return true;
+    }
+  }
+  return false;
+}
+function insertCommentOnce(node, j, comment) {
+  const hasCommentInInlineComments = existsComment(node.comments, comment);
+  const hasCommentInLeadingComments = existsComment(node.leadingComments, comment);
+  if (!hasCommentInInlineComments && !hasCommentInLeadingComments) {
+    node.comments = [...node.comments || [], j.commentBlock(` ${comment} `)];
+    return true;
+  }
+  return false;
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ERROR_PREFIX,
+  insertCommentOnce
+});
+//# sourceMappingURL=add-comment.js.map

package/dist/codemods/lib/add-comment.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/lib/add-comment.ts"],"sourcesContent":["import type { API, ASTPath } from 'jscodeshift';\n\nexport const ERROR_PREFIX = 'FIXME(mastra): ';\n\nfunction existsComment(comments: ASTPath<any>['node']['comments'], comment: string): boolean {\n  let hasComment = false;\n\n  if (comments) {\n    comments.forEach((commentNode: any) => {\n      const currentComment = commentNode.value.trim();\n      if (currentComment === comment) {\n        hasComment = true;\n      }\n    });\n\n    if (hasComment) {\n      return true;\n    }\n  }\n  return false;\n}\n\nexport function insertCommentOnce(node: ASTPath<any>['node'], j: API['j'], comment: string): boolean {\n  const hasCommentInInlineComments = existsComment(node.comments, comment);\n  const hasCommentInLeadingComments = existsComment(node.leadingComments, comment);\n\n  if (!hasCommentInInlineComments && !hasCommentInLeadingComments) {\n    node.comments = [...(node.comments || []), j.commentBlock(` ${comment} `)];\n    return true;\n  }\n\n  return false;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEO,IAAM,eAAe;AAE5B,SAAS,cAAc,UAA4C,SAA0B;AAC3F,MAAI,aAAa;AAEjB,MAAI,UAAU;AACZ,aAAS,QAAQ,CAAC,gBAAqB;AACrC,YAAM,iBAAiB,YAAY,MAAM,KAAK;AAC9C,UAAI,mBAAmB,SAAS;AAC9B,qBAAa;AAAA,MACf;AAAA,IACF,CAAC;AAED,QAAI,YAAY;AACd,aAAO;AAAA,IACT;AAAA,EACF;AACA,SAAO;AACT;AAEO,SAAS,kBAAkB,MAA4B,GAAa,SAA0B;AACnG,QAAM,6BAA6B,cAAc,KAAK,UAAU,OAAO;AACvE,QAAM,8BAA8B,cAAc,KAAK,iBAAiB,OAAO;AAE/E,MAAI,CAAC,8BAA8B,CAAC,6BAA6B;AAC/D,SAAK,WAAW,CAAC,GAAI,KAAK,YAAY,CAAC,GAAI,EAAE,aAAa,IAAI,OAAO,GAAG,CAAC;AACzE,WAAO;AAAA,EACT;AAEA,SAAO;AACT;","names":[]}

package/dist/codemods/lib/create-transformer.js

@@ -0,0 +1,45 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/codemods/lib/create-transformer.ts
+var create_transformer_exports = {};
+__export(create_transformer_exports, {
+  createTransformer: () => createTransformer
+});
+module.exports = __toCommonJS(create_transformer_exports);
+function createTransformer(transformFn) {
+  return function transformer(fileInfo, api, options) {
+    const j = api.jscodeshift;
+    const root = j(fileInfo.source);
+    const context = {
+      j,
+      root,
+      hasChanges: false,
+      messages: []
+    };
+    transformFn(fileInfo, api, options, context);
+    context.messages.forEach((message) => api.report(message));
+    return context.hasChanges ? root.toSource({ quote: "single" }) : null;
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  createTransformer
+});
+//# sourceMappingURL=create-transformer.js.map

package/dist/codemods/lib/create-transformer.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/lib/create-transformer.ts"],"sourcesContent":["// Copied from https://github.com/vercel/ai/blob/main/packages/codemod/src/codemods/lib/create-transformer.ts\n// License: Apache-2.0\n\nimport type { FileInfo, API, JSCodeshift, Collection } from 'jscodeshift';\n\ntype TransformerFunction = (fileInfo: FileInfo, api: API, options: any, context: TransformContext) => void;\n\nexport interface TransformContext {\n  /**\n   * The jscodeshift API object.\n   */\n  j: JSCodeshift;\n\n  /**\n   * The root collection of the AST.\n   */\n  root: Collection<any>;\n\n  /**\n   * Codemods should set this to true if they make any changes to the AST.\n   */\n  hasChanges: boolean;\n\n  /**\n   * Codemods can append messages to this array to report information to the user.\n   */\n  messages: string[];\n}\n\nexport function createTransformer(transformFn: TransformerFunction) {\n  // Note the return type of this function is explicitly designed to conform to\n  // the signature expected by jscodeshift. For more see\n  // https://github.com/facebook/jscodeshift\n  return function transformer(fileInfo: FileInfo, api: API, options: any) {\n    const j = api.jscodeshift;\n    const root = j(fileInfo.source);\n    const context: TransformContext = {\n      j,\n      root,\n      hasChanges: false,\n      messages: [],\n    };\n\n    // Execute the transformation\n    transformFn(fileInfo, api, options, context);\n\n    // Report any messages\n    context.messages.forEach(message => api.report(message));\n\n    // Return the transformed source code if changes were made\n    return context.hasChanges ? root.toSource({ quote: 'single' }) : null;\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AA6BO,SAAS,kBAAkB,aAAkC;AAIlE,SAAO,SAAS,YAAY,UAAoB,KAAU,SAAc;AACtE,UAAM,IAAI,IAAI;AACd,UAAM,OAAO,EAAE,SAAS,MAAM;AAC9B,UAAM,UAA4B;AAAA,MAChC;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,UAAU,CAAC;AAAA,IACb;AAGA,gBAAY,UAAU,KAAK,SAAS,OAAO;AAG3C,YAAQ,SAAS,QAAQ,aAAW,IAAI,OAAO,OAAO,CAAC;AAGvD,WAAO,QAAQ,aAAa,KAAK,SAAS,EAAE,OAAO,SAAS,CAAC,IAAI;AAAA,EACnE;AACF;","names":[]}

package/dist/codemods/v1/mastra-core-imports.js

@@ -0,0 +1,189 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/codemods/v1/mastra-core-imports.ts
+var mastra_core_imports_exports = {};
+__export(mastra_core_imports_exports, {
+  default: () => mastra_core_imports_default
+});
+module.exports = __toCommonJS(mastra_core_imports_exports);
+
+// src/codemods/lib/create-transformer.ts
+function createTransformer(transformFn) {
+  return function transformer(fileInfo, api, options) {
+    const j = api.jscodeshift;
+    const root = j(fileInfo.source);
+    const context = {
+      j,
+      root,
+      hasChanges: false,
+      messages: []
+    };
+    transformFn(fileInfo, api, options, context);
+    context.messages.forEach((message) => api.report(message));
+    return context.hasChanges ? root.toSource({ quote: "single" }) : null;
+  };
+}
+
+// src/codemods/v1/mastra-core-imports.ts
+var EXPORT_TO_SUBPATH = {
+  // Agent
+  Agent: "@mastra/core/agent",
+  // Tools
+  createTool: "@mastra/core/tools",
+  Tool: "@mastra/core/tools",
+  // Workflows
+  createWorkflow: "@mastra/core/workflows",
+  createStep: "@mastra/core/workflows",
+  Workflow: "@mastra/core/workflows",
+  Step: "@mastra/core/workflows",
+  // Request Context
+  RequestContext: "@mastra/core/request-context",
+  // Processors
+  BatchPartsProcessor: "@mastra/core/processors",
+  PIIDetector: "@mastra/core/processors",
+  ModerationProcessor: "@mastra/core/processors",
+  TokenLimiterProcessor: "@mastra/core/processors",
+  Processor: "@mastra/core/processors",
+  UnicodeNormalizer: "@mastra/core/processors",
+  SystemPromptScrubber: "@mastra/core/processors",
+  PromptInjectionDetector: "@mastra/core/processors",
+  LanguageDetector: "@mastra/core/processors",
+  // Voice
+  CompositeVoice: "@mastra/core/voice",
+  // Scorers/Evals
+  runExperiment: "@mastra/core/scores",
+  createScorer: "@mastra/core/scores",
+  // Server
+  registerApiRoute: "@mastra/core/server",
+  // AI Tracing
+  DefaultExporter: "@mastra/core/ai-tracing",
+  CloudExporter: "@mastra/core/ai-tracing",
+  // Streaming
+  ChunkType: "@mastra/core/stream",
+  MastraMessageV2: "@mastra/core/stream",
+  // LLM/Models
+  ModelRouterEmbeddingModel: "@mastra/core/llm"
+};
+var mastra_core_imports_default = createTransformer((fileInfo, api, options, context) => {
+  const { j, root } = context;
+  root.find(j.ImportDeclaration, {
+    source: { value: "@mastra/core" }
+  }).forEach((importPath) => {
+    const node = importPath.node;
+    const specifiers = node.specifiers || [];
+    const declarationImportKind = node.importKind || "value";
+    const { remainingSpecifiers, importsToMove } = categorizeImports(specifiers, declarationImportKind);
+    if (importsToMove.length === 0) return;
+    context.hasChanges = true;
+    const groupedImports = groupImportsBySubpath(importsToMove);
+    const newImports = createNewImports(j, groupedImports, context);
+    insertImports(j, importPath, newImports);
+    updateOriginalImport(j, importPath, node, remainingSpecifiers, context);
+  });
+});
+function categorizeImports(specifiers, declarationImportKind) {
+  const remainingSpecifiers = [];
+  const importsToMove = [];
+  specifiers.forEach((specifier) => {
+    if (specifier.type !== "ImportSpecifier") {
+      remainingSpecifiers.push(specifier);
+      return;
+    }
+    const imported = specifier.imported;
+    const importedName = getImportedName(imported);
+    const localName = specifier.local?.name || importedName;
+    const specifierImportKind = specifier.importKind || "value";
+    const effectiveImportKind = declarationImportKind !== "value" ? declarationImportKind : specifierImportKind;
+    const isDeclarationType = declarationImportKind !== "value";
+    const newSubpath = EXPORT_TO_SUBPATH[importedName];
+    if (newSubpath) {
+      importsToMove.push({
+        subpath: newSubpath,
+        localName,
+        importedName,
+        importKind: effectiveImportKind,
+        isDeclarationType
+      });
+    } else {
+      remainingSpecifiers.push(specifier);
+    }
+  });
+  return { remainingSpecifiers, importsToMove };
+}
+function getImportedName(imported) {
+  if (imported.type === "Identifier") {
+    return imported.name;
+  }
+  return imported.value || "";
+}
+function groupImportsBySubpath(importsToMove) {
+  const groupedImports = /* @__PURE__ */ new Map();
+  importsToMove.forEach(({ subpath, localName, importedName, importKind, isDeclarationType }) => {
+    const key = `${subpath}::${importKind}::${isDeclarationType}`;
+    if (!groupedImports.has(key)) {
+      groupedImports.set(key, []);
+    }
+    groupedImports.get(key).push({ localName, importedName, importKind, isDeclarationType });
+  });
+  return groupedImports;
+}
+function createNewImports(j, groupedImports, context) {
+  const newImports = [];
+  groupedImports.forEach((imports, key) => {
+    const [subpath, importKind] = key.split("::");
+    const newSpecifiers = imports.map(({ localName, importedName }) => {
+      if (localName === importedName) {
+        return j.importSpecifier(j.identifier(importedName));
+      } else {
+        return j.importSpecifier(j.identifier(importedName), j.identifier(localName));
+      }
+    });
+    const newImport = j.importDeclaration(newSpecifiers, j.stringLiteral(subpath));
+    if (importKind !== "value") {
+      newImport.importKind = importKind;
+    }
+    newImports.push(newImport);
+    const importList = imports.map((i) => i.importedName).join(", ");
+    const kindLabel = importKind !== "value" ? ` (${importKind})` : "";
+    context.messages.push(`Moved imports to '${subpath}'${kindLabel}: ${importList}`);
+  });
+  return newImports;
+}
+function insertImports(j, importPath, newImports) {
+  newImports.reverse().forEach((newImport) => {
+    j(importPath).insertAfter(newImport);
+  });
+}
+function updateOriginalImport(j, importPath, node, remainingSpecifiers, context) {
+  if (remainingSpecifiers.length > 0) {
+    node.specifiers = remainingSpecifiers;
+    const remainingList = extractRemainingImportNames(remainingSpecifiers);
+    if (remainingList) {
+      context.messages.push(`Kept at '@mastra/core': ${remainingList}`);
+    }
+  } else {
+    j(importPath).remove();
+    context.messages.push(`Removed original '@mastra/core' import (all imports moved to subpaths)`);
+  }
+}
+function extractRemainingImportNames(remainingSpecifiers) {
+  return remainingSpecifiers.filter((s) => s.type === "ImportSpecifier").map((s) => s.imported?.name || s.local?.name).filter(Boolean).join(", ");
+}
+//# sourceMappingURL=mastra-core-imports.js.map

package/dist/codemods/v1/mastra-core-imports.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/v1/mastra-core-imports.ts","../../../src/codemods/lib/create-transformer.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\n\n/**\n * For v1 we removed all top-level exports from \"@mastra/core\" except for `Mastra` and `type Config`.\n * All other imports should use subpath imports, e.g. `import { Agent } from \"@mastra/core/agent\"`.\n *\n * This codemod updates all imports from \"@mastra/core\" to use the new subpath imports. It leaves imports to `Mastra` and `Config` unchanged.\n */\n\n// TODO: Do not hardcode this mapping, generate it from the package's exports in the future\nconst EXPORT_TO_SUBPATH: Record<string, string> = {\n  // Agent\n  Agent: '@mastra/core/agent',\n\n  // Tools\n  createTool: '@mastra/core/tools',\n  Tool: '@mastra/core/tools',\n\n  // Workflows\n  createWorkflow: '@mastra/core/workflows',\n  createStep: '@mastra/core/workflows',\n  Workflow: '@mastra/core/workflows',\n  Step: '@mastra/core/workflows',\n\n  // Request Context\n  RequestContext: '@mastra/core/request-context',\n\n  // Processors\n  BatchPartsProcessor: '@mastra/core/processors',\n  PIIDetector: '@mastra/core/processors',\n  ModerationProcessor: '@mastra/core/processors',\n  TokenLimiterProcessor: '@mastra/core/processors',\n  Processor: '@mastra/core/processors',\n  UnicodeNormalizer: '@mastra/core/processors',\n  SystemPromptScrubber: '@mastra/core/processors',\n  PromptInjectionDetector: '@mastra/core/processors',\n  LanguageDetector: '@mastra/core/processors',\n\n  // Voice\n  CompositeVoice: '@mastra/core/voice',\n\n  // Scorers/Evals\n  runExperiment: '@mastra/core/scores',\n  createScorer: '@mastra/core/scores',\n\n  // Server\n  registerApiRoute: '@mastra/core/server',\n\n  // AI Tracing\n  DefaultExporter: '@mastra/core/ai-tracing',\n  CloudExporter: '@mastra/core/ai-tracing',\n\n  // Streaming\n  ChunkType: '@mastra/core/stream',\n  MastraMessageV2: '@mastra/core/stream',\n\n  // LLM/Models\n  ModelRouterEmbeddingModel: '@mastra/core/llm',\n};\n\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  // Find all import declarations from '@mastra/core'\n  root\n    .find(j.ImportDeclaration, {\n      source: { value: '@mastra/core' },\n    })\n    .forEach(importPath => {\n      const node = importPath.node;\n      const specifiers = node.specifiers || [];\n      const declarationImportKind = node.importKind || 'value';\n\n      // Categorize specifiers into those that stay vs those that move\n      const { remainingSpecifiers, importsToMove } = categorizeImports(specifiers, declarationImportKind);\n\n      // Early return: No imports to move\n      if (importsToMove.length === 0) return;\n\n      context.hasChanges = true;\n\n      // Group imports by their target subpath\n      const groupedImports = groupImportsBySubpath(importsToMove);\n\n      // Create new import declarations for each subpath\n      const newImports = createNewImports(j, groupedImports, context);\n\n      // Insert new imports after the current one (in reverse to maintain order)\n      insertImports(j, importPath, newImports);\n\n      // Update or remove the original import\n      updateOriginalImport(j, importPath, node, remainingSpecifiers, context);\n    });\n});\n\n/**\n * Categorize import specifiers into those that stay vs those that move\n */\nfunction categorizeImports(specifiers: any[], declarationImportKind: 'type' | 'typeof' | 'value') {\n  const remainingSpecifiers: any[] = [];\n  const importsToMove: Array<{\n    subpath: string;\n    localName: string;\n    importedName: string;\n    importKind: 'type' | 'typeof' | 'value';\n    isDeclarationType: boolean;\n  }> = [];\n\n  specifiers.forEach(specifier => {\n    // Keep default and namespace imports as-is\n    if (specifier.type !== 'ImportSpecifier') {\n      remainingSpecifiers.push(specifier);\n      return;\n    }\n\n    const imported = specifier.imported;\n    const importedName = getImportedName(imported);\n    const localName = specifier.local?.name || importedName;\n    const specifierImportKind = specifier.importKind || 'value';\n\n    // Determine effective importKind:\n    // - If declaration is \"import type {}\", use 'type' for all specifiers\n    // - Otherwise, use the specifier's own importKind\n    const effectiveImportKind = declarationImportKind !== 'value' ? declarationImportKind : specifierImportKind;\n    const isDeclarationType = declarationImportKind !== 'value';\n\n    // Check if this import should be moved to a subpath\n    const newSubpath = EXPORT_TO_SUBPATH[importedName];\n\n    if (newSubpath) {\n      importsToMove.push({\n        subpath: newSubpath,\n        localName,\n        importedName,\n        importKind: effectiveImportKind,\n        isDeclarationType,\n      });\n    } else {\n      // This import stays at '@mastra/core' (e.g., Mastra, Config)\n      remainingSpecifiers.push(specifier);\n    }\n  });\n\n  return { remainingSpecifiers, importsToMove };\n}\n\n/**\n * Extract the imported name from an import specifier\n */\nfunction getImportedName(imported: any): string {\n  if (imported.type === 'Identifier') {\n    return imported.name;\n  }\n  // Handle string literal imports (edge case)\n  return imported.value || '';\n}\n\n/**\n * Group imports by their target subpath and importKind\n */\nfunction groupImportsBySubpath(\n  importsToMove: Array<{\n    subpath: string;\n    localName: string;\n    importedName: string;\n    importKind: 'type' | 'typeof' | 'value';\n    isDeclarationType: boolean;\n  }>,\n) {\n  const groupedImports = new Map<\n    string,\n    Array<{\n      localName: string;\n      importedName: string;\n      importKind: 'type' | 'typeof' | 'value';\n      isDeclarationType: boolean;\n    }>\n  >();\n\n  importsToMove.forEach(({ subpath, localName, importedName, importKind, isDeclarationType }) => {\n    // Create a key that includes both subpath and importKind to ensure separate import declarations\n    const key = `${subpath}::${importKind}::${isDeclarationType}`;\n    if (!groupedImports.has(key)) {\n      groupedImports.set(key, []);\n    }\n    groupedImports.get(key)!.push({ localName, importedName, importKind, isDeclarationType });\n  });\n\n  return groupedImports;\n}\n\n/**\n * Create new import declarations for each subpath and importKind\n */\nfunction createNewImports(\n  j: any,\n  groupedImports: Map<\n    string,\n    Array<{\n      localName: string;\n      importedName: string;\n      importKind: 'type' | 'typeof' | 'value';\n      isDeclarationType: boolean;\n    }>\n  >,\n  context: any,\n) {\n  const newImports: any[] = [];\n\n  groupedImports.forEach((imports, key) => {\n    // Extract subpath, importKind, and isDeclarationType from the composite key\n    const [subpath, importKind] = key.split('::');\n\n    const newSpecifiers = imports.map(({ localName, importedName }) => {\n      if (localName === importedName) {\n        // import { Agent } from '@mastra/core/agent'\n        return j.importSpecifier(j.identifier(importedName));\n      } else {\n        // import { Agent as MastraAgent } from '@mastra/core/agent'\n        return j.importSpecifier(j.identifier(importedName), j.identifier(localName));\n      }\n      // Note: We don't set importKind on specifiers since we're creating\n      // separate import declarations for each importKind. All specifiers in a type\n      // import group will be in an \"import type\" declaration.\n    });\n\n    const newImport = j.importDeclaration(newSpecifiers, j.stringLiteral(subpath));\n\n    // Set importKind on declaration if this is a type import (either declaration-level or inline)\n    if (importKind !== 'value') {\n      newImport.importKind = importKind;\n    }\n\n    newImports.push(newImport);\n\n    // Log which imports were moved to which subpath\n    const importList = imports.map(i => i.importedName).join(', ');\n    const kindLabel = importKind !== 'value' ? ` (${importKind})` : '';\n    context.messages.push(`Moved imports to '${subpath}'${kindLabel}: ${importList}`);\n  });\n\n  return newImports;\n}\n\n/**\n * Insert new imports after the current import (in reverse to maintain order)\n */\nfunction insertImports(j: any, importPath: any, newImports: any[]) {\n  newImports.reverse().forEach(newImport => {\n    j(importPath).insertAfter(newImport);\n  });\n}\n\n/**\n * Update or remove the original import declaration\n */\nfunction updateOriginalImport(j: any, importPath: any, node: any, remainingSpecifiers: any[], context: any) {\n  if (remainingSpecifiers.length > 0) {\n    // Keep the original import with only the remaining specifiers\n    node.specifiers = remainingSpecifiers;\n\n    const remainingList = extractRemainingImportNames(remainingSpecifiers);\n    if (remainingList) {\n      context.messages.push(`Kept at '@mastra/core': ${remainingList}`);\n    }\n  } else {\n    // Remove the original import entirely (all imports moved)\n    j(importPath).remove();\n    context.messages.push(`Removed original '@mastra/core' import (all imports moved to subpaths)`);\n  }\n}\n\n/**\n * Extract the names of remaining imports for logging\n */\nfunction extractRemainingImportNames(remainingSpecifiers: any[]): string {\n  return remainingSpecifiers\n    .filter(s => s.type === 'ImportSpecifier')\n    .map(s => s.imported?.name || s.local?.name)\n    .filter(Boolean)\n    .join(', ');\n}\n","// Copied from https://github.com/vercel/ai/blob/main/packages/codemod/src/codemods/lib/create-transformer.ts\n// License: Apache-2.0\n\nimport type { FileInfo, API, JSCodeshift, Collection } from 'jscodeshift';\n\ntype TransformerFunction = (fileInfo: FileInfo, api: API, options: any, context: TransformContext) => void;\n\nexport interface TransformContext {\n  /**\n   * The jscodeshift API object.\n   */\n  j: JSCodeshift;\n\n  /**\n   * The root collection of the AST.\n   */\n  root: Collection<any>;\n\n  /**\n   * Codemods should set this to true if they make any changes to the AST.\n   */\n  hasChanges: boolean;\n\n  /**\n   * Codemods can append messages to this array to report information to the user.\n   */\n  messages: string[];\n}\n\nexport function createTransformer(transformFn: TransformerFunction) {\n  // Note the return type of this function is explicitly designed to conform to\n  // the signature expected by jscodeshift. For more see\n  // https://github.com/facebook/jscodeshift\n  return function transformer(fileInfo: FileInfo, api: API, options: any) {\n    const j = api.jscodeshift;\n    const root = j(fileInfo.source);\n    const context: TransformContext = {\n      j,\n      root,\n      hasChanges: false,\n      messages: [],\n    };\n\n    // Execute the transformation\n    transformFn(fileInfo, api, options, context);\n\n    // Report any messages\n    context.messages.forEach(message => api.report(message));\n\n    // Return the transformed source code if changes were made\n    return context.hasChanges ? root.toSource({ quote: 'single' }) : null;\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;AC6BO,SAAS,kBAAkB,aAAkC;AAIlE,SAAO,SAAS,YAAY,UAAoB,KAAU,SAAc;AACtE,UAAM,IAAI,IAAI;AACd,UAAM,OAAO,EAAE,SAAS,MAAM;AAC9B,UAAM,UAA4B;AAAA,MAChC;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,UAAU,CAAC;AAAA,IACb;AAGA,gBAAY,UAAU,KAAK,SAAS,OAAO;AAG3C,YAAQ,SAAS,QAAQ,aAAW,IAAI,OAAO,OAAO,CAAC;AAGvD,WAAO,QAAQ,aAAa,KAAK,SAAS,EAAE,OAAO,SAAS,CAAC,IAAI;AAAA,EACnE;AACF;;;AD1CA,IAAM,oBAA4C;AAAA;AAAA,EAEhD,OAAO;AAAA;AAAA,EAGP,YAAY;AAAA,EACZ,MAAM;AAAA;AAAA,EAGN,gBAAgB;AAAA,EAChB,YAAY;AAAA,EACZ,UAAU;AAAA,EACV,MAAM;AAAA;AAAA,EAGN,gBAAgB;AAAA;AAAA,EAGhB,qBAAqB;AAAA,EACrB,aAAa;AAAA,EACb,qBAAqB;AAAA,EACrB,uBAAuB;AAAA,EACvB,WAAW;AAAA,EACX,mBAAmB;AAAA,EACnB,sBAAsB;AAAA,EACtB,yBAAyB;AAAA,EACzB,kBAAkB;AAAA;AAAA,EAGlB,gBAAgB;AAAA;AAAA,EAGhB,eAAe;AAAA,EACf,cAAc;AAAA;AAAA,EAGd,kBAAkB;AAAA;AAAA,EAGlB,iBAAiB;AAAA,EACjB,eAAe;AAAA;AAAA,EAGf,WAAW;AAAA,EACX,iBAAiB;AAAA;AAAA,EAGjB,2BAA2B;AAC7B;AAEA,IAAO,8BAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,OACG,KAAK,EAAE,mBAAmB;AAAA,IACzB,QAAQ,EAAE,OAAO,eAAe;AAAA,EAClC,CAAC,EACA,QAAQ,gBAAc;AACrB,UAAM,OAAO,WAAW;AACxB,UAAM,aAAa,KAAK,cAAc,CAAC;AACvC,UAAM,wBAAwB,KAAK,cAAc;AAGjD,UAAM,EAAE,qBAAqB,cAAc,IAAI,kBAAkB,YAAY,qBAAqB;AAGlG,QAAI,cAAc,WAAW,EAAG;AAEhC,YAAQ,aAAa;AAGrB,UAAM,iBAAiB,sBAAsB,aAAa;AAG1D,UAAM,aAAa,iBAAiB,GAAG,gBAAgB,OAAO;AAG9D,kBAAc,GAAG,YAAY,UAAU;AAGvC,yBAAqB,GAAG,YAAY,MAAM,qBAAqB,OAAO;AAAA,EACxE,CAAC;AACL,CAAC;AAKD,SAAS,kBAAkB,YAAmB,uBAAoD;AAChG,QAAM,sBAA6B,CAAC;AACpC,QAAM,gBAMD,CAAC;AAEN,aAAW,QAAQ,eAAa;AAE9B,QAAI,UAAU,SAAS,mBAAmB;AACxC,0BAAoB,KAAK,SAAS;AAClC;AAAA,IACF;AAEA,UAAM,WAAW,UAAU;AAC3B,UAAM,eAAe,gBAAgB,QAAQ;AAC7C,UAAM,YAAY,UAAU,OAAO,QAAQ;AAC3C,UAAM,sBAAsB,UAAU,cAAc;AAKpD,UAAM,sBAAsB,0BAA0B,UAAU,wBAAwB;AACxF,UAAM,oBAAoB,0BAA0B;AAGpD,UAAM,aAAa,kBAAkB,YAAY;AAEjD,QAAI,YAAY;AACd,oBAAc,KAAK;AAAA,QACjB,SAAS;AAAA,QACT;AAAA,QACA;AAAA,QACA,YAAY;AAAA,QACZ;AAAA,MACF,CAAC;AAAA,IACH,OAAO;AAEL,0BAAoB,KAAK,SAAS;AAAA,IACpC;AAAA,EACF,CAAC;AAED,SAAO,EAAE,qBAAqB,cAAc;AAC9C;AAKA,SAAS,gBAAgB,UAAuB;AAC9C,MAAI,SAAS,SAAS,cAAc;AAClC,WAAO,SAAS;AAAA,EAClB;AAEA,SAAO,SAAS,SAAS;AAC3B;AAKA,SAAS,sBACP,eAOA;AACA,QAAM,iBAAiB,oBAAI,IAQzB;AAEF,gBAAc,QAAQ,CAAC,EAAE,SAAS,WAAW,cAAc,YAAY,kBAAkB,MAAM;AAE7F,UAAM,MAAM,GAAG,OAAO,KAAK,UAAU,KAAK,iBAAiB;AAC3D,QAAI,CAAC,eAAe,IAAI,GAAG,GAAG;AAC5B,qBAAe,IAAI,KAAK,CAAC,CAAC;AAAA,IAC5B;AACA,mBAAe,IAAI,GAAG,EAAG,KAAK,EAAE,WAAW,cAAc,YAAY,kBAAkB,CAAC;AAAA,EAC1F,CAAC;AAED,SAAO;AACT;AAKA,SAAS,iBACP,GACA,gBASA,SACA;AACA,QAAM,aAAoB,CAAC;AAE3B,iBAAe,QAAQ,CAAC,SAAS,QAAQ;AAEvC,UAAM,CAAC,SAAS,UAAU,IAAI,IAAI,MAAM,IAAI;AAE5C,UAAM,gBAAgB,QAAQ,IAAI,CAAC,EAAE,WAAW,aAAa,MAAM;AACjE,UAAI,cAAc,cAAc;AAE9B,eAAO,EAAE,gBAAgB,EAAE,WAAW,YAAY,CAAC;AAAA,MACrD,OAAO;AAEL,eAAO,EAAE,gBAAgB,EAAE,WAAW,YAAY,GAAG,EAAE,WAAW,SAAS,CAAC;AAAA,MAC9E;AAAA,IAIF,CAAC;AAED,UAAM,YAAY,EAAE,kBAAkB,eAAe,EAAE,cAAc,OAAO,CAAC;AAG7E,QAAI,eAAe,SAAS;AAC1B,gBAAU,aAAa;AAAA,IACzB;AAEA,eAAW,KAAK,SAAS;AAGzB,UAAM,aAAa,QAAQ,IAAI,OAAK,EAAE,YAAY,EAAE,KAAK,IAAI;AAC7D,UAAM,YAAY,eAAe,UAAU,KAAK,UAAU,MAAM;AAChE,YAAQ,SAAS,KAAK,qBAAqB,OAAO,IAAI,SAAS,KAAK,UAAU,EAAE;AAAA,EAClF,CAAC;AAED,SAAO;AACT;AAKA,SAAS,cAAc,GAAQ,YAAiB,YAAmB;AACjE,aAAW,QAAQ,EAAE,QAAQ,eAAa;AACxC,MAAE,UAAU,EAAE,YAAY,SAAS;AAAA,EACrC,CAAC;AACH;AAKA,SAAS,qBAAqB,GAAQ,YAAiB,MAAW,qBAA4B,SAAc;AAC1G,MAAI,oBAAoB,SAAS,GAAG;AAElC,SAAK,aAAa;AAElB,UAAM,gBAAgB,4BAA4B,mBAAmB;AACrE,QAAI,eAAe;AACjB,cAAQ,SAAS,KAAK,2BAA2B,aAAa,EAAE;AAAA,IAClE;AAAA,EACF,OAAO;AAEL,MAAE,UAAU,EAAE,OAAO;AACrB,YAAQ,SAAS,KAAK,wEAAwE;AAAA,EAChG;AACF;AAKA,SAAS,4BAA4B,qBAAoC;AACvE,SAAO,oBACJ,OAAO,OAAK,EAAE,SAAS,iBAAiB,EACxC,IAAI,OAAK,EAAE,UAAU,QAAQ,EAAE,OAAO,IAAI,EAC1C,OAAO,OAAO,EACd,KAAK,IAAI;AACd;","names":[]}

package/dist/codemods/v1/runtime-context.js

@@ -0,0 +1,166 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/codemods/v1/runtime-context.ts
+var runtime_context_exports = {};
+__export(runtime_context_exports, {
+  default: () => runtime_context_default
+});
+module.exports = __toCommonJS(runtime_context_exports);
+
+// src/codemods/lib/create-transformer.ts
+function createTransformer(transformFn) {
+  return function transformer(fileInfo, api, options) {
+    const j = api.jscodeshift;
+    const root = j(fileInfo.source);
+    const context = {
+      j,
+      root,
+      hasChanges: false,
+      messages: []
+    };
+    transformFn(fileInfo, api, options, context);
+    context.messages.forEach((message) => api.report(message));
+    return context.hasChanges ? root.toSource({ quote: "single" }) : null;
+  };
+}
+
+// src/codemods/v1/runtime-context.ts
+var runtime_context_default = createTransformer((fileInfo, api, options, context) => {
+  const { j, root } = context;
+  let hasRuntimeContextImport = false;
+  root.find(j.ImportDeclaration).forEach((importPath) => {
+    const node = importPath.node;
+    if (node.source.value !== "@mastra/core/runtime-context") return;
+    node.source.value = "@mastra/core/request-context";
+    context.hasChanges = true;
+    node.specifiers?.forEach((specifier) => {
+      if (specifier.type !== "ImportSpecifier") return;
+      const imported = specifier.imported;
+      if (imported.type === "Identifier" && imported.name === "RuntimeContext") {
+        hasRuntimeContextImport = true;
+        imported.name = "RequestContext";
+        context.messages.push(`Updated import: RuntimeContext \u2192 RequestContext from '@mastra/core/request-context'`);
+      }
+    });
+  });
+  if (!hasRuntimeContextImport) return;
+  renameIdentifiers(j, root, context, "RuntimeContext", "RequestContext", "type");
+  renameIdentifiers(j, root, context, "runtimeContext", "requestContext", "variable/parameter");
+  renameMiddlewareStringLiterals(j, root, context);
+});
+function renameIdentifiers(j, root, context, oldName, newName, description) {
+  const identifiers = root.find(j.Identifier, { name: oldName });
+  const count = identifiers.length;
+  if (count === 0) return;
+  identifiers.forEach((path) => {
+    path.node.name = newName;
+  });
+  context.hasChanges = true;
+  context.messages.push(`Renamed ${count} ${oldName} ${description} references to ${newName}`);
+}
+function renameMiddlewareStringLiterals(j, root, context) {
+  let stringLiteralCount = 0;
+  root.find(j.NewExpression, {
+    callee: { type: "Identifier", name: "Mastra" }
+  }).forEach((mastraPath) => {
+    const configArg = mastraPath.node.arguments[0];
+    if (!configArg || configArg.type !== "ObjectExpression") return;
+    const contextParamNames = /* @__PURE__ */ new Set();
+    stringLiteralCount += processNode(configArg, contextParamNames, context);
+  });
+  if (stringLiteralCount > 0) {
+    context.messages.push(
+      `Renamed ${stringLiteralCount} string literal 'runtimeContext' to 'requestContext' in Mastra server.middleware`
+    );
+  }
+}
+function processNode(node, contextParamNames, context) {
+  if (!node || typeof node !== "object") return 0;
+  let count = 0;
+  if (isHandlerProperty(node)) {
+    const paramName = extractFirstParamName(node.value);
+    if (paramName) {
+      contextParamNames.add(paramName);
+    }
+  }
+  if (isContextGetCall(node, contextParamNames)) {
+    if (renameStringLiteralArg(node, context)) {
+      count++;
+    }
+  }
+  for (const key in node) {
+    if (!shouldProcessKey(key, node)) continue;
+    const value = node[key];
+    if (Array.isArray(value)) {
+      value.forEach((item) => {
+        count += processNode(item, contextParamNames, context);
+      });
+    } else if (value && typeof value === "object") {
+      count += processNode(value, contextParamNames, context);
+    }
+  }
+  return count;
+}
+function isHandlerProperty(node) {
+  return (node.type === "Property" || node.type === "ObjectProperty") && node.key?.name === "handler";
+}
+function extractFirstParamName(handlerValue) {
+  if (!handlerValue || handlerValue.type !== "ArrowFunctionExpression" && handlerValue.type !== "FunctionExpression") {
+    return null;
+  }
+  if (!handlerValue.params || handlerValue.params.length === 0) {
+    return null;
+  }
+  const firstParam = handlerValue.params[0];
+  if (firstParam?.type === "Identifier") {
+    return firstParam.name;
+  }
+  return null;
+}
+function isContextGetCall(node, contextParamNames) {
+  if (node.type !== "CallExpression") return false;
+  const callee = node.callee;
+  if (!callee || callee.type !== "MemberExpression") return false;
+  const object = callee.object;
+  if (!object || object.type !== "Identifier") return false;
+  if (!contextParamNames.has(object.name)) return false;
+  const property = callee.property;
+  if (!property || property.type !== "Identifier" || property.name !== "get") return false;
+  return true;
+}
+function renameStringLiteralArg(node, context) {
+  const firstArg = node.arguments?.[0];
+  if (!firstArg) return false;
+  const isRuntimeContextLiteral = firstArg.type === "StringLiteral" && firstArg.value === "runtimeContext" || firstArg.type === "Literal" && firstArg.value === "runtimeContext";
+  if (!isRuntimeContextLiteral) return false;
+  firstArg.value = "requestContext";
+  if (firstArg.extra?.raw) {
+    const quote = firstArg.extra.raw.charAt(0);
+    firstArg.extra.raw = `${quote}requestContext${quote}`;
+  }
+  context.hasChanges = true;
+  return true;
+}
+function shouldProcessKey(key, node) {
+  if (!node.hasOwnProperty(key)) return false;
+  if (key === "loc" || key === "comments") return false;
+  return true;
+}
+//# sourceMappingURL=runtime-context.js.map

package/dist/codemods/v1/runtime-context.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/v1/runtime-context.ts","../../../src/codemods/lib/create-transformer.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\n\n/**\n * The `RuntimeContext` class has been renamed to `RequestContext`, and all parameter names have been updated from `runtimeContext` to `requestContext` across all APIs.\n */\n\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  // Track whether RuntimeContext was imported from @mastra/core/runtime-context\n  let hasRuntimeContextImport = false;\n\n  // 1. Update import declarations from runtime-context to request-context\n  root.find(j.ImportDeclaration).forEach(importPath => {\n    const node = importPath.node;\n\n    // Early return: Only process imports from @mastra/core/runtime-context\n    if (node.source.value !== '@mastra/core/runtime-context') return;\n\n    // Update the import path\n    node.source.value = '@mastra/core/request-context';\n    context.hasChanges = true;\n\n    // Update RuntimeContext to RequestContext in import specifiers\n    node.specifiers?.forEach(specifier => {\n      if (specifier.type !== 'ImportSpecifier') return;\n\n      const imported = specifier.imported;\n      if (imported.type === 'Identifier' && imported.name === 'RuntimeContext') {\n        hasRuntimeContextImport = true;\n        imported.name = 'RequestContext';\n        context.messages.push(`Updated import: RuntimeContext → RequestContext from '@mastra/core/request-context'`);\n      }\n    });\n  });\n\n  // Early return: Only proceed if RuntimeContext was imported from Mastra\n  if (!hasRuntimeContextImport) return;\n\n  // 2. Rename RuntimeContext type/class references\n  renameIdentifiers(j, root, context, 'RuntimeContext', 'RequestContext', 'type');\n\n  // 3. Rename runtimeContext variable/parameter identifiers\n  renameIdentifiers(j, root, context, 'runtimeContext', 'requestContext', 'variable/parameter');\n\n  // 4. Rename string literal 'runtimeContext' to 'requestContext' in Mastra middleware\n  renameMiddlewareStringLiterals(j, root, context);\n});\n\n/**\n * Helper to rename all occurrences of an identifier\n */\nfunction renameIdentifiers(j: any, root: any, context: any, oldName: string, newName: string, description: string) {\n  const identifiers = root.find(j.Identifier, { name: oldName });\n  const count = identifiers.length;\n\n  if (count === 0) return;\n\n  identifiers.forEach((path: any) => {\n    path.node.name = newName;\n  });\n\n  context.hasChanges = true;\n  context.messages.push(`Renamed ${count} ${oldName} ${description} references to ${newName}`);\n}\n\n/**\n * Helper to rename 'runtimeContext' string literals in Mastra middleware handlers\n */\nfunction renameMiddlewareStringLiterals(j: any, root: any, context: any) {\n  let stringLiteralCount = 0;\n\n  // Find all new Mastra({ ... }) expressions\n  root\n    .find(j.NewExpression, {\n      callee: { type: 'Identifier', name: 'Mastra' },\n    })\n    .forEach((mastraPath: any) => {\n      const configArg = mastraPath.node.arguments[0];\n      if (!configArg || configArg.type !== 'ObjectExpression') return;\n\n      // Process this Mastra config to find and rename context.get() calls\n      const contextParamNames = new Set<string>();\n      stringLiteralCount += processNode(configArg, contextParamNames, context);\n    });\n\n  if (stringLiteralCount > 0) {\n    context.messages.push(\n      `Renamed ${stringLiteralCount} string literal 'runtimeContext' to 'requestContext' in Mastra server.middleware`,\n    );\n  }\n}\n\n/**\n * Recursively search for handler properties and rename context.get() calls\n */\nfunction processNode(node: any, contextParamNames: Set<string>, context: any): number {\n  if (!node || typeof node !== 'object') return 0;\n\n  let count = 0;\n\n  // Check if this is a handler property\n  if (isHandlerProperty(node)) {\n    const paramName = extractFirstParamName(node.value);\n    if (paramName) {\n      contextParamNames.add(paramName);\n    }\n  }\n\n  // Check if this is a context.get('runtimeContext') call\n  if (isContextGetCall(node, contextParamNames)) {\n    if (renameStringLiteralArg(node, context)) {\n      count++;\n    }\n  }\n\n  // Recursively process all object properties\n  for (const key in node) {\n    if (!shouldProcessKey(key, node)) continue;\n\n    const value = node[key];\n    if (Array.isArray(value)) {\n      value.forEach(item => {\n        count += processNode(item, contextParamNames, context);\n      });\n    } else if (value && typeof value === 'object') {\n      count += processNode(value, contextParamNames, context);\n    }\n  }\n\n  return count;\n}\n\n/**\n * Check if a node is a handler property (Property or ObjectProperty with key 'handler')\n */\nfunction isHandlerProperty(node: any): boolean {\n  return (node.type === 'Property' || node.type === 'ObjectProperty') && node.key?.name === 'handler';\n}\n\n/**\n * Extract the first parameter name from a function expression\n */\nfunction extractFirstParamName(handlerValue: any): string | null {\n  if (\n    !handlerValue ||\n    (handlerValue.type !== 'ArrowFunctionExpression' && handlerValue.type !== 'FunctionExpression')\n  ) {\n    return null;\n  }\n\n  if (!handlerValue.params || handlerValue.params.length === 0) {\n    return null;\n  }\n\n  const firstParam = handlerValue.params[0];\n  if (firstParam?.type === 'Identifier') {\n    return firstParam.name;\n  }\n\n  return null;\n}\n\n/**\n * Check if a node is a context.get() call expression\n */\nfunction isContextGetCall(node: any, contextParamNames: Set<string>): boolean {\n  if (node.type !== 'CallExpression') return false;\n\n  const callee = node.callee;\n  if (!callee || callee.type !== 'MemberExpression') return false;\n\n  const object = callee.object;\n  if (!object || object.type !== 'Identifier') return false;\n\n  if (!contextParamNames.has(object.name)) return false;\n\n  const property = callee.property;\n  if (!property || property.type !== 'Identifier' || property.name !== 'get') return false;\n\n  return true;\n}\n\n/**\n * Rename the first string argument from 'runtimeContext' to 'requestContext'\n */\nfunction renameStringLiteralArg(node: any, context: any): boolean {\n  const firstArg = node.arguments?.[0];\n  if (!firstArg) return false;\n\n  const isRuntimeContextLiteral =\n    (firstArg.type === 'StringLiteral' && firstArg.value === 'runtimeContext') ||\n    (firstArg.type === 'Literal' && firstArg.value === 'runtimeContext');\n\n  if (!isRuntimeContextLiteral) return false;\n\n  // Rename the value\n  firstArg.value = 'requestContext';\n\n  // Update the raw value if it exists\n  if (firstArg.extra?.raw) {\n    const quote = firstArg.extra.raw.charAt(0);\n    firstArg.extra.raw = `${quote}requestContext${quote}`;\n  }\n\n  context.hasChanges = true;\n  return true;\n}\n\n/**\n * Check if we should process this object key during recursion\n */\nfunction shouldProcessKey(key: string, node: any): boolean {\n  // Skip non-own properties\n  if (!node.hasOwnProperty(key)) return false;\n\n  // Skip metadata properties that don't contain code\n  if (key === 'loc' || key === 'comments') return false;\n\n  return true;\n}\n","// Copied from https://github.com/vercel/ai/blob/main/packages/codemod/src/codemods/lib/create-transformer.ts\n// License: Apache-2.0\n\nimport type { FileInfo, API, JSCodeshift, Collection } from 'jscodeshift';\n\ntype TransformerFunction = (fileInfo: FileInfo, api: API, options: any, context: TransformContext) => void;\n\nexport interface TransformContext {\n  /**\n   * The jscodeshift API object.\n   */\n  j: JSCodeshift;\n\n  /**\n   * The root collection of the AST.\n   */\n  root: Collection<any>;\n\n  /**\n   * Codemods should set this to true if they make any changes to the AST.\n   */\n  hasChanges: boolean;\n\n  /**\n   * Codemods can append messages to this array to report information to the user.\n   */\n  messages: string[];\n}\n\nexport function createTransformer(transformFn: TransformerFunction) {\n  // Note the return type of this function is explicitly designed to conform to\n  // the signature expected by jscodeshift. For more see\n  // https://github.com/facebook/jscodeshift\n  return function transformer(fileInfo: FileInfo, api: API, options: any) {\n    const j = api.jscodeshift;\n    const root = j(fileInfo.source);\n    const context: TransformContext = {\n      j,\n      root,\n      hasChanges: false,\n      messages: [],\n    };\n\n    // Execute the transformation\n    transformFn(fileInfo, api, options, context);\n\n    // Report any messages\n    context.messages.forEach(message => api.report(message));\n\n    // Return the transformed source code if changes were made\n    return context.hasChanges ? root.toSource({ quote: 'single' }) : null;\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;AC6BO,SAAS,kBAAkB,aAAkC;AAIlE,SAAO,SAAS,YAAY,UAAoB,KAAU,SAAc;AACtE,UAAM,IAAI,IAAI;AACd,UAAM,OAAO,EAAE,SAAS,MAAM;AAC9B,UAAM,UAA4B;AAAA,MAChC;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,UAAU,CAAC;AAAA,IACb;AAGA,gBAAY,UAAU,KAAK,SAAS,OAAO;AAG3C,YAAQ,SAAS,QAAQ,aAAW,IAAI,OAAO,OAAO,CAAC;AAGvD,WAAO,QAAQ,aAAa,KAAK,SAAS,EAAE,OAAO,SAAS,CAAC,IAAI;AAAA,EACnE;AACF;;;AD9CA,IAAO,0BAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,MAAI,0BAA0B;AAG9B,OAAK,KAAK,EAAE,iBAAiB,EAAE,QAAQ,gBAAc;AACnD,UAAM,OAAO,WAAW;AAGxB,QAAI,KAAK,OAAO,UAAU,+BAAgC;AAG1D,SAAK,OAAO,QAAQ;AACpB,YAAQ,aAAa;AAGrB,SAAK,YAAY,QAAQ,eAAa;AACpC,UAAI,UAAU,SAAS,kBAAmB;AAE1C,YAAM,WAAW,UAAU;AAC3B,UAAI,SAAS,SAAS,gBAAgB,SAAS,SAAS,kBAAkB;AACxE,kCAA0B;AAC1B,iBAAS,OAAO;AAChB,gBAAQ,SAAS,KAAK,0FAAqF;AAAA,MAC7G;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAGD,MAAI,CAAC,wBAAyB;AAG9B,oBAAkB,GAAG,MAAM,SAAS,kBAAkB,kBAAkB,MAAM;AAG9E,oBAAkB,GAAG,MAAM,SAAS,kBAAkB,kBAAkB,oBAAoB;AAG5F,iCAA+B,GAAG,MAAM,OAAO;AACjD,CAAC;AAKD,SAAS,kBAAkB,GAAQ,MAAW,SAAc,SAAiB,SAAiB,aAAqB;AACjH,QAAM,cAAc,KAAK,KAAK,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAC7D,QAAM,QAAQ,YAAY;AAE1B,MAAI,UAAU,EAAG;AAEjB,cAAY,QAAQ,CAAC,SAAc;AACjC,SAAK,KAAK,OAAO;AAAA,EACnB,CAAC;AAED,UAAQ,aAAa;AACrB,UAAQ,SAAS,KAAK,WAAW,KAAK,IAAI,OAAO,IAAI,WAAW,kBAAkB,OAAO,EAAE;AAC7F;AAKA,SAAS,+BAA+B,GAAQ,MAAW,SAAc;AACvE,MAAI,qBAAqB;AAGzB,OACG,KAAK,EAAE,eAAe;AAAA,IACrB,QAAQ,EAAE,MAAM,cAAc,MAAM,SAAS;AAAA,EAC/C,CAAC,EACA,QAAQ,CAAC,eAAoB;AAC5B,UAAM,YAAY,WAAW,KAAK,UAAU,CAAC;AAC7C,QAAI,CAAC,aAAa,UAAU,SAAS,mBAAoB;AAGzD,UAAM,oBAAoB,oBAAI,IAAY;AAC1C,0BAAsB,YAAY,WAAW,mBAAmB,OAAO;AAAA,EACzE,CAAC;AAEH,MAAI,qBAAqB,GAAG;AAC1B,YAAQ,SAAS;AAAA,MACf,WAAW,kBAAkB;AAAA,IAC/B;AAAA,EACF;AACF;AAKA,SAAS,YAAY,MAAW,mBAAgC,SAAsB;AACpF,MAAI,CAAC,QAAQ,OAAO,SAAS,SAAU,QAAO;AAE9C,MAAI,QAAQ;AAGZ,MAAI,kBAAkB,IAAI,GAAG;AAC3B,UAAM,YAAY,sBAAsB,KAAK,KAAK;AAClD,QAAI,WAAW;AACb,wBAAkB,IAAI,SAAS;AAAA,IACjC;AAAA,EACF;AAGA,MAAI,iBAAiB,MAAM,iBAAiB,GAAG;AAC7C,QAAI,uBAAuB,MAAM,OAAO,GAAG;AACzC;AAAA,IACF;AAAA,EACF;AAGA,aAAW,OAAO,MAAM;AACtB,QAAI,CAAC,iBAAiB,KAAK,IAAI,EAAG;AAElC,UAAM,QAAQ,KAAK,GAAG;AACtB,QAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,YAAM,QAAQ,UAAQ;AACpB,iBAAS,YAAY,MAAM,mBAAmB,OAAO;AAAA,MACvD,CAAC;AAAA,IACH,WAAW,SAAS,OAAO,UAAU,UAAU;AAC7C,eAAS,YAAY,OAAO,mBAAmB,OAAO;AAAA,IACxD;AAAA,EACF;AAEA,SAAO;AACT;AAKA,SAAS,kBAAkB,MAAoB;AAC7C,UAAQ,KAAK,SAAS,cAAc,KAAK,SAAS,qBAAqB,KAAK,KAAK,SAAS;AAC5F;AAKA,SAAS,sBAAsB,cAAkC;AAC/D,MACE,CAAC,gBACA,aAAa,SAAS,6BAA6B,aAAa,SAAS,sBAC1E;AACA,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,aAAa,UAAU,aAAa,OAAO,WAAW,GAAG;AAC5D,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,aAAa,OAAO,CAAC;AACxC,MAAI,YAAY,SAAS,cAAc;AACrC,WAAO,WAAW;AAAA,EACpB;AAEA,SAAO;AACT;AAKA,SAAS,iBAAiB,MAAW,mBAAyC;AAC5E,MAAI,KAAK,SAAS,iBAAkB,QAAO;AAE3C,QAAM,SAAS,KAAK;AACpB,MAAI,CAAC,UAAU,OAAO,SAAS,mBAAoB,QAAO;AAE1D,QAAM,SAAS,OAAO;AACtB,MAAI,CAAC,UAAU,OAAO,SAAS,aAAc,QAAO;AAEpD,MAAI,CAAC,kBAAkB,IAAI,OAAO,IAAI,EAAG,QAAO;AAEhD,QAAM,WAAW,OAAO;AACxB,MAAI,CAAC,YAAY,SAAS,SAAS,gBAAgB,SAAS,SAAS,MAAO,QAAO;AAEnF,SAAO;AACT;AAKA,SAAS,uBAAuB,MAAW,SAAuB;AAChE,QAAM,WAAW,KAAK,YAAY,CAAC;AACnC,MAAI,CAAC,SAAU,QAAO;AAEtB,QAAM,0BACH,SAAS,SAAS,mBAAmB,SAAS,UAAU,oBACxD,SAAS,SAAS,aAAa,SAAS,UAAU;AAErD,MAAI,CAAC,wBAAyB,QAAO;AAGrC,WAAS,QAAQ;AAGjB,MAAI,SAAS,OAAO,KAAK;AACvB,UAAM,QAAQ,SAAS,MAAM,IAAI,OAAO,CAAC;AACzC,aAAS,MAAM,MAAM,GAAG,KAAK,iBAAiB,KAAK;AAAA,EACrD;AAEA,UAAQ,aAAa;AACrB,SAAO;AACT;AAKA,SAAS,iBAAiB,KAAa,MAAoB;AAEzD,MAAI,CAAC,KAAK,eAAe,GAAG,EAAG,QAAO;AAGtC,MAAI,QAAQ,SAAS,QAAQ,WAAY,QAAO;AAEhD,SAAO;AACT;","names":[]}

package/dist/index.js

@@ -0,0 +1,107 @@
+#! /usr/bin/env node
+'use strict';
+
+var commander = require('commander');
+var debug = require('debug');
+var child_process = require('child_process');
+var fs = require('fs');
+var path = require('path');
+
+function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
+
+var debug__default = /*#__PURE__*/_interopDefault(debug);
+var fs__default = /*#__PURE__*/_interopDefault(fs);
+var path__default = /*#__PURE__*/_interopDefault(path);
+
+var log = debug__default.default("codemod:transform");
+var error = debug__default.default("codemod:transform:error");
+function getJscodeshift() {
+  const localJscodeshift = path__default.default.resolve(__dirname, "../node_modules/.bin/jscodeshift");
+  return fs__default.default.existsSync(localJscodeshift) ? localJscodeshift : "jscodeshift";
+}
+function buildCommand(codemodPath, targetPath, jscodeshift, options) {
+  let command = `${jscodeshift} -t ${codemodPath} ${targetPath}     --parser tsx     --ignore-pattern="**/node_modules/**"     --ignore-pattern="**/.*/**"     --ignore-pattern="**/dist/**"     --ignore-pattern="**/build/**"     --ignore-pattern="**/*.min.js"     --ignore-pattern="**/*.bundle.js"`;
+  if (options.dry) {
+    command += " --dry";
+  }
+  if (options.print) {
+    command += " --print";
+  }
+  if (options.verbose) {
+    command += " --verbose";
+  }
+  if (options.jscodeshift) {
+    command += ` ${options.jscodeshift}`;
+  }
+  return command;
+}
+function parseErrors(transform2, output) {
+  const errors = [];
+  const errorRegex = /ERR (.+) Transformation error/g;
+  const syntaxErrorRegex = /SyntaxError: .+/g;
+  let match;
+  while ((match = errorRegex.exec(output)) !== null) {
+    const filename = match[1];
+    const syntaxErrorMatch = syntaxErrorRegex.exec(output);
+    if (syntaxErrorMatch) {
+      const summary = syntaxErrorMatch[0];
+      errors.push({ transform: transform2, filename, summary });
+    }
+  }
+  return errors;
+}
+function parseNotImplementedErrors(transform2, output) {
+  const notImplementedErrors = [];
+  const notImplementedRegex = /Not Implemented (.+): (.+)/g;
+  let match;
+  while ((match = notImplementedRegex.exec(output)) !== null) {
+    const filename = match[1];
+    const summary = match[2];
+    notImplementedErrors.push({ transform: transform2, filename, summary });
+  }
+  return notImplementedErrors;
+}
+function transform(codemod, source, transformOptions, options = { logStatus: true }) {
+  if (options.logStatus) {
+    log(`Applying codemod '${codemod}': ${source}`);
+  }
+  const codemodPath = path__default.default.resolve(__dirname, `./codemods/${codemod}.js`);
+  const targetPath = path__default.default.resolve(source);
+  const jscodeshift = getJscodeshift();
+  const command = buildCommand(codemodPath, targetPath, jscodeshift, transformOptions);
+  const stdout = child_process.execSync(command, { encoding: "utf8", stdio: "pipe" });
+  const errors = parseErrors(codemod, stdout);
+  const notImplementedErrors = parseNotImplementedErrors(codemod, stdout);
+  if (options.logStatus) {
+    if (errors.length > 0) {
+      errors.forEach(({ transform: transform2, filename, summary }) => {
+        error(`Error applying codemod [codemod=${transform2}, path=${filename}, summary=${summary}]`);
+      });
+    }
+    if (notImplementedErrors.length > 0) {
+      log(
+        `Some files require manual changes. Please search your codebase for \`FIXME(mastra): \` comments and follow the instructions to complete the upgrade.`
+      );
+      notImplementedErrors.forEach(({ transform: transform2, filename, summary }) => {
+        log(`Not Implemented [codemod=${transform2}, path=${filename}, summary=${summary}]`);
+      });
+    }
+  }
+  return { errors, notImplementedErrors };
+}
+
+// src/index.ts
+var error2 = debug__default.default("codemod:error");
+debug__default.default.enable("codemod:*");
+var program = new commander.Command();
+program.name("codemod").description("CLI for running Mastra codemods").argument("<codemod>", "Codemod to run").argument("<source>", "Path to source files or directory").option("-d, --dry", "Dry run (no changes are made to files)").option("-p, --print", "Print transformed files to stdout").option("--verbose", "Show more information about the transform process").option("-j, --jscodeshift <options>", "Pass options directly to jscodeshift").action((codemod, source, options) => {
+  try {
+    transform(codemod, source, options);
+  } catch (err) {
+    error2(`Error transforming: ${err}`);
+    process.exit(1);
+  }
+});
+program.parse(process.argv);
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/lib/transform.ts","../src/index.ts"],"names":["debug","path","fs","transform","execSync","error","Command"],"mappings":";;;;;;;;;;;;;;;AAeA,IAAM,GAAA,GAAMA,uBAAM,mBAAmB,CAAA;AACrC,IAAM,KAAA,GAAQA,uBAAM,yBAAyB,CAAA;AAE7C,SAAS,cAAA,GAAyB;AAChC,EAAA,MAAM,gBAAA,GAAmBC,qBAAA,CAAK,OAAA,CAAQ,SAAA,EAAW,kCAAkC,CAAA;AACnF,EAAA,OAAOC,mBAAA,CAAG,UAAA,CAAW,gBAAgB,CAAA,GAAI,gBAAA,GAAmB,aAAA;AAC9D;AAEA,SAAS,YAAA,CAAa,WAAA,EAAqB,UAAA,EAAoB,WAAA,EAAqB,OAAA,EAAmC;AAGrH,EAAA,IAAI,UAAU,CAAA,EAAG,WAAW,CAAA,IAAA,EAAO,WAAW,IAAI,UAAU,CAAA,yOAAA,CAAA;AAS5D,EAAA,IAAI,QAAQ,GAAA,EAAK;AACf,IAAA,OAAA,IAAW,QAAA;AAAA,EACb;AAEA,EAAA,IAAI,QAAQ,KAAA,EAAO;AACjB,IAAA,OAAA,IAAW,UAAA;AAAA,EACb;AAEA,EAAA,IAAI,QAAQ,OAAA,EAAS;AACnB,IAAA,OAAA,IAAW,YAAA;AAAA,EACb;AAEA,EAAA,IAAI,QAAQ,WAAA,EAAa;AACvB,IAAA,OAAA,IAAW,CAAA,CAAA,EAAI,QAAQ,WAAW,CAAA,CAAA;AAAA,EACpC;AAEA,EAAA,OAAO,OAAA;AACT;AAQA,SAAS,WAAA,CAAYC,YAAmB,MAAA,EAAiC;AACvE,EAAA,MAAM,SAA0B,EAAC;AACjC,EAAA,MAAM,UAAA,GAAa,gCAAA;AACnB,EAAA,MAAM,gBAAA,GAAmB,kBAAA;AAEzB,EAAA,IAAI,KAAA;AACJ,EAAA,OAAA,CAAQ,KAAA,GAAQ,UAAA,CAAW,IAAA,CAAK,MAAM,OAAO,IAAA,EAAM;AACjD,IAAA,MAAM,QAAA,GAAW,MAAM,CAAC,CAAA;AACxB,IAAA,MAAM,gBAAA,GAAmB,gBAAA,CAAiB,IAAA,CAAK,MAAM,CAAA;AACrD,IAAA,IAAI,gBAAA,EAAkB;AACpB,MAAA,MAAM,OAAA,GAAU,iBAAiB,CAAC,CAAA;AAClC,MAAA,MAAA,CAAO,KAAK,EAAE,SAAA,EAAAA,UAAAA,EAAW,QAAA,EAAU,SAAS,CAAA;AAAA,IAC9C;AAAA,EACF;AAEA,EAAA,OAAO,MAAA;AACT;AAEA,SAAS,yBAAA,CAA0BA,YAAmB,MAAA,EAAiC;AACrF,EAAA,MAAM,uBAAwC,EAAC;AAC/C,EAAA,MAAM,mBAAA,GAAsB,6BAAA;AAE5B,EAAA,IAAI,KAAA;AACJ,EAAA,OAAA,CAAQ,KAAA,GAAQ,mBAAA,CAAoB,IAAA,CAAK,MAAM,OAAO,IAAA,EAAM;AAC1D,IAAA,MAAM,QAAA,GAAW,MAAM,CAAC,CAAA;AACxB,IAAA,MAAM,OAAA,GAAU,MAAM,CAAC,CAAA;AACvB,IAAA,oBAAA,CAAqB,KAAK,EAAE,SAAA,EAAAA,UAAAA,EAAW,QAAA,EAAU,SAAS,CAAA;AAAA,EAC5D;AAEA,EAAA,OAAO,oBAAA;AACT;AAEO,SAAS,SAAA,CACd,SACA,MAAA,EACA,gBAAA,EACA,UAAkC,EAAE,SAAA,EAAW,MAAK,EACgB;AACpE,EAAA,IAAI,QAAQ,SAAA,EAAW;AACrB,IAAA,GAAA,CAAI,CAAA,kBAAA,EAAqB,OAAO,CAAA,GAAA,EAAM,MAAM,CAAA,CAAE,CAAA;AAAA,EAChD;AACA,EAAA,MAAM,cAAcF,qBAAA,CAAK,OAAA,CAAQ,SAAA,EAAW,CAAA,WAAA,EAAc,OAAO,CAAA,GAAA,CAAK,CAAA;AACtE,EAAA,MAAM,UAAA,GAAaA,qBAAA,CAAK,OAAA,CAAQ,MAAM,CAAA;AACtC,EAAA,MAAM,cAAc,cAAA,EAAe;AACnC,EAAA,MAAM,OAAA,GAAU,YAAA,CAAa,WAAA,EAAa,UAAA,EAAY,aAAa,gBAAgB,CAAA;AACnF,EAAA,MAAM,MAAA,GAASG,uBAAS,OAAA,EAAS,EAAE,UAAU,MAAA,EAAQ,KAAA,EAAO,QAAQ,CAAA;AACpE,EAAA,MAAM,MAAA,GAAS,WAAA,CAAY,OAAA,EAAS,MAAM,CAAA;AAC1C,EAAA,MAAM,oBAAA,GAAuB,yBAAA,CAA0B,OAAA,EAAS,MAAM,CAAA;AACtE,EAAA,IAAI,QAAQ,SAAA,EAAW;AACrB,IAAA,IAAI,MAAA,CAAO,SAAS,CAAA,EAAG;AACrB,MAAA,MAAA,CAAO,QAAQ,CAAC,EAAE,WAAAD,UAAAA,EAAW,QAAA,EAAU,SAAQ,KAAM;AACnD,QAAA,KAAA,CAAM,mCAAmCA,UAAS,CAAA,OAAA,EAAU,QAAQ,CAAA,UAAA,EAAa,OAAO,CAAA,CAAA,CAAG,CAAA;AAAA,MAC7F,CAAC,CAAA;AAAA,IACH;AAEA,IAAA,IAAI,oBAAA,CAAqB,SAAS,CAAA,EAAG;AACnC,MAAA,GAAA;AAAA,QACE,CAAA,oJAAA;AAAA,OACF;AACA,MAAA,oBAAA,CAAqB,QAAQ,CAAC,EAAE,WAAAA,UAAAA,EAAW,QAAA,EAAU,SAAQ,KAAM;AACjE,QAAA,GAAA,CAAI,4BAA4BA,UAAS,CAAA,OAAA,EAAU,QAAQ,CAAA,UAAA,EAAa,OAAO,CAAA,CAAA,CAAG,CAAA;AAAA,MACpF,CAAC,CAAA;AAAA,IACH;AAAA,EACF;AAEA,EAAA,OAAO,EAAE,QAAQ,oBAAA,EAAqB;AACxC;;;ACxHA,IAAME,MAAAA,GAAQL,uBAAM,eAAe,CAAA;AACnCA,sBAAAA,CAAM,OAAO,WAAW,CAAA;AAExB,IAAM,OAAA,GAAU,IAAIM,iBAAA,EAAQ;AAE5B,OAAA,CACG,IAAA,CAAK,SAAS,CAAA,CACd,WAAA,CAAY,iCAAiC,CAAA,CAC7C,QAAA,CAAS,WAAA,EAAa,gBAAgB,EACtC,QAAA,CAAS,UAAA,EAAY,mCAAmC,CAAA,CACxD,OAAO,WAAA,EAAa,wCAAwC,CAAA,CAC5D,MAAA,CAAO,aAAA,EAAe,mCAAmC,CAAA,CACzD,MAAA,CAAO,aAAa,mDAAmD,CAAA,CACvE,MAAA,CAAO,6BAAA,EAA+B,sCAAsC,CAAA,CAC5E,MAAA,CAAO,CAAC,OAAA,EAAS,QAAQ,OAAA,KAAY;AACpC,EAAA,IAAI;AACF,IAAA,SAAA,CAAU,OAAA,EAAS,QAAQ,OAAO,CAAA;AAAA,EACpC,SAAS,GAAA,EAAU;AACjB,IAAAD,MAAAA,CAAM,CAAA,oBAAA,EAAuB,GAAG,CAAA,CAAE,CAAA;AAClC,IAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAAA,EAChB;AACF,CAAC,CAAA;AAEH,OAAA,CAAQ,KAAA,CAAM,QAAQ,IAAI,CAAA","file":"index.js","sourcesContent":["// Adjusted from https://github.com/vercel/ai/blob/main/packages/codemod/src/lib/transform.ts\n// License: Apache-2.0\n\nimport { execSync } from 'child_process';\nimport fs from 'fs';\nimport path from 'path';\nimport debug from 'debug';\n\ninterface TransformOptions {\n  dry?: boolean;\n  print?: boolean;\n  verbose?: boolean;\n  jscodeshift?: string;\n}\n\nconst log = debug('codemod:transform');\nconst error = debug('codemod:transform:error');\n\nfunction getJscodeshift(): string {\n  const localJscodeshift = path.resolve(__dirname, '../node_modules/.bin/jscodeshift');\n  return fs.existsSync(localJscodeshift) ? localJscodeshift : 'jscodeshift';\n}\n\nfunction buildCommand(codemodPath: string, targetPath: string, jscodeshift: string, options: TransformOptions): string {\n  // Ignoring everything under `.*/` covers `.mastra/` along with any other\n  // framework build related or otherwise intended-to-be-hidden directories.\n  let command = `${jscodeshift} -t ${codemodPath} ${targetPath} \\\n    --parser tsx \\\n    --ignore-pattern=\"**/node_modules/**\" \\\n    --ignore-pattern=\"**/.*/**\" \\\n    --ignore-pattern=\"**/dist/**\" \\\n    --ignore-pattern=\"**/build/**\" \\\n    --ignore-pattern=\"**/*.min.js\" \\\n    --ignore-pattern=\"**/*.bundle.js\"`;\n\n  if (options.dry) {\n    command += ' --dry';\n  }\n\n  if (options.print) {\n    command += ' --print';\n  }\n\n  if (options.verbose) {\n    command += ' --verbose';\n  }\n\n  if (options.jscodeshift) {\n    command += ` ${options.jscodeshift}`;\n  }\n\n  return command;\n}\n\nexport type TransformErrors = {\n  transform: string;\n  filename: string;\n  summary: string;\n}[];\n\nfunction parseErrors(transform: string, output: string): TransformErrors {\n  const errors: TransformErrors = [];\n  const errorRegex = /ERR (.+) Transformation error/g;\n  const syntaxErrorRegex = /SyntaxError: .+/g;\n\n  let match;\n  while ((match = errorRegex.exec(output)) !== null) {\n    const filename = match[1]!;\n    const syntaxErrorMatch = syntaxErrorRegex.exec(output);\n    if (syntaxErrorMatch) {\n      const summary = syntaxErrorMatch[0];\n      errors.push({ transform, filename, summary });\n    }\n  }\n\n  return errors;\n}\n\nfunction parseNotImplementedErrors(transform: string, output: string): TransformErrors {\n  const notImplementedErrors: TransformErrors = [];\n  const notImplementedRegex = /Not Implemented (.+): (.+)/g;\n\n  let match;\n  while ((match = notImplementedRegex.exec(output)) !== null) {\n    const filename = match[1]!;\n    const summary = match[2]!;\n    notImplementedErrors.push({ transform, filename, summary });\n  }\n\n  return notImplementedErrors;\n}\n\nexport function transform(\n  codemod: string,\n  source: string,\n  transformOptions: TransformOptions,\n  options: { logStatus: boolean } = { logStatus: true },\n): { errors: TransformErrors; notImplementedErrors: TransformErrors } {\n  if (options.logStatus) {\n    log(`Applying codemod '${codemod}': ${source}`);\n  }\n  const codemodPath = path.resolve(__dirname, `./codemods/${codemod}.js`);\n  const targetPath = path.resolve(source);\n  const jscodeshift = getJscodeshift();\n  const command = buildCommand(codemodPath, targetPath, jscodeshift, transformOptions);\n  const stdout = execSync(command, { encoding: 'utf8', stdio: 'pipe' });\n  const errors = parseErrors(codemod, stdout);\n  const notImplementedErrors = parseNotImplementedErrors(codemod, stdout);\n  if (options.logStatus) {\n    if (errors.length > 0) {\n      errors.forEach(({ transform, filename, summary }) => {\n        error(`Error applying codemod [codemod=${transform}, path=${filename}, summary=${summary}]`);\n      });\n    }\n\n    if (notImplementedErrors.length > 0) {\n      log(\n        `Some files require manual changes. Please search your codebase for \\`FIXME(mastra): \\` comments and follow the instructions to complete the upgrade.`,\n      );\n      notImplementedErrors.forEach(({ transform, filename, summary }) => {\n        log(`Not Implemented [codemod=${transform}, path=${filename}, summary=${summary}]`);\n      });\n    }\n  }\n\n  return { errors, notImplementedErrors };\n}\n","#! /usr/bin/env node\n\nimport { Command } from 'commander';\nimport debug from 'debug';\nimport { transform } from './lib/transform';\n\nconst error = debug('codemod:error');\ndebug.enable('codemod:*');\n\nconst program = new Command();\n\nprogram\n  .name('codemod')\n  .description('CLI for running Mastra codemods')\n  .argument('<codemod>', 'Codemod to run')\n  .argument('<source>', 'Path to source files or directory')\n  .option('-d, --dry', 'Dry run (no changes are made to files)')\n  .option('-p, --print', 'Print transformed files to stdout')\n  .option('--verbose', 'Show more information about the transform process')\n  .option('-j, --jscodeshift <options>', 'Pass options directly to jscodeshift')\n  .action((codemod, source, options) => {\n    try {\n      transform(codemod, source, options);\n    } catch (err: any) {\n      error(`Error transforming: ${err}`);\n      process.exit(1);\n    }\n  });\n\nprogram.parse(process.argv);\n"]}

package/package.json

@@ -0,0 +1,56 @@
+{
+  "name": "@mastra/codemod",
+  "version": "0.0.0-fix-9244-clickhouse-metadata-20251105010900",
+  "license": "Apache-2.0",
+  "description": "Codemod CLI for Mastra",
+  "bin": {
+    "codemod": "./dist/index.js"
+  },
+  "files": [
+    "dist",
+    "CHANGELOG.md"
+  ],
+  "keywords": [
+    "mastra",
+    "cli",
+    "ai",
+    "codemod"
+  ],
+  "dependencies": {
+    "commander": "^14.0.2",
+    "debug": "^4.4.3",
+    "jscodeshift": "^17.3.0"
+  },
+  "devDependencies": {
+    "@commander-js/extra-typings": "^14.0.0",
+    "@types/debug": "^4.1.12",
+    "@types/jscodeshift": "^17.3.0",
+    "@types/node": "^24.9.2",
+    "eslint": "^9.37.0",
+    "tsup": "^8.5.0",
+    "typescript": "^5.8.3",
+    "vitest": "^3.2.4",
+    "@internal/lint": "0.0.0-fix-9244-clickhouse-metadata-20251105010900"
+  },
+  "homepage": "https://mastra.ai",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/mastra-ai/mastra.git",
+    "directory": "packages/codemod"
+  },
+  "bugs": {
+    "url": "https://github.com/mastra-ai/mastra/issues"
+  },
+  "engines": {
+    "node": ">=20"
+  },
+  "scripts": {
+    "dev": "tsup --watch",
+    "build": "tsup --silent --config tsup.config.ts",
+    "test": "vitest run",
+    "test:watch": "vitest watch",
+    "typecheck": "tsc --noEmit --incremental",
+    "lint": "eslint .",
+    "scaffold": "npx tsx scripts/scaffold-codemod.ts"
+  }
+}