@mastra/codemod 0.1.0-beta.1 → 0.1.0-beta.3

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @mastra/codemod
 
+## 0.1.0-beta.3
+
+### Patch Changes
+
+- Add `v1/workflow-stream-vnext` codemod. This codemod renames `streamVNext()`, `resumeStreamVNext()`, and `observeStreamVNext()` to their "non-VNext" counterparts. ([#10802](https://github.com/mastra-ai/mastra/pull/10802))
+
+## 0.1.0-beta.2
+
+### Patch Changes
+
+- Fix `mastra-required-id`, `mcp-get-toolsets`, and `mcp-get-tools` codemods to add missing imports and instances. ([#10221](https://github.com/mastra-ai/mastra/pull/10221))
+
 ## 0.1.0-beta.1
 
 ### Patch Changes

package/README.md

@@ -70,6 +70,7 @@ npx @mastra/codemod@beta v1/mastra-core-imports .
 | `v1/workflow-create-run-async`       | Renames `workflow.createRunAsync()` → `workflow.createRun()`                                                       |
 | `v1/workflow-list-runs`              | Renames `workflow.getWorkflowRuns()` → `workflow.listWorkflowRuns()`                                               |
 | `v1/workflow-run-count`              | Renames `context.runCount` → `context.retryCount` in step execution functions                                      |
+| `v1/workflow-stream-vnext`           | Renames `streamVNext()`, `resumeStreamVNext()`, and `observeStreamVNext()`                                         |
 
 ## CLI Options
 

package/dist/codemods/v1/mcp-get-tools.js

@@ -9,7 +9,9 @@ import {
 // src/codemods/v1/mcp-get-tools.ts
 var mcp_get_tools_default = createTransformer((_fileInfo, _api, _options, context) => {
   const { j, root } = context;
-  const mcpInstances = trackClassInstances(j, root, "MCPServer");
+  const mcpServerInstances = trackClassInstances(j, root, "MCPServer");
+  const mcpClientInstances = trackClassInstances(j, root, "MCPClient");
+  const mcpInstances = /* @__PURE__ */ new Set([...mcpServerInstances, ...mcpClientInstances]);
   const count = renameMethod(j, root, mcpInstances, "getTools", "listTools");
   if (count > 0) {
     context.hasChanges = true;

package/dist/codemods/v1/mcp-get-tools.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/codemods/v1/mcp-get-tools.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\nimport { trackClassInstances, renameMethod } from '../lib/utils';\n\n/**\n * Transforms MCPServer getTools method to listTools:\n * - mcp.getTools() → mcp.listTools()\n *\n * Only transforms methods on variables that were instantiated with `new MCPServer(...)`\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  // Track MCPServer instances and rename method in a single optimized pass\n  const mcpInstances = trackClassInstances(j, root, 'MCPServer');\n  const count = renameMethod(j, root, mcpInstances, 'getTools', 'listTools');\n\n  if (count > 0) {\n    context.hasChanges = true;\n    context.messages.push(`Transformed MCPServer method: getTools → listTools`);\n  }\n});\n"],"mappings":";;;;;;;;;AASA,IAAO,wBAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,eAAe,oBAAoB,GAAG,MAAM,WAAW;AAC7D,QAAM,QAAQ,aAAa,GAAG,MAAM,cAAc,YAAY,WAAW;AAEzE,MAAI,QAAQ,GAAG;AACb,YAAQ,aAAa;AACrB,YAAQ,SAAS,KAAK,yDAAoD;AAAA,EAC5E;AACF,CAAC;","names":[]}
+{"version":3,"sources":["../../../src/codemods/v1/mcp-get-tools.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\nimport { trackClassInstances, renameMethod } from '../lib/utils';\n\n/**\n * Transforms MCPServer and MCPClient getTools method to listTools:\n * - mcp.getTools() → mcp.listTools()\n *\n * Only transforms methods on variables that were instantiated with `new MCPServer(...)` or `new MCPClient(...)`\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  // Track MCPServer and MCPClient instances and rename method in a single optimized pass\n  const mcpServerInstances = trackClassInstances(j, root, 'MCPServer');\n  const mcpClientInstances = trackClassInstances(j, root, 'MCPClient');\n  const mcpInstances = new Set([...mcpServerInstances, ...mcpClientInstances]);\n  const count = renameMethod(j, root, mcpInstances, 'getTools', 'listTools');\n\n  if (count > 0) {\n    context.hasChanges = true;\n    context.messages.push(`Transformed MCPServer method: getTools → listTools`);\n  }\n});\n"],"mappings":";;;;;;;;;AASA,IAAO,wBAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,qBAAqB,oBAAoB,GAAG,MAAM,WAAW;AACnE,QAAM,qBAAqB,oBAAoB,GAAG,MAAM,WAAW;AACnE,QAAM,eAAe,oBAAI,IAAI,CAAC,GAAG,oBAAoB,GAAG,kBAAkB,CAAC;AAC3E,QAAM,QAAQ,aAAa,GAAG,MAAM,cAAc,YAAY,WAAW;AAEzE,MAAI,QAAQ,GAAG;AACb,YAAQ,aAAa;AACrB,YAAQ,SAAS,KAAK,yDAAoD;AAAA,EAC5E;AACF,CAAC;","names":[]}

package/dist/codemods/v1/mcp-get-toolsets.js

@@ -9,7 +9,9 @@ import {
 // src/codemods/v1/mcp-get-toolsets.ts
 var mcp_get_toolsets_default = createTransformer((_fileInfo, _api, _options, context) => {
   const { j, root } = context;
-  const mcpInstances = trackClassInstances(j, root, "MCPServer");
+  const mcpServerInstances = trackClassInstances(j, root, "MCPServer");
+  const mcpClientInstances = trackClassInstances(j, root, "MCPClient");
+  const mcpInstances = /* @__PURE__ */ new Set([...mcpServerInstances, ...mcpClientInstances]);
   const count = renameMethod(j, root, mcpInstances, "getToolsets", "listToolsets");
   if (count > 0) {
     context.hasChanges = true;

package/dist/codemods/v1/mcp-get-toolsets.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/codemods/v1/mcp-get-toolsets.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\nimport { trackClassInstances, renameMethod } from '../lib/utils';\n\n/**\n * Transforms MCPServer getToolsets method to listToolsets:\n * - mcp.getToolsets() → mcp.listToolsets()\n *\n * Only transforms methods on variables that were instantiated with `new MCPServer(...)`\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  // Track MCPServer instances and rename method in a single optimized pass\n  const mcpInstances = trackClassInstances(j, root, 'MCPServer');\n  const count = renameMethod(j, root, mcpInstances, 'getToolsets', 'listToolsets');\n\n  if (count > 0) {\n    context.hasChanges = true;\n    context.messages.push(`Transformed MCPServer method: getToolsets → listToolsets`);\n  }\n});\n"],"mappings":";;;;;;;;;AASA,IAAO,2BAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,eAAe,oBAAoB,GAAG,MAAM,WAAW;AAC7D,QAAM,QAAQ,aAAa,GAAG,MAAM,cAAc,eAAe,cAAc;AAE/E,MAAI,QAAQ,GAAG;AACb,YAAQ,aAAa;AACrB,YAAQ,SAAS,KAAK,+DAA0D;AAAA,EAClF;AACF,CAAC;","names":[]}
+{"version":3,"sources":["../../../src/codemods/v1/mcp-get-toolsets.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\nimport { trackClassInstances, renameMethod } from '../lib/utils';\n\n/**\n * Transforms MCPServer and MCPClient getToolsets method to listToolsets:\n * - mcp.getToolsets() → mcp.listToolsets()\n *\n * Only transforms methods on variables that were instantiated with `new MCPServer(...)` or `new MCPClient(...)`\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  // Track MCPServer and MCPClient instances and rename method in a single optimized pass\n  const mcpServerInstances = trackClassInstances(j, root, 'MCPServer');\n  const mcpClientInstances = trackClassInstances(j, root, 'MCPClient');\n  const mcpInstances = new Set([...mcpServerInstances, ...mcpClientInstances]);\n  const count = renameMethod(j, root, mcpInstances, 'getToolsets', 'listToolsets');\n\n  if (count > 0) {\n    context.hasChanges = true;\n    context.messages.push(`Transformed MCPServer method: getToolsets → listToolsets`);\n  }\n});\n"],"mappings":";;;;;;;;;AASA,IAAO,2BAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,qBAAqB,oBAAoB,GAAG,MAAM,WAAW;AACnE,QAAM,qBAAqB,oBAAoB,GAAG,MAAM,WAAW;AACnE,QAAM,eAAe,oBAAI,IAAI,CAAC,GAAG,oBAAoB,GAAG,kBAAkB,CAAC;AAC3E,QAAM,QAAQ,aAAa,GAAG,MAAM,cAAc,eAAe,cAAc;AAE/E,MAAI,QAAQ,GAAG;AACb,YAAQ,aAAa;AACrB,YAAQ,SAAS,KAAK,+DAA0D;AAAA,EAClF;AACF,CAAC;","names":[]}

package/dist/codemods/v1/not-implemented/mastra-required-id.js

@@ -15,6 +15,7 @@ var mastra_required_id_default = createTransformer((fileInfo, api, options, cont
     "D1Store",
     "MongoDBStore",
     "DynamoDBStore",
+    "LibSQLVector",
     "PgVector",
     "ChromaVector",
     "PineconeVector",

package/dist/codemods/v1/not-implemented/mastra-required-id.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/codemods/v1/not-implemented/mastra-required-id.ts"],"sourcesContent":["/* eslint-disable no-warning-comments */\nimport { insertCommentOnce } from '../../lib/add-comment';\nimport { createTransformer } from '../../lib/create-transformer';\n\n/**\n * Adds FIXME comments to Mastra primitives that now require an `id` parameter.\n * This includes storages, vector stores, agents, workflows, tools, scorers, and MCP servers.\n *\n * Before:\n * const agent = new Agent({ name: 'Support Agent' });\n * const tool = createTool({ description: 'Get weather' });\n *\n * After:\n * /* FIXME(mastra): Add a unique `id` parameter. See: ... *\\/\n * const agent = new Agent({ name: 'Support Agent' });\n * /* FIXME(mastra): Add a unique `id` parameter. See: ... *\\/\n * const tool = createTool({ description: 'Get weather' });\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  const COMMENT_MESSAGE =\n    'FIXME(mastra): Add a unique `id` parameter. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/mastra#required-id-parameter-for-all-mastra-primitives';\n\n  // List of class names that require id\n  const storageClasses = [\n    'LibSQLStore',\n    'PostgresStore',\n    'D1Store',\n    'MongoDBStore',\n    'DynamoDBStore',\n    'PgVector',\n    'ChromaVector',\n    'PineconeVector',\n    'QdrantVector',\n    'LanceVector',\n    'Agent',\n    'MCPServer',\n  ];\n\n  // List of function names that require id\n  const createFunctions = ['createWorkflow', 'createTool', 'createScorer'];\n\n  // Find NewExpression for classes\n  root.find(j.NewExpression).forEach(path => {\n    if (path.value.callee.type === 'Identifier') {\n      const className = path.value.callee.name;\n\n      if (storageClasses.includes(className)) {\n        // Check if the config already has an id property\n        const hasId = path.value.arguments.some(arg => {\n          if (arg.type === 'ObjectExpression') {\n            return arg.properties?.some(\n              prop =>\n                (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n                prop.key?.type === 'Identifier' &&\n                prop.key.name === 'id',\n            );\n          }\n          return false;\n        });\n\n        if (!hasId) {\n          // Find the parent statement to add comment\n          let parent = path.parent;\n          const statementTypes = new Set([\n            'VariableDeclaration',\n            'ExpressionStatement',\n            'ReturnStatement',\n            'ExportDefaultDeclaration',\n            'ExportNamedDeclaration',\n            'Program',\n          ]);\n\n          while (parent && !statementTypes.has(parent.value.type)) {\n            parent = parent.parent;\n          }\n\n          if (parent && parent.value) {\n            // For export declarations, add comment to the export itself\n            let targetNode = parent.value;\n            if (\n              targetNode.type !== 'ExportDefaultDeclaration' &&\n              targetNode.type !== 'ExportNamedDeclaration' &&\n              parent.parent &&\n              (parent.parent.value.type === 'ExportDefaultDeclaration' ||\n                parent.parent.value.type === 'ExportNamedDeclaration')\n            ) {\n              targetNode = parent.parent.value;\n            }\n\n            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n            if (added) {\n              context.hasChanges = true;\n            }\n          }\n        }\n      }\n    }\n  });\n\n  // Find CallExpression for create functions\n  root.find(j.CallExpression).forEach(path => {\n    if (path.value.callee.type === 'Identifier') {\n      const functionName = path.value.callee.name;\n\n      if (createFunctions.includes(functionName)) {\n        // Check if the config already has an id property\n        const hasId = path.value.arguments.some(arg => {\n          if (arg.type === 'ObjectExpression') {\n            return arg.properties?.some(\n              prop =>\n                (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n                prop.key?.type === 'Identifier' &&\n                prop.key.name === 'id',\n            );\n          }\n          return false;\n        });\n\n        if (!hasId) {\n          // Find the parent statement to add comment\n          let parent = path.parent;\n          const statementTypes = new Set([\n            'VariableDeclaration',\n            'ExpressionStatement',\n            'ReturnStatement',\n            'ExportDefaultDeclaration',\n            'ExportNamedDeclaration',\n            'Program',\n          ]);\n\n          while (parent && !statementTypes.has(parent.value.type)) {\n            parent = parent.parent;\n          }\n\n          if (parent && parent.value) {\n            // For export declarations, add comment to the export itself\n            let targetNode = parent.value;\n            if (\n              targetNode.type !== 'ExportDefaultDeclaration' &&\n              targetNode.type !== 'ExportNamedDeclaration' &&\n              parent.parent &&\n              (parent.parent.value.type === 'ExportDefaultDeclaration' ||\n                parent.parent.value.type === 'ExportNamedDeclaration')\n            ) {\n              targetNode = parent.parent.value;\n            }\n\n            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n            if (added) {\n              context.hasChanges = true;\n            }\n          }\n        }\n      }\n    }\n  });\n\n  if (context.hasChanges) {\n    context.messages.push(`Not Implemented ${fileInfo.path}: Mastra primitives now require a unique id parameter.`);\n  }\n});\n"],"mappings":";;;;;;;;AAkBA,IAAO,6BAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,kBACJ;AAGF,QAAM,iBAAiB;AAAA,IACrB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAGA,QAAM,kBAAkB,CAAC,kBAAkB,cAAc,cAAc;AAGvE,OAAK,KAAK,EAAE,aAAa,EAAE,QAAQ,UAAQ;AACzC,QAAI,KAAK,MAAM,OAAO,SAAS,cAAc;AAC3C,YAAM,YAAY,KAAK,MAAM,OAAO;AAEpC,UAAI,eAAe,SAAS,SAAS,GAAG;AAEtC,cAAM,QAAQ,KAAK,MAAM,UAAU,KAAK,SAAO;AAC7C,cAAI,IAAI,SAAS,oBAAoB;AACnC,mBAAO,IAAI,YAAY;AAAA,cACrB,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS;AAAA,YACtB;AAAA,UACF;AACA,iBAAO;AAAA,QACT,CAAC;AAED,YAAI,CAAC,OAAO;AAEV,cAAI,SAAS,KAAK;AAClB,gBAAM,iBAAiB,oBAAI,IAAI;AAAA,YAC7B;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAED,iBAAO,UAAU,CAAC,eAAe,IAAI,OAAO,MAAM,IAAI,GAAG;AACvD,qBAAS,OAAO;AAAA,UAClB;AAEA,cAAI,UAAU,OAAO,OAAO;AAE1B,gBAAI,aAAa,OAAO;AACxB,gBACE,WAAW,SAAS,8BACpB,WAAW,SAAS,4BACpB,OAAO,WACN,OAAO,OAAO,MAAM,SAAS,8BAC5B,OAAO,OAAO,MAAM,SAAS,2BAC/B;AACA,2BAAa,OAAO,OAAO;AAAA,YAC7B;AAEA,kBAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,gBAAI,OAAO;AACT,sBAAQ,aAAa;AAAA,YACvB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAGD,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,QAAI,KAAK,MAAM,OAAO,SAAS,cAAc;AAC3C,YAAM,eAAe,KAAK,MAAM,OAAO;AAEvC,UAAI,gBAAgB,SAAS,YAAY,GAAG;AAE1C,cAAM,QAAQ,KAAK,MAAM,UAAU,KAAK,SAAO;AAC7C,cAAI,IAAI,SAAS,oBAAoB;AACnC,mBAAO,IAAI,YAAY;AAAA,cACrB,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS;AAAA,YACtB;AAAA,UACF;AACA,iBAAO;AAAA,QACT,CAAC;AAED,YAAI,CAAC,OAAO;AAEV,cAAI,SAAS,KAAK;AAClB,gBAAM,iBAAiB,oBAAI,IAAI;AAAA,YAC7B;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAED,iBAAO,UAAU,CAAC,eAAe,IAAI,OAAO,MAAM,IAAI,GAAG;AACvD,qBAAS,OAAO;AAAA,UAClB;AAEA,cAAI,UAAU,OAAO,OAAO;AAE1B,gBAAI,aAAa,OAAO;AACxB,gBACE,WAAW,SAAS,8BACpB,WAAW,SAAS,4BACpB,OAAO,WACN,OAAO,OAAO,MAAM,SAAS,8BAC5B,OAAO,OAAO,MAAM,SAAS,2BAC/B;AACA,2BAAa,OAAO,OAAO;AAAA,YAC7B;AAEA,kBAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,gBAAI,OAAO;AACT,sBAAQ,aAAa;AAAA,YACvB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,mBAAmB,SAAS,IAAI,wDAAwD;AAAA,EAChH;AACF,CAAC;","names":[]}
+{"version":3,"sources":["../../../../src/codemods/v1/not-implemented/mastra-required-id.ts"],"sourcesContent":["/* eslint-disable no-warning-comments */\nimport { insertCommentOnce } from '../../lib/add-comment';\nimport { createTransformer } from '../../lib/create-transformer';\n\n/**\n * Adds FIXME comments to Mastra primitives that now require an `id` parameter.\n * This includes storages, vector stores, agents, workflows, tools, scorers, and MCP servers.\n *\n * Before:\n * const agent = new Agent({ name: 'Support Agent' });\n * const tool = createTool({ description: 'Get weather' });\n *\n * After:\n * /* FIXME(mastra): Add a unique `id` parameter. See: ... *\\/\n * const agent = new Agent({ name: 'Support Agent' });\n * /* FIXME(mastra): Add a unique `id` parameter. See: ... *\\/\n * const tool = createTool({ description: 'Get weather' });\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  const COMMENT_MESSAGE =\n    'FIXME(mastra): Add a unique `id` parameter. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/mastra#required-id-parameter-for-all-mastra-primitives';\n\n  // List of class names that require id\n  const storageClasses = [\n    'LibSQLStore',\n    'PostgresStore',\n    'D1Store',\n    'MongoDBStore',\n    'DynamoDBStore',\n    'LibSQLVector',\n    'PgVector',\n    'ChromaVector',\n    'PineconeVector',\n    'QdrantVector',\n    'LanceVector',\n    'Agent',\n    'MCPServer',\n  ];\n\n  // List of function names that require id\n  const createFunctions = ['createWorkflow', 'createTool', 'createScorer'];\n\n  // Find NewExpression for classes\n  root.find(j.NewExpression).forEach(path => {\n    if (path.value.callee.type === 'Identifier') {\n      const className = path.value.callee.name;\n\n      if (storageClasses.includes(className)) {\n        // Check if the config already has an id property\n        const hasId = path.value.arguments.some(arg => {\n          if (arg.type === 'ObjectExpression') {\n            return arg.properties?.some(\n              prop =>\n                (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n                prop.key?.type === 'Identifier' &&\n                prop.key.name === 'id',\n            );\n          }\n          return false;\n        });\n\n        if (!hasId) {\n          // Find the parent statement to add comment\n          let parent = path.parent;\n          const statementTypes = new Set([\n            'VariableDeclaration',\n            'ExpressionStatement',\n            'ReturnStatement',\n            'ExportDefaultDeclaration',\n            'ExportNamedDeclaration',\n            'Program',\n          ]);\n\n          while (parent && !statementTypes.has(parent.value.type)) {\n            parent = parent.parent;\n          }\n\n          if (parent && parent.value) {\n            // For export declarations, add comment to the export itself\n            let targetNode = parent.value;\n            if (\n              targetNode.type !== 'ExportDefaultDeclaration' &&\n              targetNode.type !== 'ExportNamedDeclaration' &&\n              parent.parent &&\n              (parent.parent.value.type === 'ExportDefaultDeclaration' ||\n                parent.parent.value.type === 'ExportNamedDeclaration')\n            ) {\n              targetNode = parent.parent.value;\n            }\n\n            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n            if (added) {\n              context.hasChanges = true;\n            }\n          }\n        }\n      }\n    }\n  });\n\n  // Find CallExpression for create functions\n  root.find(j.CallExpression).forEach(path => {\n    if (path.value.callee.type === 'Identifier') {\n      const functionName = path.value.callee.name;\n\n      if (createFunctions.includes(functionName)) {\n        // Check if the config already has an id property\n        const hasId = path.value.arguments.some(arg => {\n          if (arg.type === 'ObjectExpression') {\n            return arg.properties?.some(\n              prop =>\n                (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n                prop.key?.type === 'Identifier' &&\n                prop.key.name === 'id',\n            );\n          }\n          return false;\n        });\n\n        if (!hasId) {\n          // Find the parent statement to add comment\n          let parent = path.parent;\n          const statementTypes = new Set([\n            'VariableDeclaration',\n            'ExpressionStatement',\n            'ReturnStatement',\n            'ExportDefaultDeclaration',\n            'ExportNamedDeclaration',\n            'Program',\n          ]);\n\n          while (parent && !statementTypes.has(parent.value.type)) {\n            parent = parent.parent;\n          }\n\n          if (parent && parent.value) {\n            // For export declarations, add comment to the export itself\n            let targetNode = parent.value;\n            if (\n              targetNode.type !== 'ExportDefaultDeclaration' &&\n              targetNode.type !== 'ExportNamedDeclaration' &&\n              parent.parent &&\n              (parent.parent.value.type === 'ExportDefaultDeclaration' ||\n                parent.parent.value.type === 'ExportNamedDeclaration')\n            ) {\n              targetNode = parent.parent.value;\n            }\n\n            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n            if (added) {\n              context.hasChanges = true;\n            }\n          }\n        }\n      }\n    }\n  });\n\n  if (context.hasChanges) {\n    context.messages.push(`Not Implemented ${fileInfo.path}: Mastra primitives now require a unique id parameter.`);\n  }\n});\n"],"mappings":";;;;;;;;AAkBA,IAAO,6BAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,kBACJ;AAGF,QAAM,iBAAiB;AAAA,IACrB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAGA,QAAM,kBAAkB,CAAC,kBAAkB,cAAc,cAAc;AAGvE,OAAK,KAAK,EAAE,aAAa,EAAE,QAAQ,UAAQ;AACzC,QAAI,KAAK,MAAM,OAAO,SAAS,cAAc;AAC3C,YAAM,YAAY,KAAK,MAAM,OAAO;AAEpC,UAAI,eAAe,SAAS,SAAS,GAAG;AAEtC,cAAM,QAAQ,KAAK,MAAM,UAAU,KAAK,SAAO;AAC7C,cAAI,IAAI,SAAS,oBAAoB;AACnC,mBAAO,IAAI,YAAY;AAAA,cACrB,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS;AAAA,YACtB;AAAA,UACF;AACA,iBAAO;AAAA,QACT,CAAC;AAED,YAAI,CAAC,OAAO;AAEV,cAAI,SAAS,KAAK;AAClB,gBAAM,iBAAiB,oBAAI,IAAI;AAAA,YAC7B;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAED,iBAAO,UAAU,CAAC,eAAe,IAAI,OAAO,MAAM,IAAI,GAAG;AACvD,qBAAS,OAAO;AAAA,UAClB;AAEA,cAAI,UAAU,OAAO,OAAO;AAE1B,gBAAI,aAAa,OAAO;AACxB,gBACE,WAAW,SAAS,8BACpB,WAAW,SAAS,4BACpB,OAAO,WACN,OAAO,OAAO,MAAM,SAAS,8BAC5B,OAAO,OAAO,MAAM,SAAS,2BAC/B;AACA,2BAAa,OAAO,OAAO;AAAA,YAC7B;AAEA,kBAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,gBAAI,OAAO;AACT,sBAAQ,aAAa;AAAA,YACvB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAGD,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,QAAI,KAAK,MAAM,OAAO,SAAS,cAAc;AAC3C,YAAM,eAAe,KAAK,MAAM,OAAO;AAEvC,UAAI,gBAAgB,SAAS,YAAY,GAAG;AAE1C,cAAM,QAAQ,KAAK,MAAM,UAAU,KAAK,SAAO;AAC7C,cAAI,IAAI,SAAS,oBAAoB;AACnC,mBAAO,IAAI,YAAY;AAAA,cACrB,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS;AAAA,YACtB;AAAA,UACF;AACA,iBAAO;AAAA,QACT,CAAC;AAED,YAAI,CAAC,OAAO;AAEV,cAAI,SAAS,KAAK;AAClB,gBAAM,iBAAiB,oBAAI,IAAI;AAAA,YAC7B;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAED,iBAAO,UAAU,CAAC,eAAe,IAAI,OAAO,MAAM,IAAI,GAAG;AACvD,qBAAS,OAAO;AAAA,UAClB;AAEA,cAAI,UAAU,OAAO,OAAO;AAE1B,gBAAI,aAAa,OAAO;AACxB,gBACE,WAAW,SAAS,8BACpB,WAAW,SAAS,4BACpB,OAAO,WACN,OAAO,OAAO,MAAM,SAAS,8BAC5B,OAAO,OAAO,MAAM,SAAS,2BAC/B;AACA,2BAAa,OAAO,OAAO;AAAA,YAC7B;AAEA,kBAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,gBAAI,OAAO;AACT,sBAAQ,aAAa;AAAA,YACvB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,mBAAmB,SAAS,IAAI,wDAAwD;AAAA,EAChH;AACF,CAAC;","names":[]}

package/dist/codemods/v1/workflow-stream-vnext.js

@@ -0,0 +1,35 @@
+import {
+  createTransformer
+} from "../chunk-JNFQ6J6B.js";
+
+// src/codemods/v1/workflow-stream-vnext.ts
+var workflow_stream_vnext_default = createTransformer((fileInfo, api, options, context) => {
+  const { j, root } = context;
+  const methodRenames = {
+    streamVNext: "stream",
+    resumeStreamVNext: "resumeStream",
+    observeStreamVNext: "observeStream"
+  };
+  let count = 0;
+  root.find(j.CallExpression).forEach((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "MemberExpression") return;
+    if (callee.property.type !== "Identifier") return;
+    const oldName = callee.property.name;
+    const newName = methodRenames[oldName];
+    if (newName) {
+      callee.property.name = newName;
+      count++;
+    }
+  });
+  if (count > 0) {
+    context.hasChanges = true;
+    context.messages.push(
+      `Renamed workflow run VNext methods: streamVNext/resumeStreamVNext/observeStreamVNext \u2192 stream/resumeStream/observeStream`
+    );
+  }
+});
+export {
+  workflow_stream_vnext_default as default
+};
+//# sourceMappingURL=workflow-stream-vnext.js.map

package/dist/codemods/v1/workflow-stream-vnext.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/v1/workflow-stream-vnext.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\n\n/**\n * Transforms workflow run VNext methods to their standard names:\n * - run.streamVNext() → run.stream()\n * - run.resumeStreamVNext() → run.resumeStream()\n * - run.observeStreamVNext() → run.observeStream()\n *\n * These methods are called on the result of workflow.createRun().\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  // Map of old method names to new method names\n  const methodRenames: Record<string, string> = {\n    streamVNext: 'stream',\n    resumeStreamVNext: 'resumeStream',\n    observeStreamVNext: 'observeStream',\n  };\n\n  let count = 0;\n\n  // Find all call expressions and rename VNext methods\n  // These method names are unique enough to Mastra workflow runs\n  // that we can safely rename them globally\n  root.find(j.CallExpression).forEach(path => {\n    const { callee } = path.value;\n    if (callee.type !== 'MemberExpression') return;\n    if (callee.property.type !== 'Identifier') return;\n\n    const oldName = callee.property.name;\n    const newName = methodRenames[oldName];\n\n    if (newName) {\n      callee.property.name = newName;\n      count++;\n    }\n  });\n\n  if (count > 0) {\n    context.hasChanges = true;\n    context.messages.push(\n      `Renamed workflow run VNext methods: streamVNext/resumeStreamVNext/observeStreamVNext → stream/resumeStream/observeStream`,\n    );\n  }\n});\n"],"mappings":";;;;;AAUA,IAAO,gCAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,gBAAwC;AAAA,IAC5C,aAAa;AAAA,IACb,mBAAmB;AAAA,IACnB,oBAAoB;AAAA,EACtB;AAEA,MAAI,QAAQ;AAKZ,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,UAAM,EAAE,OAAO,IAAI,KAAK;AACxB,QAAI,OAAO,SAAS,mBAAoB;AACxC,QAAI,OAAO,SAAS,SAAS,aAAc;AAE3C,UAAM,UAAU,OAAO,SAAS;AAChC,UAAM,UAAU,cAAc,OAAO;AAErC,QAAI,SAAS;AACX,aAAO,SAAS,OAAO;AACvB;AAAA,IACF;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,GAAG;AACb,YAAQ,aAAa;AACrB,YAAQ,SAAS;AAAA,MACf;AAAA,IACF;AAAA,EACF;AACF,CAAC;","names":[]}

package/dist/index.js

@@ -1,11 +1,11 @@
 #! /usr/bin/env node
 import { Command } from 'commander';
 import debug from 'debug';
-import fs from 'fs';
 import child_process from 'child_process';
+import fs from 'fs';
+import path from 'path';
 import { fileURLToPath } from 'url';
 import util from 'util';
-import path from 'path';
 import { intro, spinner, outro } from '@clack/prompts';
 
 var exec = util.promisify(child_process.exec);
@@ -116,6 +116,7 @@ var BUNDLE = [
   "v1/workflow-create-run-async",
   "v1/workflow-run-count",
   "v1/workflow-list-runs",
+  "v1/workflow-stream-vnext",
   "v1/memory-query-to-recall",
   "v1/memory-vector-search-param",
   "v1/memory-message-v2-type",

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/lib/transform.ts","../src/lib/bundle.ts","../src/lib/upgrade.ts","../src/index.ts"],"names":["transform","log","debug","error"],"mappings":";;;;;;;;;;AAOA,IAAM,IAAA,GAAO,IAAA,CAAK,SAAA,CAAU,aAAA,CAAc,IAAI,CAAA;AAS9C,IAAM,GAAA,GAAM,MAAM,mBAAmB,CAAA;AACrC,IAAM,KAAA,GAAQ,MAAM,yBAAyB,CAAA;AAC7C,IAAM,UAAA,GAAa,aAAA,CAAc,MAAA,CAAA,IAAA,CAAY,GAAG,CAAA;AAChD,IAAM,SAAA,GAAY,IAAA,CAAK,OAAA,CAAQ,UAAU,CAAA;AAEzC,SAAS,cAAA,GAAyB;AAChC,EAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,OAAA,CAAQ,SAAA,EAAW,kCAAkC,CAAA;AACnF,EAAA,OAAO,EAAA,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,CAAYA,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;AAEA,eAAsB,SAAA,CACpB,SACA,MAAA,EACA,gBAAA,EACA,UAAkC,EAAE,SAAA,EAAW,MAAK,EACyB;AAC7E,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,cAAc,IAAA,CAAK,OAAA,CAAQ,SAAA,EAAW,CAAA,WAAA,EAAc,OAAO,CAAA,GAAA,CAAK,CAAA;AACtE,EAAA,MAAM,UAAA,GAAa,IAAA,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,EAAE,QAAO,GAAI,MAAM,KAAK,OAAA,EAAS,EAAE,QAAA,EAAU,MAAA,EAAQ,CAAA;AAC3D,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,WAAAA,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;;;AChIO,IAAM,MAAA,GAAS;AAAA,EACpB,wBAAA;AAAA,EACA,oBAAA;AAAA,EACA,kCAAA;AAAA,EACA,uBAAA;AAAA,EACA,uCAAA;AAAA,EACA,0BAAA;AAAA,EACA,gBAAA;AAAA,EACA,4BAAA;AAAA,EACA,iCAAA;AAAA,EACA,uBAAA;AAAA,EACA,2CAAA;AAAA,EACA,kCAAA;AAAA,EACA,yBAAA;AAAA,EACA,kBAAA;AAAA,EACA,qBAAA;AAAA,EACA,qBAAA;AAAA,EACA,4BAAA;AAAA,EACA,wBAAA;AAAA,EACA,6BAAA;AAAA,EACA,sBAAA;AAAA,EACA,yBAAA;AAAA,EACA,yBAAA;AAAA,EACA,2BAAA;AAAA,EACA,8BAAA;AAAA,EACA,uBAAA;AAAA,EACA,uBAAA;AAAA,EACA,2BAAA;AAAA,EACA,+BAAA;AAAA,EACA,2BAAA;AAAA,EACA,oCAAA;AAAA,EACA,gCAAA;AAAA,EACA,iCAAA;AAAA,EACA,mCAAA;AAAA,EACA,+BAAA;AAAA,EACA;AACF,CAAA;;;ACxBA,IAAMC,IAAAA,GAAMC,MAAM,iBAAiB,CAAA;AACnC,IAAMC,MAAAA,GAAQD,MAAM,uBAAuB,CAAA;AAG3C,IAAM,WAAW,MAAA,CAAO,MAAA,CAAO,aAAW,OAAA,CAAQ,UAAA,CAAW,KAAK,CAAC,CAAA;AAEnE,eAAe,WAAA,CAAY,QAAA,EAAoB,OAAA,EAA2B,YAAA,EAAsB;AAC9F,EAAA,MAAM,GAAA,GAAM,QAAQ,GAAA,EAAI;AACxB,EAAA,KAAA,CAAM,CAAA,SAAA,EAAY,YAAY,CAAA,SAAA,CAAW,CAAA;AACzC,EAAA,MAAM,WAAW,QAAA,CAAS,MAAA;AAC1B,EAAA,MAAM,IAAI,OAAA,EAAQ;AAElB,EAAA,CAAA,CAAE,KAAA,CAAM,CAAA,QAAA,EAAW,QAAQ,CAAA,CAAA,EAAI,YAAY,CAAA,SAAA,CAAW,CAAA;AAEtD,EAAA,MAAM,YAA6B,EAAC;AACpC,EAAA,IAAI,uBAAA,GAA0B,KAAA;AAC9B,EAAA,IAAI,KAAA,GAAQ,CAAA;AACZ,EAAA,KAAA,MAAW,CAAC,CAAA,EAAG,OAAO,CAAA,IAAK,QAAA,CAAS,SAAQ,EAAG;AAC7C,IAAA,MAAM,EAAE,QAAQ,oBAAA,EAAqB,GAAI,MAAM,SAAA,CAAU,OAAA,EAAS,KAAK,OAAA,EAAS;AAAA,MAC9E,SAAA,EAAW;AAAA,KACZ,CAAA;AACD,IAAA,SAAA,CAAU,IAAA,CAAK,GAAG,MAAM,CAAA;AACxB,IAAA,IAAI,oBAAA,CAAqB,SAAS,CAAA,EAAG;AACnC,MAAA,uBAAA,GAA0B,IAAA;AAAA,IAC5B;AACA,IAAA,KAAA,EAAA;AACA,IAAA,CAAA,CAAE,QAAQ,CAAA,QAAA,EAAW,KAAK,IAAI,QAAQ,CAAA,EAAA,EAAK,OAAO,CAAA,CAAA,CAAG,CAAA;AAAA,EACvD;AACA,EAAA,CAAA,CAAE,IAAA,CAAK,CAAA,IAAA,EAAO,KAAK,CAAA,CAAA,EAAI,QAAQ,CAAA,UAAA,CAAY,CAAA;AAE3C,EAAA,IAAI,SAAA,CAAU,SAAS,CAAA,EAAG;AACxB,IAAAD,IAAAA,CAAI,CAAA,KAAA,EAAQ,YAAY,CAAA,2DAAA,CAA6D,CAAA;AACrF,IAAA,SAAA,CAAU,QAAQ,CAAC,EAAE,WAAAD,UAAAA,EAAW,QAAA,EAAU,SAAQ,KAAM;AACtD,MAAAG,OAAM,CAAA,QAAA,EAAWH,UAAS,UAAU,QAAQ,CAAA,UAAA,EAAa,OAAO,CAAA,CAAE,CAAA;AAAA,IACpE,CAAC,CAAA;AAAA,EACH;AAEA,EAAA,IAAI,uBAAA,EAAyB;AAC3B,IAAAC,IAAAA;AAAA,MACE,QAAQ,YAAY,CAAA,mJAAA;AAAA,KACtB;AAAA,EACF;AAEA,EAAA,KAAA,CAAM,CAAA,EAAG,YAAY,CAAA,mBAAA,CAAqB,CAAA;AAC5C;AAEA,eAAsB,UAAU,OAAA,EAA2B;AACzD,EAAA,MAAM,WAAA,CAAY,QAAA,EAAU,OAAA,EAAS,IAAI,CAAA;AAC3C;;;ACtDA,IAAME,MAAAA,GAAQD,MAAM,eAAe,CAAA;AACnCA,KAAAA,CAAM,OAAO,WAAW,CAAA;AAExB,IAAM,OAAA,GAAU,IAAI,OAAA,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,OAAO,OAAA,EAAS,QAAQ,OAAA,KAAY;AAC1C,EAAA,IAAI;AACF,IAAA,MAAM,SAAA,CAAU,OAAA,EAAS,MAAA,EAAQ,OAAO,CAAA;AAAA,EAC1C,SAAS,GAAA,EAAU;AACjB,IAAAC,MAAAA,CAAM,CAAA,oBAAA,EAAuB,GAAG,CAAA,CAAE,CAAA;AAClC,IAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAAA,EAChB;AACF,CAAC,CAAA;AAEH,OAAA,CACG,OAAA,CAAQ,IAAI,CAAA,CACZ,WAAA,CAAY,oCAAoC,EAChD,MAAA,CAAO,WAAA,EAAa,wCAAwC,CAAA,CAC5D,MAAA,CAAO,aAAA,EAAe,mCAAmC,CAAA,CACzD,MAAA,CAAO,WAAA,EAAa,mDAAmD,CAAA,CACvE,MAAA,CAAO,+BAA+B,sCAAsC,CAAA,CAC5E,MAAA,CAAO,OAAM,OAAA,KAAW;AACvB,EAAA,IAAI;AACF,IAAA,MAAM,UAAU,OAAO,CAAA;AAAA,EACzB,SAAS,GAAA,EAAU;AACjB,IAAAA,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":["import fs from 'fs';\nimport child_process from 'node:child_process';\nimport { fileURLToPath } from 'node:url';\nimport util from 'node:util';\nimport path from 'path';\nimport debug from 'debug';\n\nconst exec = util.promisify(child_process.exec);\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');\nconst __filename = fileURLToPath(import.meta.url);\nconst __dirname = path.dirname(__filename);\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 async function transform(\n  codemod: string,\n  source: string,\n  transformOptions: TransformOptions,\n  options: { logStatus: boolean } = { logStatus: true },\n): Promise<{ 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 } = await exec(command, { encoding: 'utf8' });\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","// List of all codemods\nexport const BUNDLE = [\n  'v1/mastra-core-imports',\n  'v1/runtime-context',\n  'v1/not-implemented/mastra-memory',\n  'v1/mastra-plural-apis',\n  'v1/not-implemented/mastra-required-id',\n  'v1/agent-property-access',\n  'v1/agent-voice',\n  'v1/agent-processor-methods',\n  'v1/agent-generate-stream-v-next',\n  'v1/agent-abort-signal',\n  'v1/not-implemented/agent-format-parameter',\n  'v1/not-implemented/agent-to-step',\n  'v1/voice-property-names',\n  'v1/mcp-get-tools',\n  'v1/mcp-get-toolsets',\n  'v1/client-sdk-types',\n  'v1/client-to-ai-sdk-format',\n  'v1/client-offset-limit',\n  'v1/client-get-memory-thread',\n  'v1/experimental-auth',\n  'v1/evals-run-experiment',\n  'v1/evals-scorer-by-name',\n  'v1/evals-prebuilt-imports',\n  'v1/workflow-create-run-async',\n  'v1/workflow-run-count',\n  'v1/workflow-list-runs',\n  'v1/memory-query-to-recall',\n  'v1/memory-vector-search-param',\n  'v1/memory-message-v2-type',\n  'v1/storage-get-threads-by-resource',\n  'v1/storage-list-messages-by-id',\n  'v1/storage-postgres-schema-name',\n  'v1/storage-get-messages-paginated',\n  'v1/storage-list-workflow-runs',\n  'v1/vector-pg-constructor',\n];\n","import { spinner, intro, outro } from '@clack/prompts';\nimport debug from 'debug';\nimport { BUNDLE } from './bundle';\nimport type { TransformErrors } from './transform';\nimport { transform } from './transform';\n\ninterface TransformOptions {\n  dry?: true;\n  print?: true;\n  verbose?: true;\n  jscodeshift?: string;\n}\n\nconst log = debug('codemod:upgrade');\nconst error = debug('codemod:upgrade:error');\n\n// Extract v1 codemods from the bundle\nconst v1Bundle = BUNDLE.filter(codemod => codemod.startsWith('v1/'));\n\nasync function runCodemods(codemods: string[], options: TransformOptions, versionLabel: string) {\n  const cwd = process.cwd();\n  intro(`Starting ${versionLabel} codemods`);\n  const modCount = codemods.length;\n  const s = spinner();\n\n  s.start(`Running ${modCount} ${versionLabel} codemods`);\n\n  const allErrors: TransformErrors = [];\n  let notImplementedAvailable = false;\n  let count = 0;\n  for (const [_, codemod] of codemods.entries()) {\n    const { errors, notImplementedErrors } = await transform(codemod, cwd, options, {\n      logStatus: false,\n    });\n    allErrors.push(...errors);\n    if (notImplementedErrors.length > 0) {\n      notImplementedAvailable = true;\n    }\n    count++;\n    s.message(`Codemod ${count}/${modCount} (${codemod})`);\n  }\n  s.stop(`Ran ${count}/${modCount} codemods.`);\n\n  if (allErrors.length > 0) {\n    log(`Some ${versionLabel} codemods did not apply successfully to all files. Details:`);\n    allErrors.forEach(({ transform, filename, summary }) => {\n      error(`codemod=${transform}, path=${filename}, summary=${summary}`);\n    });\n  }\n\n  if (notImplementedAvailable) {\n    log(\n      `Some ${versionLabel} codemods require manual changes. Please search your codebase for \\`FIXME(mastra): \\` comments and follow the instructions to complete the upgrade.`,\n    );\n  }\n\n  outro(`${versionLabel} codemods complete.`);\n}\n\nexport async function upgradeV1(options: TransformOptions) {\n  await runCodemods(v1Bundle, options, 'v1');\n}\n","#! /usr/bin/env node\n\nimport { Command } from 'commander';\nimport debug from 'debug';\nimport { transform } from './lib/transform';\nimport { upgradeV1 } from './lib/upgrade';\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(async (codemod, source, options) => {\n    try {\n      await transform(codemod, source, options);\n    } catch (err: any) {\n      error(`Error transforming: ${err}`);\n      process.exit(1);\n    }\n  });\n\nprogram\n  .command('v1')\n  .description('Apply all v1 codemods (v0.x to v1)')\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(async options => {\n    try {\n      await upgradeV1(options);\n    } catch (err: any) {\n      error(`Error transforming: ${err}`);\n      process.exit(1);\n    }\n  });\n\nprogram.parse(process.argv);\n"]}
+{"version":3,"sources":["../src/lib/transform.ts","../src/lib/bundle.ts","../src/lib/upgrade.ts","../src/index.ts"],"names":["transform","log","debug","error"],"mappings":";;;;;;;;;;AAOA,IAAM,IAAA,GAAO,IAAA,CAAK,SAAA,CAAU,aAAA,CAAc,IAAI,CAAA;AAS9C,IAAM,GAAA,GAAM,MAAM,mBAAmB,CAAA;AACrC,IAAM,KAAA,GAAQ,MAAM,yBAAyB,CAAA;AAC7C,IAAM,UAAA,GAAa,aAAA,CAAc,MAAA,CAAA,IAAA,CAAY,GAAG,CAAA;AAChD,IAAM,SAAA,GAAY,IAAA,CAAK,OAAA,CAAQ,UAAU,CAAA;AAEzC,SAAS,cAAA,GAAyB;AAChC,EAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,OAAA,CAAQ,SAAA,EAAW,kCAAkC,CAAA;AACnF,EAAA,OAAO,EAAA,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,CAAYA,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;AAEA,eAAsB,SAAA,CACpB,SACA,MAAA,EACA,gBAAA,EACA,UAAkC,EAAE,SAAA,EAAW,MAAK,EACyB;AAC7E,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,cAAc,IAAA,CAAK,OAAA,CAAQ,SAAA,EAAW,CAAA,WAAA,EAAc,OAAO,CAAA,GAAA,CAAK,CAAA;AACtE,EAAA,MAAM,UAAA,GAAa,IAAA,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,EAAE,QAAO,GAAI,MAAM,KAAK,OAAA,EAAS,EAAE,QAAA,EAAU,MAAA,EAAQ,CAAA;AAC3D,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,WAAAA,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;;;AChIO,IAAM,MAAA,GAAS;AAAA,EACpB,wBAAA;AAAA,EACA,oBAAA;AAAA,EACA,kCAAA;AAAA,EACA,uBAAA;AAAA,EACA,uCAAA;AAAA,EACA,0BAAA;AAAA,EACA,gBAAA;AAAA,EACA,4BAAA;AAAA,EACA,iCAAA;AAAA,EACA,uBAAA;AAAA,EACA,2CAAA;AAAA,EACA,kCAAA;AAAA,EACA,yBAAA;AAAA,EACA,kBAAA;AAAA,EACA,qBAAA;AAAA,EACA,qBAAA;AAAA,EACA,4BAAA;AAAA,EACA,wBAAA;AAAA,EACA,6BAAA;AAAA,EACA,sBAAA;AAAA,EACA,yBAAA;AAAA,EACA,yBAAA;AAAA,EACA,2BAAA;AAAA,EACA,8BAAA;AAAA,EACA,uBAAA;AAAA,EACA,uBAAA;AAAA,EACA,0BAAA;AAAA,EACA,2BAAA;AAAA,EACA,+BAAA;AAAA,EACA,2BAAA;AAAA,EACA,oCAAA;AAAA,EACA,gCAAA;AAAA,EACA,iCAAA;AAAA,EACA,mCAAA;AAAA,EACA,+BAAA;AAAA,EACA;AACF,CAAA;;;ACzBA,IAAMC,IAAAA,GAAMC,MAAM,iBAAiB,CAAA;AACnC,IAAMC,MAAAA,GAAQD,MAAM,uBAAuB,CAAA;AAG3C,IAAM,WAAW,MAAA,CAAO,MAAA,CAAO,aAAW,OAAA,CAAQ,UAAA,CAAW,KAAK,CAAC,CAAA;AAEnE,eAAe,WAAA,CAAY,QAAA,EAAoB,OAAA,EAA2B,YAAA,EAAsB;AAC9F,EAAA,MAAM,GAAA,GAAM,QAAQ,GAAA,EAAI;AACxB,EAAA,KAAA,CAAM,CAAA,SAAA,EAAY,YAAY,CAAA,SAAA,CAAW,CAAA;AACzC,EAAA,MAAM,WAAW,QAAA,CAAS,MAAA;AAC1B,EAAA,MAAM,IAAI,OAAA,EAAQ;AAElB,EAAA,CAAA,CAAE,KAAA,CAAM,CAAA,QAAA,EAAW,QAAQ,CAAA,CAAA,EAAI,YAAY,CAAA,SAAA,CAAW,CAAA;AAEtD,EAAA,MAAM,YAA6B,EAAC;AACpC,EAAA,IAAI,uBAAA,GAA0B,KAAA;AAC9B,EAAA,IAAI,KAAA,GAAQ,CAAA;AACZ,EAAA,KAAA,MAAW,CAAC,CAAA,EAAG,OAAO,CAAA,IAAK,QAAA,CAAS,SAAQ,EAAG;AAC7C,IAAA,MAAM,EAAE,QAAQ,oBAAA,EAAqB,GAAI,MAAM,SAAA,CAAU,OAAA,EAAS,KAAK,OAAA,EAAS;AAAA,MAC9E,SAAA,EAAW;AAAA,KACZ,CAAA;AACD,IAAA,SAAA,CAAU,IAAA,CAAK,GAAG,MAAM,CAAA;AACxB,IAAA,IAAI,oBAAA,CAAqB,SAAS,CAAA,EAAG;AACnC,MAAA,uBAAA,GAA0B,IAAA;AAAA,IAC5B;AACA,IAAA,KAAA,EAAA;AACA,IAAA,CAAA,CAAE,QAAQ,CAAA,QAAA,EAAW,KAAK,IAAI,QAAQ,CAAA,EAAA,EAAK,OAAO,CAAA,CAAA,CAAG,CAAA;AAAA,EACvD;AACA,EAAA,CAAA,CAAE,IAAA,CAAK,CAAA,IAAA,EAAO,KAAK,CAAA,CAAA,EAAI,QAAQ,CAAA,UAAA,CAAY,CAAA;AAE3C,EAAA,IAAI,SAAA,CAAU,SAAS,CAAA,EAAG;AACxB,IAAAD,IAAAA,CAAI,CAAA,KAAA,EAAQ,YAAY,CAAA,2DAAA,CAA6D,CAAA;AACrF,IAAA,SAAA,CAAU,QAAQ,CAAC,EAAE,WAAAD,UAAAA,EAAW,QAAA,EAAU,SAAQ,KAAM;AACtD,MAAAG,OAAM,CAAA,QAAA,EAAWH,UAAS,UAAU,QAAQ,CAAA,UAAA,EAAa,OAAO,CAAA,CAAE,CAAA;AAAA,IACpE,CAAC,CAAA;AAAA,EACH;AAEA,EAAA,IAAI,uBAAA,EAAyB;AAC3B,IAAAC,IAAAA;AAAA,MACE,QAAQ,YAAY,CAAA,mJAAA;AAAA,KACtB;AAAA,EACF;AAEA,EAAA,KAAA,CAAM,CAAA,EAAG,YAAY,CAAA,mBAAA,CAAqB,CAAA;AAC5C;AAEA,eAAsB,UAAU,OAAA,EAA2B;AACzD,EAAA,MAAM,WAAA,CAAY,QAAA,EAAU,OAAA,EAAS,IAAI,CAAA;AAC3C;;;ACtDA,IAAME,MAAAA,GAAQD,MAAM,eAAe,CAAA;AACnCA,KAAAA,CAAM,OAAO,WAAW,CAAA;AAExB,IAAM,OAAA,GAAU,IAAI,OAAA,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,OAAO,OAAA,EAAS,QAAQ,OAAA,KAAY;AAC1C,EAAA,IAAI;AACF,IAAA,MAAM,SAAA,CAAU,OAAA,EAAS,MAAA,EAAQ,OAAO,CAAA;AAAA,EAC1C,SAAS,GAAA,EAAU;AACjB,IAAAC,MAAAA,CAAM,CAAA,oBAAA,EAAuB,GAAG,CAAA,CAAE,CAAA;AAClC,IAAA,OAAA,CAAQ,KAAK,CAAC,CAAA;AAAA,EAChB;AACF,CAAC,CAAA;AAEH,OAAA,CACG,OAAA,CAAQ,IAAI,CAAA,CACZ,WAAA,CAAY,oCAAoC,EAChD,MAAA,CAAO,WAAA,EAAa,wCAAwC,CAAA,CAC5D,MAAA,CAAO,aAAA,EAAe,mCAAmC,CAAA,CACzD,MAAA,CAAO,WAAA,EAAa,mDAAmD,CAAA,CACvE,MAAA,CAAO,+BAA+B,sCAAsC,CAAA,CAC5E,MAAA,CAAO,OAAM,OAAA,KAAW;AACvB,EAAA,IAAI;AACF,IAAA,MAAM,UAAU,OAAO,CAAA;AAAA,EACzB,SAAS,GAAA,EAAU;AACjB,IAAAA,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":["import child_process from 'node:child_process';\nimport fs from 'node:fs';\nimport path from 'node:path';\nimport { fileURLToPath } from 'node:url';\nimport util from 'node:util';\nimport debug from 'debug';\n\nconst exec = util.promisify(child_process.exec);\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');\nconst __filename = fileURLToPath(import.meta.url);\nconst __dirname = path.dirname(__filename);\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 async function transform(\n  codemod: string,\n  source: string,\n  transformOptions: TransformOptions,\n  options: { logStatus: boolean } = { logStatus: true },\n): Promise<{ 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 } = await exec(command, { encoding: 'utf8' });\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","// List of all codemods\nexport const BUNDLE = [\n  'v1/mastra-core-imports',\n  'v1/runtime-context',\n  'v1/not-implemented/mastra-memory',\n  'v1/mastra-plural-apis',\n  'v1/not-implemented/mastra-required-id',\n  'v1/agent-property-access',\n  'v1/agent-voice',\n  'v1/agent-processor-methods',\n  'v1/agent-generate-stream-v-next',\n  'v1/agent-abort-signal',\n  'v1/not-implemented/agent-format-parameter',\n  'v1/not-implemented/agent-to-step',\n  'v1/voice-property-names',\n  'v1/mcp-get-tools',\n  'v1/mcp-get-toolsets',\n  'v1/client-sdk-types',\n  'v1/client-to-ai-sdk-format',\n  'v1/client-offset-limit',\n  'v1/client-get-memory-thread',\n  'v1/experimental-auth',\n  'v1/evals-run-experiment',\n  'v1/evals-scorer-by-name',\n  'v1/evals-prebuilt-imports',\n  'v1/workflow-create-run-async',\n  'v1/workflow-run-count',\n  'v1/workflow-list-runs',\n  'v1/workflow-stream-vnext',\n  'v1/memory-query-to-recall',\n  'v1/memory-vector-search-param',\n  'v1/memory-message-v2-type',\n  'v1/storage-get-threads-by-resource',\n  'v1/storage-list-messages-by-id',\n  'v1/storage-postgres-schema-name',\n  'v1/storage-get-messages-paginated',\n  'v1/storage-list-workflow-runs',\n  'v1/vector-pg-constructor',\n];\n","import { spinner, intro, outro } from '@clack/prompts';\nimport debug from 'debug';\nimport { BUNDLE } from './bundle';\nimport type { TransformErrors } from './transform';\nimport { transform } from './transform';\n\ninterface TransformOptions {\n  dry?: true;\n  print?: true;\n  verbose?: true;\n  jscodeshift?: string;\n}\n\nconst log = debug('codemod:upgrade');\nconst error = debug('codemod:upgrade:error');\n\n// Extract v1 codemods from the bundle\nconst v1Bundle = BUNDLE.filter(codemod => codemod.startsWith('v1/'));\n\nasync function runCodemods(codemods: string[], options: TransformOptions, versionLabel: string) {\n  const cwd = process.cwd();\n  intro(`Starting ${versionLabel} codemods`);\n  const modCount = codemods.length;\n  const s = spinner();\n\n  s.start(`Running ${modCount} ${versionLabel} codemods`);\n\n  const allErrors: TransformErrors = [];\n  let notImplementedAvailable = false;\n  let count = 0;\n  for (const [_, codemod] of codemods.entries()) {\n    const { errors, notImplementedErrors } = await transform(codemod, cwd, options, {\n      logStatus: false,\n    });\n    allErrors.push(...errors);\n    if (notImplementedErrors.length > 0) {\n      notImplementedAvailable = true;\n    }\n    count++;\n    s.message(`Codemod ${count}/${modCount} (${codemod})`);\n  }\n  s.stop(`Ran ${count}/${modCount} codemods.`);\n\n  if (allErrors.length > 0) {\n    log(`Some ${versionLabel} codemods did not apply successfully to all files. Details:`);\n    allErrors.forEach(({ transform, filename, summary }) => {\n      error(`codemod=${transform}, path=${filename}, summary=${summary}`);\n    });\n  }\n\n  if (notImplementedAvailable) {\n    log(\n      `Some ${versionLabel} codemods require manual changes. Please search your codebase for \\`FIXME(mastra): \\` comments and follow the instructions to complete the upgrade.`,\n    );\n  }\n\n  outro(`${versionLabel} codemods complete.`);\n}\n\nexport async function upgradeV1(options: TransformOptions) {\n  await runCodemods(v1Bundle, options, 'v1');\n}\n","#! /usr/bin/env node\n\nimport { Command } from 'commander';\nimport debug from 'debug';\nimport { transform } from './lib/transform';\nimport { upgradeV1 } from './lib/upgrade';\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(async (codemod, source, options) => {\n    try {\n      await transform(codemod, source, options);\n    } catch (err: any) {\n      error(`Error transforming: ${err}`);\n      process.exit(1);\n    }\n  });\n\nprogram\n  .command('v1')\n  .description('Apply all v1 codemods (v0.x to v1)')\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(async options => {\n    try {\n      await upgradeV1(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

@@ -1,7 +1,7 @@
 {
   "name": "@mastra/codemod",
   "type": "module",
-  "version": "0.1.0-beta.1",
+  "version": "0.1.0-beta.3",
   "license": "Apache-2.0",
   "description": "Codemod CLI for Mastra",
   "bin": {
@@ -27,11 +27,13 @@
     "@commander-js/extra-typings": "^14.0.0",
     "@types/debug": "^4.1.12",
     "@types/jscodeshift": "^17.3.0",
-    "@types/node": "^24.9.2",
+    "@types/node": "22.13.17",
+    "@vitest/coverage-v8": "4.0.12",
+    "@vitest/ui": "4.0.12",
     "eslint": "^9.37.0",
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
-    "vitest": "^3.2.4",
+    "vitest": "4.0.12",
     "@internal/lint": "0.0.53"
   },
   "homepage": "https://mastra.ai",