@aws/nx-plugin 1.0.0-rc.3 → 1.0.0-rc.31

package/src/ts/react-website/agui/files/common/src/hooks/useAgui__agentNameClassName__.tsx.template

@@ -1,7 +1,6 @@
 <%_ if (auth === 'iam') { _%>
 import { HttpAgent } from '@ag-ui/client';
 import type { AbstractAgent, RunAgentInput } from '@ag-ui/client';
-import { Observable } from 'rxjs';
 import { useMemo } from 'react';
 import { useRuntimeConfig } from './useRuntimeConfig';
 import { useSigV4, type SigV4Client } from './useSigV4';
@@ -36,7 +35,7 @@ class SigV4HttpAgent extends HttpAgent {
     config: ConstructorParameters<typeof HttpAgent>[0],
     sigV4Client: SigV4Client,
   ) {
-    super(config);
+    super({ ...config, fetch: sigV4Client.fetch as (url: string, init: RequestInit) => Promise<Response> });
     this.sigV4Client = sigV4Client;
   }
 
@@ -49,20 +48,6 @@ class SigV4HttpAgent extends HttpAgent {
     return init;
   }
 
-  // Remove this override once https://github.com/ag-ui-protocol/ag-ui/issues/1316 is fixed.
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  override run(input: RunAgentInput): any {
-    const sigV4Fetch = this.sigV4Client.fetch as typeof window.fetch;
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    return new Observable<any>((subscriber) => {
-      const originalFetch = window.fetch;
-      window.fetch = sigV4Fetch;
-      const sub = super.run(input).subscribe(subscriber);
-      window.fetch = originalFetch;
-      return () => sub.unsubscribe();
-    });
-  }
-
   override clone(): SigV4HttpAgent {
     const cloned = super.clone() as SigV4HttpAgent;
     cloned.sigV4Client = this.sigV4Client;

package/src/ts/react-website/agui/generator.d.ts

@@ -2,7 +2,7 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
-import { ProjectConfiguration, Tree } from '@nx/devkit';
+import { type ProjectConfiguration, type Tree } from '@nx/devkit';
 export type AgUiAuth = 'iam' | 'cognito' | 'none';
 export interface AgUiReactConnectionOptions {
     /** The React website project to connect FROM */

package/src/ts/react-website/agui/generator.js

@@ -1,19 +1,15 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.addAgUiReactConnection = void 0;
 /**
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
- */
-const devkit_1 = require("@nx/devkit");
-const generator_1 = require("../runtime-config/generator");
-const versions_1 = require("../../../utils/versions");
-const ast_1 = require("../../../utils/ast");
-const agent_runtime_config_1 = require("../../../connection/agent-runtime-config");
-const names_1 = require("../../../utils/names");
-const shared_shadcn_1 = require("../../../utils/shared-shadcn");
-const npm_scope_1 = require("../../../utils/npm-scope");
-const pnpm_workspace_1 = require("../../../utils/pnpm-workspace");
+ */ import { addDependenciesToPackageJson, generateFiles, joinPathFragments, OverwriteStrategy } from "@nx/devkit";
+import { addAgentRuntimeToConnectionNamespace } from "../../../connection/agent-runtime-config.js";
+import { addDestructuredImport, addSingleImport, applyGritQL } from "../../../utils/ast.js";
+import { kebabCase } from "../../../utils/names.js";
+import { getNpmScopePrefix, toScopeAlias } from "../../../utils/npm-scope.js";
+import { registerPnpmBuiltDependencies } from "../../../utils/pnpm-workspace.js";
+import { sharedShadcnGenerator } from "../../../utils/shared-shadcn.js";
+import { withVersions } from "../../../utils/versions.js";
+import { runtimeConfigGenerator } from "../runtime-config/generator.js";
 /**
  * Wires a React website up to an AG-UI agent using CopilotKit and
  * @ag-ui/client's HttpAgent. Agent-server-language agnostic.
@@ -25,69 +21,86 @@ const pnpm_workspace_1 = require("../../../utils/pnpm-workspace");
  *
  * Also vends a `src/components/copilot` theme module picked from the
  * website's `metadata.ux` (cloudscape / shadcn / default).
- */
-const addAgUiReactConnection = async (tree, options) => {
+ */ export const addAgUiReactConnection = async (tree, options)=>{
     const { frontendProjectConfig, agentName, agentNameClassName, auth } = options;
     const theme = resolveAgUiTheme(frontendProjectConfig);
-    const scopeAlias = (0, npm_scope_1.toScopeAlias)((0, npm_scope_1.getNpmScopePrefix)(tree));
+    const scopeAlias = toScopeAlias(getNpmScopePrefix(tree));
     // Generates `src/components/AguiProvider.tsx` (if absent) and
     // `src/hooks/useAgui<AgentName>.tsx`. Existing files are kept so re-running
     // the generator is idempotent.
-    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', 'common'), frontendProjectConfig.root, { agentName, agentNameClassName, auth }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', 'common'), frontendProjectConfig.root, {
+        agentName,
+        agentNameClassName,
+        auth
+    }, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
     // Shadcn theme imports from the shared shadcn library, so it must exist.
     if (theme === 'shadcn') {
-        await (0, shared_shadcn_1.sharedShadcnGenerator)(tree);
+        await sharedShadcnGenerator(tree);
     }
-    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', theme), frontendProjectConfig.root, { scopeAlias }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', theme), frontendProjectConfig.root, {
+        scopeAlias
+    }, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
     if (auth === 'iam') {
-        (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, '../../../utils/files/website/hooks/sigv4'), (0, devkit_1.joinPathFragments)(frontendProjectConfig.sourceRoot, 'hooks'), {}, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
+        generateFiles(tree, joinPathFragments(import.meta.dirname, '../../../utils/files/website/hooks/sigv4'), joinPathFragments(frontendProjectConfig.sourceRoot, 'hooks'), {}, {
+            overwriteStrategy: OverwriteStrategy.KeepExisting
+        });
     }
-    await (0, generator_1.runtimeConfigGenerator)(tree, {
-        project: frontendProjectConfig.name,
+    await runtimeConfigGenerator(tree, {
+        project: frontendProjectConfig.name
     });
     // AST-patch AguiProvider to register this agent's hook
     await registerAgentHookInProvider(tree, frontendProjectConfig, agentNameClassName);
     // Wrap <App /> in <AguiProvider> in main.tsx (idempotent)
-    const mainTsxPath = (0, devkit_1.joinPathFragments)(frontendProjectConfig.sourceRoot, 'main.tsx');
-    await (0, ast_1.addSingleImport)(tree, mainTsxPath, 'AguiProvider', './components/AguiProvider');
-    await (0, ast_1.applyGritQL)(tree, mainTsxPath, `\`<App />\` => \`<AguiProvider><App /></AguiProvider>\` where { $program <: not contains \`<AguiProvider>$_</AguiProvider>\` }`);
+    const mainTsxPath = joinPathFragments(frontendProjectConfig.sourceRoot, 'main.tsx');
+    await addSingleImport(tree, mainTsxPath, 'AguiProvider', './components/AguiProvider');
+    await applyGritQL(tree, mainTsxPath, `\`<App />\` => \`<AguiProvider><App /></AguiProvider>\` where { $program <: not contains \`<AguiProvider>$_</AguiProvider>\` }`);
     // @copilotkit/react-core transitively pulls in @scarf/scarf (telemetry),
     // which has a postinstall script. Under pnpm 11's default
     // `strictDepBuilds=true` that script is treated as an unreviewed build and
     // fails the install. Register it as an explicitly-rejected build so pnpm
     // knows we've seen it and will skip it instead of erroring.
-    (0, pnpm_workspace_1.registerPnpmBuiltDependencies)(tree, { '@scarf/scarf': false });
-    (0, devkit_1.addDependenciesToPackageJson)(tree, (0, versions_1.withVersions)([
+    registerPnpmBuiltDependencies(tree, {
+        '@scarf/scarf': false
+    });
+    addDependenciesToPackageJson(tree, withVersions([
         '@copilotkit/react-core',
         '@ag-ui/client',
-        ...(theme === 'cloudscape'
-            ? ['@cloudscape-design/chat-components']
-            : []),
-        ...(theme === 'shadcn' ? ['lucide-react'] : []),
-        ...(auth === 'iam'
-            ? [
-                'oidc-client-ts',
-                'aws4fetch',
-                '@aws-sdk/credential-provider-cognito-identity',
-                'react-oidc-context',
-                'rxjs',
-            ]
-            : []),
-        ...(auth === 'cognito' ? ['react-oidc-context'] : []),
-    ]), (0, versions_1.withVersions)([...(auth === 'iam' ? ['@smithy/types'] : [])]));
+        ...theme === 'cloudscape' ? [
+            '@cloudscape-design/chat-components'
+        ] : [],
+        ...theme === 'shadcn' ? [
+            'lucide-react'
+        ] : [],
+        ...auth === 'iam' ? [
+            'oidc-client-ts',
+            'aws4fetch',
+            '@aws-sdk/credential-provider-cognito-identity',
+            'react-oidc-context'
+        ] : [],
+        ...auth === 'cognito' ? [
+            'react-oidc-context'
+        ] : []
+    ]), withVersions([
+        ...auth === 'iam' ? [
+            '@smithy/types'
+        ] : []
+    ]));
     // Agents only publish their runtime ARN to the 'agentcore' namespace by
     // default, which isn't exposed to the website. Patch the agent's CDK/TF
     // construct to also publish under 'connection' so the browser can read it
     // from runtime-config.json.
-    await (0, agent_runtime_config_1.addAgentRuntimeToConnectionNamespace)(tree, {
-        agentNameKebabCase: (0, names_1.kebabCase)(agentNameClassName),
-        agentNameClassName,
+    await addAgentRuntimeToConnectionNamespace(tree, {
+        agentNameKebabCase: kebabCase(agentNameClassName),
+        agentNameClassName
     });
 };
-exports.addAgUiReactConnection = addAgUiReactConnection;
-const resolveAgUiTheme = (frontendProjectConfig) => {
+const resolveAgUiTheme = (frontendProjectConfig)=>{
     const ux = frontendProjectConfig.metadata?.ux?.toLowerCase();
-    switch (ux) {
+    switch(ux){
         case 'cloudscape':
             return 'cloudscape';
         case 'shadcn':
@@ -106,27 +119,21 @@ const resolveAgUiTheme = (frontendProjectConfig) => {
  *   - `<varName>` into the `useMemo` deps array
  *
  * All steps are guarded so re-running produces byte-identical output.
- */
-const registerAgentHookInProvider = async (tree, frontendProjectConfig, agentNameClassName) => {
-    const providerPath = (0, devkit_1.joinPathFragments)(frontendProjectConfig.sourceRoot, 'components', 'AguiProvider.tsx');
+ */ const registerAgentHookInProvider = async (tree, frontendProjectConfig, agentNameClassName)=>{
+    const providerPath = joinPathFragments(frontendProjectConfig.sourceRoot, 'components', 'AguiProvider.tsx');
     const hookName = `useAgui${agentNameClassName}`;
     // useAguiStoryAgent -> storyAgentAgents
     const varName = `${agentNameClassName.charAt(0).toLowerCase()}${agentNameClassName.slice(1)}Agents`;
     // 1. Add the hook import
-    await (0, ast_1.addDestructuredImport)(tree, providerPath, [hookName], `../hooks/${hookName}`);
+    await addDestructuredImport(tree, providerPath, [
+        hookName
+    ], `../hooks/${hookName}`);
     // 2. Insert `const <varName> = <hookName>();` before the useMemo declaration
-    await (0, ast_1.applyGritQL)(tree, providerPath, `\`const selfManagedAgents = useMemo<Record<string, AbstractAgent>>($body, $deps);\` => ` +
-        `\`const ${varName} = ${hookName}();\nconst selfManagedAgents = useMemo<Record<string, AbstractAgent>>($body, $deps);\` ` +
-        `where { $program <: not contains \`const ${varName} = ${hookName}()\` }`);
+    await applyGritQL(tree, providerPath, `\`const selfManagedAgents = useMemo<Record<string, AbstractAgent>>($body, $deps);\` => ` + `\`const ${varName} = ${hookName}();\nconst selfManagedAgents = useMemo<Record<string, AbstractAgent>>($body, $deps);\` ` + `where { $program <: not contains \`const ${varName} = ${hookName}()\` }`);
     // 3. Spread `...<varName>` into the memoised object and add `<varName>` to
     //    the deps array. Branches on `{}`/`[]` vs populated so both the
     //    first-agent and nth-agent cases produce clean output.
-    await (0, ast_1.applyGritQL)(tree, providerPath, `\`useMemo<Record<string, AbstractAgent>>(() => ($obj), $deps)\` where {` +
-        ` $obj <: not contains \`...${varName}\`,` +
-        ` if ($obj <: \`{}\`) { $obj => \`{ ...${varName} }\` }` +
-        ` else { $obj <: \`{$items}\` where { $items += \`, ...${varName}\` } },` +
-        ` if ($deps <: \`[]\`) { $deps => \`[${varName}]\` }` +
-        ` else { $deps <: \`[$dd]\` where { $dd += \`, ${varName}\` } }` +
-        ` }`);
+    await applyGritQL(tree, providerPath, `\`useMemo<Record<string, AbstractAgent>>(() => ($obj), $deps)\` where {` + ` $obj <: not contains \`...${varName}\`,` + ` if ($obj <: \`{}\`) { $obj => \`{ ...${varName} }\` }` + ` else { $obj <: \`{$items}\` where { $items += \`, ...${varName}\` } },` + ` if ($deps <: \`[]\`) { $deps => \`[${varName}]\` }` + ` else { $deps <: \`[$dd]\` where { $dd += \`, ${varName}\` } }` + ` }`);
 };
+
 //# sourceMappingURL=generator.js.map

package/src/ts/react-website/agui/generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/ts/react-website/agui/generator.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACH,uCAOoB;AACpB,2DAAqE;AACrE,sDAAuD;AACvD,4CAI4B;AAC5B,mFAAgG;AAChG,gDAAiD;AACjD,gEAAqE;AACrE,wDAA2E;AAC3E,kEAA8E;AAiB9E;;;;;;;;;;;GAWG;AACI,MAAM,sBAAsB,GAAG,KAAK,EACzC,IAAU,EACV,OAAmC,EACnC,EAAE;IACF,MAAM,EAAE,qBAAqB,EAAE,SAAS,EAAE,kBAAkB,EAAE,IAAI,EAAE,GAClE,OAAO,CAAC;IAEV,MAAM,KAAK,GAAG,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,IAAA,wBAAY,EAAC,IAAA,6BAAiB,EAAC,IAAI,CAAC,CAAC,CAAC;IAEzD,8DAA8D;IAC9D,4EAA4E;IAC5E,+BAA+B;IAC/B,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,EAC/C,qBAAqB,CAAC,IAAI,EAC1B,EAAE,SAAS,EAAE,kBAAkB,EAAE,IAAI,EAAE,EACvC,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;IAEF,yEAAyE;IACzE,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvB,MAAM,IAAA,qCAAqB,EAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IACD,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,EAC5C,qBAAqB,CAAC,IAAI,EAC1B,EAAE,UAAU,EAAE,EACd,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;IAEF,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;QACnB,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,0CAA0C,CAAC,EACxE,IAAA,0BAAiB,EAAC,qBAAqB,CAAC,UAAU,EAAE,OAAO,CAAC,EAC5D,EAAE,EACF,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;IACJ,CAAC;IAED,MAAM,IAAA,kCAAsB,EAAC,IAAI,EAAE;QACjC,OAAO,EAAE,qBAAqB,CAAC,IAAI;KACpC,CAAC,CAAC;IAEH,uDAAuD;IACvD,MAAM,2BAA2B,CAC/B,IAAI,EACJ,qBAAqB,EACrB,kBAAkB,CACnB,CAAC;IAEF,0DAA0D;IAC1D,MAAM,WAAW,GAAG,IAAA,0BAAiB,EACnC,qBAAqB,CAAC,UAAU,EAChC,UAAU,CACX,CAAC;IACF,MAAM,IAAA,qBAAe,EACnB,IAAI,EACJ,WAAW,EACX,cAAc,EACd,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,WAAW,EACX,gIAAgI,CACjI,CAAC;IAEF,yEAAyE;IACzE,0DAA0D;IAC1D,2EAA2E;IAC3E,yEAAyE;IACzE,4DAA4D;IAC5D,IAAA,8CAA6B,EAAC,IAAI,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;IAE/D,IAAA,qCAA4B,EAC1B,IAAI,EACJ,IAAA,uBAAY,EAAC;QACX,wBAAwB;QACxB,eAAe;QACf,GAAI,CAAC,KAAK,KAAK,YAAY;YACzB,CAAC,CAAC,CAAC,oCAAoC,CAAC;YACxC,CAAC,CAAC,EAAE,CAAS;QACf,GAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,CAAS;QACxD,GAAI,CAAC,IAAI,KAAK,KAAK;YACjB,CAAC,CAAC;gBACE,gBAAgB;gBAChB,WAAW;gBACX,+CAA+C;gBAC/C,oBAAoB;gBACpB,MAAM;aACP;YACH,CAAC,CAAC,EAAE,CAAS;QACf,GAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAS;KAC/D,CAAC,EACF,IAAA,uBAAY,EAAC,CAAC,GAAI,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAS,CAAC,CAAC,CACtE,CAAC;IAEF,wEAAwE;IACxE,wEAAwE;IACxE,0EAA0E;IAC1E,4BAA4B;IAC5B,MAAM,IAAA,2DAAoC,EAAC,IAAI,EAAE;QAC/C,kBAAkB,EAAE,IAAA,iBAAS,EAAC,kBAAkB,CAAC;QACjD,kBAAkB;KACnB,CAAC,CAAC;AACL,CAAC,CAAC;AA7GW,QAAA,sBAAsB,0BA6GjC;AAEF,MAAM,gBAAgB,GAAG,CACvB,qBAA2C,EAChC,EAAE;IACb,MAAM,EAAE,GACL,qBAAqB,CAAC,QAAgB,EAAE,EAC1C,EAAE,WAAW,EAAE,CAAC;IACjB,QAAQ,EAAE,EAAE,CAAC;QACX,KAAK,YAAY;YACf,OAAO,YAAY,CAAC;QACtB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC,CAAC;AAEF;;;;;;;;;;GAUG;AACH,MAAM,2BAA2B,GAAG,KAAK,EACvC,IAAU,EACV,qBAA2C,EAC3C,kBAA0B,EAC1B,EAAE;IACF,MAAM,YAAY,GAAG,IAAA,0BAAiB,EACpC,qBAAqB,CAAC,UAAU,EAChC,YAAY,EACZ,kBAAkB,CACnB,CAAC;IACF,MAAM,QAAQ,GAAG,UAAU,kBAAkB,EAAE,CAAC;IAChD,wCAAwC;IACxC,MAAM,OAAO,GAAG,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;IAEpG,yBAAyB;IACzB,MAAM,IAAA,2BAAqB,EACzB,IAAI,EACJ,YAAY,EACZ,CAAC,QAAQ,CAAC,EACV,YAAY,QAAQ,EAAE,CACvB,CAAC;IAEF,6EAA6E;IAC7E,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,YAAY,EACZ,yFAAyF;QACvF,WAAW,OAAO,MAAM,QAAQ,yFAAyF;QACzH,4CAA4C,OAAO,MAAM,QAAQ,QAAQ,CAC5E,CAAC;IAEF,2EAA2E;IAC3E,oEAAoE;IACpE,2DAA2D;IAC3D,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,YAAY,EACZ,yEAAyE;QACvE,8BAA8B,OAAO,KAAK;QAC1C,yCAAyC,OAAO,QAAQ;QACxD,yDAAyD,OAAO,SAAS;QACzE,uCAAuC,OAAO,OAAO;QACrD,iDAAiD,OAAO,QAAQ;QAChE,IAAI,CACP,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/ts/react-website/agui/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  addDependenciesToPackageJson,\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type ProjectConfiguration,\n  type Tree,\n} from '@nx/devkit';\nimport { addAgentRuntimeToConnectionNamespace } from '../../../connection/agent-runtime-config';\nimport {\n  addDestructuredImport,\n  addSingleImport,\n  applyGritQL,\n} from '../../../utils/ast';\nimport { kebabCase } from '../../../utils/names';\nimport { getNpmScopePrefix, toScopeAlias } from '../../../utils/npm-scope';\nimport { registerPnpmBuiltDependencies } from '../../../utils/pnpm-workspace';\nimport { sharedShadcnGenerator } from '../../../utils/shared-shadcn';\nimport { withVersions } from '../../../utils/versions';\nimport { runtimeConfigGenerator } from '../runtime-config/generator';\n\nexport type AgUiAuth = 'iam' | 'cognito' | 'none';\n\ntype AgUiTheme = 'cloudscape' | 'shadcn' | 'default';\n\nexport interface AgUiReactConnectionOptions {\n  /** The React website project to connect FROM */\n  frontendProjectConfig: ProjectConfiguration;\n  /** The AG-UI agent's exposed name (used as CopilotKit's agent key) */\n  agentName: string;\n  /** PascalCase class name used for component names and runtime config keys */\n  agentNameClassName: string;\n  /** Auth scheme used by the agent */\n  auth: AgUiAuth;\n}\n\n/**\n * Wires a React website up to an AG-UI agent using CopilotKit and\n * @ag-ui/client's HttpAgent. Agent-server-language agnostic.\n *\n * On first invocation creates a shared `AguiProvider` component (with an empty\n * `selfManagedAgents` registry) and wraps `<App />` in it. Each invocation\n * AST-patches `AguiProvider` to register one more agent hook — running the\n * generator multiple times is idempotent and additive.\n *\n * Also vends a `src/components/copilot` theme module picked from the\n * website's `metadata.ux` (cloudscape / shadcn / default).\n */\nexport const addAgUiReactConnection = async (\n  tree: Tree,\n  options: AgUiReactConnectionOptions,\n) => {\n  const { frontendProjectConfig, agentName, agentNameClassName, auth } =\n    options;\n\n  const theme = resolveAgUiTheme(frontendProjectConfig);\n  const scopeAlias = toScopeAlias(getNpmScopePrefix(tree));\n\n  // Generates `src/components/AguiProvider.tsx` (if absent) and\n  // `src/hooks/useAgui<AgentName>.tsx`. Existing files are kept so re-running\n  // the generator is idempotent.\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files', 'common'),\n    frontendProjectConfig.root,\n    { agentName, agentNameClassName, auth },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Shadcn theme imports from the shared shadcn library, so it must exist.\n  if (theme === 'shadcn') {\n    await sharedShadcnGenerator(tree);\n  }\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files', theme),\n    frontendProjectConfig.root,\n    { scopeAlias },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  if (auth === 'iam') {\n    generateFiles(\n      tree,\n      joinPathFragments(\n        import.meta.dirname,\n        '../../../utils/files/website/hooks/sigv4',\n      ),\n      joinPathFragments(frontendProjectConfig.sourceRoot, 'hooks'),\n      {},\n      { overwriteStrategy: OverwriteStrategy.KeepExisting },\n    );\n  }\n\n  await runtimeConfigGenerator(tree, {\n    project: frontendProjectConfig.name,\n  });\n\n  // AST-patch AguiProvider to register this agent's hook\n  await registerAgentHookInProvider(\n    tree,\n    frontendProjectConfig,\n    agentNameClassName,\n  );\n\n  // Wrap <App /> in <AguiProvider> in main.tsx (idempotent)\n  const mainTsxPath = joinPathFragments(\n    frontendProjectConfig.sourceRoot,\n    'main.tsx',\n  );\n  await addSingleImport(\n    tree,\n    mainTsxPath,\n    'AguiProvider',\n    './components/AguiProvider',\n  );\n  await applyGritQL(\n    tree,\n    mainTsxPath,\n    `\\`<App />\\` => \\`<AguiProvider><App /></AguiProvider>\\` where { $program <: not contains \\`<AguiProvider>$_</AguiProvider>\\` }`,\n  );\n\n  // @copilotkit/react-core transitively pulls in @scarf/scarf (telemetry),\n  // which has a postinstall script. Under pnpm 11's default\n  // `strictDepBuilds=true` that script is treated as an unreviewed build and\n  // fails the install. Register it as an explicitly-rejected build so pnpm\n  // knows we've seen it and will skip it instead of erroring.\n  registerPnpmBuiltDependencies(tree, { '@scarf/scarf': false });\n\n  addDependenciesToPackageJson(\n    tree,\n    withVersions([\n      '@copilotkit/react-core',\n      '@ag-ui/client',\n      ...((theme === 'cloudscape'\n        ? ['@cloudscape-design/chat-components']\n        : []) as any),\n      ...((theme === 'shadcn' ? ['lucide-react'] : []) as any),\n      ...((auth === 'iam'\n        ? [\n            'oidc-client-ts',\n            'aws4fetch',\n            '@aws-sdk/credential-provider-cognito-identity',\n            'react-oidc-context',\n          ]\n        : []) as any),\n      ...((auth === 'cognito' ? ['react-oidc-context'] : []) as any),\n    ]),\n    withVersions([...((auth === 'iam' ? ['@smithy/types'] : []) as any)]),\n  );\n\n  // Agents only publish their runtime ARN to the 'agentcore' namespace by\n  // default, which isn't exposed to the website. Patch the agent's CDK/TF\n  // construct to also publish under 'connection' so the browser can read it\n  // from runtime-config.json.\n  await addAgentRuntimeToConnectionNamespace(tree, {\n    agentNameKebabCase: kebabCase(agentNameClassName),\n    agentNameClassName,\n  });\n};\n\nconst resolveAgUiTheme = (\n  frontendProjectConfig: ProjectConfiguration,\n): AgUiTheme => {\n  const ux = (\n    (frontendProjectConfig.metadata as any)?.ux as string | undefined\n  )?.toLowerCase();\n  switch (ux) {\n    case 'cloudscape':\n      return 'cloudscape';\n    case 'shadcn':\n      return 'shadcn';\n    default:\n      return 'default';\n  }\n};\n\n/**\n * Register this agent's hook in the shared `AguiProvider` using GritQL AST\n * patches. The provider starts out with an empty `selfManagedAgents` registry\n * (from the template); each call appends:\n *   - `import { useAgui<Name> } from '../hooks/useAgui<Name>';`\n *   - `const <varName> = useAgui<Name>();` before the `useMemo` call\n *   - `...<varName>` into the memoised object literal\n *   - `<varName>` into the `useMemo` deps array\n *\n * All steps are guarded so re-running produces byte-identical output.\n */\nconst registerAgentHookInProvider = async (\n  tree: Tree,\n  frontendProjectConfig: ProjectConfiguration,\n  agentNameClassName: string,\n) => {\n  const providerPath = joinPathFragments(\n    frontendProjectConfig.sourceRoot,\n    'components',\n    'AguiProvider.tsx',\n  );\n  const hookName = `useAgui${agentNameClassName}`;\n  // useAguiStoryAgent -> storyAgentAgents\n  const varName = `${agentNameClassName.charAt(0).toLowerCase()}${agentNameClassName.slice(1)}Agents`;\n\n  // 1. Add the hook import\n  await addDestructuredImport(\n    tree,\n    providerPath,\n    [hookName],\n    `../hooks/${hookName}`,\n  );\n\n  // 2. Insert `const <varName> = <hookName>();` before the useMemo declaration\n  await applyGritQL(\n    tree,\n    providerPath,\n    `\\`const selfManagedAgents = useMemo<Record<string, AbstractAgent>>($body, $deps);\\` => ` +\n      `\\`const ${varName} = ${hookName}();\\nconst selfManagedAgents = useMemo<Record<string, AbstractAgent>>($body, $deps);\\` ` +\n      `where { $program <: not contains \\`const ${varName} = ${hookName}()\\` }`,\n  );\n\n  // 3. Spread `...<varName>` into the memoised object and add `<varName>` to\n  //    the deps array. Branches on `{}`/`[]` vs populated so both the\n  //    first-agent and nth-agent cases produce clean output.\n  await applyGritQL(\n    tree,\n    providerPath,\n    `\\`useMemo<Record<string, AbstractAgent>>(() => ($obj), $deps)\\` where {` +\n      ` $obj <: not contains \\`...${varName}\\`,` +\n      ` if ($obj <: \\`{}\\`) { $obj => \\`{ ...${varName} }\\` }` +\n      ` else { $obj <: \\`{$items}\\` where { $items += \\`, ...${varName}\\` } },` +\n      ` if ($deps <: \\`[]\\`) { $deps => \\`[${varName}]\\` }` +\n      ` else { $deps <: \\`[$dd]\\` where { $dd += \\`, ${varName}\\` } }` +\n      ` }`,\n  );\n};\n"],"names":["addDependenciesToPackageJson","generateFiles","joinPathFragments","OverwriteStrategy","addAgentRuntimeToConnectionNamespace","addDestructuredImport","addSingleImport","applyGritQL","kebabCase","getNpmScopePrefix","toScopeAlias","registerPnpmBuiltDependencies","sharedShadcnGenerator","withVersions","runtimeConfigGenerator","addAgUiReactConnection","tree","options","frontendProjectConfig","agentName","agentNameClassName","auth","theme","resolveAgUiTheme","scopeAlias","dirname","root","overwriteStrategy","KeepExisting","sourceRoot","project","name","registerAgentHookInProvider","mainTsxPath","agentNameKebabCase","ux","metadata","toLowerCase","providerPath","hookName","varName","charAt","slice"],"mappings":"AAAA;;;CAGC,GACD,SACEA,4BAA4B,EAC5BC,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,QAGZ,aAAa;AACpB,SAASC,oCAAoC,QAAQ,8CAA2C;AAChG,SACEC,qBAAqB,EACrBC,eAAe,EACfC,WAAW,QACN,wBAAqB;AAC5B,SAASC,SAAS,QAAQ,0BAAuB;AACjD,SAASC,iBAAiB,EAAEC,YAAY,QAAQ,8BAA2B;AAC3E,SAASC,6BAA6B,QAAQ,mCAAgC;AAC9E,SAASC,qBAAqB,QAAQ,kCAA+B;AACrE,SAASC,YAAY,QAAQ,6BAA0B;AACvD,SAASC,sBAAsB,QAAQ,iCAA8B;AAiBrE;;;;;;;;;;;CAWC,GACD,OAAO,MAAMC,yBAAyB,OACpCC,MACAC;IAEA,MAAM,EAAEC,qBAAqB,EAAEC,SAAS,EAAEC,kBAAkB,EAAEC,IAAI,EAAE,GAClEJ;IAEF,MAAMK,QAAQC,iBAAiBL;IAC/B,MAAMM,aAAad,aAAaD,kBAAkBO;IAElD,8DAA8D;IAC9D,4EAA4E;IAC5E,+BAA+B;IAC/Bf,cACEe,MACAd,kBAAkB,YAAYuB,OAAO,EAAE,SAAS,WAChDP,sBAAsBQ,IAAI,EAC1B;QAAEP;QAAWC;QAAoBC;IAAK,GACtC;QAAEM,mBAAmBxB,kBAAkByB,YAAY;IAAC;IAGtD,yEAAyE;IACzE,IAAIN,UAAU,UAAU;QACtB,MAAMV,sBAAsBI;IAC9B;IACAf,cACEe,MACAd,kBAAkB,YAAYuB,OAAO,EAAE,SAASH,QAChDJ,sBAAsBQ,IAAI,EAC1B;QAAEF;IAAW,GACb;QAAEG,mBAAmBxB,kBAAkByB,YAAY;IAAC;IAGtD,IAAIP,SAAS,OAAO;QAClBpB,cACEe,MACAd,kBACE,YAAYuB,OAAO,EACnB,6CAEFvB,kBAAkBgB,sBAAsBW,UAAU,EAAE,UACpD,CAAC,GACD;YAAEF,mBAAmBxB,kBAAkByB,YAAY;QAAC;IAExD;IAEA,MAAMd,uBAAuBE,MAAM;QACjCc,SAASZ,sBAAsBa,IAAI;IACrC;IAEA,uDAAuD;IACvD,MAAMC,4BACJhB,MACAE,uBACAE;IAGF,0DAA0D;IAC1D,MAAMa,cAAc/B,kBAClBgB,sBAAsBW,UAAU,EAChC;IAEF,MAAMvB,gBACJU,MACAiB,aACA,gBACA;IAEF,MAAM1B,YACJS,MACAiB,aACA,CAAC,8HAA8H,CAAC;IAGlI,yEAAyE;IACzE,0DAA0D;IAC1D,2EAA2E;IAC3E,yEAAyE;IACzE,4DAA4D;IAC5DtB,8BAA8BK,MAAM;QAAE,gBAAgB;IAAM;IAE5DhB,6BACEgB,MACAH,aAAa;QACX;QACA;WACKS,UAAU,eACX;YAAC;SAAqC,GACtC,EAAE;WACDA,UAAU,WAAW;YAAC;SAAe,GAAG,EAAE;WAC1CD,SAAS,QACV;YACE;YACA;YACA;YACA;SACD,GACD,EAAE;WACDA,SAAS,YAAY;YAAC;SAAqB,GAAG,EAAE;KACtD,GACDR,aAAa;WAAMQ,SAAS,QAAQ;YAAC;SAAgB,GAAG,EAAE;KAAU;IAGtE,wEAAwE;IACxE,wEAAwE;IACxE,0EAA0E;IAC1E,4BAA4B;IAC5B,MAAMjB,qCAAqCY,MAAM;QAC/CkB,oBAAoB1B,UAAUY;QAC9BA;IACF;AACF,EAAE;AAEF,MAAMG,mBAAmB,CACvBL;IAEA,MAAMiB,KACHjB,sBAAsBkB,QAAQ,EAAUD,IACxCE;IACH,OAAQF;QACN,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;QACT;YACE,OAAO;IACX;AACF;AAEA;;;;;;;;;;CAUC,GACD,MAAMH,8BAA8B,OAClChB,MACAE,uBACAE;IAEA,MAAMkB,eAAepC,kBACnBgB,sBAAsBW,UAAU,EAChC,cACA;IAEF,MAAMU,WAAW,CAAC,OAAO,EAAEnB,oBAAoB;IAC/C,wCAAwC;IACxC,MAAMoB,UAAU,GAAGpB,mBAAmBqB,MAAM,CAAC,GAAGJ,WAAW,KAAKjB,mBAAmBsB,KAAK,CAAC,GAAG,MAAM,CAAC;IAEnG,yBAAyB;IACzB,MAAMrC,sBACJW,MACAsB,cACA;QAACC;KAAS,EACV,CAAC,SAAS,EAAEA,UAAU;IAGxB,6EAA6E;IAC7E,MAAMhC,YACJS,MACAsB,cACA,CAAC,uFAAuF,CAAC,GACvF,CAAC,QAAQ,EAAEE,QAAQ,GAAG,EAAED,SAAS,uFAAuF,CAAC,GACzH,CAAC,yCAAyC,EAAEC,QAAQ,GAAG,EAAED,SAAS,MAAM,CAAC;IAG7E,2EAA2E;IAC3E,oEAAoE;IACpE,2DAA2D;IAC3D,MAAMhC,YACJS,MACAsB,cACA,CAAC,uEAAuE,CAAC,GACvE,CAAC,2BAA2B,EAAEE,QAAQ,GAAG,CAAC,GAC1C,CAAC,sCAAsC,EAAEA,QAAQ,MAAM,CAAC,GACxD,CAAC,sDAAsD,EAAEA,QAAQ,OAAO,CAAC,GACzE,CAAC,oCAAoC,EAAEA,QAAQ,KAAK,CAAC,GACrD,CAAC,8CAA8C,EAAEA,QAAQ,MAAM,CAAC,GAChE,CAAC,EAAE,CAAC;AAEV"}