@aws/nx-plugin 1.0.0-rc.4 → 1.0.0-rc.41

package/src/ts/rdb/trpc-connection/schema.d.ts

@@ -9,4 +9,5 @@
 export interface TsRdbTrpcConnectionGeneratorSchema {
   sourceProject: string;
   targetProject: string;
+  preferInstallDependencies?: boolean;
 }

package/src/ts/rdb/trpc-connection/schema.json

@@ -12,6 +12,11 @@
     "targetProject": {
       "type": "string",
       "description": "The ts#rdb project to connect to"
+    },
+    "preferInstallDependencies": {
+      "type": "boolean",
+      "description": "Whether to prefer installing dependencies after the generator runs. Set to false to defer installing when batching multiple generators (an install still runs if needed so subsequent generators can compute the Nx project graph); install once at the end.",
+      "default": true
     }
   },
   "required": ["sourceProject", "targetProject"]

package/src/ts/rdb/utils.d.ts

@@ -2,5 +2,5 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
-import { Tree } from '@nx/devkit';
+import type { Tree } from '@nx/devkit';
 export declare const injectRdsCaBundleIntoDockerfile: (tree: Tree, dockerfilePath: string) => void;

package/src/ts/rdb/utils.js

@@ -1,20 +1,18 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.injectRdsCaBundleIntoDockerfile = void 0;
-const RDS_CA_RUN = [
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ const RDS_CA_RUN = [
     'RUN apt-get update && apt-get install -y --no-install-recommends curl ca-certificates && \\',
     '    rm -rf /var/lib/apt/lists/* && \\',
     '    curl -fsSL "https://truststore.pki.rds.amazonaws.com/global/global-bundle.pem" \\',
     '    -o /usr/local/share/ca-certificates/rds-bundle.crt && \\',
-    '    update-ca-certificates',
+    '    update-ca-certificates'
 ].join('\n');
-const injectRdsCaBundleIntoDockerfile = (tree, dockerfilePath) => {
-    if (!tree.exists(dockerfilePath))
-        return;
+export const injectRdsCaBundleIntoDockerfile = (tree, dockerfilePath)=>{
+    if (!tree.exists(dockerfilePath)) return;
     const content = tree.read(dockerfilePath, 'utf-8');
-    if (content.includes('rds-bundle.crt'))
-        return;
+    if (content.includes('rds-bundle.crt')) return;
     tree.write(dockerfilePath, content.replace(/^(EXPOSE \d+)/m, `${RDS_CA_RUN}\n\n$1`));
 };
-exports.injectRdsCaBundleIntoDockerfile = injectRdsCaBundleIntoDockerfile;
+
 //# sourceMappingURL=utils.js.map

package/src/ts/rdb/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../../../../packages/nx-plugin/src/ts/rdb/utils.ts"],"names":[],"mappings":";;;AAMA,MAAM,UAAU,GAAG;IACjB,6FAA6F;IAC7F,uCAAuC;IACvC,uFAAuF;IACvF,8DAA8D;IAC9D,4BAA4B;CAC7B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEN,MAAM,+BAA+B,GAAG,CAC7C,IAAU,EACV,cAAsB,EAChB,EAAE;IACR,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;QAAE,OAAO;IACzC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAE,CAAC;IACpD,IAAI,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QAAE,OAAO;IAC/C,IAAI,CAAC,KAAK,CACR,cAAc,EACd,OAAO,CAAC,OAAO,CAAC,gBAAgB,EAAE,GAAG,UAAU,QAAQ,CAAC,CACzD,CAAC;AACJ,CAAC,CAAC;AAXW,QAAA,+BAA+B,mCAW1C"}
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/ts/rdb/utils.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport type { Tree } from '@nx/devkit';\n\nconst RDS_CA_RUN = [\n  'RUN apt-get update && apt-get install -y --no-install-recommends curl ca-certificates && \\\\',\n  '    rm -rf /var/lib/apt/lists/* && \\\\',\n  '    curl -fsSL \"https://truststore.pki.rds.amazonaws.com/global/global-bundle.pem\" \\\\',\n  '    -o /usr/local/share/ca-certificates/rds-bundle.crt && \\\\',\n  '    update-ca-certificates',\n].join('\\n');\n\nexport const injectRdsCaBundleIntoDockerfile = (\n  tree: Tree,\n  dockerfilePath: string,\n): void => {\n  if (!tree.exists(dockerfilePath)) return;\n  const content = tree.read(dockerfilePath, 'utf-8')!;\n  if (content.includes('rds-bundle.crt')) return;\n  tree.write(\n    dockerfilePath,\n    content.replace(/^(EXPOSE \\d+)/m, `${RDS_CA_RUN}\\n\\n$1`),\n  );\n};\n"],"names":["RDS_CA_RUN","join","injectRdsCaBundleIntoDockerfile","tree","dockerfilePath","exists","content","read","includes","write","replace"],"mappings":"AAAA;;;CAGC,GAGD,MAAMA,aAAa;IACjB;IACA;IACA;IACA;IACA;CACD,CAACC,IAAI,CAAC;AAEP,OAAO,MAAMC,kCAAkC,CAC7CC,MACAC;IAEA,IAAI,CAACD,KAAKE,MAAM,CAACD,iBAAiB;IAClC,MAAME,UAAUH,KAAKI,IAAI,CAACH,gBAAgB;IAC1C,IAAIE,QAAQE,QAAQ,CAAC,mBAAmB;IACxCL,KAAKM,KAAK,CACRL,gBACAE,QAAQI,OAAO,CAAC,kBAAkB,GAAGV,WAAW,MAAM,CAAC;AAE3D,EAAE"}

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"}