@aws/nx-plugin 1.0.0-rc.7 → 1.0.0-rc.71

package/src/migrations/v1.0.0-rc.57/0001-ts-agent-session-management-support/migration.js

@@ -0,0 +1,449 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ import { getProjects, joinPathFragments, visitNotIgnoredFiles } from "@nx/devkit";
+import { TS_AGENT_GENERATOR_INFO } from "../../../ts/agent/generator.js";
+import { AGENT_CONNECTION_PROJECT_DIR } from "../../../utils/agent-connection/agent-connection.js";
+import { applyGritQL, captureGritQLVariable, matchGritQL } from "../../../utils/ast.js";
+import { formatFilesInSubtree } from "../../../utils/format.js";
+import { isEsmWorkspace } from "../../../utils/module-format.js";
+import { kebabCase } from "../../../utils/names.js";
+import { getRelativePathToRootByDirectory, toProjectRelativePath } from "../../../utils/paths.js";
+/**
+ * Add session management support to ts#agent.
+ *
+ * How to write a migration:
+ * - https://nx.dev/docs/kb/migration-generators
+ * - What `nextSteps` means: https://nx.dev/docs/reference/devkit/MigrationReturnObject
+ *
+ * Guardrails:
+ * - Pattern-match before writing: skip files that have diverged from the shape
+ *   your generators produce and report them via `nextSteps`, or consider a
+ *   hybrid migration, rather than clobbering the user's changes.
+ * - Idempotent: re-running must be a no-op.
+ * - Format what you write: finish with `formatFilesInSubtree` so the files your
+ *   migration wrote are formatted correctly.
+ */ // Reshapes an `agentRuntimes` rc.set entry from a bare ARN to `{ arn }`. Only
+// the 'agentcore' namespace — 'connection' (client-facing) stays a string,
+// since it shouldn't carry the session bucket name to the browser. No
+// `session` field added here; a fresh session.ts (created below) defaults to
+// in-memory. Idempotent: only matches the pre-migration shape.
+const CDK_RC_SET_PATTERN = `\`rc.set('agentcore', 'agentRuntimes', { ...rc.get('agentcore').agentRuntimes, $name: this.agentCoreRuntime.agentRuntimeArn });\` => raw\`rc.set('agentcore', 'agentRuntimes', {
+  ...rc.get('agentcore').agentRuntimes,
+  $name: {
+    arn: this.agentCoreRuntime.agentRuntimeArn,
+  },
+});\``;
+// Terraform equivalent, anchored on the module's fixed name/shape so it
+// doesn't also match the 'connection' module's identical-looking `value` line.
+const TF_VALUE_PATTERN = `language hcl\n\`module "add_agent_runtime_to_runtime_config" {
+  source = "../../../core/runtime-config/entry"
+
+  namespace = "agentcore"
+  key       = "agentRuntimes"
+  value     = { "$name" = module.agent_core_runtime.agent_core_runtime_arn }
+
+  depends_on = [module.agent_core_runtime]
+}\` => \`module "add_agent_runtime_to_runtime_config" {
+  source = "../../../core/runtime-config/entry"
+
+  namespace = "agentcore"
+  key       = "agentRuntimes"
+  value     = { "$name" = { arn = module.agent_core_runtime.agent_core_runtime_arn } }
+
+  depends_on = [module.agent_core_runtime]
+}\``;
+// Shared `AgentCoreRuntimeConfig` interface, vended into
+// `packages/common/agent-connection/src/core/runtime-config.ts`.
+const TS_RUNTIME_CONFIG_INTERFACE_PATTERN = `\`export interface AgentCoreRuntimeConfig {
+  agentRuntimes?: Record<string, string>;
+  gateways?: Record<string, string>;
+}\` => raw\`export interface AgentRuntimeEntry {
+  arn: string;
+  /** Session storage details. Only set when the agent has S3 session storage configured. */
+  session?: {
+    /** Name of the S3 bucket storing session data. */
+    bucketName: string;
+  };
+}
+
+export interface AgentCoreRuntimeConfig {
+  agentRuntimes?: Record<string, AgentRuntimeEntry>;
+  gateways?: Record<string, string>;
+}\``;
+// `config.agentRuntimes?.['Name']` as read by generated TS a2a/mcp clients and
+// the agent-chat CLI. Idempotent via the `not contains ?.arn` guard.
+const TS_CLIENT_ARN_PATTERNS = [
+    '`config.agentRuntimes?.[$name]` => `config.agentRuntimes?.[$name]?.arn` where { $program <: not contains `config.agentRuntimes?.[$name]?.arn` }',
+    '`config?.agentRuntimes?.[$name]` => `config?.agentRuntimes?.[$name]?.arn` where { $program <: not contains `config?.agentRuntimes?.[$name]?.arn` }'
+];
+// The agent-chat CLI's agentcore.ts calls `getAppConfig` directly rather than
+// the shared `AgentCoreRuntimeConfig` type, so its own inline cast needs the
+// same { arn } reshape TS_RUNTIME_CONFIG_INTERFACE_PATTERN gives the shared
+// interface.
+const AGENTCORE_CHAT_SCRIPT_TYPE_PATTERN = '`{ agentRuntimes?: Record<string, string> }` => `{ agentRuntimes?: Record<string, { arn: string }> }`';
+// Python equivalent (Strands + LangChain client templates).
+const PY_CLIENT_ARN_PATTERN = 'language python\n`agent_runtime_arn = config.get("agentRuntimes", {}).get($name)` => `agent_runtime = config.get("agentRuntimes", {}).get($name)\nagent_runtime_arn = agent_runtime.get("arn") if agent_runtime else None`';
+// Removes the imperative logModelErrors(agent)/logToolErrors(agent) calls.
+// Required for AG-UI (its adapter clones the template agent per-thread, and
+// hooks added directly to the template don't carry over — the equivalent
+// behaviour is restored via the AGUI_INDEX_* plugins below). For HTTP/A2A
+// it's a style unification, since the imperative form already worked there.
+const AGENT_TS_LOG_MODEL_CALL_PATTERN = '`logModelErrors(agent);` => .';
+const AGENT_TS_LOG_TOOL_CALL_PATTERN = '`logToolErrors(agent);` => .';
+// HTTP/A2A agents build one Agent per session, so wiring `sessionManager`
+// directly into the constructor is safe. Generic over `$props` so this works
+// regardless of customised systemPrompt/tools, as long as it's an inline
+// object literal. Prepended (not appended) to sidestep a trailing-comma
+// double-comma when `$props` is captured as raw source text.
+const AGENT_TS_SESSION_MANAGER_CONSTRUCTOR_PATTERN = '`new Agent({ $props })` => `new Agent({ sessionManager: await getSessionManager(), $props })` where { $props <: not contains `sessionManager` }';
+// Match-only gate for the rewrites above/below: only applies when the Agent
+// is an inline object literal, not e.g. a variable.
+const AGENT_TS_CONSTRUCTOR_MATCH_PATTERN = '`new Agent({ $props })`';
+// Moves the model/tool error hooks to Agent constructor plugins, mirroring
+// the AG-UI rewrite for consistency (HTTP/A2A already worked with the
+// imperative form).
+const AGENT_TS_PLUGINS_CONSTRUCTOR_PATTERN = '`new Agent({ $props })` => `new Agent({ plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()], $props })` where { $props <: not contains `plugins` }';
+// Captures the agent-connection package specifier so the new imports (and the
+// session.ts this migration creates) target the same module. Generic over
+// `$names` since a connection generator may have merged its own import here.
+const AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN = "`import { $names } from '$mod';` where { $names <: contains `logModelErrors`, $names <: contains `logToolErrors` }";
+// Removes `logModelErrors`/`logToolErrors` from the import, preserving any
+// other merged specifiers (or the whole statement if none remain). Decomposed
+// via `import_clause(name=named_imports($imports))` since a naive
+// `$rest`-based rewrite silently fails once 2+ other specifiers are merged in.
+const AGENT_TS_REMOVE_LOG_ERRORS_IMPORT_PATTERN = "`import $clause from '$mod';` as $import where { $clause <: import_clause(name=named_imports($imports)), $imports <: contains `logModelErrors`, $imports <: contains `logToolErrors`, if ($imports <: [`logModelErrors`, `logToolErrors`]) { $import => . } else { $imports <: some import_specifier(name=or { `logModelErrors`, `logToolErrors` }) => . } }";
+// Existing agents predate session.ts, so there's no prior storage to
+// preserve — default to in-memory.
+const legacySessionManagerContent = (agentConnectionModule, localSessionsDir)=>`import { SessionManager } from '@strands-agents/sdk';
+import { InMemoryStorage, LocalFileStorage } from '@strands-agents/sdk/storage';
+import { getCurrentSessionId } from '${agentConnectionModule}';
+
+/**
+ * Returns a SessionManager for persisting conversation state across
+ * invocations. Local development always uses local file storage for
+ * convenience, regardless of the configured session option. Without a
+ * configured session option, conversation state is kept in memory only and
+ * does not survive process restarts.
+ */
+export const getSessionManager = async (): Promise<SessionManager> => {
+  const sessionId = getCurrentSessionId();
+  if (!sessionId) {
+    throw new Error(
+      'No current session id — cannot resolve a SessionManager outside of a request scope.',
+    );
+  }
+  if (process.env.LOCAL_DEV === 'true') {
+    return new SessionManager({
+      sessionId,
+      storage: new LocalFileStorage('${localSessionsDir}'),
+    });
+  }
+  return new SessionManager({ sessionId, storage: new InMemoryStorage() });
+};
+`;
+/** The Nx project owning `dirPath`, if any (the longest-matching project.root). */ const findOwningProject = (tree, dirPath)=>{
+    let best;
+    for (const project of getProjects(tree).values()){
+        if ((dirPath === project.root || dirPath.startsWith(`${project.root}/`)) && (!best || project.root.length > best.root.length)) {
+            best = project;
+        }
+    }
+    return best;
+};
+/** This agent's own ComponentMetadata entry, as written by the ts#agent generator. */ const findAgentComponentMetadata = (project, dirRelativeToProjectRoot)=>project.metadata?.components?.find((component)=>component.generator === TS_AGENT_GENERATOR_INFO.id && component.path === dirRelativeToProjectRoot);
+/** This agent's real kebab-case name, from ComponentMetadata's `rc` (class-name) field. */ const agentTmpNameFor = (project, dir)=>{
+    const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);
+    const rc = findAgentComponentMetadata(project, dirRelativeToProjectRoot)?.rc;
+    return rc ? kebabCase(rc) : undefined;
+};
+/** This agent's protocol, from ComponentMetadata's `protocol` field. */ const agentProtocolFor = (project, dir)=>{
+    const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);
+    return findAgentComponentMetadata(project, dirRelativeToProjectRoot)?.protocol;
+};
+/** The relative path from `projectRoot` up to this agent's workspace-root-level local session storage. */ const localSessionsDirFor = (projectRoot, agentTmpName)=>joinPathFragments(getRelativePathToRootByDirectory(projectRoot), `tmp/agents/strands/${agentTmpName}`);
+// Wires ModelErrorLoggingPlugin/ToolErrorLoggingPlugin into the StrandsAgent
+// constructor, restoring the error-logging behaviour removed from agent.ts.
+const AGUI_INDEX_IMPORT_PATTERN = "`import { $names } from '$mod'` => `import { ModelErrorLoggingPlugin, ToolErrorLoggingPlugin, $names } from '$mod'` where { $names <: contains `runWithSessionId`, $names <: not contains `ModelErrorLoggingPlugin` }";
+// Match-only form of the target shape, used to gate before touching the
+// sibling agent.ts, and before committing to the rewrite here.
+const AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN = '`new StrandsAgent({ agent, name: $name, description: $desc })`';
+const AGUI_INDEX_CONSTRUCTOR_PATTERN = `${AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN} => \`new StrandsAgent({ agent, name: $name, description: $desc, plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()] })\``;
+const AGUI_INDEX_IMPORT_CAPTURE_PATTERN = "`import { $names } from '$mod';` where { $names <: contains `runWithSessionId` }";
+// AG-UI's StrandsAgent adapter needs a sessionManagerProvider (called once
+// per threadId), mirroring AGENT_TS_SESSION_MANAGER_CONSTRUCTOR_PATTERN.
+const aguiIndexSessionManagerImportPattern = (esm)=>`\`import { getAgent } from '$mod';\` => raw\`import { getAgent } from '$mod';
+import { getSessionManager } from './session${esm ? '.js' : ''}';\` where { $program <: not contains \`getSessionManager\` }`;
+// Anchored on the exact 4-property shape produced above, so this only fires
+// once plugins are wired in. Used both to gate and to rewrite.
+const AGUI_INDEX_SESSION_MANAGER_MATCH_PATTERN = '`new StrandsAgent({ agent, name: $name, description: $desc, plugins: $plugins })`';
+const AGUI_INDEX_SESSION_MANAGER_CONSTRUCTOR_PATTERN = `${AGUI_INDEX_SESSION_MANAGER_MATCH_PATTERN} => \`new StrandsAgent({ agent, name: $name, description: $desc, plugins: $plugins, config: { sessionManagerProvider: getSessionManager } })\``;
+// The plugin classes referenced above only exist once model-errors-strands.ts
+// / tool-errors-strands.ts are converted from their plain logXErrors function
+// form (see checkPluginClass/applyPluginClass below). Anchored on the
+// `agent.addHook(...)` call so this doesn't re-match its own output.
+const MODEL_ERRORS_IMPORT_PATTERN = "`import { AfterModelCallEvent, type LocalAgent } from '@strands-agents/sdk';` => `import { AfterModelCallEvent, type LocalAgent, type Plugin } from '@strands-agents/sdk';`";
+// Match-only form of the target shape, checked before committing to either
+// this or the sibling tool-errors conversion — see checkPluginClass.
+const MODEL_ERRORS_CLASS_MATCH_PATTERN = '`export const logModelErrors = (agent: LocalAgent): void => { agent.addHook(AfterModelCallEvent, $callback); };`';
+const MODEL_ERRORS_CLASS_PATTERN = `${MODEL_ERRORS_CLASS_MATCH_PATTERN} => raw\`export class ModelErrorLoggingPlugin implements Plugin {\n  readonly name = 'model-error-logging';\n\n  initAgent(agent: LocalAgent): void {\n    agent.addHook(AfterModelCallEvent, $callback);\n  }\n}\``;
+const TOOL_ERRORS_IMPORT_PATTERN = "`import { AfterToolCallEvent, type LocalAgent, TextBlock } from '@strands-agents/sdk';` => `import { AfterToolCallEvent, type LocalAgent, type Plugin, TextBlock } from '@strands-agents/sdk';`";
+const TOOL_ERRORS_CLASS_MATCH_PATTERN = '`export const logToolErrors = (agent: LocalAgent): void => { agent.addHook(AfterToolCallEvent, $callback); };`';
+const TOOL_ERRORS_CLASS_PATTERN = `${TOOL_ERRORS_CLASS_MATCH_PATTERN} => raw\`export class ToolErrorLoggingPlugin implements Plugin {\n  readonly name = 'tool-error-logging';\n\n  initAgent(agent: LocalAgent): void {\n    agent.addHook(AfterToolCallEvent, $callback);\n  }\n}\``;
+/**
+ * Checks (without writing) whether `filePath` still needs converting from its
+ * plain `logXErrors` function to `className`, and whether the conversion
+ * pattern would actually apply. `'done'` covers both "already converted" and
+ * "file doesn't exist" — either way there's nothing to do and nothing
+ * blocking the sibling conversion.
+ */ async function checkPluginClass(tree, filePath, className, classMatchPattern) {
+    if (!tree.exists(filePath)) {
+        return 'done';
+    }
+    if ((tree.read(filePath, 'utf-8') ?? '').includes(`class ${className}`)) {
+        return 'done';
+    }
+    return await matchGritQL(tree, filePath, classMatchPattern) ? 'ready' : 'diverged';
+}
+/** Applies the class + import rewrite for a file already confirmed `'ready'`. */ async function applyPluginClass(tree, filePath, importPattern, classPattern) {
+    // The class rewrite is applied first, and the import only added once it's
+    // confirmed to have actually happened here — not because either could fail
+    // at this point (checkPluginClass already confirmed the pattern matches),
+    // but to keep this file's own edit order self-evidently safe to read.
+    await applyGritQL(tree, filePath, classPattern);
+    await applyGritQL(tree, filePath, importPattern);
+}
+export default async function migration(tree) {
+    const nextSteps = [];
+    let anyChanges = false;
+    // Both classes live in one shared agent-connection package, so their
+    // readiness is resolved once up front rather than per agent.
+    const modelErrorsPath = joinPathFragments(AGENT_CONNECTION_PROJECT_DIR, 'src', 'core', 'model-errors-strands.ts');
+    const toolErrorsPath = joinPathFragments(AGENT_CONNECTION_PROJECT_DIR, 'src', 'core', 'tool-errors-strands.ts');
+    // Checked together, before writing either: all-or-nothing, so a run either
+    // converts both or leaves both untouched — never just one.
+    const modelPluginStatus = await checkPluginClass(tree, modelErrorsPath, 'ModelErrorLoggingPlugin', MODEL_ERRORS_CLASS_MATCH_PATTERN);
+    const toolPluginStatus = await checkPluginClass(tree, toolErrorsPath, 'ToolErrorLoggingPlugin', TOOL_ERRORS_CLASS_MATCH_PATTERN);
+    const pluginsReady = modelPluginStatus !== 'diverged' && toolPluginStatus !== 'diverged';
+    if (pluginsReady) {
+        if (modelPluginStatus === 'ready') {
+            await applyPluginClass(tree, modelErrorsPath, MODEL_ERRORS_IMPORT_PATTERN, MODEL_ERRORS_CLASS_PATTERN);
+        }
+        if (toolPluginStatus === 'ready') {
+            await applyPluginClass(tree, toolErrorsPath, TOOL_ERRORS_IMPORT_PATTERN, TOOL_ERRORS_CLASS_PATTERN);
+        }
+    } else {
+        if (modelPluginStatus === 'diverged') {
+            nextSteps.push(`${modelErrorsPath}: diverged from the generated shape - left as-is. Manually convert it to a \`ModelErrorLoggingPlugin\` class implementing \`Plugin\` (see the ts#agent generator's template); agent.ts/index.ts files that reference it are left unmigrated until both this and ${toolErrorsPath} are converted.`);
+        }
+        if (toolPluginStatus === 'diverged') {
+            nextSteps.push(`${toolErrorsPath}: diverged from the generated shape - left as-is. Manually convert it to a \`ToolErrorLoggingPlugin\` class implementing \`Plugin\` (see the ts#agent generator's template); agent.ts/index.ts files that reference it are left unmigrated until both this and ${modelErrorsPath} are converted.`);
+        }
+    }
+    const filePaths = [];
+    visitNotIgnoredFiles(tree, '', (filePath)=>filePaths.push(filePath));
+    for (const filePath of filePaths){
+        if (filePath === modelErrorsPath || filePath === toolErrorsPath) {
+            continue;
+        }
+        if (filePath.endsWith('.d.ts')) {
+            continue;
+        }
+        // session.ts already targets the new { arn, session } shape via
+        // `config.agentRuntimes?.[$name]?.session?.bucketName` — the generic
+        // reshape branch below would otherwise corrupt that into `?.arn?.session?.bucketName`.
+        if (filePath.endsWith('/session.ts')) {
+            continue;
+        }
+        if (filePath.endsWith('/agent.ts')) {
+            const dir = filePath.split('/').slice(0, -1).join('/');
+            const project = findOwningProject(tree, dir);
+            // Without a registered project there's no ComponentMetadata, so this
+            // agent's protocol/name/agent-connection module can't be resolved — and
+            // a pre-migration AG-UI agent.ts is textually identical to an
+            // HTTP/A2A one, so there's nothing left to safely guess from.
+            if (!project) {
+                if (await matchGritQL(tree, filePath, AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN)) {
+                    nextSteps.push(`${filePath}: could not determine the project root — manually verify whether this is an AG-UI or HTTP/A2A agent and complete its session-manager/error-logging-plugin migration accordingly (see the ts#agent generator's template).`);
+                }
+                continue;
+            }
+            const protocol = agentProtocolFor(project, dir);
+            if (protocol === 'ag-ui') {
+                const hasLogCalls = await matchGritQL(tree, filePath, AGENT_TS_LOG_MODEL_CALL_PATTERN) || await matchGritQL(tree, filePath, AGENT_TS_LOG_TOOL_CALL_PATTERN);
+                if (!hasLogCalls) {
+                    continue;
+                }
+                // The sibling index.ts restores error-logging via StrandsAgent
+                // plugins — removing the calls here without that succeeding would
+                // drop error logging rather than move it.
+                const indexPath = `${dir}/index.ts`;
+                const indexReady = pluginsReady && await matchGritQL(tree, indexPath, AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN);
+                if (!indexReady) {
+                    nextSteps.push(`${filePath}: left the logModelErrors/logToolErrors calls in place — ${indexPath} could not be automatically wired with the equivalent StrandsAgent plugins. Move error logging to plugins there manually (see the ts#agent generator's template), then remove these calls.`);
+                    continue;
+                }
+                const rewroteImport = await applyGritQL(tree, filePath, AGENT_TS_REMOVE_LOG_ERRORS_IMPORT_PATTERN);
+                const rewroteModelCall = await applyGritQL(tree, filePath, AGENT_TS_LOG_MODEL_CALL_PATTERN);
+                const rewroteToolCall = await applyGritQL(tree, filePath, AGENT_TS_LOG_TOOL_CALL_PATTERN);
+                if (rewroteImport || rewroteModelCall || rewroteToolCall) {
+                    anyChanges = true;
+                }
+                continue;
+            }
+            if (!protocol) {
+                if (await matchGritQL(tree, filePath, AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN)) {
+                    nextSteps.push(`${filePath}: could not determine this agent's protocol from its ComponentMetadata — manually verify whether it's AG-UI or HTTP/A2A and wire sessionManager/the error-logging plugins in accordingly (see the ts#agent generator's template).`);
+                }
+                continue;
+            }
+            // HTTP/A2A: wire sessionManager into the Agent constructor and create
+            // the sibling session.ts if it doesn't exist yet.
+            if (!await matchGritQL(tree, filePath, AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN)) {
+                continue;
+            }
+            const sessionPath = `${dir}/session.ts`;
+            // Checked before any writes: if the plugin classes aren't available, or
+            // the Agent isn't an inline object literal, leave the file untouched
+            // entirely rather than ending up half-migrated.
+            if (!pluginsReady) {
+                nextSteps.push(`${filePath}: left as-is — ModelErrorLoggingPlugin/ToolErrorLoggingPlugin are not available yet (see the model-errors-strands.ts/tool-errors-strands.ts next step above). Wire sessionManager and the plugins in manually once those classes exist.`);
+                continue;
+            }
+            if (!await matchGritQL(tree, filePath, AGENT_TS_CONSTRUCTOR_MATCH_PATTERN)) {
+                nextSteps.push(`${filePath}: the Agent is not constructed with an inline object literal (\`new Agent({ ... })\`), so sessionManager and the error-logging plugins could not be wired in automatically. Manually add \`sessionManager: await getSessionManager()\` and \`plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()]\` to its constructor (see the ts#agent generator's template), creating ${sessionPath} first if it doesn't already exist.`);
+                continue;
+            }
+            if (!tree.exists(sessionPath)) {
+                const mod = await captureGritQLVariable(tree, filePath, AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN, 'mod');
+                const agentName = agentTmpNameFor(project, dir);
+                if (mod && agentName) {
+                    tree.write(sessionPath, legacySessionManagerContent(mod, localSessionsDirFor(project.root, agentName)));
+                } else {
+                    nextSteps.push(`${filePath}: could not determine the agent-connection module or this agent's name from its ComponentMetadata — manually create a session.ts (see the ts#agent generator's template) and wire \`sessionManager: await getSessionManager()\` into the Agent constructor.`);
+                    continue;
+                }
+            }
+            // Merges the plugin imports into the existing logModelErrors/
+            // logToolErrors import clause (rather than a redundant second import),
+            // and adds getSessionManager as its own line. The old specifiers are
+            // stripped separately below, after this rewrite anchors on them.
+            const esm = isEsmWorkspace(tree);
+            await applyGritQL(tree, filePath, `\`import { $names } from '$mod';\` => raw\`import { ModelErrorLoggingPlugin, ToolErrorLoggingPlugin, $names } from '$mod';
+import { getSessionManager } from './session${esm ? '.js' : ''}';\` where { $names <: contains \`logModelErrors\`, $names <: contains \`logToolErrors\`, $program <: not contains \`getSessionManager\` }`);
+            // Only rewrite the constructor once the import is confirmed present.
+            if (!(tree.read(filePath, 'utf-8') ?? '').includes('getSessionManager')) {
+                nextSteps.push(`${filePath}: found ${sessionPath} but couldn't confirm the getSessionManager import — wire \`sessionManager: await getSessionManager()\` into the Agent constructor manually.`);
+                continue;
+            }
+            await applyGritQL(tree, filePath, AGENT_TS_LOG_MODEL_CALL_PATTERN);
+            await applyGritQL(tree, filePath, AGENT_TS_LOG_TOOL_CALL_PATTERN);
+            await applyGritQL(tree, filePath, AGENT_TS_REMOVE_LOG_ERRORS_IMPORT_PATTERN);
+            const rewrotePlugins = await applyGritQL(tree, filePath, AGENT_TS_PLUGINS_CONSTRUCTOR_PATTERN);
+            const rewroteConstructor = await applyGritQL(tree, filePath, AGENT_TS_SESSION_MANAGER_CONSTRUCTOR_PATTERN);
+            if (rewrotePlugins || rewroteConstructor) {
+                anyChanges = true;
+            }
+            continue;
+        }
+        if (filePath.endsWith('/index.ts') && (tree.read(filePath, 'utf-8') ?? '').includes('@ag-ui/aws-strands')) {
+            const alreadyWired = (tree.read(filePath, 'utf-8') ?? '').includes('ModelErrorLoggingPlugin');
+            if (!alreadyWired) {
+                // Import and constructor only apply cleanly together — adding the
+                // import while the constructor can't be rewritten would leave an
+                // unused import and no plugins wired in.
+                const needsPluginWiring = await matchGritQL(tree, filePath, AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN);
+                if (!pluginsReady) {
+                    if (needsPluginWiring) {
+                        nextSteps.push(`${filePath}: left the StrandsAgent constructor as-is — ModelErrorLoggingPlugin/ToolErrorLoggingPlugin are not available yet (see the model-errors-strands.ts/tool-errors-strands.ts next step above). Wire them in manually once those classes exist.`);
+                    }
+                    continue;
+                }
+                if (!needsPluginWiring) {
+                    nextSteps.push(`${filePath}: the StrandsAgent constructor has diverged from the generated shape - left as-is. Manually add \`plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()]\` to it (see the ts#agent generator's template).`);
+                    continue;
+                }
+                const rewroteImport = await applyGritQL(tree, filePath, AGUI_INDEX_IMPORT_PATTERN);
+                const rewroteConstructor = await applyGritQL(tree, filePath, AGUI_INDEX_CONSTRUCTOR_PATTERN);
+                if (rewroteImport || rewroteConstructor) {
+                    anyChanges = true;
+                }
+            }
+            // Create the sibling session.ts (once plugins are present) and wire in
+            // sessionManagerProvider. Gated on the exact post-plugins shape, so a
+            // customised constructor is left alone rather than creating an unused
+            // session.ts.
+            if (!await matchGritQL(tree, filePath, AGUI_INDEX_SESSION_MANAGER_MATCH_PATTERN)) {
+                continue;
+            }
+            const dir = filePath.split('/').slice(0, -1).join('/');
+            const sessionPath = `${dir}/session.ts`;
+            if (!tree.exists(sessionPath)) {
+                const project = findOwningProject(tree, dir);
+                const mod = await captureGritQLVariable(tree, filePath, AGUI_INDEX_IMPORT_CAPTURE_PATTERN, 'mod');
+                const agentName = project && agentTmpNameFor(project, dir);
+                if (project && mod && agentName) {
+                    tree.write(sessionPath, legacySessionManagerContent(mod, localSessionsDirFor(project.root, agentName)));
+                } else {
+                    nextSteps.push(`${filePath}: could not determine the project root, agent-connection module, or this agent's name from its ComponentMetadata — manually create a session.ts (see the ts#agent generator's template) and wire \`config: { sessionManagerProvider: getSessionManager }\` into the StrandsAgent constructor.`);
+                    continue;
+                }
+            }
+            const esm = isEsmWorkspace(tree);
+            await applyGritQL(tree, filePath, aguiIndexSessionManagerImportPattern(esm));
+            if (!(tree.read(filePath, 'utf-8') ?? '').includes('getSessionManager')) {
+                nextSteps.push(`${filePath}: found ${sessionPath} but couldn't confirm the getSessionManager import — wire \`config: { sessionManagerProvider: getSessionManager }\` into the StrandsAgent constructor manually.`);
+                continue;
+            }
+            const rewroteSessionManagerConstructor = await applyGritQL(tree, filePath, AGUI_INDEX_SESSION_MANAGER_CONSTRUCTOR_PATTERN);
+            if (rewroteSessionManagerConstructor) {
+                anyChanges = true;
+            }
+            continue;
+        }
+        if (filePath.endsWith('.ts') || filePath.endsWith('.tsx')) {
+            const contents = tree.read(filePath, 'utf-8') ?? '';
+            if (!contents.includes('agentRuntimes')) {
+                continue;
+            }
+            // Only the 'agentcore' namespace is reshaped — see CDK_RC_SET_PATTERN.
+            const rewroteAgentcoreRcSet = await applyGritQL(tree, filePath, CDK_RC_SET_PATTERN);
+            const rewroteInterface = await applyGritQL(tree, filePath, TS_RUNTIME_CONFIG_INTERFACE_PATTERN);
+            let rewroteClientArn = false;
+            for (const pattern of TS_CLIENT_ARN_PATTERNS){
+                rewroteClientArn = await applyGritQL(tree, filePath, pattern) ? true : rewroteClientArn;
+            }
+            const rewroteChatScriptType = await applyGritQL(tree, filePath, AGENTCORE_CHAT_SCRIPT_TYPE_PATTERN);
+            if (rewroteAgentcoreRcSet || rewroteInterface || rewroteClientArn || rewroteChatScriptType) {
+                anyChanges = true;
+            }
+        } else if (filePath.endsWith('.tf')) {
+            const contents = tree.read(filePath, 'utf-8') ?? '';
+            if (!contents.includes('"agentRuntimes"')) {
+                continue;
+            }
+            const rewrote = await applyGritQL(tree, filePath, TF_VALUE_PATTERN);
+            if (rewrote) {
+                anyChanges = true;
+            }
+        } else if (filePath.endsWith('.py')) {
+            const contents = tree.read(filePath, 'utf-8') ?? '';
+            if (!contents.includes('agentRuntimes')) {
+                continue;
+            }
+            const rewrote = await applyGritQL(tree, filePath, PY_CLIENT_ARN_PATTERN);
+            if (rewrote) {
+                anyChanges = true;
+            }
+        }
+    }
+    if (anyChanges) {
+        nextSteps.push('If you have custom code reading the runtime config `agentcore.agentRuntimes` entries ' + 'as plain ARN strings (e.g. custom Lambda handlers or agent code), update it to read ' + 'the ARN from `.arn` on the new `{ arn }` shape.');
+    }
+    await formatFilesInSubtree(tree);
+    return {
+        nextSteps
+    };
+}
+
+//# sourceMappingURL=migration.js.map

package/src/migrations/v1.0.0-rc.57/0001-ts-agent-session-management-support/migration.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/migrations/v1.0.0-rc.57/0001-ts-agent-session-management-support/migration.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  getProjects,\n  joinPathFragments,\n  type MigrationReturnObject,\n  type ProjectConfiguration,\n  type Tree,\n  visitNotIgnoredFiles,\n} from '@nx/devkit';\nimport { TS_AGENT_GENERATOR_INFO } from '../../../ts/agent/generator';\nimport { AGENT_CONNECTION_PROJECT_DIR } from '../../../utils/agent-connection/agent-connection';\nimport {\n  applyGritQL,\n  captureGritQLVariable,\n  matchGritQL,\n} from '../../../utils/ast';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { isEsmWorkspace } from '../../../utils/module-format';\nimport { kebabCase } from '../../../utils/names';\nimport type { ComponentMetadata } from '../../../utils/nx';\nimport {\n  getRelativePathToRootByDirectory,\n  toProjectRelativePath,\n} from '../../../utils/paths';\n\n/**\n * Add session management support to ts#agent.\n *\n * How to write a migration:\n * - https://nx.dev/docs/kb/migration-generators\n * - What `nextSteps` means: https://nx.dev/docs/reference/devkit/MigrationReturnObject\n *\n * Guardrails:\n * - Pattern-match before writing: skip files that have diverged from the shape\n *   your generators produce and report them via `nextSteps`, or consider a\n *   hybrid migration, rather than clobbering the user's changes.\n * - Idempotent: re-running must be a no-op.\n * - Format what you write: finish with `formatFilesInSubtree` so the files your\n *   migration wrote are formatted correctly.\n */\n\n// Reshapes an `agentRuntimes` rc.set entry from a bare ARN to `{ arn }`. Only\n// the 'agentcore' namespace — 'connection' (client-facing) stays a string,\n// since it shouldn't carry the session bucket name to the browser. No\n// `session` field added here; a fresh session.ts (created below) defaults to\n// in-memory. Idempotent: only matches the pre-migration shape.\nconst CDK_RC_SET_PATTERN = `\\`rc.set('agentcore', 'agentRuntimes', { ...rc.get('agentcore').agentRuntimes, $name: this.agentCoreRuntime.agentRuntimeArn });\\` => raw\\`rc.set('agentcore', 'agentRuntimes', {\n  ...rc.get('agentcore').agentRuntimes,\n  $name: {\n    arn: this.agentCoreRuntime.agentRuntimeArn,\n  },\n});\\``;\n\n// Terraform equivalent, anchored on the module's fixed name/shape so it\n// doesn't also match the 'connection' module's identical-looking `value` line.\nconst TF_VALUE_PATTERN = `language hcl\\n\\`module \"add_agent_runtime_to_runtime_config\" {\n  source = \"../../../core/runtime-config/entry\"\n\n  namespace = \"agentcore\"\n  key       = \"agentRuntimes\"\n  value     = { \"$name\" = module.agent_core_runtime.agent_core_runtime_arn }\n\n  depends_on = [module.agent_core_runtime]\n}\\` => \\`module \"add_agent_runtime_to_runtime_config\" {\n  source = \"../../../core/runtime-config/entry\"\n\n  namespace = \"agentcore\"\n  key       = \"agentRuntimes\"\n  value     = { \"$name\" = { arn = module.agent_core_runtime.agent_core_runtime_arn } }\n\n  depends_on = [module.agent_core_runtime]\n}\\``;\n\n// Shared `AgentCoreRuntimeConfig` interface, vended into\n// `packages/common/agent-connection/src/core/runtime-config.ts`.\nconst TS_RUNTIME_CONFIG_INTERFACE_PATTERN = `\\`export interface AgentCoreRuntimeConfig {\n  agentRuntimes?: Record<string, string>;\n  gateways?: Record<string, string>;\n}\\` => raw\\`export interface AgentRuntimeEntry {\n  arn: string;\n  /** Session storage details. Only set when the agent has S3 session storage configured. */\n  session?: {\n    /** Name of the S3 bucket storing session data. */\n    bucketName: string;\n  };\n}\n\nexport interface AgentCoreRuntimeConfig {\n  agentRuntimes?: Record<string, AgentRuntimeEntry>;\n  gateways?: Record<string, string>;\n}\\``;\n\n// `config.agentRuntimes?.['Name']` as read by generated TS a2a/mcp clients and\n// the agent-chat CLI. Idempotent via the `not contains ?.arn` guard.\nconst TS_CLIENT_ARN_PATTERNS = [\n  '`config.agentRuntimes?.[$name]` => `config.agentRuntimes?.[$name]?.arn` where { $program <: not contains `config.agentRuntimes?.[$name]?.arn` }',\n  '`config?.agentRuntimes?.[$name]` => `config?.agentRuntimes?.[$name]?.arn` where { $program <: not contains `config?.agentRuntimes?.[$name]?.arn` }',\n];\n\n// The agent-chat CLI's agentcore.ts calls `getAppConfig` directly rather than\n// the shared `AgentCoreRuntimeConfig` type, so its own inline cast needs the\n// same { arn } reshape TS_RUNTIME_CONFIG_INTERFACE_PATTERN gives the shared\n// interface.\nconst AGENTCORE_CHAT_SCRIPT_TYPE_PATTERN =\n  '`{ agentRuntimes?: Record<string, string> }` => `{ agentRuntimes?: Record<string, { arn: string }> }`';\n\n// Python equivalent (Strands + LangChain client templates).\nconst PY_CLIENT_ARN_PATTERN =\n  'language python\\n`agent_runtime_arn = config.get(\"agentRuntimes\", {}).get($name)` => `agent_runtime = config.get(\"agentRuntimes\", {}).get($name)\\nagent_runtime_arn = agent_runtime.get(\"arn\") if agent_runtime else None`';\n\n// Removes the imperative logModelErrors(agent)/logToolErrors(agent) calls.\n// Required for AG-UI (its adapter clones the template agent per-thread, and\n// hooks added directly to the template don't carry over — the equivalent\n// behaviour is restored via the AGUI_INDEX_* plugins below). For HTTP/A2A\n// it's a style unification, since the imperative form already worked there.\nconst AGENT_TS_LOG_MODEL_CALL_PATTERN = '`logModelErrors(agent);` => .';\nconst AGENT_TS_LOG_TOOL_CALL_PATTERN = '`logToolErrors(agent);` => .';\n\n// HTTP/A2A agents build one Agent per session, so wiring `sessionManager`\n// directly into the constructor is safe. Generic over `$props` so this works\n// regardless of customised systemPrompt/tools, as long as it's an inline\n// object literal. Prepended (not appended) to sidestep a trailing-comma\n// double-comma when `$props` is captured as raw source text.\nconst AGENT_TS_SESSION_MANAGER_CONSTRUCTOR_PATTERN =\n  '`new Agent({ $props })` => `new Agent({ sessionManager: await getSessionManager(), $props })` where { $props <: not contains `sessionManager` }';\n\n// Match-only gate for the rewrites above/below: only applies when the Agent\n// is an inline object literal, not e.g. a variable.\nconst AGENT_TS_CONSTRUCTOR_MATCH_PATTERN = '`new Agent({ $props })`';\n\n// Moves the model/tool error hooks to Agent constructor plugins, mirroring\n// the AG-UI rewrite for consistency (HTTP/A2A already worked with the\n// imperative form).\nconst AGENT_TS_PLUGINS_CONSTRUCTOR_PATTERN =\n  '`new Agent({ $props })` => `new Agent({ plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()], $props })` where { $props <: not contains `plugins` }';\n\n// Captures the agent-connection package specifier so the new imports (and the\n// session.ts this migration creates) target the same module. Generic over\n// `$names` since a connection generator may have merged its own import here.\nconst AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN =\n  \"`import { $names } from '$mod';` where { $names <: contains `logModelErrors`, $names <: contains `logToolErrors` }\";\n\n// Removes `logModelErrors`/`logToolErrors` from the import, preserving any\n// other merged specifiers (or the whole statement if none remain). Decomposed\n// via `import_clause(name=named_imports($imports))` since a naive\n// `$rest`-based rewrite silently fails once 2+ other specifiers are merged in.\nconst AGENT_TS_REMOVE_LOG_ERRORS_IMPORT_PATTERN =\n  \"`import $clause from '$mod';` as $import where { $clause <: import_clause(name=named_imports($imports)), $imports <: contains `logModelErrors`, $imports <: contains `logToolErrors`, if ($imports <: [`logModelErrors`, `logToolErrors`]) { $import => . } else { $imports <: some import_specifier(name=or { `logModelErrors`, `logToolErrors` }) => . } }\";\n\n// Existing agents predate session.ts, so there's no prior storage to\n// preserve — default to in-memory.\nconst legacySessionManagerContent = (\n  agentConnectionModule: string,\n  localSessionsDir: string,\n): string => `import { SessionManager } from '@strands-agents/sdk';\nimport { InMemoryStorage, LocalFileStorage } from '@strands-agents/sdk/storage';\nimport { getCurrentSessionId } from '${agentConnectionModule}';\n\n/**\n * Returns a SessionManager for persisting conversation state across\n * invocations. Local development always uses local file storage for\n * convenience, regardless of the configured session option. Without a\n * configured session option, conversation state is kept in memory only and\n * does not survive process restarts.\n */\nexport const getSessionManager = async (): Promise<SessionManager> => {\n  const sessionId = getCurrentSessionId();\n  if (!sessionId) {\n    throw new Error(\n      'No current session id — cannot resolve a SessionManager outside of a request scope.',\n    );\n  }\n  if (process.env.LOCAL_DEV === 'true') {\n    return new SessionManager({\n      sessionId,\n      storage: new LocalFileStorage('${localSessionsDir}'),\n    });\n  }\n  return new SessionManager({ sessionId, storage: new InMemoryStorage() });\n};\n`;\n\n/** The Nx project owning `dirPath`, if any (the longest-matching project.root). */\nconst findOwningProject = (\n  tree: Tree,\n  dirPath: string,\n): ProjectConfiguration | undefined => {\n  let best: ProjectConfiguration | undefined;\n  for (const project of getProjects(tree).values()) {\n    if (\n      (dirPath === project.root || dirPath.startsWith(`${project.root}/`)) &&\n      (!best || project.root.length > best.root.length)\n    ) {\n      best = project;\n    }\n  }\n  return best;\n};\n\n/** This agent's own ComponentMetadata entry, as written by the ts#agent generator. */\nconst findAgentComponentMetadata = (\n  project: ProjectConfiguration,\n  dirRelativeToProjectRoot: string,\n): ComponentMetadata | undefined =>\n  (project.metadata as { components?: ComponentMetadata[] })?.components?.find(\n    (component) =>\n      component.generator === TS_AGENT_GENERATOR_INFO.id &&\n      component.path === dirRelativeToProjectRoot,\n  );\n\n/** This agent's real kebab-case name, from ComponentMetadata's `rc` (class-name) field. */\nconst agentTmpNameFor = (\n  project: ProjectConfiguration,\n  dir: string,\n): string | undefined => {\n  const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);\n  const rc = findAgentComponentMetadata(project, dirRelativeToProjectRoot)?.rc;\n  return rc ? kebabCase(rc) : undefined;\n};\n\n/** This agent's protocol, from ComponentMetadata's `protocol` field. */\nconst agentProtocolFor = (\n  project: ProjectConfiguration,\n  dir: string,\n): string | undefined => {\n  const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);\n  return findAgentComponentMetadata(project, dirRelativeToProjectRoot)\n    ?.protocol;\n};\n\n/** The relative path from `projectRoot` up to this agent's workspace-root-level local session storage. */\nconst localSessionsDirFor = (\n  projectRoot: string,\n  agentTmpName: string,\n): string =>\n  joinPathFragments(\n    getRelativePathToRootByDirectory(projectRoot),\n    `tmp/agents/strands/${agentTmpName}`,\n  );\n\n// Wires ModelErrorLoggingPlugin/ToolErrorLoggingPlugin into the StrandsAgent\n// constructor, restoring the error-logging behaviour removed from agent.ts.\nconst AGUI_INDEX_IMPORT_PATTERN =\n  \"`import { $names } from '$mod'` => `import { ModelErrorLoggingPlugin, ToolErrorLoggingPlugin, $names } from '$mod'` where { $names <: contains `runWithSessionId`, $names <: not contains `ModelErrorLoggingPlugin` }\";\n// Match-only form of the target shape, used to gate before touching the\n// sibling agent.ts, and before committing to the rewrite here.\nconst AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN =\n  '`new StrandsAgent({ agent, name: $name, description: $desc })`';\nconst AGUI_INDEX_CONSTRUCTOR_PATTERN = `${AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN} => \\`new StrandsAgent({ agent, name: $name, description: $desc, plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()] })\\``;\n\nconst AGUI_INDEX_IMPORT_CAPTURE_PATTERN =\n  \"`import { $names } from '$mod';` where { $names <: contains `runWithSessionId` }\";\n\n// AG-UI's StrandsAgent adapter needs a sessionManagerProvider (called once\n// per threadId), mirroring AGENT_TS_SESSION_MANAGER_CONSTRUCTOR_PATTERN.\nconst aguiIndexSessionManagerImportPattern = (esm: boolean) =>\n  `\\`import { getAgent } from '$mod';\\` => raw\\`import { getAgent } from '$mod';\nimport { getSessionManager } from './session${esm ? '.js' : ''}';\\` where { $program <: not contains \\`getSessionManager\\` }`;\n// Anchored on the exact 4-property shape produced above, so this only fires\n// once plugins are wired in. Used both to gate and to rewrite.\nconst AGUI_INDEX_SESSION_MANAGER_MATCH_PATTERN =\n  '`new StrandsAgent({ agent, name: $name, description: $desc, plugins: $plugins })`';\nconst AGUI_INDEX_SESSION_MANAGER_CONSTRUCTOR_PATTERN = `${AGUI_INDEX_SESSION_MANAGER_MATCH_PATTERN} => \\`new StrandsAgent({ agent, name: $name, description: $desc, plugins: $plugins, config: { sessionManagerProvider: getSessionManager } })\\``;\n\n// The plugin classes referenced above only exist once model-errors-strands.ts\n// / tool-errors-strands.ts are converted from their plain logXErrors function\n// form (see checkPluginClass/applyPluginClass below). Anchored on the\n// `agent.addHook(...)` call so this doesn't re-match its own output.\nconst MODEL_ERRORS_IMPORT_PATTERN =\n  \"`import { AfterModelCallEvent, type LocalAgent } from '@strands-agents/sdk';` => `import { AfterModelCallEvent, type LocalAgent, type Plugin } from '@strands-agents/sdk';`\";\n// Match-only form of the target shape, checked before committing to either\n// this or the sibling tool-errors conversion — see checkPluginClass.\nconst MODEL_ERRORS_CLASS_MATCH_PATTERN =\n  '`export const logModelErrors = (agent: LocalAgent): void => { agent.addHook(AfterModelCallEvent, $callback); };`';\nconst MODEL_ERRORS_CLASS_PATTERN = `${MODEL_ERRORS_CLASS_MATCH_PATTERN} => raw\\`export class ModelErrorLoggingPlugin implements Plugin {\\n  readonly name = 'model-error-logging';\\n\\n  initAgent(agent: LocalAgent): void {\\n    agent.addHook(AfterModelCallEvent, $callback);\\n  }\\n}\\``;\n\nconst TOOL_ERRORS_IMPORT_PATTERN =\n  \"`import { AfterToolCallEvent, type LocalAgent, TextBlock } from '@strands-agents/sdk';` => `import { AfterToolCallEvent, type LocalAgent, type Plugin, TextBlock } from '@strands-agents/sdk';`\";\nconst TOOL_ERRORS_CLASS_MATCH_PATTERN =\n  '`export const logToolErrors = (agent: LocalAgent): void => { agent.addHook(AfterToolCallEvent, $callback); };`';\nconst TOOL_ERRORS_CLASS_PATTERN = `${TOOL_ERRORS_CLASS_MATCH_PATTERN} => raw\\`export class ToolErrorLoggingPlugin implements Plugin {\\n  readonly name = 'tool-error-logging';\\n\\n  initAgent(agent: LocalAgent): void {\\n    agent.addHook(AfterToolCallEvent, $callback);\\n  }\\n}\\``;\n\ntype PluginClassStatus = 'ready' | 'done' | 'diverged';\n\n/**\n * Checks (without writing) whether `filePath` still needs converting from its\n * plain `logXErrors` function to `className`, and whether the conversion\n * pattern would actually apply. `'done'` covers both \"already converted\" and\n * \"file doesn't exist\" — either way there's nothing to do and nothing\n * blocking the sibling conversion.\n */\nasync function checkPluginClass(\n  tree: Tree,\n  filePath: string,\n  className: string,\n  classMatchPattern: string,\n): Promise<PluginClassStatus> {\n  if (!tree.exists(filePath)) {\n    return 'done';\n  }\n  if ((tree.read(filePath, 'utf-8') ?? '').includes(`class ${className}`)) {\n    return 'done';\n  }\n  return (await matchGritQL(tree, filePath, classMatchPattern))\n    ? 'ready'\n    : 'diverged';\n}\n\n/** Applies the class + import rewrite for a file already confirmed `'ready'`. */\nasync function applyPluginClass(\n  tree: Tree,\n  filePath: string,\n  importPattern: string,\n  classPattern: string,\n): Promise<void> {\n  // The class rewrite is applied first, and the import only added once it's\n  // confirmed to have actually happened here — not because either could fail\n  // at this point (checkPluginClass already confirmed the pattern matches),\n  // but to keep this file's own edit order self-evidently safe to read.\n  await applyGritQL(tree, filePath, classPattern);\n  await applyGritQL(tree, filePath, importPattern);\n}\n\nexport default async function migration(\n  tree: Tree,\n): Promise<MigrationReturnObject> {\n  const nextSteps: string[] = [];\n  let anyChanges = false;\n\n  // Both classes live in one shared agent-connection package, so their\n  // readiness is resolved once up front rather than per agent.\n  const modelErrorsPath = joinPathFragments(\n    AGENT_CONNECTION_PROJECT_DIR,\n    'src',\n    'core',\n    'model-errors-strands.ts',\n  );\n  const toolErrorsPath = joinPathFragments(\n    AGENT_CONNECTION_PROJECT_DIR,\n    'src',\n    'core',\n    'tool-errors-strands.ts',\n  );\n  // Checked together, before writing either: all-or-nothing, so a run either\n  // converts both or leaves both untouched — never just one.\n  const modelPluginStatus = await checkPluginClass(\n    tree,\n    modelErrorsPath,\n    'ModelErrorLoggingPlugin',\n    MODEL_ERRORS_CLASS_MATCH_PATTERN,\n  );\n  const toolPluginStatus = await checkPluginClass(\n    tree,\n    toolErrorsPath,\n    'ToolErrorLoggingPlugin',\n    TOOL_ERRORS_CLASS_MATCH_PATTERN,\n  );\n  const pluginsReady =\n    modelPluginStatus !== 'diverged' && toolPluginStatus !== 'diverged';\n\n  if (pluginsReady) {\n    if (modelPluginStatus === 'ready') {\n      await applyPluginClass(\n        tree,\n        modelErrorsPath,\n        MODEL_ERRORS_IMPORT_PATTERN,\n        MODEL_ERRORS_CLASS_PATTERN,\n      );\n    }\n    if (toolPluginStatus === 'ready') {\n      await applyPluginClass(\n        tree,\n        toolErrorsPath,\n        TOOL_ERRORS_IMPORT_PATTERN,\n        TOOL_ERRORS_CLASS_PATTERN,\n      );\n    }\n  } else {\n    if (modelPluginStatus === 'diverged') {\n      nextSteps.push(\n        `${modelErrorsPath}: diverged from the generated shape - left as-is. Manually convert it to a \\`ModelErrorLoggingPlugin\\` class implementing \\`Plugin\\` (see the ts#agent generator's template); agent.ts/index.ts files that reference it are left unmigrated until both this and ${toolErrorsPath} are converted.`,\n      );\n    }\n    if (toolPluginStatus === 'diverged') {\n      nextSteps.push(\n        `${toolErrorsPath}: diverged from the generated shape - left as-is. Manually convert it to a \\`ToolErrorLoggingPlugin\\` class implementing \\`Plugin\\` (see the ts#agent generator's template); agent.ts/index.ts files that reference it are left unmigrated until both this and ${modelErrorsPath} are converted.`,\n      );\n    }\n  }\n\n  const filePaths: string[] = [];\n  visitNotIgnoredFiles(tree, '', (filePath) => filePaths.push(filePath));\n\n  for (const filePath of filePaths) {\n    if (filePath === modelErrorsPath || filePath === toolErrorsPath) {\n      continue;\n    }\n    if (filePath.endsWith('.d.ts')) {\n      continue;\n    }\n\n    // session.ts already targets the new { arn, session } shape via\n    // `config.agentRuntimes?.[$name]?.session?.bucketName` — the generic\n    // reshape branch below would otherwise corrupt that into `?.arn?.session?.bucketName`.\n    if (filePath.endsWith('/session.ts')) {\n      continue;\n    }\n\n    if (filePath.endsWith('/agent.ts')) {\n      const dir = filePath.split('/').slice(0, -1).join('/');\n      const project = findOwningProject(tree, dir);\n\n      // Without a registered project there's no ComponentMetadata, so this\n      // agent's protocol/name/agent-connection module can't be resolved — and\n      // a pre-migration AG-UI agent.ts is textually identical to an\n      // HTTP/A2A one, so there's nothing left to safely guess from.\n      if (!project) {\n        if (\n          await matchGritQL(tree, filePath, AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN)\n        ) {\n          nextSteps.push(\n            `${filePath}: could not determine the project root — manually verify whether this is an AG-UI or HTTP/A2A agent and complete its session-manager/error-logging-plugin migration accordingly (see the ts#agent generator's template).`,\n          );\n        }\n        continue;\n      }\n\n      const protocol = agentProtocolFor(project, dir);\n\n      if (protocol === 'ag-ui') {\n        const hasLogCalls =\n          (await matchGritQL(\n            tree,\n            filePath,\n            AGENT_TS_LOG_MODEL_CALL_PATTERN,\n          )) ||\n          (await matchGritQL(tree, filePath, AGENT_TS_LOG_TOOL_CALL_PATTERN));\n        if (!hasLogCalls) {\n          continue;\n        }\n\n        // The sibling index.ts restores error-logging via StrandsAgent\n        // plugins — removing the calls here without that succeeding would\n        // drop error logging rather than move it.\n        const indexPath = `${dir}/index.ts`;\n        const indexReady =\n          pluginsReady &&\n          (await matchGritQL(\n            tree,\n            indexPath,\n            AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN,\n          ));\n\n        if (!indexReady) {\n          nextSteps.push(\n            `${filePath}: left the logModelErrors/logToolErrors calls in place — ${indexPath} could not be automatically wired with the equivalent StrandsAgent plugins. Move error logging to plugins there manually (see the ts#agent generator's template), then remove these calls.`,\n          );\n          continue;\n        }\n\n        const rewroteImport = await applyGritQL(\n          tree,\n          filePath,\n          AGENT_TS_REMOVE_LOG_ERRORS_IMPORT_PATTERN,\n        );\n        const rewroteModelCall = await applyGritQL(\n          tree,\n          filePath,\n          AGENT_TS_LOG_MODEL_CALL_PATTERN,\n        );\n        const rewroteToolCall = await applyGritQL(\n          tree,\n          filePath,\n          AGENT_TS_LOG_TOOL_CALL_PATTERN,\n        );\n        if (rewroteImport || rewroteModelCall || rewroteToolCall) {\n          anyChanges = true;\n        }\n        continue;\n      }\n\n      if (!protocol) {\n        if (\n          await matchGritQL(tree, filePath, AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN)\n        ) {\n          nextSteps.push(\n            `${filePath}: could not determine this agent's protocol from its ComponentMetadata — manually verify whether it's AG-UI or HTTP/A2A and wire sessionManager/the error-logging plugins in accordingly (see the ts#agent generator's template).`,\n          );\n        }\n        continue;\n      }\n\n      // HTTP/A2A: wire sessionManager into the Agent constructor and create\n      // the sibling session.ts if it doesn't exist yet.\n      if (\n        !(await matchGritQL(\n          tree,\n          filePath,\n          AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN,\n        ))\n      ) {\n        continue;\n      }\n\n      const sessionPath = `${dir}/session.ts`;\n\n      // Checked before any writes: if the plugin classes aren't available, or\n      // the Agent isn't an inline object literal, leave the file untouched\n      // entirely rather than ending up half-migrated.\n      if (!pluginsReady) {\n        nextSteps.push(\n          `${filePath}: left as-is — ModelErrorLoggingPlugin/ToolErrorLoggingPlugin are not available yet (see the model-errors-strands.ts/tool-errors-strands.ts next step above). Wire sessionManager and the plugins in manually once those classes exist.`,\n        );\n        continue;\n      }\n      if (\n        !(await matchGritQL(tree, filePath, AGENT_TS_CONSTRUCTOR_MATCH_PATTERN))\n      ) {\n        nextSteps.push(\n          `${filePath}: the Agent is not constructed with an inline object literal (\\`new Agent({ ... })\\`), so sessionManager and the error-logging plugins could not be wired in automatically. Manually add \\`sessionManager: await getSessionManager()\\` and \\`plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()]\\` to its constructor (see the ts#agent generator's template), creating ${sessionPath} first if it doesn't already exist.`,\n        );\n        continue;\n      }\n\n      if (!tree.exists(sessionPath)) {\n        const mod = await captureGritQLVariable(\n          tree,\n          filePath,\n          AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN,\n          'mod',\n        );\n        const agentName = agentTmpNameFor(project, dir);\n\n        if (mod && agentName) {\n          tree.write(\n            sessionPath,\n            legacySessionManagerContent(\n              mod,\n              localSessionsDirFor(project.root, agentName),\n            ),\n          );\n        } else {\n          nextSteps.push(\n            `${filePath}: could not determine the agent-connection module or this agent's name from its ComponentMetadata — manually create a session.ts (see the ts#agent generator's template) and wire \\`sessionManager: await getSessionManager()\\` into the Agent constructor.`,\n          );\n          continue;\n        }\n      }\n\n      // Merges the plugin imports into the existing logModelErrors/\n      // logToolErrors import clause (rather than a redundant second import),\n      // and adds getSessionManager as its own line. The old specifiers are\n      // stripped separately below, after this rewrite anchors on them.\n      const esm = isEsmWorkspace(tree);\n      await applyGritQL(\n        tree,\n        filePath,\n        `\\`import { $names } from '$mod';\\` => raw\\`import { ModelErrorLoggingPlugin, ToolErrorLoggingPlugin, $names } from '$mod';\nimport { getSessionManager } from './session${esm ? '.js' : ''}';\\` where { $names <: contains \\`logModelErrors\\`, $names <: contains \\`logToolErrors\\`, $program <: not contains \\`getSessionManager\\` }`,\n      );\n\n      // Only rewrite the constructor once the import is confirmed present.\n      if (!(tree.read(filePath, 'utf-8') ?? '').includes('getSessionManager')) {\n        nextSteps.push(\n          `${filePath}: found ${sessionPath} but couldn't confirm the getSessionManager import — wire \\`sessionManager: await getSessionManager()\\` into the Agent constructor manually.`,\n        );\n        continue;\n      }\n\n      await applyGritQL(tree, filePath, AGENT_TS_LOG_MODEL_CALL_PATTERN);\n      await applyGritQL(tree, filePath, AGENT_TS_LOG_TOOL_CALL_PATTERN);\n      await applyGritQL(\n        tree,\n        filePath,\n        AGENT_TS_REMOVE_LOG_ERRORS_IMPORT_PATTERN,\n      );\n\n      const rewrotePlugins = await applyGritQL(\n        tree,\n        filePath,\n        AGENT_TS_PLUGINS_CONSTRUCTOR_PATTERN,\n      );\n      const rewroteConstructor = await applyGritQL(\n        tree,\n        filePath,\n        AGENT_TS_SESSION_MANAGER_CONSTRUCTOR_PATTERN,\n      );\n      if (rewrotePlugins || rewroteConstructor) {\n        anyChanges = true;\n      }\n      continue;\n    }\n\n    if (\n      filePath.endsWith('/index.ts') &&\n      (tree.read(filePath, 'utf-8') ?? '').includes('@ag-ui/aws-strands')\n    ) {\n      const alreadyWired = (tree.read(filePath, 'utf-8') ?? '').includes(\n        'ModelErrorLoggingPlugin',\n      );\n\n      if (!alreadyWired) {\n        // Import and constructor only apply cleanly together — adding the\n        // import while the constructor can't be rewritten would leave an\n        // unused import and no plugins wired in.\n        const needsPluginWiring = await matchGritQL(\n          tree,\n          filePath,\n          AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN,\n        );\n        if (!pluginsReady) {\n          if (needsPluginWiring) {\n            nextSteps.push(\n              `${filePath}: left the StrandsAgent constructor as-is — ModelErrorLoggingPlugin/ToolErrorLoggingPlugin are not available yet (see the model-errors-strands.ts/tool-errors-strands.ts next step above). Wire them in manually once those classes exist.`,\n            );\n          }\n          continue;\n        }\n        if (!needsPluginWiring) {\n          nextSteps.push(\n            `${filePath}: the StrandsAgent constructor has diverged from the generated shape - left as-is. Manually add \\`plugins: [new ModelErrorLoggingPlugin(), new ToolErrorLoggingPlugin()]\\` to it (see the ts#agent generator's template).`,\n          );\n          continue;\n        }\n\n        const rewroteImport = await applyGritQL(\n          tree,\n          filePath,\n          AGUI_INDEX_IMPORT_PATTERN,\n        );\n        const rewroteConstructor = await applyGritQL(\n          tree,\n          filePath,\n          AGUI_INDEX_CONSTRUCTOR_PATTERN,\n        );\n        if (rewroteImport || rewroteConstructor) {\n          anyChanges = true;\n        }\n      }\n\n      // Create the sibling session.ts (once plugins are present) and wire in\n      // sessionManagerProvider. Gated on the exact post-plugins shape, so a\n      // customised constructor is left alone rather than creating an unused\n      // session.ts.\n      if (\n        !(await matchGritQL(\n          tree,\n          filePath,\n          AGUI_INDEX_SESSION_MANAGER_MATCH_PATTERN,\n        ))\n      ) {\n        continue;\n      }\n\n      const dir = filePath.split('/').slice(0, -1).join('/');\n      const sessionPath = `${dir}/session.ts`;\n\n      if (!tree.exists(sessionPath)) {\n        const project = findOwningProject(tree, dir);\n        const mod = await captureGritQLVariable(\n          tree,\n          filePath,\n          AGUI_INDEX_IMPORT_CAPTURE_PATTERN,\n          'mod',\n        );\n        const agentName = project && agentTmpNameFor(project, dir);\n\n        if (project && mod && agentName) {\n          tree.write(\n            sessionPath,\n            legacySessionManagerContent(\n              mod,\n              localSessionsDirFor(project.root, agentName),\n            ),\n          );\n        } else {\n          nextSteps.push(\n            `${filePath}: could not determine the project root, agent-connection module, or this agent's name from its ComponentMetadata — manually create a session.ts (see the ts#agent generator's template) and wire \\`config: { sessionManagerProvider: getSessionManager }\\` into the StrandsAgent constructor.`,\n          );\n          continue;\n        }\n      }\n\n      const esm = isEsmWorkspace(tree);\n      await applyGritQL(\n        tree,\n        filePath,\n        aguiIndexSessionManagerImportPattern(esm),\n      );\n\n      if (!(tree.read(filePath, 'utf-8') ?? '').includes('getSessionManager')) {\n        nextSteps.push(\n          `${filePath}: found ${sessionPath} but couldn't confirm the getSessionManager import — wire \\`config: { sessionManagerProvider: getSessionManager }\\` into the StrandsAgent constructor manually.`,\n        );\n        continue;\n      }\n\n      const rewroteSessionManagerConstructor = await applyGritQL(\n        tree,\n        filePath,\n        AGUI_INDEX_SESSION_MANAGER_CONSTRUCTOR_PATTERN,\n      );\n      if (rewroteSessionManagerConstructor) {\n        anyChanges = true;\n      }\n      continue;\n    }\n\n    if (filePath.endsWith('.ts') || filePath.endsWith('.tsx')) {\n      const contents = tree.read(filePath, 'utf-8') ?? '';\n      if (!contents.includes('agentRuntimes')) {\n        continue;\n      }\n      // Only the 'agentcore' namespace is reshaped — see CDK_RC_SET_PATTERN.\n      const rewroteAgentcoreRcSet = await applyGritQL(\n        tree,\n        filePath,\n        CDK_RC_SET_PATTERN,\n      );\n      const rewroteInterface = await applyGritQL(\n        tree,\n        filePath,\n        TS_RUNTIME_CONFIG_INTERFACE_PATTERN,\n      );\n      let rewroteClientArn = false;\n      for (const pattern of TS_CLIENT_ARN_PATTERNS) {\n        rewroteClientArn = (await applyGritQL(tree, filePath, pattern))\n          ? true\n          : rewroteClientArn;\n      }\n      const rewroteChatScriptType = await applyGritQL(\n        tree,\n        filePath,\n        AGENTCORE_CHAT_SCRIPT_TYPE_PATTERN,\n      );\n\n      if (\n        rewroteAgentcoreRcSet ||\n        rewroteInterface ||\n        rewroteClientArn ||\n        rewroteChatScriptType\n      ) {\n        anyChanges = true;\n      }\n    } else if (filePath.endsWith('.tf')) {\n      const contents = tree.read(filePath, 'utf-8') ?? '';\n      if (!contents.includes('\"agentRuntimes\"')) {\n        continue;\n      }\n      const rewrote = await applyGritQL(tree, filePath, TF_VALUE_PATTERN);\n      if (rewrote) {\n        anyChanges = true;\n      }\n    } else if (filePath.endsWith('.py')) {\n      const contents = tree.read(filePath, 'utf-8') ?? '';\n      if (!contents.includes('agentRuntimes')) {\n        continue;\n      }\n      const rewrote = await applyGritQL(tree, filePath, PY_CLIENT_ARN_PATTERN);\n      if (rewrote) {\n        anyChanges = true;\n      }\n    }\n  }\n\n  if (anyChanges) {\n    nextSteps.push(\n      'If you have custom code reading the runtime config `agentcore.agentRuntimes` entries ' +\n        'as plain ARN strings (e.g. custom Lambda handlers or agent code), update it to read ' +\n        'the ARN from `.arn` on the new `{ arn }` shape.',\n    );\n  }\n\n  await formatFilesInSubtree(tree);\n\n  return { nextSteps };\n}\n"],"names":["getProjects","joinPathFragments","visitNotIgnoredFiles","TS_AGENT_GENERATOR_INFO","AGENT_CONNECTION_PROJECT_DIR","applyGritQL","captureGritQLVariable","matchGritQL","formatFilesInSubtree","isEsmWorkspace","kebabCase","getRelativePathToRootByDirectory","toProjectRelativePath","CDK_RC_SET_PATTERN","TF_VALUE_PATTERN","TS_RUNTIME_CONFIG_INTERFACE_PATTERN","TS_CLIENT_ARN_PATTERNS","AGENTCORE_CHAT_SCRIPT_TYPE_PATTERN","PY_CLIENT_ARN_PATTERN","AGENT_TS_LOG_MODEL_CALL_PATTERN","AGENT_TS_LOG_TOOL_CALL_PATTERN","AGENT_TS_SESSION_MANAGER_CONSTRUCTOR_PATTERN","AGENT_TS_CONSTRUCTOR_MATCH_PATTERN","AGENT_TS_PLUGINS_CONSTRUCTOR_PATTERN","AGENT_TS_LOG_IMPORT_CAPTURE_PATTERN","AGENT_TS_REMOVE_LOG_ERRORS_IMPORT_PATTERN","legacySessionManagerContent","agentConnectionModule","localSessionsDir","findOwningProject","tree","dirPath","best","project","values","root","startsWith","length","findAgentComponentMetadata","dirRelativeToProjectRoot","metadata","components","find","component","generator","id","path","agentTmpNameFor","dir","rc","undefined","agentProtocolFor","protocol","localSessionsDirFor","projectRoot","agentTmpName","AGUI_INDEX_IMPORT_PATTERN","AGUI_INDEX_CONSTRUCTOR_MATCH_PATTERN","AGUI_INDEX_CONSTRUCTOR_PATTERN","AGUI_INDEX_IMPORT_CAPTURE_PATTERN","aguiIndexSessionManagerImportPattern","esm","AGUI_INDEX_SESSION_MANAGER_MATCH_PATTERN","AGUI_INDEX_SESSION_MANAGER_CONSTRUCTOR_PATTERN","MODEL_ERRORS_IMPORT_PATTERN","MODEL_ERRORS_CLASS_MATCH_PATTERN","MODEL_ERRORS_CLASS_PATTERN","TOOL_ERRORS_IMPORT_PATTERN","TOOL_ERRORS_CLASS_MATCH_PATTERN","TOOL_ERRORS_CLASS_PATTERN","checkPluginClass","filePath","className","classMatchPattern","exists","read","includes","applyPluginClass","importPattern","classPattern","migration","nextSteps","anyChanges","modelErrorsPath","toolErrorsPath","modelPluginStatus","toolPluginStatus","pluginsReady","push","filePaths","endsWith","split","slice","join","hasLogCalls","indexPath","indexReady","rewroteImport","rewroteModelCall","rewroteToolCall","sessionPath","mod","agentName","write","rewrotePlugins","rewroteConstructor","alreadyWired","needsPluginWiring","rewroteSessionManagerConstructor","contents","rewroteAgentcoreRcSet","rewroteInterface","rewroteClientArn","pattern","rewroteChatScriptType","rewrote"],"mappings":"AAAA;;;CAGC,GACD,SACEA,WAAW,EACXC,iBAAiB,EAIjBC,oBAAoB,QACf,aAAa;AACpB,SAASC,uBAAuB,QAAQ,iCAA8B;AACtE,SAASC,4BAA4B,QAAQ,sDAAmD;AAChG,SACEC,WAAW,EACXC,qBAAqB,EACrBC,WAAW,QACN,wBAAqB;AAC5B,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,cAAc,QAAQ,kCAA+B;AAC9D,SAASC,SAAS,QAAQ,0BAAuB;AAEjD,SACEC,gCAAgC,EAChCC,qBAAqB,QAChB,0BAAuB;AAE9B;;;;;;;;;;;;;;CAcC,GAED,8EAA8E;AAC9E,2EAA2E;AAC3E,sEAAsE;AACtE,6EAA6E;AAC7E,+DAA+D;AAC/D,MAAMC,qBAAqB,CAAC;;;;;KAKvB,CAAC;AAEN,wEAAwE;AACxE,+EAA+E;AAC/E,MAAMC,mBAAmB,CAAC;;;;;;;;;;;;;;;;GAgBvB,CAAC;AAEJ,yDAAyD;AACzD,iEAAiE;AACjE,MAAMC,sCAAsC,CAAC;;;;;;;;;;;;;;;GAe1C,CAAC;AAEJ,+EAA+E;AAC/E,qEAAqE;AACrE,MAAMC,yBAAyB;IAC7B;IACA;CACD;AAED,8EAA8E;AAC9E,6EAA6E;AAC7E,4EAA4E;AAC5E,aAAa;AACb,MAAMC,qCACJ;AAEF,4DAA4D;AAC5D,MAAMC,wBACJ;AAEF,2EAA2E;AAC3E,4EAA4E;AAC5E,yEAAyE;AACzE,0EAA0E;AAC1E,4EAA4E;AAC5E,MAAMC,kCAAkC;AACxC,MAAMC,iCAAiC;AAEvC,0EAA0E;AAC1E,6EAA6E;AAC7E,yEAAyE;AACzE,wEAAwE;AACxE,6DAA6D;AAC7D,MAAMC,+CACJ;AAEF,4EAA4E;AAC5E,oDAAoD;AACpD,MAAMC,qCAAqC;AAE3C,2EAA2E;AAC3E,sEAAsE;AACtE,oBAAoB;AACpB,MAAMC,uCACJ;AAEF,8EAA8E;AAC9E,0EAA0E;AAC1E,6EAA6E;AAC7E,MAAMC,sCACJ;AAEF,2EAA2E;AAC3E,8EAA8E;AAC9E,kEAAkE;AAClE,+EAA+E;AAC/E,MAAMC,4CACJ;AAEF,qEAAqE;AACrE,mCAAmC;AACnC,MAAMC,8BAA8B,CAClCC,uBACAC,mBACW,CAAC;;qCAEuB,EAAED,sBAAsB;;;;;;;;;;;;;;;;;;;qCAmBxB,EAAEC,iBAAiB;;;;;AAKxD,CAAC;AAED,iFAAiF,GACjF,MAAMC,oBAAoB,CACxBC,MACAC;IAEA,IAAIC;IACJ,KAAK,MAAMC,WAAWjC,YAAY8B,MAAMI,MAAM,GAAI;QAChD,IACE,AAACH,CAAAA,YAAYE,QAAQE,IAAI,IAAIJ,QAAQK,UAAU,CAAC,GAAGH,QAAQE,IAAI,CAAC,CAAC,CAAC,CAAA,KACjE,CAAA,CAACH,QAAQC,QAAQE,IAAI,CAACE,MAAM,GAAGL,KAAKG,IAAI,CAACE,MAAM,AAAD,GAC/C;YACAL,OAAOC;QACT;IACF;IACA,OAAOD;AACT;AAEA,oFAAoF,GACpF,MAAMM,6BAA6B,CACjCL,SACAM,2BAECN,QAAQO,QAAQ,EAA2CC,YAAYC,KACtE,CAACC,YACCA,UAAUC,SAAS,KAAKzC,wBAAwB0C,EAAE,IAClDF,UAAUG,IAAI,KAAKP;AAGzB,yFAAyF,GACzF,MAAMQ,kBAAkB,CACtBd,SACAe;IAEA,MAAMT,2BAA2B3B,sBAAsBqB,SAASe;IAChE,MAAMC,KAAKX,2BAA2BL,SAASM,2BAA2BU;IAC1E,OAAOA,KAAKvC,UAAUuC,MAAMC;AAC9B;AAEA,sEAAsE,GACtE,MAAMC,mBAAmB,CACvBlB,SACAe;IAEA,MAAMT,2BAA2B3B,sBAAsBqB,SAASe;IAChE,OAAOV,2BAA2BL,SAASM,2BACvCa;AACN;AAEA,wGAAwG,GACxG,MAAMC,sBAAsB,CAC1BC,aACAC,eAEAtD,kBACEU,iCAAiC2C,cACjC,CAAC,mBAAmB,EAAEC,cAAc;AAGxC,6EAA6E;AAC7E,4EAA4E;AAC5E,MAAMC,4BACJ;AACF,wEAAwE;AACxE,+DAA+D;AAC/D,MAAMC,uCACJ;AACF,MAAMC,iCAAiC,GAAGD,qCAAqC,4IAA4I,CAAC;AAE5N,MAAME,oCACJ;AAEF,2EAA2E;AAC3E,yEAAyE;AACzE,MAAMC,uCAAuC,CAACC,MAC5C,CAAC;4CACyC,EAAEA,MAAM,QAAQ,GAAG,6DAA6D,CAAC;AAC7H,4EAA4E;AAC5E,+DAA+D;AAC/D,MAAMC,2CACJ;AACF,MAAMC,iDAAiD,GAAGD,yCAAyC,8IAA8I,CAAC;AAElP,8EAA8E;AAC9E,8EAA8E;AAC9E,sEAAsE;AACtE,qEAAqE;AACrE,MAAME,8BACJ;AACF,2EAA2E;AAC3E,qEAAqE;AACrE,MAAMC,mCACJ;AACF,MAAMC,6BAA6B,GAAGD,iCAAiC,mNAAmN,CAAC;AAE3R,MAAME,6BACJ;AACF,MAAMC,kCACJ;AACF,MAAMC,4BAA4B,GAAGD,gCAAgC,gNAAgN,CAAC;AAItR;;;;;;CAMC,GACD,eAAeE,iBACbxC,IAAU,EACVyC,QAAgB,EAChBC,SAAiB,EACjBC,iBAAyB;IAEzB,IAAI,CAAC3C,KAAK4C,MAAM,CAACH,WAAW;QAC1B,OAAO;IACT;IACA,IAAI,AAACzC,CAAAA,KAAK6C,IAAI,CAACJ,UAAU,YAAY,EAAC,EAAGK,QAAQ,CAAC,CAAC,MAAM,EAAEJ,WAAW,GAAG;QACvE,OAAO;IACT;IACA,OAAO,AAAC,MAAMjE,YAAYuB,MAAMyC,UAAUE,qBACtC,UACA;AACN;AAEA,+EAA+E,GAC/E,eAAeI,iBACb/C,IAAU,EACVyC,QAAgB,EAChBO,aAAqB,EACrBC,YAAoB;IAEpB,0EAA0E;IAC1E,2EAA2E;IAC3E,0EAA0E;IAC1E,sEAAsE;IACtE,MAAM1E,YAAYyB,MAAMyC,UAAUQ;IAClC,MAAM1E,YAAYyB,MAAMyC,UAAUO;AACpC;AAEA,eAAe,eAAeE,UAC5BlD,IAAU;IAEV,MAAMmD,YAAsB,EAAE;IAC9B,IAAIC,aAAa;IAEjB,qEAAqE;IACrE,6DAA6D;IAC7D,MAAMC,kBAAkBlF,kBACtBG,8BACA,OACA,QACA;IAEF,MAAMgF,iBAAiBnF,kBACrBG,8BACA,OACA,QACA;IAEF,2EAA2E;IAC3E,2DAA2D;IAC3D,MAAMiF,oBAAoB,MAAMf,iBAC9BxC,MACAqD,iBACA,2BACAlB;IAEF,MAAMqB,mBAAmB,MAAMhB,iBAC7BxC,MACAsD,gBACA,0BACAhB;IAEF,MAAMmB,eACJF,sBAAsB,cAAcC,qBAAqB;IAE3D,IAAIC,cAAc;QAChB,IAAIF,sBAAsB,SAAS;YACjC,MAAMR,iBACJ/C,MACAqD,iBACAnB,6BACAE;QAEJ;QACA,IAAIoB,qBAAqB,SAAS;YAChC,MAAMT,iBACJ/C,MACAsD,gBACAjB,4BACAE;QAEJ;IACF,OAAO;QACL,IAAIgB,sBAAsB,YAAY;YACpCJ,UAAUO,IAAI,CACZ,GAAGL,gBAAgB,gQAAgQ,EAAEC,eAAe,eAAe,CAAC;QAExT;QACA,IAAIE,qBAAqB,YAAY;YACnCL,UAAUO,IAAI,CACZ,GAAGJ,eAAe,+PAA+P,EAAED,gBAAgB,eAAe,CAAC;QAEvT;IACF;IAEA,MAAMM,YAAsB,EAAE;IAC9BvF,qBAAqB4B,MAAM,IAAI,CAACyC,WAAakB,UAAUD,IAAI,CAACjB;IAE5D,KAAK,MAAMA,YAAYkB,UAAW;QAChC,IAAIlB,aAAaY,mBAAmBZ,aAAaa,gBAAgB;YAC/D;QACF;QACA,IAAIb,SAASmB,QAAQ,CAAC,UAAU;YAC9B;QACF;QAEA,gEAAgE;QAChE,qEAAqE;QACrE,uFAAuF;QACvF,IAAInB,SAASmB,QAAQ,CAAC,gBAAgB;YACpC;QACF;QAEA,IAAInB,SAASmB,QAAQ,CAAC,cAAc;YAClC,MAAM1C,MAAMuB,SAASoB,KAAK,CAAC,KAAKC,KAAK,CAAC,GAAG,CAAC,GAAGC,IAAI,CAAC;YAClD,MAAM5D,UAAUJ,kBAAkBC,MAAMkB;YAExC,qEAAqE;YACrE,wEAAwE;YACxE,8DAA8D;YAC9D,8DAA8D;YAC9D,IAAI,CAACf,SAAS;gBACZ,IACE,MAAM1B,YAAYuB,MAAMyC,UAAU/C,sCAClC;oBACAyD,UAAUO,IAAI,CACZ,GAAGjB,SAAS,wNAAwN,CAAC;gBAEzO;gBACA;YACF;YAEA,MAAMnB,WAAWD,iBAAiBlB,SAASe;YAE3C,IAAII,aAAa,SAAS;gBACxB,MAAM0C,cACJ,AAAC,MAAMvF,YACLuB,MACAyC,UACApD,oCAED,MAAMZ,YAAYuB,MAAMyC,UAAUnD;gBACrC,IAAI,CAAC0E,aAAa;oBAChB;gBACF;gBAEA,+DAA+D;gBAC/D,kEAAkE;gBAClE,0CAA0C;gBAC1C,MAAMC,YAAY,GAAG/C,IAAI,SAAS,CAAC;gBACnC,MAAMgD,aACJT,gBACC,MAAMhF,YACLuB,MACAiE,WACAtC;gBAGJ,IAAI,CAACuC,YAAY;oBACff,UAAUO,IAAI,CACZ,GAAGjB,SAAS,yDAAyD,EAAEwB,UAAU,0LAA0L,CAAC;oBAE9Q;gBACF;gBAEA,MAAME,gBAAgB,MAAM5F,YAC1ByB,MACAyC,UACA9C;gBAEF,MAAMyE,mBAAmB,MAAM7F,YAC7ByB,MACAyC,UACApD;gBAEF,MAAMgF,kBAAkB,MAAM9F,YAC5ByB,MACAyC,UACAnD;gBAEF,IAAI6E,iBAAiBC,oBAAoBC,iBAAiB;oBACxDjB,aAAa;gBACf;gBACA;YACF;YAEA,IAAI,CAAC9B,UAAU;gBACb,IACE,MAAM7C,YAAYuB,MAAMyC,UAAU/C,sCAClC;oBACAyD,UAAUO,IAAI,CACZ,GAAGjB,SAAS,iOAAiO,CAAC;gBAElP;gBACA;YACF;YAEA,sEAAsE;YACtE,kDAAkD;YAClD,IACE,CAAE,MAAMhE,YACNuB,MACAyC,UACA/C,sCAEF;gBACA;YACF;YAEA,MAAM4E,cAAc,GAAGpD,IAAI,WAAW,CAAC;YAEvC,wEAAwE;YACxE,qEAAqE;YACrE,gDAAgD;YAChD,IAAI,CAACuC,cAAc;gBACjBN,UAAUO,IAAI,CACZ,GAAGjB,SAAS,uOAAuO,CAAC;gBAEtP;YACF;YACA,IACE,CAAE,MAAMhE,YAAYuB,MAAMyC,UAAUjD,qCACpC;gBACA2D,UAAUO,IAAI,CACZ,GAAGjB,SAAS,2XAA2X,EAAE6B,YAAY,mCAAmC,CAAC;gBAE3b;YACF;YAEA,IAAI,CAACtE,KAAK4C,MAAM,CAAC0B,cAAc;gBAC7B,MAAMC,MAAM,MAAM/F,sBAChBwB,MACAyC,UACA/C,qCACA;gBAEF,MAAM8E,YAAYvD,gBAAgBd,SAASe;gBAE3C,IAAIqD,OAAOC,WAAW;oBACpBxE,KAAKyE,KAAK,CACRH,aACA1E,4BACE2E,KACAhD,oBAAoBpB,QAAQE,IAAI,EAAEmE;gBAGxC,OAAO;oBACLrB,UAAUO,IAAI,CACZ,GAAGjB,SAAS,2PAA2P,CAAC;oBAE1Q;gBACF;YACF;YAEA,8DAA8D;YAC9D,uEAAuE;YACvE,qEAAqE;YACrE,iEAAiE;YACjE,MAAMV,MAAMpD,eAAeqB;YAC3B,MAAMzB,YACJyB,MACAyC,UACA,CAAC;4CACmC,EAAEV,MAAM,QAAQ,GAAG,0IAA0I,CAAC;YAGpM,qEAAqE;YACrE,IAAI,CAAC,AAAC/B,CAAAA,KAAK6C,IAAI,CAACJ,UAAU,YAAY,EAAC,EAAGK,QAAQ,CAAC,sBAAsB;gBACvEK,UAAUO,IAAI,CACZ,GAAGjB,SAAS,QAAQ,EAAE6B,YAAY,4IAA4I,CAAC;gBAEjL;YACF;YAEA,MAAM/F,YAAYyB,MAAMyC,UAAUpD;YAClC,MAAMd,YAAYyB,MAAMyC,UAAUnD;YAClC,MAAMf,YACJyB,MACAyC,UACA9C;YAGF,MAAM+E,iBAAiB,MAAMnG,YAC3ByB,MACAyC,UACAhD;YAEF,MAAMkF,qBAAqB,MAAMpG,YAC/ByB,MACAyC,UACAlD;YAEF,IAAImF,kBAAkBC,oBAAoB;gBACxCvB,aAAa;YACf;YACA;QACF;QAEA,IACEX,SAASmB,QAAQ,CAAC,gBAClB,AAAC5D,CAAAA,KAAK6C,IAAI,CAACJ,UAAU,YAAY,EAAC,EAAGK,QAAQ,CAAC,uBAC9C;YACA,MAAM8B,eAAe,AAAC5E,CAAAA,KAAK6C,IAAI,CAACJ,UAAU,YAAY,EAAC,EAAGK,QAAQ,CAChE;YAGF,IAAI,CAAC8B,cAAc;gBACjB,kEAAkE;gBAClE,iEAAiE;gBACjE,yCAAyC;gBACzC,MAAMC,oBAAoB,MAAMpG,YAC9BuB,MACAyC,UACAd;gBAEF,IAAI,CAAC8B,cAAc;oBACjB,IAAIoB,mBAAmB;wBACrB1B,UAAUO,IAAI,CACZ,GAAGjB,SAAS,0OAA0O,CAAC;oBAE3P;oBACA;gBACF;gBACA,IAAI,CAACoC,mBAAmB;oBACtB1B,UAAUO,IAAI,CACZ,GAAGjB,SAAS,yNAAyN,CAAC;oBAExO;gBACF;gBAEA,MAAM0B,gBAAgB,MAAM5F,YAC1ByB,MACAyC,UACAf;gBAEF,MAAMiD,qBAAqB,MAAMpG,YAC/ByB,MACAyC,UACAb;gBAEF,IAAIuC,iBAAiBQ,oBAAoB;oBACvCvB,aAAa;gBACf;YACF;YAEA,uEAAuE;YACvE,sEAAsE;YACtE,sEAAsE;YACtE,cAAc;YACd,IACE,CAAE,MAAM3E,YACNuB,MACAyC,UACAT,2CAEF;gBACA;YACF;YAEA,MAAMd,MAAMuB,SAASoB,KAAK,CAAC,KAAKC,KAAK,CAAC,GAAG,CAAC,GAAGC,IAAI,CAAC;YAClD,MAAMO,cAAc,GAAGpD,IAAI,WAAW,CAAC;YAEvC,IAAI,CAAClB,KAAK4C,MAAM,CAAC0B,cAAc;gBAC7B,MAAMnE,UAAUJ,kBAAkBC,MAAMkB;gBACxC,MAAMqD,MAAM,MAAM/F,sBAChBwB,MACAyC,UACAZ,mCACA;gBAEF,MAAM2C,YAAYrE,WAAWc,gBAAgBd,SAASe;gBAEtD,IAAIf,WAAWoE,OAAOC,WAAW;oBAC/BxE,KAAKyE,KAAK,CACRH,aACA1E,4BACE2E,KACAhD,oBAAoBpB,QAAQE,IAAI,EAAEmE;gBAGxC,OAAO;oBACLrB,UAAUO,IAAI,CACZ,GAAGjB,SAAS,6RAA6R,CAAC;oBAE5S;gBACF;YACF;YAEA,MAAMV,MAAMpD,eAAeqB;YAC3B,MAAMzB,YACJyB,MACAyC,UACAX,qCAAqCC;YAGvC,IAAI,CAAC,AAAC/B,CAAAA,KAAK6C,IAAI,CAACJ,UAAU,YAAY,EAAC,EAAGK,QAAQ,CAAC,sBAAsB;gBACvEK,UAAUO,IAAI,CACZ,GAAGjB,SAAS,QAAQ,EAAE6B,YAAY,+JAA+J,CAAC;gBAEpM;YACF;YAEA,MAAMQ,mCAAmC,MAAMvG,YAC7CyB,MACAyC,UACAR;YAEF,IAAI6C,kCAAkC;gBACpC1B,aAAa;YACf;YACA;QACF;QAEA,IAAIX,SAASmB,QAAQ,CAAC,UAAUnB,SAASmB,QAAQ,CAAC,SAAS;YACzD,MAAMmB,WAAW/E,KAAK6C,IAAI,CAACJ,UAAU,YAAY;YACjD,IAAI,CAACsC,SAASjC,QAAQ,CAAC,kBAAkB;gBACvC;YACF;YACA,uEAAuE;YACvE,MAAMkC,wBAAwB,MAAMzG,YAClCyB,MACAyC,UACA1D;YAEF,MAAMkG,mBAAmB,MAAM1G,YAC7ByB,MACAyC,UACAxD;YAEF,IAAIiG,mBAAmB;YACvB,KAAK,MAAMC,WAAWjG,uBAAwB;gBAC5CgG,mBAAmB,AAAC,MAAM3G,YAAYyB,MAAMyC,UAAU0C,WAClD,OACAD;YACN;YACA,MAAME,wBAAwB,MAAM7G,YAClCyB,MACAyC,UACAtD;YAGF,IACE6F,yBACAC,oBACAC,oBACAE,uBACA;gBACAhC,aAAa;YACf;QACF,OAAO,IAAIX,SAASmB,QAAQ,CAAC,QAAQ;YACnC,MAAMmB,WAAW/E,KAAK6C,IAAI,CAACJ,UAAU,YAAY;YACjD,IAAI,CAACsC,SAASjC,QAAQ,CAAC,oBAAoB;gBACzC;YACF;YACA,MAAMuC,UAAU,MAAM9G,YAAYyB,MAAMyC,UAAUzD;YAClD,IAAIqG,SAAS;gBACXjC,aAAa;YACf;QACF,OAAO,IAAIX,SAASmB,QAAQ,CAAC,QAAQ;YACnC,MAAMmB,WAAW/E,KAAK6C,IAAI,CAACJ,UAAU,YAAY;YACjD,IAAI,CAACsC,SAASjC,QAAQ,CAAC,kBAAkB;gBACvC;YACF;YACA,MAAMuC,UAAU,MAAM9G,YAAYyB,MAAMyC,UAAUrD;YAClD,IAAIiG,SAAS;gBACXjC,aAAa;YACf;QACF;IACF;IAEA,IAAIA,YAAY;QACdD,UAAUO,IAAI,CACZ,0FACE,yFACA;IAEN;IAEA,MAAMhF,qBAAqBsB;IAE3B,OAAO;QAAEmD;IAAU;AACrB"}

package/src/migrations/v1.0.0-rc.58/0001-smithy-ssdk-bundle-pins/metadata.json

@@ -0,0 +1,3 @@
+{
+  "description": "Bump the rolldown and dts plugin pins in the Smithy build.Dockerfile so the bundled SSDK type declaration resolves"
+}

package/src/migrations/v1.0.0-rc.58/0001-smithy-ssdk-bundle-pins/migration.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type MigrationReturnObject, type Tree } from '@nx/devkit';
+export default function migration(tree: Tree): Promise<MigrationReturnObject>;