@aws/nx-plugin 1.0.0-rc.96 → 1.0.0-rc.98

package/src/ts/rdb/generator.d.ts

@@ -93,6 +93,14 @@ export declare const DEPENDENCIES: import("../../utils/declared-dependencies.js"
     readonly name: "minimatch";
 } & {
     versionOnly: true;
+}, {
+    readonly name: "deepmerge-ts";
+} & {
+    versionOnly: true;
+}, {
+    readonly name: "mysql2";
+} & {
+    versionOnly: true;
 }], readonly import("../../utils/declared-dependencies.js").DeclaredPyDependency<"ruff" | "a2a-sdk" | "ag-ui-langgraph" | "ag-ui-protocol" | "ag-ui-strands" | "aiosqlite" | "aws-lambda-powertools" | "aws-lambda-powertools[tracer]" | "aws-lambda-powertools[parser]" | "aws-opentelemetry-distro" | "bedrock-agentcore" | "boto3" | "checkov" | "fastapi" | "fastapi[standard]" | "httpx" | "langchain" | "langchain-aws" | "langchain-mcp-adapters" | "langgraph" | "langgraph-checkpoint-aws" | "langgraph-checkpoint-sqlite" | "mcp" | "pip-check-updates" | "pip-licenses" | "strands-agents" | "strands-agents[a2a]" | "strands-agents-tools" | "ty" | "pynamodb" | "uvicorn" | "sqlmodel" | "alembic" | "aiomysql" | "asyncpg" | "greenlet", TsRdbMetadata>[]>;
 export declare const TS_RDB_GENERATOR_INFO: NxGeneratorInfo;
 export declare const tsRdbGenerator: (tree: Tree, options: TsRdbGeneratorSchema) => Promise<GeneratorCallback>;

package/src/ts/rdb/generator.js

@@ -8,7 +8,7 @@ import { addStarExport } from "../../utils/ast.js";
 import { addTypeScriptBundleTarget, BUNDLE_DEPENDENCIES } from "../../utils/bundle/bundle.js";
 import { resolveContainers } from "../../utils/containers.js";
 import { declareDependencies, ownedElsewhere } from "../../utils/declared-dependencies.js";
-import { addDockerScanTarget, DOCKER_DEPENDENCIES, IMAGE_BUILD_CACHE, NODE_IMAGE_DEPENDENCIES, nodeImageVersions } from "../../utils/docker.js";
+import { addDockerScanTarget, DOCKER_DEPENDENCIES, IMAGE_BUILD_CACHE, NODE_IMAGE_DEPENDENCIES, nodeImageVersions, PRISMA_IMAGE_DEPENDENCIES } from "../../utils/docker.js";
 import { formatFilesInSubtree } from "../../utils/format.js";
 import { FS_DEPENDENCIES, FsCommands } from "../../utils/fs.js";
 import { updateGitIgnore } from "../../utils/git.js";
@@ -88,7 +88,8 @@ export const DEPENDENCIES = declareDependencies()({
         ...ownedElsewhere(DOCKER_DEPENDENCIES),
         ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES),
         ...ownedElsewhere(SHARED_RDB_SCRIPTS_DEPENDENCIES),
-        ...ownedElsewhere(NODE_IMAGE_DEPENDENCIES)
+        ...ownedElsewhere(NODE_IMAGE_DEPENDENCIES),
+        ...ownedElsewhere(PRISMA_IMAGE_DEPENDENCIES)
     ]
 });
 export const TS_RDB_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
@@ -138,6 +139,8 @@ export const tsRdbGenerator = async (tree, options)=>{
         databasePackageAlias: fullyQualifiedName,
         databaseProvider: options.engine === 'mysql' ? 'mysql' : 'postgresql',
         prismaVersion: TS_VERSIONS.prisma,
+        deepmergeTsVersion: TS_VERSIONS['deepmerge-ts'],
+        mysql2Version: TS_VERSIONS.mysql2,
         ...nodeImageVersions(),
         prismaAdapterPackage: options.engine === 'mysql' ? '@prisma/adapter-mariadb' : '@prisma/adapter-pg',
         prismaAdapterClassName: options.engine === 'mysql' ? 'PrismaMariaDb' : 'PrismaPg',
@@ -276,28 +279,29 @@ export const tsRdbGenerator = async (tree, options)=>{
     // which case neither provider's packages were added.
     const iac = options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;
     if (options.infra !== 'none') {
-        if (iac === 'terraform') {
-            projectConfig.targets['docker'] = {
-                cache: IMAGE_BUILD_CACHE,
-                executor: 'nx:run-commands',
-                options: {
-                    command: `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`
-                },
-                dependsOn: [
-                    'bundle'
-                ]
-            };
-            addArtifactDependencyToTargets(projectConfig, 'docker');
-            addDockerScanTarget(tree, {
-                project: projectConfig,
-                containerEngine,
-                trivyTargetName: 'trivy',
-                dockerTargetName: 'docker',
-                imageTags: [
-                    migrationDockerImageTag
-                ]
-            }, DEPENDENCIES);
-        }
+        // The migration image ships under both providers — Terraform pushes the
+        // locally built tag, CDK builds the same context as an image asset — so the
+        // build and its scan are wired up for either.
+        projectConfig.targets['docker'] = {
+            cache: IMAGE_BUILD_CACHE,
+            executor: 'nx:run-commands',
+            options: {
+                command: `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`
+            },
+            dependsOn: [
+                'bundle'
+            ]
+        };
+        addArtifactDependencyToTargets(projectConfig, 'docker');
+        addDockerScanTarget(tree, {
+            project: projectConfig,
+            containerEngine,
+            trivyTargetName: 'trivy',
+            dockerTargetName: 'docker',
+            imageTags: [
+                migrationDockerImageTag
+            ]
+        }, DEPENDENCIES);
         addArtifactDependencyToTargets(projectConfig, 'bundle');
     }
     addDependencyToTargetIfNotPresent(projectConfig, 'compile', 'generate');

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../packages/nx-plugin/src/ts/rdb/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport { relative } from 'node:path';\nimport {\n  detectPackageManager,\n  type GeneratorCallback,\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  readProjectConfiguration,\n  type Tree,\n  updateJson,\n  updateProjectConfiguration,\n  writeJson,\n} from '@nx/devkit';\nimport { addTsDependencies } from '../../utils/add-dependencies.js';\nimport { addStarExport } from '../../utils/ast.js';\nimport {\n  addTypeScriptBundleTarget,\n  BUNDLE_DEPENDENCIES,\n} from '../../utils/bundle/bundle.js';\nimport { resolveContainers } from '../../utils/containers.js';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../utils/declared-dependencies.js';\nimport {\n  addDockerScanTarget,\n  DOCKER_DEPENDENCIES,\n  IMAGE_BUILD_CACHE,\n  NODE_IMAGE_DEPENDENCIES,\n  nodeImageVersions,\n} from '../../utils/docker.js';\nimport { formatFilesInSubtree } from '../../utils/format.js';\nimport { FS_DEPENDENCIES, FsCommands } from '../../utils/fs.js';\nimport { updateGitIgnore } from '../../utils/git.js';\nimport { resolveIac } from '../../utils/iac.js';\nimport { installDependencies } from '../../utils/install.js';\nimport { addGeneratorMetricsIfApplicable } from '../../utils/metrics.js';\nimport { esmVars } from '../../utils/module-format.js';\nimport { kebabCase, snakeCase, toClassName } from '../../utils/names.js';\nimport { getNpmScope } from '../../utils/npm-scope.js';\nimport {\n  addArtifactDependencyToTargets,\n  addDependencyToTargetIfNotPresent,\n  addGeneratorMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n} from '../../utils/nx.js';\nimport { getRelativePathToRootByDirectory } from '../../utils/paths.js';\nimport { registerPnpmBuiltDependencies } from '../../utils/pnpm-workspace.js';\nimport { assignPort } from '../../utils/port.js';\nimport { addRdbInfra } from '../../utils/rdb-constructs/rdb-constructs.js';\nimport {\n  SHARED_CONSTRUCTS_DEPENDENCIES,\n  sharedConstructsGenerator,\n} from '../../utils/shared-constructs.js';\nimport {\n  type IacMetadata,\n  PACKAGES_DIR,\n  SHARED_SCRIPTS_DIR,\n} from '../../utils/shared-constructs-constants.js';\nimport {\n  SHARED_RDB_SCRIPTS_DEPENDENCIES,\n  sharedRdbScriptsGenerator,\n} from '../../utils/shared-rdb-scripts.js';\nimport { TS_VERSIONS } from '../../utils/versions.js';\nimport tsProjectGenerator, { getTsLibDetails } from '../lib/generator.js';\nimport type { TsRdbGeneratorSchema } from './schema';\n\n/** The metadata this generator records, which its predicates read. */\nexport interface TsRdbMetadata extends IacMetadata {\n  readonly engine: TsRdbGeneratorSchema['engine'];\n}\n\n// Each entry names the engine branch it belongs to, so the same declaration\n// drives both adding and the version sync.\nexport const DEPENDENCIES = declareDependencies<TsRdbMetadata>()({\n  ts: [\n    { name: '@aws-lambda-powertools/parameters' },\n    { name: '@aws-sdk/client-appconfigdata' },\n    { name: '@aws-sdk/client-secrets-manager' },\n    { name: '@aws-sdk/rds-signer' },\n    { name: '@prisma/client' },\n    { name: '@prisma/adapter-mariadb', when: (m) => m.engine === 'mysql' },\n    { name: 'mariadb', when: (m) => m.engine === 'mysql' },\n    { name: '@prisma/adapter-pg', when: (m) => m.engine !== 'mysql' },\n    { name: 'pg', when: (m) => m.engine !== 'mysql' },\n    { name: '@types/aws-lambda', dev: true },\n    { name: '@types/pg', when: (m) => m.engine !== 'mysql', dev: true },\n    // The prisma CLI and tsx run migration/seed scripts from the root.\n    { name: 'prisma', dev: true, root: true },\n    { name: 'tsx', dev: true, root: true },\n    ...ownedElsewhere(BUNDLE_DEPENDENCIES),\n    ...ownedElsewhere(FS_DEPENDENCIES),\n    ...ownedElsewhere(DOCKER_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_RDB_SCRIPTS_DEPENDENCIES),\n    ...ownedElsewhere(NODE_IMAGE_DEPENDENCIES),\n  ],\n});\n\nexport const TS_RDB_GENERATOR_INFO: NxGeneratorInfo = getGeneratorInfo(\n  import.meta.filename,\n);\n\nexport const tsRdbGenerator = async (\n  tree: Tree,\n  options: TsRdbGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const nameKebabCase = kebabCase(options.name);\n  const nameClassName = toClassName(options.name);\n  const databaseUser = options.databaseUser ?? 'dbadmin';\n  const databaseName = snakeCase(options.databaseName ?? options.name);\n  const containerEngine = await resolveContainers(tree, 'inherit');\n  const { fullyQualifiedName, dir } = getTsLibDetails(tree, {\n    name: options.name,\n    directory: options.directory,\n    subDirectory: options.subDirectory,\n  });\n\n  let projectExists: boolean;\n  try {\n    readProjectConfiguration(tree, fullyQualifiedName);\n    projectExists = true;\n  } catch {\n    projectExists = false;\n  }\n\n  if (!projectExists) {\n    await tsProjectGenerator(tree, {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n      preferInstallDependencies: false,\n    });\n  }\n\n  updateJson(tree, joinPathFragments(dir, 'tsconfig.lib.json'), (tsConfig) => ({\n    ...tsConfig,\n    include: ['src/**/*.ts', 'generated/prisma/**/*.ts'],\n  }));\n\n  const projectConfig = readProjectConfiguration(tree, fullyQualifiedName);\n  const localDbPort = assignPort(\n    tree,\n    projectConfig,\n    options.engine === 'mysql' ? 3306 : 5432,\n  );\n  const localDbHost = 'localhost';\n  const localDbUser = options.engine === 'mysql' ? 'root' : 'dbadmin';\n  const localDbPassword = 'password';\n  const containerName = `${getNpmScope(tree)}-${databaseName}`;\n  const dockerImage =\n    options.engine === 'mysql'\n      ? 'public.ecr.aws/docker/library/mysql:8.0.44'\n      : 'public.ecr.aws/docker/library/postgres:17.7';\n\n  const templateOptions = {\n    engine: options.engine,\n    runtimeConfigKey: nameClassName,\n    databasePackageAlias: fullyQualifiedName,\n    databaseProvider: options.engine === 'mysql' ? 'mysql' : 'postgresql',\n    prismaVersion: TS_VERSIONS.prisma,\n    ...nodeImageVersions(),\n    prismaAdapterPackage:\n      options.engine === 'mysql'\n        ? '@prisma/adapter-mariadb'\n        : '@prisma/adapter-pg',\n    prismaAdapterClassName:\n      options.engine === 'mysql' ? 'PrismaMariaDb' : 'PrismaPg',\n    localDbPort,\n    localDbHost,\n    localDbName: databaseName,\n    localDbUser,\n    localDbPassword,\n    containerEngine,\n    containerName,\n    dockerImage,\n    ...esmVars(tree),\n  };\n\n  // The Prisma schema and models, the handlers and the prisma client module are\n  // all user-editable, so a re-run (to add infrastructure, or after a failed\n  // run) must not clobber the user's schema and edits.\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    dir,\n    templateOptions,\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // The barrel re-exports prisma.ts. `tsProjectGenerator` writes a stub index.ts\n  // when it creates the project, so the re-export is added to whatever is there\n  // rather than replacing it — which also leaves a user's own additions to the\n  // barrel alone on a re-run.\n  await addStarExport(\n    tree,\n    joinPathFragments(dir, 'src', 'index.ts'),\n    './prisma.js',\n  );\n\n  // config.json is framework-owned: it carries values derived from the\n  // generator's options, so it converges on a re-run with changed options.\n  writeJson(tree, joinPathFragments(dir, 'config.json'), {\n    runtimeConfigKey: nameClassName,\n    localDev: {\n      containerEngine,\n      containerName,\n      image: dockerImage,\n      port: localDbPort,\n      host: localDbHost,\n      dbName: databaseName,\n      dbUser: localDbUser,\n      dbPassword: localDbPassword,\n      dbEngine: options.engine,\n    },\n  });\n  updateGitIgnore(tree, dir, (patterns) => [...patterns, 'generated/prisma']);\n  await sharedRdbScriptsGenerator(tree, options.engine, DEPENDENCIES);\n  const waitForDbScript =\n    options.engine === 'mysql'\n      ? 'wait-for-mysql-db.ts'\n      : 'wait-for-postgres-db.ts';\n  const scriptsDir = relative(\n    dir,\n    joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'rdb'),\n  );\n  const relativePathToRoot = getRelativePathToRootByDirectory(\n    projectConfig.root,\n  );\n  const fs = new FsCommands(tree, DEPENDENCIES);\n  const migrationBundleDir = joinPathFragments(\n    'dist',\n    projectConfig.root,\n    'bundle',\n    'migration',\n  );\n  const migrationDockerImageTag = `${getNpmScope(tree)}-${kebabCase(options.name)}-migration:latest`;\n\n  if (options.infra !== 'none') {\n    await addTypeScriptBundleTarget(\n      tree,\n      projectConfig,\n      {\n        targetFilePath: 'src/migration-handler.ts',\n        bundleOutputDir: 'migration',\n      },\n      DEPENDENCIES,\n    );\n    await addTypeScriptBundleTarget(\n      tree,\n      projectConfig,\n      {\n        targetFilePath: 'src/create-db-user-handler.ts',\n        bundleOutputDir: 'create-db-user',\n      },\n      DEPENDENCIES,\n    );\n    const bundleTarget = projectConfig.targets['bundle'];\n    // The bundle target starts with a single rolldown `command`. Wrap it with\n    // the migration asset copy steps, unless it has already been transformed\n    // into a `commands` array on a previous run.\n    if (!bundleTarget.options.commands) {\n      const rolldownCommand = bundleTarget.options.command;\n      delete bundleTarget.options.command;\n      bundleTarget.options = {\n        ...bundleTarget.options,\n        commands: [\n          fs.rm(`${relativePathToRoot}dist/{projectRoot}/bundle/migration`),\n          fs.mkdir(`${relativePathToRoot}dist/{projectRoot}/bundle/migration`),\n          fs.cpDir(\n            'prisma',\n            `${relativePathToRoot}dist/{projectRoot}/bundle/migration/prisma`,\n          ),\n          fs.cpFile(\n            'prisma.config.ts',\n            `${relativePathToRoot}dist/{projectRoot}/bundle/migration/prisma.config.ts`,\n          ),\n          fs.cpFile(\n            'Dockerfile',\n            `${relativePathToRoot}dist/{projectRoot}/bundle/migration/Dockerfile`,\n          ),\n          rolldownCommand,\n        ],\n        parallel: false,\n      };\n    }\n  }\n\n  projectConfig.targets.generate = {\n    executor: 'nx:run-commands',\n    outputs: ['{projectRoot}/generated/prisma'],\n    options: {\n      command: 'prisma generate',\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets['pull-image'] = {\n    executor: 'nx:run-commands',\n    options: {\n      command: `tsx ${scriptsDir}/pull-image.ts`,\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets['dev'] = {\n    executor: 'nx:run-commands',\n    dependsOn: ['pull-image'],\n    options: {\n      command: `tsx ${scriptsDir}/start-container.ts`,\n      cwd: '{projectRoot}',\n    },\n    continuous: true,\n  };\n  projectConfig.targets['wait-for-db'] = {\n    executor: 'nx:run-commands',\n    dependsOn: ['dev'],\n    options: {\n      command: `tsx ${scriptsDir}/${waitForDbScript}`,\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets.prisma = {\n    executor: 'nx:run-commands',\n    dependsOn: ['dev', 'wait-for-db'],\n    options: {\n      cwd: '{projectRoot}',\n      command: 'prisma',\n      env: {\n        LOCAL_DEV: 'true',\n      },\n    },\n  };\n  // Recorded in the metadata below so the version sync can tell a CDK project\n  // from a Terraform one; undefined when no infrastructure was generated, in\n  // which case neither provider's packages were added.\n  const iac =\n    options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;\n\n  if (options.infra !== 'none') {\n    if (iac === 'terraform') {\n      projectConfig.targets['docker'] = {\n        cache: IMAGE_BUILD_CACHE,\n        executor: 'nx:run-commands',\n        options: {\n          command: `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`,\n        },\n        dependsOn: ['bundle'],\n      };\n      addArtifactDependencyToTargets(projectConfig, 'docker');\n\n      addDockerScanTarget(\n        tree,\n        {\n          project: projectConfig,\n          containerEngine,\n          trivyTargetName: 'trivy',\n          dockerTargetName: 'docker',\n          imageTags: [migrationDockerImageTag],\n        },\n        DEPENDENCIES,\n      );\n    }\n    addArtifactDependencyToTargets(projectConfig, 'bundle');\n  }\n  addDependencyToTargetIfNotPresent(projectConfig, 'compile', 'generate');\n  updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);\n  // Recorded here and read by the declaration's predicates, so the packages\n  // added below are exactly the ones the version sync will own.\n  const metadata: TsRdbMetadata = {\n    engine: options.engine,\n    ...(iac ? { iac } : {}),\n  };\n  addGeneratorMetadata(\n    tree,\n    fullyQualifiedName,\n    TS_RDB_GENERATOR_INFO,\n    metadata,\n  );\n\n  if (options.infra !== 'none') {\n    const iac = await resolveIac(tree, options.iac);\n    await sharedConstructsGenerator(tree, { iac }, DEPENDENCIES);\n    await addRdbInfra(tree, {\n      iac,\n      projectName: fullyQualifiedName,\n      projectRoot: dir,\n      nameClassName,\n      nameKebabCase,\n      databasePackageAlias: fullyQualifiedName,\n      databaseName,\n      adminUser: databaseUser,\n      engine: options.engine,\n      migrationBundleDir,\n      createDbUserBundleDir: joinPathFragments(\n        'dist',\n        projectConfig.root,\n        'bundle',\n        'create-db-user',\n      ),\n      framework: options.framework,\n      migrationDockerImageTag,\n      containerEngine,\n    });\n  }\n\n  addTsDependencies(tree, DEPENDENCIES, {\n    metadata,\n    projectRoot: projectConfig.root,\n  });\n\n  // @prisma/adapter-pg depends on @types/pg ^8.16.0. Yarn does not dedupe it to\n  // the workspace's pinned @types/pg, so it installs a separate copy under the\n  // adapter's own node_modules whenever a newer @types/pg is published. The two\n  // copies declare structurally incompatible Pool types, so passing our `new\n  // Pool(...)` to `new PrismaPg(...)` fails to compile. Scope the resolution to\n  // the adapter so other @types/pg consumers are unaffected. Classic yarn only\n  // honours the `**/`-prefixed descriptor in a workspace and berry only the\n  // bare one, so declare both.\n  if (options.engine === 'postgres' && detectPackageManager() === 'yarn') {\n    updateJson(tree, 'package.json', (packageJson) => {\n      packageJson.resolutions = {\n        ...packageJson.resolutions,\n        '**/@prisma/adapter-pg/@types/pg': TS_VERSIONS['@types/pg'],\n        '@prisma/adapter-pg/@types/pg': TS_VERSIONS['@types/pg'],\n      };\n      return packageJson;\n    });\n  }\n\n  registerPnpmBuiltDependencies(tree, {\n    '@prisma/engines': false,\n    prisma: false,\n  });\n\n  await addGeneratorMetricsIfApplicable(tree, [TS_RDB_GENERATOR_INFO]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript'],\n    });\n};\n\nexport default tsRdbGenerator;\n"],"names":["relative","detectPackageManager","generateFiles","joinPathFragments","OverwriteStrategy","readProjectConfiguration","updateJson","updateProjectConfiguration","writeJson","addTsDependencies","addStarExport","addTypeScriptBundleTarget","BUNDLE_DEPENDENCIES","resolveContainers","declareDependencies","ownedElsewhere","addDockerScanTarget","DOCKER_DEPENDENCIES","IMAGE_BUILD_CACHE","NODE_IMAGE_DEPENDENCIES","nodeImageVersions","formatFilesInSubtree","FS_DEPENDENCIES","FsCommands","updateGitIgnore","resolveIac","installDependencies","addGeneratorMetricsIfApplicable","esmVars","kebabCase","snakeCase","toClassName","getNpmScope","addArtifactDependencyToTargets","addDependencyToTargetIfNotPresent","addGeneratorMetadata","getGeneratorInfo","getRelativePathToRootByDirectory","registerPnpmBuiltDependencies","assignPort","addRdbInfra","SHARED_CONSTRUCTS_DEPENDENCIES","sharedConstructsGenerator","PACKAGES_DIR","SHARED_SCRIPTS_DIR","SHARED_RDB_SCRIPTS_DEPENDENCIES","sharedRdbScriptsGenerator","TS_VERSIONS","tsProjectGenerator","getTsLibDetails","DEPENDENCIES","ts","name","when","m","engine","dev","root","TS_RDB_GENERATOR_INFO","filename","tsRdbGenerator","tree","options","nameKebabCase","nameClassName","databaseUser","databaseName","containerEngine","fullyQualifiedName","dir","directory","subDirectory","projectExists","preferInstallDependencies","tsConfig","include","projectConfig","localDbPort","localDbHost","localDbUser","localDbPassword","containerName","dockerImage","templateOptions","runtimeConfigKey","databasePackageAlias","databaseProvider","prismaVersion","prisma","prismaAdapterPackage","prismaAdapterClassName","localDbName","dirname","overwriteStrategy","KeepExisting","localDev","image","port","host","dbName","dbUser","dbPassword","dbEngine","patterns","waitForDbScript","scriptsDir","relativePathToRoot","fs","migrationBundleDir","migrationDockerImageTag","infra","targetFilePath","bundleOutputDir","bundleTarget","targets","commands","rolldownCommand","command","rm","mkdir","cpDir","cpFile","parallel","generate","executor","outputs","cwd","dependsOn","continuous","env","LOCAL_DEV","iac","undefined","cache","project","trivyTargetName","dockerTargetName","imageTags","metadata","projectName","projectRoot","adminUser","createDbUserBundleDir","framework","packageJson","resolutions","languages"],"mappings":"AAAA;;;CAGC,GACD,SAASA,QAAQ,QAAQ,YAAY;AACrC,SACEC,oBAAoB,EAEpBC,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EACjBC,wBAAwB,EAExBC,UAAU,EACVC,0BAA0B,EAC1BC,SAAS,QACJ,aAAa;AACpB,SAASC,iBAAiB,QAAQ,kCAAkC;AACpE,SAASC,aAAa,QAAQ,qBAAqB;AACnD,SACEC,yBAAyB,EACzBC,mBAAmB,QACd,+BAA+B;AACtC,SAASC,iBAAiB,QAAQ,4BAA4B;AAC9D,SACEC,mBAAmB,EACnBC,cAAc,QACT,uCAAuC;AAC9C,SACEC,mBAAmB,EACnBC,mBAAmB,EACnBC,iBAAiB,EACjBC,uBAAuB,EACvBC,iBAAiB,QACZ,wBAAwB;AAC/B,SAASC,oBAAoB,QAAQ,wBAAwB;AAC7D,SAASC,eAAe,EAAEC,UAAU,QAAQ,oBAAoB;AAChE,SAASC,eAAe,QAAQ,qBAAqB;AACrD,SAASC,UAAU,QAAQ,qBAAqB;AAChD,SAASC,mBAAmB,QAAQ,yBAAyB;AAC7D,SAASC,+BAA+B,QAAQ,yBAAyB;AACzE,SAASC,OAAO,QAAQ,+BAA+B;AACvD,SAASC,SAAS,EAAEC,SAAS,EAAEC,WAAW,QAAQ,uBAAuB;AACzE,SAASC,WAAW,QAAQ,2BAA2B;AACvD,SACEC,8BAA8B,EAC9BC,iCAAiC,EACjCC,oBAAoB,EACpBC,gBAAgB,QAEX,oBAAoB;AAC3B,SAASC,gCAAgC,QAAQ,uBAAuB;AACxE,SAASC,6BAA6B,QAAQ,gCAAgC;AAC9E,SAASC,UAAU,QAAQ,sBAAsB;AACjD,SAASC,WAAW,QAAQ,+CAA+C;AAC3E,SACEC,8BAA8B,EAC9BC,yBAAyB,QACpB,mCAAmC;AAC1C,SAEEC,YAAY,EACZC,kBAAkB,QACb,6CAA6C;AACpD,SACEC,+BAA+B,EAC/BC,yBAAyB,QACpB,oCAAoC;AAC3C,SAASC,WAAW,QAAQ,0BAA0B;AACtD,OAAOC,sBAAsBC,eAAe,QAAQ,sBAAsB;AAQ1E,4EAA4E;AAC5E,2CAA2C;AAC3C,OAAO,MAAMC,eAAepC,sBAAqC;IAC/DqC,IAAI;QACF;YAAEC,MAAM;QAAoC;QAC5C;YAAEA,MAAM;QAAgC;QACxC;YAAEA,MAAM;QAAkC;QAC1C;YAAEA,MAAM;QAAsB;QAC9B;YAAEA,MAAM;QAAiB;QACzB;YAAEA,MAAM;YAA2BC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QACrE;YAAEH,MAAM;YAAWC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QACrD;YAAEH,MAAM;YAAsBC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QAChE;YAAEH,MAAM;YAAMC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QAChD;YAAEH,MAAM;YAAqBI,KAAK;QAAK;QACvC;YAAEJ,MAAM;YAAaC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;YAASC,KAAK;QAAK;QAClE,mEAAmE;QACnE;YAAEJ,MAAM;YAAUI,KAAK;YAAMC,MAAM;QAAK;QACxC;YAAEL,MAAM;YAAOI,KAAK;YAAMC,MAAM;QAAK;WAClC1C,eAAeH;WACfG,eAAeO;WACfP,eAAeE;WACfF,eAAe0B;WACf1B,eAAe8B;WACf9B,eAAeI;KACnB;AACH,GAAG;AAEH,OAAO,MAAMuC,wBAAyCtB,iBACpD,YAAYuB,QAAQ,EACpB;AAEF,OAAO,MAAMC,iBAAiB,OAC5BC,MACAC;IAEA,MAAMC,gBAAgBlC,UAAUiC,QAAQV,IAAI;IAC5C,MAAMY,gBAAgBjC,YAAY+B,QAAQV,IAAI;IAC9C,MAAMa,eAAeH,QAAQG,YAAY,IAAI;IAC7C,MAAMC,eAAepC,UAAUgC,QAAQI,YAAY,IAAIJ,QAAQV,IAAI;IACnE,MAAMe,kBAAkB,MAAMtD,kBAAkBgD,MAAM;IACtD,MAAM,EAAEO,kBAAkB,EAAEC,GAAG,EAAE,GAAGpB,gBAAgBY,MAAM;QACxDT,MAAMU,QAAQV,IAAI;QAClBkB,WAAWR,QAAQQ,SAAS;QAC5BC,cAAcT,QAAQS,YAAY;IACpC;IAEA,IAAIC;IACJ,IAAI;QACFnE,yBAAyBwD,MAAMO;QAC/BI,gBAAgB;IAClB,EAAE,OAAM;QACNA,gBAAgB;IAClB;IAEA,IAAI,CAACA,eAAe;QAClB,MAAMxB,mBAAmBa,MAAM;YAC7BT,MAAMU,QAAQV,IAAI;YAClBkB,WAAWR,QAAQQ,SAAS;YAC5BC,cAAcT,QAAQS,YAAY;YAClCE,2BAA2B;QAC7B;IACF;IAEAnE,WAAWuD,MAAM1D,kBAAkBkE,KAAK,sBAAsB,CAACK,WAAc,CAAA;YAC3E,GAAGA,QAAQ;YACXC,SAAS;gBAAC;gBAAe;aAA2B;QACtD,CAAA;IAEA,MAAMC,gBAAgBvE,yBAAyBwD,MAAMO;IACrD,MAAMS,cAActC,WAClBsB,MACAe,eACAd,QAAQP,MAAM,KAAK,UAAU,OAAO;IAEtC,MAAMuB,cAAc;IACpB,MAAMC,cAAcjB,QAAQP,MAAM,KAAK,UAAU,SAAS;IAC1D,MAAMyB,kBAAkB;IACxB,MAAMC,gBAAgB,GAAGjD,YAAY6B,MAAM,CAAC,EAAEK,cAAc;IAC5D,MAAMgB,cACJpB,QAAQP,MAAM,KAAK,UACf,+CACA;IAEN,MAAM4B,kBAAkB;QACtB5B,QAAQO,QAAQP,MAAM;QACtB6B,kBAAkBpB;QAClBqB,sBAAsBjB;QACtBkB,kBAAkBxB,QAAQP,MAAM,KAAK,UAAU,UAAU;QACzDgC,eAAexC,YAAYyC,MAAM;QACjC,GAAGpE,mBAAmB;QACtBqE,sBACE3B,QAAQP,MAAM,KAAK,UACf,4BACA;QACNmC,wBACE5B,QAAQP,MAAM,KAAK,UAAU,kBAAkB;QACjDsB;QACAC;QACAa,aAAazB;QACba;QACAC;QACAb;QACAc;QACAC;QACA,GAAGtD,QAAQiC,KAAK;IAClB;IAEA,8EAA8E;IAC9E,2EAA2E;IAC3E,qDAAqD;IACrD3D,cACE2D,MACA1D,kBAAkB,YAAYyF,OAAO,EAAE,UACvCvB,KACAc,iBACA;QAAEU,mBAAmBzF,kBAAkB0F,YAAY;IAAC;IAGtD,+EAA+E;IAC/E,8EAA8E;IAC9E,6EAA6E;IAC7E,4BAA4B;IAC5B,MAAMpF,cACJmD,MACA1D,kBAAkBkE,KAAK,OAAO,aAC9B;IAGF,qEAAqE;IACrE,yEAAyE;IACzE7D,UAAUqD,MAAM1D,kBAAkBkE,KAAK,gBAAgB;QACrDe,kBAAkBpB;QAClB+B,UAAU;YACR5B;YACAc;YACAe,OAAOd;YACPe,MAAMpB;YACNqB,MAAMpB;YACNqB,QAAQjC;YACRkC,QAAQrB;YACRsB,YAAYrB;YACZsB,UAAUxC,QAAQP,MAAM;QAC1B;IACF;IACA/B,gBAAgBqC,MAAMQ,KAAK,CAACkC,WAAa;eAAIA;YAAU;SAAmB;IAC1E,MAAMzD,0BAA0Be,MAAMC,QAAQP,MAAM,EAAEL;IACtD,MAAMsD,kBACJ1C,QAAQP,MAAM,KAAK,UACf,yBACA;IACN,MAAMkD,aAAazG,SACjBqE,KACAlE,kBAAkBwC,cAAcC,oBAAoB,OAAO;IAE7D,MAAM8D,qBAAqBrE,iCACzBuC,cAAcnB,IAAI;IAEpB,MAAMkD,KAAK,IAAIpF,WAAWsC,MAAMX;IAChC,MAAM0D,qBAAqBzG,kBACzB,QACAyE,cAAcnB,IAAI,EAClB,UACA;IAEF,MAAMoD,0BAA0B,GAAG7E,YAAY6B,MAAM,CAAC,EAAEhC,UAAUiC,QAAQV,IAAI,EAAE,iBAAiB,CAAC;IAElG,IAAIU,QAAQgD,KAAK,KAAK,QAAQ;QAC5B,MAAMnG,0BACJkD,MACAe,eACA;YACEmC,gBAAgB;YAChBC,iBAAiB;QACnB,GACA9D;QAEF,MAAMvC,0BACJkD,MACAe,eACA;YACEmC,gBAAgB;YAChBC,iBAAiB;QACnB,GACA9D;QAEF,MAAM+D,eAAerC,cAAcsC,OAAO,CAAC,SAAS;QACpD,0EAA0E;QAC1E,yEAAyE;QACzE,6CAA6C;QAC7C,IAAI,CAACD,aAAanD,OAAO,CAACqD,QAAQ,EAAE;YAClC,MAAMC,kBAAkBH,aAAanD,OAAO,CAACuD,OAAO;YACpD,OAAOJ,aAAanD,OAAO,CAACuD,OAAO;YACnCJ,aAAanD,OAAO,GAAG;gBACrB,GAAGmD,aAAanD,OAAO;gBACvBqD,UAAU;oBACRR,GAAGW,EAAE,CAAC,GAAGZ,mBAAmB,mCAAmC,CAAC;oBAChEC,GAAGY,KAAK,CAAC,GAAGb,mBAAmB,mCAAmC,CAAC;oBACnEC,GAAGa,KAAK,CACN,UACA,GAAGd,mBAAmB,0CAA0C,CAAC;oBAEnEC,GAAGc,MAAM,CACP,oBACA,GAAGf,mBAAmB,oDAAoD,CAAC;oBAE7EC,GAAGc,MAAM,CACP,cACA,GAAGf,mBAAmB,8CAA8C,CAAC;oBAEvEU;iBACD;gBACDM,UAAU;YACZ;QACF;IACF;IAEA9C,cAAcsC,OAAO,CAACS,QAAQ,GAAG;QAC/BC,UAAU;QACVC,SAAS;YAAC;SAAiC;QAC3C/D,SAAS;YACPuD,SAAS;YACTS,KAAK;QACP;IACF;IACAlD,cAAcsC,OAAO,CAAC,aAAa,GAAG;QACpCU,UAAU;QACV9D,SAAS;YACPuD,SAAS,CAAC,IAAI,EAAEZ,WAAW,cAAc,CAAC;YAC1CqB,KAAK;QACP;IACF;IACAlD,cAAcsC,OAAO,CAAC,MAAM,GAAG;QAC7BU,UAAU;QACVG,WAAW;YAAC;SAAa;QACzBjE,SAAS;YACPuD,SAAS,CAAC,IAAI,EAAEZ,WAAW,mBAAmB,CAAC;YAC/CqB,KAAK;QACP;QACAE,YAAY;IACd;IACApD,cAAcsC,OAAO,CAAC,cAAc,GAAG;QACrCU,UAAU;QACVG,WAAW;YAAC;SAAM;QAClBjE,SAAS;YACPuD,SAAS,CAAC,IAAI,EAAEZ,WAAW,CAAC,EAAED,iBAAiB;YAC/CsB,KAAK;QACP;IACF;IACAlD,cAAcsC,OAAO,CAAC1B,MAAM,GAAG;QAC7BoC,UAAU;QACVG,WAAW;YAAC;YAAO;SAAc;QACjCjE,SAAS;YACPgE,KAAK;YACLT,SAAS;YACTY,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IACA,4EAA4E;IAC5E,2EAA2E;IAC3E,qDAAqD;IACrD,MAAMC,MACJrE,QAAQgD,KAAK,KAAK,SAAS,MAAMrF,WAAWoC,MAAMC,QAAQqE,GAAG,IAAIC;IAEnE,IAAItE,QAAQgD,KAAK,KAAK,QAAQ;QAC5B,IAAIqB,QAAQ,aAAa;YACvBvD,cAAcsC,OAAO,CAAC,SAAS,GAAG;gBAChCmB,OAAOnH;gBACP0G,UAAU;gBACV9D,SAAS;oBACPuD,SAAS,GAAGlD,gBAAgB,oDAAoD,EAAE0C,wBAAwB,CAAC,EAAED,oBAAoB;gBACnI;gBACAmB,WAAW;oBAAC;iBAAS;YACvB;YACA9F,+BAA+B2C,eAAe;YAE9C5D,oBACE6C,MACA;gBACEyE,SAAS1D;gBACTT;gBACAoE,iBAAiB;gBACjBC,kBAAkB;gBAClBC,WAAW;oBAAC5B;iBAAwB;YACtC,GACA3D;QAEJ;QACAjB,+BAA+B2C,eAAe;IAChD;IACA1C,kCAAkC0C,eAAe,WAAW;IAC5DrE,2BAA2BsD,MAAMO,oBAAoBQ;IACrD,0EAA0E;IAC1E,8DAA8D;IAC9D,MAAM8D,WAA0B;QAC9BnF,QAAQO,QAAQP,MAAM;QACtB,GAAI4E,MAAM;YAAEA;QAAI,IAAI,CAAC,CAAC;IACxB;IACAhG,qBACE0B,MACAO,oBACAV,uBACAgF;IAGF,IAAI5E,QAAQgD,KAAK,KAAK,QAAQ;QAC5B,MAAMqB,MAAM,MAAM1G,WAAWoC,MAAMC,QAAQqE,GAAG;QAC9C,MAAMzF,0BAA0BmB,MAAM;YAAEsE;QAAI,GAAGjF;QAC/C,MAAMV,YAAYqB,MAAM;YACtBsE;YACAQ,aAAavE;YACbwE,aAAavE;YACbL;YACAD;YACAsB,sBAAsBjB;YACtBF;YACA2E,WAAW5E;YACXV,QAAQO,QAAQP,MAAM;YACtBqD;YACAkC,uBAAuB3I,kBACrB,QACAyE,cAAcnB,IAAI,EAClB,UACA;YAEFsF,WAAWjF,QAAQiF,SAAS;YAC5BlC;YACA1C;QACF;IACF;IAEA1D,kBAAkBoD,MAAMX,cAAc;QACpCwF;QACAE,aAAahE,cAAcnB,IAAI;IACjC;IAEA,8EAA8E;IAC9E,6EAA6E;IAC7E,8EAA8E;IAC9E,2EAA2E;IAC3E,8EAA8E;IAC9E,6EAA6E;IAC7E,0EAA0E;IAC1E,6BAA6B;IAC7B,IAAIK,QAAQP,MAAM,KAAK,cAActD,2BAA2B,QAAQ;QACtEK,WAAWuD,MAAM,gBAAgB,CAACmF;YAChCA,YAAYC,WAAW,GAAG;gBACxB,GAAGD,YAAYC,WAAW;gBAC1B,mCAAmClG,WAAW,CAAC,YAAY;gBAC3D,gCAAgCA,WAAW,CAAC,YAAY;YAC1D;YACA,OAAOiG;QACT;IACF;IAEA1G,8BAA8BuB,MAAM;QAClC,mBAAmB;QACnB2B,QAAQ;IACV;IAEA,MAAM7D,gCAAgCkC,MAAM;QAACH;KAAsB;IAEnE,MAAMrC,qBAAqBwC;IAC3B,OAAO,IACLnC,oBAAoBmC,MAAMC,QAAQW,yBAAyB,EAAE;YAC3DyE,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF,eAAetF,eAAe"}
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/ts/rdb/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport { relative } from 'node:path';\nimport {\n  detectPackageManager,\n  type GeneratorCallback,\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  readProjectConfiguration,\n  type Tree,\n  updateJson,\n  updateProjectConfiguration,\n  writeJson,\n} from '@nx/devkit';\nimport { addTsDependencies } from '../../utils/add-dependencies.js';\nimport { addStarExport } from '../../utils/ast.js';\nimport {\n  addTypeScriptBundleTarget,\n  BUNDLE_DEPENDENCIES,\n} from '../../utils/bundle/bundle.js';\nimport { resolveContainers } from '../../utils/containers.js';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../utils/declared-dependencies.js';\nimport {\n  addDockerScanTarget,\n  DOCKER_DEPENDENCIES,\n  IMAGE_BUILD_CACHE,\n  NODE_IMAGE_DEPENDENCIES,\n  nodeImageVersions,\n  PRISMA_IMAGE_DEPENDENCIES,\n} from '../../utils/docker.js';\nimport { formatFilesInSubtree } from '../../utils/format.js';\nimport { FS_DEPENDENCIES, FsCommands } from '../../utils/fs.js';\nimport { updateGitIgnore } from '../../utils/git.js';\nimport { resolveIac } from '../../utils/iac.js';\nimport { installDependencies } from '../../utils/install.js';\nimport { addGeneratorMetricsIfApplicable } from '../../utils/metrics.js';\nimport { esmVars } from '../../utils/module-format.js';\nimport { kebabCase, snakeCase, toClassName } from '../../utils/names.js';\nimport { getNpmScope } from '../../utils/npm-scope.js';\nimport {\n  addArtifactDependencyToTargets,\n  addDependencyToTargetIfNotPresent,\n  addGeneratorMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n} from '../../utils/nx.js';\nimport { getRelativePathToRootByDirectory } from '../../utils/paths.js';\nimport { registerPnpmBuiltDependencies } from '../../utils/pnpm-workspace.js';\nimport { assignPort } from '../../utils/port.js';\nimport { addRdbInfra } from '../../utils/rdb-constructs/rdb-constructs.js';\nimport {\n  SHARED_CONSTRUCTS_DEPENDENCIES,\n  sharedConstructsGenerator,\n} from '../../utils/shared-constructs.js';\nimport {\n  type IacMetadata,\n  PACKAGES_DIR,\n  SHARED_SCRIPTS_DIR,\n} from '../../utils/shared-constructs-constants.js';\nimport {\n  SHARED_RDB_SCRIPTS_DEPENDENCIES,\n  sharedRdbScriptsGenerator,\n} from '../../utils/shared-rdb-scripts.js';\nimport { TS_VERSIONS } from '../../utils/versions.js';\nimport tsProjectGenerator, { getTsLibDetails } from '../lib/generator.js';\nimport type { TsRdbGeneratorSchema } from './schema';\n\n/** The metadata this generator records, which its predicates read. */\nexport interface TsRdbMetadata extends IacMetadata {\n  readonly engine: TsRdbGeneratorSchema['engine'];\n}\n\n// Each entry names the engine branch it belongs to, so the same declaration\n// drives both adding and the version sync.\nexport const DEPENDENCIES = declareDependencies<TsRdbMetadata>()({\n  ts: [\n    { name: '@aws-lambda-powertools/parameters' },\n    { name: '@aws-sdk/client-appconfigdata' },\n    { name: '@aws-sdk/client-secrets-manager' },\n    { name: '@aws-sdk/rds-signer' },\n    { name: '@prisma/client' },\n    { name: '@prisma/adapter-mariadb', when: (m) => m.engine === 'mysql' },\n    { name: 'mariadb', when: (m) => m.engine === 'mysql' },\n    { name: '@prisma/adapter-pg', when: (m) => m.engine !== 'mysql' },\n    { name: 'pg', when: (m) => m.engine !== 'mysql' },\n    { name: '@types/aws-lambda', dev: true },\n    { name: '@types/pg', when: (m) => m.engine !== 'mysql', dev: true },\n    // The prisma CLI and tsx run migration/seed scripts from the root.\n    { name: 'prisma', dev: true, root: true },\n    { name: 'tsx', dev: true, root: true },\n    ...ownedElsewhere(BUNDLE_DEPENDENCIES),\n    ...ownedElsewhere(FS_DEPENDENCIES),\n    ...ownedElsewhere(DOCKER_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_RDB_SCRIPTS_DEPENDENCIES),\n    ...ownedElsewhere(NODE_IMAGE_DEPENDENCIES),\n    ...ownedElsewhere(PRISMA_IMAGE_DEPENDENCIES),\n  ],\n});\n\nexport const TS_RDB_GENERATOR_INFO: NxGeneratorInfo = getGeneratorInfo(\n  import.meta.filename,\n);\n\nexport const tsRdbGenerator = async (\n  tree: Tree,\n  options: TsRdbGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const nameKebabCase = kebabCase(options.name);\n  const nameClassName = toClassName(options.name);\n  const databaseUser = options.databaseUser ?? 'dbadmin';\n  const databaseName = snakeCase(options.databaseName ?? options.name);\n  const containerEngine = await resolveContainers(tree, 'inherit');\n  const { fullyQualifiedName, dir } = getTsLibDetails(tree, {\n    name: options.name,\n    directory: options.directory,\n    subDirectory: options.subDirectory,\n  });\n\n  let projectExists: boolean;\n  try {\n    readProjectConfiguration(tree, fullyQualifiedName);\n    projectExists = true;\n  } catch {\n    projectExists = false;\n  }\n\n  if (!projectExists) {\n    await tsProjectGenerator(tree, {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n      preferInstallDependencies: false,\n    });\n  }\n\n  updateJson(tree, joinPathFragments(dir, 'tsconfig.lib.json'), (tsConfig) => ({\n    ...tsConfig,\n    include: ['src/**/*.ts', 'generated/prisma/**/*.ts'],\n  }));\n\n  const projectConfig = readProjectConfiguration(tree, fullyQualifiedName);\n  const localDbPort = assignPort(\n    tree,\n    projectConfig,\n    options.engine === 'mysql' ? 3306 : 5432,\n  );\n  const localDbHost = 'localhost';\n  const localDbUser = options.engine === 'mysql' ? 'root' : 'dbadmin';\n  const localDbPassword = 'password';\n  const containerName = `${getNpmScope(tree)}-${databaseName}`;\n  const dockerImage =\n    options.engine === 'mysql'\n      ? 'public.ecr.aws/docker/library/mysql:8.0.44'\n      : 'public.ecr.aws/docker/library/postgres:17.7';\n\n  const templateOptions = {\n    engine: options.engine,\n    runtimeConfigKey: nameClassName,\n    databasePackageAlias: fullyQualifiedName,\n    databaseProvider: options.engine === 'mysql' ? 'mysql' : 'postgresql',\n    prismaVersion: TS_VERSIONS.prisma,\n    deepmergeTsVersion: TS_VERSIONS['deepmerge-ts'],\n    mysql2Version: TS_VERSIONS.mysql2,\n    ...nodeImageVersions(),\n    prismaAdapterPackage:\n      options.engine === 'mysql'\n        ? '@prisma/adapter-mariadb'\n        : '@prisma/adapter-pg',\n    prismaAdapterClassName:\n      options.engine === 'mysql' ? 'PrismaMariaDb' : 'PrismaPg',\n    localDbPort,\n    localDbHost,\n    localDbName: databaseName,\n    localDbUser,\n    localDbPassword,\n    containerEngine,\n    containerName,\n    dockerImage,\n    ...esmVars(tree),\n  };\n\n  // The Prisma schema and models, the handlers and the prisma client module are\n  // all user-editable, so a re-run (to add infrastructure, or after a failed\n  // run) must not clobber the user's schema and edits.\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    dir,\n    templateOptions,\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // The barrel re-exports prisma.ts. `tsProjectGenerator` writes a stub index.ts\n  // when it creates the project, so the re-export is added to whatever is there\n  // rather than replacing it — which also leaves a user's own additions to the\n  // barrel alone on a re-run.\n  await addStarExport(\n    tree,\n    joinPathFragments(dir, 'src', 'index.ts'),\n    './prisma.js',\n  );\n\n  // config.json is framework-owned: it carries values derived from the\n  // generator's options, so it converges on a re-run with changed options.\n  writeJson(tree, joinPathFragments(dir, 'config.json'), {\n    runtimeConfigKey: nameClassName,\n    localDev: {\n      containerEngine,\n      containerName,\n      image: dockerImage,\n      port: localDbPort,\n      host: localDbHost,\n      dbName: databaseName,\n      dbUser: localDbUser,\n      dbPassword: localDbPassword,\n      dbEngine: options.engine,\n    },\n  });\n  updateGitIgnore(tree, dir, (patterns) => [...patterns, 'generated/prisma']);\n  await sharedRdbScriptsGenerator(tree, options.engine, DEPENDENCIES);\n  const waitForDbScript =\n    options.engine === 'mysql'\n      ? 'wait-for-mysql-db.ts'\n      : 'wait-for-postgres-db.ts';\n  const scriptsDir = relative(\n    dir,\n    joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'rdb'),\n  );\n  const relativePathToRoot = getRelativePathToRootByDirectory(\n    projectConfig.root,\n  );\n  const fs = new FsCommands(tree, DEPENDENCIES);\n  const migrationBundleDir = joinPathFragments(\n    'dist',\n    projectConfig.root,\n    'bundle',\n    'migration',\n  );\n  const migrationDockerImageTag = `${getNpmScope(tree)}-${kebabCase(options.name)}-migration:latest`;\n\n  if (options.infra !== 'none') {\n    await addTypeScriptBundleTarget(\n      tree,\n      projectConfig,\n      {\n        targetFilePath: 'src/migration-handler.ts',\n        bundleOutputDir: 'migration',\n      },\n      DEPENDENCIES,\n    );\n    await addTypeScriptBundleTarget(\n      tree,\n      projectConfig,\n      {\n        targetFilePath: 'src/create-db-user-handler.ts',\n        bundleOutputDir: 'create-db-user',\n      },\n      DEPENDENCIES,\n    );\n    const bundleTarget = projectConfig.targets['bundle'];\n    // The bundle target starts with a single rolldown `command`. Wrap it with\n    // the migration asset copy steps, unless it has already been transformed\n    // into a `commands` array on a previous run.\n    if (!bundleTarget.options.commands) {\n      const rolldownCommand = bundleTarget.options.command;\n      delete bundleTarget.options.command;\n      bundleTarget.options = {\n        ...bundleTarget.options,\n        commands: [\n          fs.rm(`${relativePathToRoot}dist/{projectRoot}/bundle/migration`),\n          fs.mkdir(`${relativePathToRoot}dist/{projectRoot}/bundle/migration`),\n          fs.cpDir(\n            'prisma',\n            `${relativePathToRoot}dist/{projectRoot}/bundle/migration/prisma`,\n          ),\n          fs.cpFile(\n            'prisma.config.ts',\n            `${relativePathToRoot}dist/{projectRoot}/bundle/migration/prisma.config.ts`,\n          ),\n          fs.cpFile(\n            'Dockerfile',\n            `${relativePathToRoot}dist/{projectRoot}/bundle/migration/Dockerfile`,\n          ),\n          rolldownCommand,\n        ],\n        parallel: false,\n      };\n    }\n  }\n\n  projectConfig.targets.generate = {\n    executor: 'nx:run-commands',\n    outputs: ['{projectRoot}/generated/prisma'],\n    options: {\n      command: 'prisma generate',\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets['pull-image'] = {\n    executor: 'nx:run-commands',\n    options: {\n      command: `tsx ${scriptsDir}/pull-image.ts`,\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets['dev'] = {\n    executor: 'nx:run-commands',\n    dependsOn: ['pull-image'],\n    options: {\n      command: `tsx ${scriptsDir}/start-container.ts`,\n      cwd: '{projectRoot}',\n    },\n    continuous: true,\n  };\n  projectConfig.targets['wait-for-db'] = {\n    executor: 'nx:run-commands',\n    dependsOn: ['dev'],\n    options: {\n      command: `tsx ${scriptsDir}/${waitForDbScript}`,\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets.prisma = {\n    executor: 'nx:run-commands',\n    dependsOn: ['dev', 'wait-for-db'],\n    options: {\n      cwd: '{projectRoot}',\n      command: 'prisma',\n      env: {\n        LOCAL_DEV: 'true',\n      },\n    },\n  };\n  // Recorded in the metadata below so the version sync can tell a CDK project\n  // from a Terraform one; undefined when no infrastructure was generated, in\n  // which case neither provider's packages were added.\n  const iac =\n    options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;\n\n  if (options.infra !== 'none') {\n    // The migration image ships under both providers — Terraform pushes the\n    // locally built tag, CDK builds the same context as an image asset — so the\n    // build and its scan are wired up for either.\n    projectConfig.targets['docker'] = {\n      cache: IMAGE_BUILD_CACHE,\n      executor: 'nx:run-commands',\n      options: {\n        command: `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`,\n      },\n      dependsOn: ['bundle'],\n    };\n    addArtifactDependencyToTargets(projectConfig, 'docker');\n\n    addDockerScanTarget(\n      tree,\n      {\n        project: projectConfig,\n        containerEngine,\n        trivyTargetName: 'trivy',\n        dockerTargetName: 'docker',\n        imageTags: [migrationDockerImageTag],\n      },\n      DEPENDENCIES,\n    );\n    addArtifactDependencyToTargets(projectConfig, 'bundle');\n  }\n  addDependencyToTargetIfNotPresent(projectConfig, 'compile', 'generate');\n  updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);\n  // Recorded here and read by the declaration's predicates, so the packages\n  // added below are exactly the ones the version sync will own.\n  const metadata: TsRdbMetadata = {\n    engine: options.engine,\n    ...(iac ? { iac } : {}),\n  };\n  addGeneratorMetadata(\n    tree,\n    fullyQualifiedName,\n    TS_RDB_GENERATOR_INFO,\n    metadata,\n  );\n\n  if (options.infra !== 'none') {\n    const iac = await resolveIac(tree, options.iac);\n    await sharedConstructsGenerator(tree, { iac }, DEPENDENCIES);\n    await addRdbInfra(tree, {\n      iac,\n      projectName: fullyQualifiedName,\n      projectRoot: dir,\n      nameClassName,\n      nameKebabCase,\n      databasePackageAlias: fullyQualifiedName,\n      databaseName,\n      adminUser: databaseUser,\n      engine: options.engine,\n      migrationBundleDir,\n      createDbUserBundleDir: joinPathFragments(\n        'dist',\n        projectConfig.root,\n        'bundle',\n        'create-db-user',\n      ),\n      framework: options.framework,\n      migrationDockerImageTag,\n      containerEngine,\n    });\n  }\n\n  addTsDependencies(tree, DEPENDENCIES, {\n    metadata,\n    projectRoot: projectConfig.root,\n  });\n\n  // @prisma/adapter-pg depends on @types/pg ^8.16.0. Yarn does not dedupe it to\n  // the workspace's pinned @types/pg, so it installs a separate copy under the\n  // adapter's own node_modules whenever a newer @types/pg is published. The two\n  // copies declare structurally incompatible Pool types, so passing our `new\n  // Pool(...)` to `new PrismaPg(...)` fails to compile. Scope the resolution to\n  // the adapter so other @types/pg consumers are unaffected. Classic yarn only\n  // honours the `**/`-prefixed descriptor in a workspace and berry only the\n  // bare one, so declare both.\n  if (options.engine === 'postgres' && detectPackageManager() === 'yarn') {\n    updateJson(tree, 'package.json', (packageJson) => {\n      packageJson.resolutions = {\n        ...packageJson.resolutions,\n        '**/@prisma/adapter-pg/@types/pg': TS_VERSIONS['@types/pg'],\n        '@prisma/adapter-pg/@types/pg': TS_VERSIONS['@types/pg'],\n      };\n      return packageJson;\n    });\n  }\n\n  registerPnpmBuiltDependencies(tree, {\n    '@prisma/engines': false,\n    prisma: false,\n  });\n\n  await addGeneratorMetricsIfApplicable(tree, [TS_RDB_GENERATOR_INFO]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript'],\n    });\n};\n\nexport default tsRdbGenerator;\n"],"names":["relative","detectPackageManager","generateFiles","joinPathFragments","OverwriteStrategy","readProjectConfiguration","updateJson","updateProjectConfiguration","writeJson","addTsDependencies","addStarExport","addTypeScriptBundleTarget","BUNDLE_DEPENDENCIES","resolveContainers","declareDependencies","ownedElsewhere","addDockerScanTarget","DOCKER_DEPENDENCIES","IMAGE_BUILD_CACHE","NODE_IMAGE_DEPENDENCIES","nodeImageVersions","PRISMA_IMAGE_DEPENDENCIES","formatFilesInSubtree","FS_DEPENDENCIES","FsCommands","updateGitIgnore","resolveIac","installDependencies","addGeneratorMetricsIfApplicable","esmVars","kebabCase","snakeCase","toClassName","getNpmScope","addArtifactDependencyToTargets","addDependencyToTargetIfNotPresent","addGeneratorMetadata","getGeneratorInfo","getRelativePathToRootByDirectory","registerPnpmBuiltDependencies","assignPort","addRdbInfra","SHARED_CONSTRUCTS_DEPENDENCIES","sharedConstructsGenerator","PACKAGES_DIR","SHARED_SCRIPTS_DIR","SHARED_RDB_SCRIPTS_DEPENDENCIES","sharedRdbScriptsGenerator","TS_VERSIONS","tsProjectGenerator","getTsLibDetails","DEPENDENCIES","ts","name","when","m","engine","dev","root","TS_RDB_GENERATOR_INFO","filename","tsRdbGenerator","tree","options","nameKebabCase","nameClassName","databaseUser","databaseName","containerEngine","fullyQualifiedName","dir","directory","subDirectory","projectExists","preferInstallDependencies","tsConfig","include","projectConfig","localDbPort","localDbHost","localDbUser","localDbPassword","containerName","dockerImage","templateOptions","runtimeConfigKey","databasePackageAlias","databaseProvider","prismaVersion","prisma","deepmergeTsVersion","mysql2Version","mysql2","prismaAdapterPackage","prismaAdapterClassName","localDbName","dirname","overwriteStrategy","KeepExisting","localDev","image","port","host","dbName","dbUser","dbPassword","dbEngine","patterns","waitForDbScript","scriptsDir","relativePathToRoot","fs","migrationBundleDir","migrationDockerImageTag","infra","targetFilePath","bundleOutputDir","bundleTarget","targets","commands","rolldownCommand","command","rm","mkdir","cpDir","cpFile","parallel","generate","executor","outputs","cwd","dependsOn","continuous","env","LOCAL_DEV","iac","undefined","cache","project","trivyTargetName","dockerTargetName","imageTags","metadata","projectName","projectRoot","adminUser","createDbUserBundleDir","framework","packageJson","resolutions","languages"],"mappings":"AAAA;;;CAGC,GACD,SAASA,QAAQ,QAAQ,YAAY;AACrC,SACEC,oBAAoB,EAEpBC,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EACjBC,wBAAwB,EAExBC,UAAU,EACVC,0BAA0B,EAC1BC,SAAS,QACJ,aAAa;AACpB,SAASC,iBAAiB,QAAQ,kCAAkC;AACpE,SAASC,aAAa,QAAQ,qBAAqB;AACnD,SACEC,yBAAyB,EACzBC,mBAAmB,QACd,+BAA+B;AACtC,SAASC,iBAAiB,QAAQ,4BAA4B;AAC9D,SACEC,mBAAmB,EACnBC,cAAc,QACT,uCAAuC;AAC9C,SACEC,mBAAmB,EACnBC,mBAAmB,EACnBC,iBAAiB,EACjBC,uBAAuB,EACvBC,iBAAiB,EACjBC,yBAAyB,QACpB,wBAAwB;AAC/B,SAASC,oBAAoB,QAAQ,wBAAwB;AAC7D,SAASC,eAAe,EAAEC,UAAU,QAAQ,oBAAoB;AAChE,SAASC,eAAe,QAAQ,qBAAqB;AACrD,SAASC,UAAU,QAAQ,qBAAqB;AAChD,SAASC,mBAAmB,QAAQ,yBAAyB;AAC7D,SAASC,+BAA+B,QAAQ,yBAAyB;AACzE,SAASC,OAAO,QAAQ,+BAA+B;AACvD,SAASC,SAAS,EAAEC,SAAS,EAAEC,WAAW,QAAQ,uBAAuB;AACzE,SAASC,WAAW,QAAQ,2BAA2B;AACvD,SACEC,8BAA8B,EAC9BC,iCAAiC,EACjCC,oBAAoB,EACpBC,gBAAgB,QAEX,oBAAoB;AAC3B,SAASC,gCAAgC,QAAQ,uBAAuB;AACxE,SAASC,6BAA6B,QAAQ,gCAAgC;AAC9E,SAASC,UAAU,QAAQ,sBAAsB;AACjD,SAASC,WAAW,QAAQ,+CAA+C;AAC3E,SACEC,8BAA8B,EAC9BC,yBAAyB,QACpB,mCAAmC;AAC1C,SAEEC,YAAY,EACZC,kBAAkB,QACb,6CAA6C;AACpD,SACEC,+BAA+B,EAC/BC,yBAAyB,QACpB,oCAAoC;AAC3C,SAASC,WAAW,QAAQ,0BAA0B;AACtD,OAAOC,sBAAsBC,eAAe,QAAQ,sBAAsB;AAQ1E,4EAA4E;AAC5E,2CAA2C;AAC3C,OAAO,MAAMC,eAAerC,sBAAqC;IAC/DsC,IAAI;QACF;YAAEC,MAAM;QAAoC;QAC5C;YAAEA,MAAM;QAAgC;QACxC;YAAEA,MAAM;QAAkC;QAC1C;YAAEA,MAAM;QAAsB;QAC9B;YAAEA,MAAM;QAAiB;QACzB;YAAEA,MAAM;YAA2BC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QACrE;YAAEH,MAAM;YAAWC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QACrD;YAAEH,MAAM;YAAsBC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QAChE;YAAEH,MAAM;YAAMC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QAChD;YAAEH,MAAM;YAAqBI,KAAK;QAAK;QACvC;YAAEJ,MAAM;YAAaC,MAAM,CAACC,IAAMA,EAAEC,MAAM,KAAK;YAASC,KAAK;QAAK;QAClE,mEAAmE;QACnE;YAAEJ,MAAM;YAAUI,KAAK;YAAMC,MAAM;QAAK;QACxC;YAAEL,MAAM;YAAOI,KAAK;YAAMC,MAAM;QAAK;WAClC3C,eAAeH;WACfG,eAAeQ;WACfR,eAAeE;WACfF,eAAe2B;WACf3B,eAAe+B;WACf/B,eAAeI;WACfJ,eAAeM;KACnB;AACH,GAAG;AAEH,OAAO,MAAMsC,wBAAyCtB,iBACpD,YAAYuB,QAAQ,EACpB;AAEF,OAAO,MAAMC,iBAAiB,OAC5BC,MACAC;IAEA,MAAMC,gBAAgBlC,UAAUiC,QAAQV,IAAI;IAC5C,MAAMY,gBAAgBjC,YAAY+B,QAAQV,IAAI;IAC9C,MAAMa,eAAeH,QAAQG,YAAY,IAAI;IAC7C,MAAMC,eAAepC,UAAUgC,QAAQI,YAAY,IAAIJ,QAAQV,IAAI;IACnE,MAAMe,kBAAkB,MAAMvD,kBAAkBiD,MAAM;IACtD,MAAM,EAAEO,kBAAkB,EAAEC,GAAG,EAAE,GAAGpB,gBAAgBY,MAAM;QACxDT,MAAMU,QAAQV,IAAI;QAClBkB,WAAWR,QAAQQ,SAAS;QAC5BC,cAAcT,QAAQS,YAAY;IACpC;IAEA,IAAIC;IACJ,IAAI;QACFpE,yBAAyByD,MAAMO;QAC/BI,gBAAgB;IAClB,EAAE,OAAM;QACNA,gBAAgB;IAClB;IAEA,IAAI,CAACA,eAAe;QAClB,MAAMxB,mBAAmBa,MAAM;YAC7BT,MAAMU,QAAQV,IAAI;YAClBkB,WAAWR,QAAQQ,SAAS;YAC5BC,cAAcT,QAAQS,YAAY;YAClCE,2BAA2B;QAC7B;IACF;IAEApE,WAAWwD,MAAM3D,kBAAkBmE,KAAK,sBAAsB,CAACK,WAAc,CAAA;YAC3E,GAAGA,QAAQ;YACXC,SAAS;gBAAC;gBAAe;aAA2B;QACtD,CAAA;IAEA,MAAMC,gBAAgBxE,yBAAyByD,MAAMO;IACrD,MAAMS,cAActC,WAClBsB,MACAe,eACAd,QAAQP,MAAM,KAAK,UAAU,OAAO;IAEtC,MAAMuB,cAAc;IACpB,MAAMC,cAAcjB,QAAQP,MAAM,KAAK,UAAU,SAAS;IAC1D,MAAMyB,kBAAkB;IACxB,MAAMC,gBAAgB,GAAGjD,YAAY6B,MAAM,CAAC,EAAEK,cAAc;IAC5D,MAAMgB,cACJpB,QAAQP,MAAM,KAAK,UACf,+CACA;IAEN,MAAM4B,kBAAkB;QACtB5B,QAAQO,QAAQP,MAAM;QACtB6B,kBAAkBpB;QAClBqB,sBAAsBjB;QACtBkB,kBAAkBxB,QAAQP,MAAM,KAAK,UAAU,UAAU;QACzDgC,eAAexC,YAAYyC,MAAM;QACjCC,oBAAoB1C,WAAW,CAAC,eAAe;QAC/C2C,eAAe3C,YAAY4C,MAAM;QACjC,GAAGxE,mBAAmB;QACtByE,sBACE9B,QAAQP,MAAM,KAAK,UACf,4BACA;QACNsC,wBACE/B,QAAQP,MAAM,KAAK,UAAU,kBAAkB;QACjDsB;QACAC;QACAgB,aAAa5B;QACba;QACAC;QACAb;QACAc;QACAC;QACA,GAAGtD,QAAQiC,KAAK;IAClB;IAEA,8EAA8E;IAC9E,2EAA2E;IAC3E,qDAAqD;IACrD5D,cACE4D,MACA3D,kBAAkB,YAAY6F,OAAO,EAAE,UACvC1B,KACAc,iBACA;QAAEa,mBAAmB7F,kBAAkB8F,YAAY;IAAC;IAGtD,+EAA+E;IAC/E,8EAA8E;IAC9E,6EAA6E;IAC7E,4BAA4B;IAC5B,MAAMxF,cACJoD,MACA3D,kBAAkBmE,KAAK,OAAO,aAC9B;IAGF,qEAAqE;IACrE,yEAAyE;IACzE9D,UAAUsD,MAAM3D,kBAAkBmE,KAAK,gBAAgB;QACrDe,kBAAkBpB;QAClBkC,UAAU;YACR/B;YACAc;YACAkB,OAAOjB;YACPkB,MAAMvB;YACNwB,MAAMvB;YACNwB,QAAQpC;YACRqC,QAAQxB;YACRyB,YAAYxB;YACZyB,UAAU3C,QAAQP,MAAM;QAC1B;IACF;IACA/B,gBAAgBqC,MAAMQ,KAAK,CAACqC,WAAa;eAAIA;YAAU;SAAmB;IAC1E,MAAM5D,0BAA0Be,MAAMC,QAAQP,MAAM,EAAEL;IACtD,MAAMyD,kBACJ7C,QAAQP,MAAM,KAAK,UACf,yBACA;IACN,MAAMqD,aAAa7G,SACjBsE,KACAnE,kBAAkByC,cAAcC,oBAAoB,OAAO;IAE7D,MAAMiE,qBAAqBxE,iCACzBuC,cAAcnB,IAAI;IAEpB,MAAMqD,KAAK,IAAIvF,WAAWsC,MAAMX;IAChC,MAAM6D,qBAAqB7G,kBACzB,QACA0E,cAAcnB,IAAI,EAClB,UACA;IAEF,MAAMuD,0BAA0B,GAAGhF,YAAY6B,MAAM,CAAC,EAAEhC,UAAUiC,QAAQV,IAAI,EAAE,iBAAiB,CAAC;IAElG,IAAIU,QAAQmD,KAAK,KAAK,QAAQ;QAC5B,MAAMvG,0BACJmD,MACAe,eACA;YACEsC,gBAAgB;YAChBC,iBAAiB;QACnB,GACAjE;QAEF,MAAMxC,0BACJmD,MACAe,eACA;YACEsC,gBAAgB;YAChBC,iBAAiB;QACnB,GACAjE;QAEF,MAAMkE,eAAexC,cAAcyC,OAAO,CAAC,SAAS;QACpD,0EAA0E;QAC1E,yEAAyE;QACzE,6CAA6C;QAC7C,IAAI,CAACD,aAAatD,OAAO,CAACwD,QAAQ,EAAE;YAClC,MAAMC,kBAAkBH,aAAatD,OAAO,CAAC0D,OAAO;YACpD,OAAOJ,aAAatD,OAAO,CAAC0D,OAAO;YACnCJ,aAAatD,OAAO,GAAG;gBACrB,GAAGsD,aAAatD,OAAO;gBACvBwD,UAAU;oBACRR,GAAGW,EAAE,CAAC,GAAGZ,mBAAmB,mCAAmC,CAAC;oBAChEC,GAAGY,KAAK,CAAC,GAAGb,mBAAmB,mCAAmC,CAAC;oBACnEC,GAAGa,KAAK,CACN,UACA,GAAGd,mBAAmB,0CAA0C,CAAC;oBAEnEC,GAAGc,MAAM,CACP,oBACA,GAAGf,mBAAmB,oDAAoD,CAAC;oBAE7EC,GAAGc,MAAM,CACP,cACA,GAAGf,mBAAmB,8CAA8C,CAAC;oBAEvEU;iBACD;gBACDM,UAAU;YACZ;QACF;IACF;IAEAjD,cAAcyC,OAAO,CAACS,QAAQ,GAAG;QAC/BC,UAAU;QACVC,SAAS;YAAC;SAAiC;QAC3ClE,SAAS;YACP0D,SAAS;YACTS,KAAK;QACP;IACF;IACArD,cAAcyC,OAAO,CAAC,aAAa,GAAG;QACpCU,UAAU;QACVjE,SAAS;YACP0D,SAAS,CAAC,IAAI,EAAEZ,WAAW,cAAc,CAAC;YAC1CqB,KAAK;QACP;IACF;IACArD,cAAcyC,OAAO,CAAC,MAAM,GAAG;QAC7BU,UAAU;QACVG,WAAW;YAAC;SAAa;QACzBpE,SAAS;YACP0D,SAAS,CAAC,IAAI,EAAEZ,WAAW,mBAAmB,CAAC;YAC/CqB,KAAK;QACP;QACAE,YAAY;IACd;IACAvD,cAAcyC,OAAO,CAAC,cAAc,GAAG;QACrCU,UAAU;QACVG,WAAW;YAAC;SAAM;QAClBpE,SAAS;YACP0D,SAAS,CAAC,IAAI,EAAEZ,WAAW,CAAC,EAAED,iBAAiB;YAC/CsB,KAAK;QACP;IACF;IACArD,cAAcyC,OAAO,CAAC7B,MAAM,GAAG;QAC7BuC,UAAU;QACVG,WAAW;YAAC;YAAO;SAAc;QACjCpE,SAAS;YACPmE,KAAK;YACLT,SAAS;YACTY,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IACA,4EAA4E;IAC5E,2EAA2E;IAC3E,qDAAqD;IACrD,MAAMC,MACJxE,QAAQmD,KAAK,KAAK,SAAS,MAAMxF,WAAWoC,MAAMC,QAAQwE,GAAG,IAAIC;IAEnE,IAAIzE,QAAQmD,KAAK,KAAK,QAAQ;QAC5B,wEAAwE;QACxE,4EAA4E;QAC5E,8CAA8C;QAC9CrC,cAAcyC,OAAO,CAAC,SAAS,GAAG;YAChCmB,OAAOvH;YACP8G,UAAU;YACVjE,SAAS;gBACP0D,SAAS,GAAGrD,gBAAgB,oDAAoD,EAAE6C,wBAAwB,CAAC,EAAED,oBAAoB;YACnI;YACAmB,WAAW;gBAAC;aAAS;QACvB;QACAjG,+BAA+B2C,eAAe;QAE9C7D,oBACE8C,MACA;YACE4E,SAAS7D;YACTT;YACAuE,iBAAiB;YACjBC,kBAAkB;YAClBC,WAAW;gBAAC5B;aAAwB;QACtC,GACA9D;QAEFjB,+BAA+B2C,eAAe;IAChD;IACA1C,kCAAkC0C,eAAe,WAAW;IAC5DtE,2BAA2BuD,MAAMO,oBAAoBQ;IACrD,0EAA0E;IAC1E,8DAA8D;IAC9D,MAAMiE,WAA0B;QAC9BtF,QAAQO,QAAQP,MAAM;QACtB,GAAI+E,MAAM;YAAEA;QAAI,IAAI,CAAC,CAAC;IACxB;IACAnG,qBACE0B,MACAO,oBACAV,uBACAmF;IAGF,IAAI/E,QAAQmD,KAAK,KAAK,QAAQ;QAC5B,MAAMqB,MAAM,MAAM7G,WAAWoC,MAAMC,QAAQwE,GAAG;QAC9C,MAAM5F,0BAA0BmB,MAAM;YAAEyE;QAAI,GAAGpF;QAC/C,MAAMV,YAAYqB,MAAM;YACtByE;YACAQ,aAAa1E;YACb2E,aAAa1E;YACbL;YACAD;YACAsB,sBAAsBjB;YACtBF;YACA8E,WAAW/E;YACXV,QAAQO,QAAQP,MAAM;YACtBwD;YACAkC,uBAAuB/I,kBACrB,QACA0E,cAAcnB,IAAI,EAClB,UACA;YAEFyF,WAAWpF,QAAQoF,SAAS;YAC5BlC;YACA7C;QACF;IACF;IAEA3D,kBAAkBqD,MAAMX,cAAc;QACpC2F;QACAE,aAAanE,cAAcnB,IAAI;IACjC;IAEA,8EAA8E;IAC9E,6EAA6E;IAC7E,8EAA8E;IAC9E,2EAA2E;IAC3E,8EAA8E;IAC9E,6EAA6E;IAC7E,0EAA0E;IAC1E,6BAA6B;IAC7B,IAAIK,QAAQP,MAAM,KAAK,cAAcvD,2BAA2B,QAAQ;QACtEK,WAAWwD,MAAM,gBAAgB,CAACsF;YAChCA,YAAYC,WAAW,GAAG;gBACxB,GAAGD,YAAYC,WAAW;gBAC1B,mCAAmCrG,WAAW,CAAC,YAAY;gBAC3D,gCAAgCA,WAAW,CAAC,YAAY;YAC1D;YACA,OAAOoG;QACT;IACF;IAEA7G,8BAA8BuB,MAAM;QAClC,mBAAmB;QACnB2B,QAAQ;IACV;IAEA,MAAM7D,gCAAgCkC,MAAM;QAACH;KAAsB;IAEnE,MAAMrC,qBAAqBwC;IAC3B,OAAO,IACLnC,oBAAoBmC,MAAMC,QAAQW,yBAAyB,EAAE;YAC3D4E,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF,eAAezF,eAAe"}

package/src/ts/rdb/trpc-connection/__snapshots__/generator.spec.ts.snap

@@ -82,3 +82,42 @@ export const createDbPlugin = () => {
 };
 "
 `;
+
+exports[`ts#rdb trpc-connection generator > should widen the root context in init.ts 1`] = `
+"import { IDbContext } from './middleware/db.js';
+import { initTRPC } from '@trpc/server';
+import {
+  createErrorPlugin,
+  createLoggerPlugin,
+  IMiddlewareContext,
+} from './middleware/index.js';
+
+export type Context = IMiddlewareContext & IDbContext;
+
+export const t = initTRPC.context<Context>().create();
+
+export const publicProcedure = t.procedure
+  .concat(createLoggerPlugin())
+  .concat(createErrorPlugin());
+"
+`;
+
+exports[`ts#rdb trpc-connection generator > should widen the root context once per database 1`] = `
+"import { IOtherDbContext } from './middleware/other-db.js';
+import { IDbContext } from './middleware/db.js';
+import { initTRPC } from '@trpc/server';
+import {
+  createErrorPlugin,
+  createLoggerPlugin,
+  IMiddlewareContext,
+} from './middleware/index.js';
+
+export type Context = IMiddlewareContext & IDbContext & IOtherDbContext;
+
+export const t = initTRPC.context<Context>().create();
+
+export const publicProcedure = t.procedure
+  .concat(createLoggerPlugin())
+  .concat(createErrorPlugin());
+"
+`;

package/src/ts/rdb/trpc-connection/generator.js

@@ -3,6 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */ import { generateFiles, joinPathFragments, OverwriteStrategy, updateProjectConfiguration } from "@nx/devkit";
 import camelCase from "lodash.camelcase";
+import { addDestructuredImport, applyGritQL, matchGritQL } from "../../../utils/ast.js";
 import { formatFilesInSubtree } from "../../../utils/format.js";
 import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
 import { kebabCase, pascalCase } from "../../../utils/names.js";
@@ -35,6 +36,18 @@ export const tsRdbTrpcConnectionGenerator = async (tree, options)=>{
     }, {
         overwriteStrategy: OverwriteStrategy.KeepExisting
     });
+    // Widen the root context so procedures composed with the plugin typecheck.
+    const initPath = joinPathFragments(sourceProject.root, 'src', 'init.ts');
+    const contextInterface = `I${rdbNamePascal}Context`;
+    if (tree.exists(initPath)) {
+        await applyGritQL(tree, initPath, `\`export type Context = $ctx;\` => \`export type Context = $ctx & ${contextInterface};\` where { $ctx <: not contains \`${contextInterface}\` }`);
+        const contextIncludesDb = await matchGritQL(tree, initPath, `\`export type Context = $ctx;\` where { $ctx <: contains \`${contextInterface}\` }`);
+        if (contextIncludesDb) {
+            await addDestructuredImport(tree, initPath, [
+                contextInterface
+            ], `./middleware/${rdbNameKebab}.js`);
+        }
+    }
     // Recorded so the version sync can identify this connection.
     addComponentGeneratorMetadata(tree, sourceProject.name, TS_RDB_TRPC_CONNECTION_GENERATOR_INFO, joinPathFragments('src', 'middleware', `${rdbNameKebab}.ts`), rdbNameCamel);
     await addGeneratorMetricsIfApplicable(tree, [

package/src/ts/rdb/trpc-connection/generator.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/ts/rdb/trpc-connection/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport camelCase from 'lodash.camelcase';\nimport { formatFilesInSubtree } from '../../../utils/format.js';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics.js';\nimport { kebabCase, pascalCase } from '../../../utils/names.js';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx.js';\nimport type { TsRdbTrpcConnectionGeneratorSchema } from './schema';\n\nexport const TS_RDB_TRPC_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const tsRdbTrpcConnectionGenerator = async (\n  tree: Tree,\n  options: TsRdbTrpcConnectionGeneratorSchema,\n): Promise<void> => {\n  const sourceProject = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const targetProject = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  if (sourceProject.targets?.['dev']) {\n    addDependencyToTargetIfNotPresent(sourceProject, 'dev', {\n      projects: [targetProject.name],\n      target: 'dev',\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  const rdbBaseName = targetProject.name.split('/').pop();\n  const rdbNameKebab = kebabCase(rdbBaseName);\n  const rdbNameCamel = camelCase(rdbBaseName);\n  const rdbNamePascal = pascalCase(rdbBaseName);\n  const rdbPackageAlias = targetProject.name;\n  const engine = ((targetProject.metadata as any) ?? {}).engine ?? 'PostgreSQL';\n\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    sourceProject.root,\n    { rdbNameKebab, rdbNameCamel, rdbNamePascal, rdbPackageAlias, engine },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Recorded so the version sync can identify this connection.\n  addComponentGeneratorMetadata(\n    tree,\n    sourceProject.name,\n    TS_RDB_TRPC_CONNECTION_GENERATOR_INFO,\n    joinPathFragments('src', 'middleware', `${rdbNameKebab}.ts`),\n    rdbNameCamel,\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    TS_RDB_TRPC_CONNECTION_GENERATOR_INFO,\n  ]);\n  await formatFilesInSubtree(tree);\n};\n\nexport default tsRdbTrpcConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","camelCase","formatFilesInSubtree","addGeneratorMetricsIfApplicable","kebabCase","pascalCase","addComponentGeneratorMetadata","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","TS_RDB_TRPC_CONNECTION_GENERATOR_INFO","filename","tsRdbTrpcConnectionGenerator","tree","options","sourceProject","targetProject","targets","projects","name","target","rdbBaseName","split","pop","rdbNameKebab","rdbNameCamel","rdbNamePascal","rdbPackageAlias","engine","metadata","dirname","root","overwriteStrategy","KeepExisting"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,OAAOC,eAAe,mBAAmB;AACzC,SAASC,oBAAoB,QAAQ,2BAA2B;AAChE,SAASC,+BAA+B,QAAQ,4BAA4B;AAC5E,SAASC,SAAS,EAAEC,UAAU,QAAQ,0BAA0B;AAChE,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAuB;AAG9B,OAAO,MAAMC,wCACXF,iBAAiB,YAAYG,QAAQ,EAAE;AAEzC,OAAO,MAAMC,+BAA+B,OAC1CC,MACAC;IAEA,MAAMC,gBAAgBN,oCACpBI,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBP,oCACpBI,MACAC,QAAQE,aAAa;IAGvB,IAAID,cAAcE,OAAO,EAAE,CAAC,MAAM,EAAE;QAClCV,kCAAkCQ,eAAe,OAAO;YACtDG,UAAU;gBAACF,cAAcG,IAAI;aAAC;YAC9BC,QAAQ;QACV;QACApB,2BAA2Ba,MAAME,cAAcI,IAAI,EAAEJ;IACvD;IAEA,MAAMM,cAAcL,cAAcG,IAAI,CAACG,KAAK,CAAC,KAAKC,GAAG;IACrD,MAAMC,eAAepB,UAAUiB;IAC/B,MAAMI,eAAexB,UAAUoB;IAC/B,MAAMK,gBAAgBrB,WAAWgB;IACjC,MAAMM,kBAAkBX,cAAcG,IAAI;IAC1C,MAAMS,SAAS,AAAC,CAAA,AAACZ,cAAca,QAAQ,IAAY,CAAC,CAAA,EAAGD,MAAM,IAAI;IAEjE/B,cACEgB,MACAf,kBAAkB,YAAYgC,OAAO,EAAE,UACvCf,cAAcgB,IAAI,EAClB;QAAEP;QAAcC;QAAcC;QAAeC;QAAiBC;IAAO,GACrE;QAAEI,mBAAmBjC,kBAAkBkC,YAAY;IAAC;IAGtD,6DAA6D;IAC7D3B,8BACEO,MACAE,cAAcI,IAAI,EAClBT,uCACAZ,kBAAkB,OAAO,cAAc,GAAG0B,aAAa,GAAG,CAAC,GAC3DC;IAGF,MAAMtB,gCAAgCU,MAAM;QAC1CH;KACD;IACD,MAAMR,qBAAqBW;AAC7B,EAAE;AAEF,eAAeD,6BAA6B"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/ts/rdb/trpc-connection/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport camelCase from 'lodash.camelcase';\nimport {\n  addDestructuredImport,\n  applyGritQL,\n  matchGritQL,\n} from '../../../utils/ast.js';\nimport { formatFilesInSubtree } from '../../../utils/format.js';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics.js';\nimport { kebabCase, pascalCase } from '../../../utils/names.js';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx.js';\nimport type { TsRdbTrpcConnectionGeneratorSchema } from './schema';\n\nexport const TS_RDB_TRPC_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const tsRdbTrpcConnectionGenerator = async (\n  tree: Tree,\n  options: TsRdbTrpcConnectionGeneratorSchema,\n): Promise<void> => {\n  const sourceProject = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const targetProject = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  if (sourceProject.targets?.['dev']) {\n    addDependencyToTargetIfNotPresent(sourceProject, 'dev', {\n      projects: [targetProject.name],\n      target: 'dev',\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  const rdbBaseName = targetProject.name.split('/').pop();\n  const rdbNameKebab = kebabCase(rdbBaseName);\n  const rdbNameCamel = camelCase(rdbBaseName);\n  const rdbNamePascal = pascalCase(rdbBaseName);\n  const rdbPackageAlias = targetProject.name;\n  const engine = ((targetProject.metadata as any) ?? {}).engine ?? 'PostgreSQL';\n\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    sourceProject.root,\n    { rdbNameKebab, rdbNameCamel, rdbNamePascal, rdbPackageAlias, engine },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Widen the root context so procedures composed with the plugin typecheck.\n  const initPath = joinPathFragments(sourceProject.root, 'src', 'init.ts');\n  const contextInterface = `I${rdbNamePascal}Context`;\n  if (tree.exists(initPath)) {\n    await applyGritQL(\n      tree,\n      initPath,\n      `\\`export type Context = $ctx;\\` => \\`export type Context = $ctx & ${contextInterface};\\` where { $ctx <: not contains \\`${contextInterface}\\` }`,\n    );\n\n    const contextIncludesDb = await matchGritQL(\n      tree,\n      initPath,\n      `\\`export type Context = $ctx;\\` where { $ctx <: contains \\`${contextInterface}\\` }`,\n    );\n    if (contextIncludesDb) {\n      await addDestructuredImport(\n        tree,\n        initPath,\n        [contextInterface],\n        `./middleware/${rdbNameKebab}.js`,\n      );\n    }\n  }\n\n  // Recorded so the version sync can identify this connection.\n  addComponentGeneratorMetadata(\n    tree,\n    sourceProject.name,\n    TS_RDB_TRPC_CONNECTION_GENERATOR_INFO,\n    joinPathFragments('src', 'middleware', `${rdbNameKebab}.ts`),\n    rdbNameCamel,\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    TS_RDB_TRPC_CONNECTION_GENERATOR_INFO,\n  ]);\n  await formatFilesInSubtree(tree);\n};\n\nexport default tsRdbTrpcConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","camelCase","addDestructuredImport","applyGritQL","matchGritQL","formatFilesInSubtree","addGeneratorMetricsIfApplicable","kebabCase","pascalCase","addComponentGeneratorMetadata","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","TS_RDB_TRPC_CONNECTION_GENERATOR_INFO","filename","tsRdbTrpcConnectionGenerator","tree","options","sourceProject","targetProject","targets","projects","name","target","rdbBaseName","split","pop","rdbNameKebab","rdbNameCamel","rdbNamePascal","rdbPackageAlias","engine","metadata","dirname","root","overwriteStrategy","KeepExisting","initPath","contextInterface","exists","contextIncludesDb"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,OAAOC,eAAe,mBAAmB;AACzC,SACEC,qBAAqB,EACrBC,WAAW,EACXC,WAAW,QACN,wBAAwB;AAC/B,SAASC,oBAAoB,QAAQ,2BAA2B;AAChE,SAASC,+BAA+B,QAAQ,4BAA4B;AAC5E,SAASC,SAAS,EAAEC,UAAU,QAAQ,0BAA0B;AAChE,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAuB;AAG9B,OAAO,MAAMC,wCACXF,iBAAiB,YAAYG,QAAQ,EAAE;AAEzC,OAAO,MAAMC,+BAA+B,OAC1CC,MACAC;IAEA,MAAMC,gBAAgBN,oCACpBI,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBP,oCACpBI,MACAC,QAAQE,aAAa;IAGvB,IAAID,cAAcE,OAAO,EAAE,CAAC,MAAM,EAAE;QAClCV,kCAAkCQ,eAAe,OAAO;YACtDG,UAAU;gBAACF,cAAcG,IAAI;aAAC;YAC9BC,QAAQ;QACV;QACAvB,2BAA2BgB,MAAME,cAAcI,IAAI,EAAEJ;IACvD;IAEA,MAAMM,cAAcL,cAAcG,IAAI,CAACG,KAAK,CAAC,KAAKC,GAAG;IACrD,MAAMC,eAAepB,UAAUiB;IAC/B,MAAMI,eAAe3B,UAAUuB;IAC/B,MAAMK,gBAAgBrB,WAAWgB;IACjC,MAAMM,kBAAkBX,cAAcG,IAAI;IAC1C,MAAMS,SAAS,AAAC,CAAA,AAACZ,cAAca,QAAQ,IAAY,CAAC,CAAA,EAAGD,MAAM,IAAI;IAEjElC,cACEmB,MACAlB,kBAAkB,YAAYmC,OAAO,EAAE,UACvCf,cAAcgB,IAAI,EAClB;QAAEP;QAAcC;QAAcC;QAAeC;QAAiBC;IAAO,GACrE;QAAEI,mBAAmBpC,kBAAkBqC,YAAY;IAAC;IAGtD,2EAA2E;IAC3E,MAAMC,WAAWvC,kBAAkBoB,cAAcgB,IAAI,EAAE,OAAO;IAC9D,MAAMI,mBAAmB,CAAC,CAAC,EAAET,cAAc,OAAO,CAAC;IACnD,IAAIb,KAAKuB,MAAM,CAACF,WAAW;QACzB,MAAMlC,YACJa,MACAqB,UACA,CAAC,kEAAkE,EAAEC,iBAAiB,mCAAmC,EAAEA,iBAAiB,IAAI,CAAC;QAGnJ,MAAME,oBAAoB,MAAMpC,YAC9BY,MACAqB,UACA,CAAC,2DAA2D,EAAEC,iBAAiB,IAAI,CAAC;QAEtF,IAAIE,mBAAmB;YACrB,MAAMtC,sBACJc,MACAqB,UACA;gBAACC;aAAiB,EAClB,CAAC,aAAa,EAAEX,aAAa,GAAG,CAAC;QAErC;IACF;IAEA,6DAA6D;IAC7DlB,8BACEO,MACAE,cAAcI,IAAI,EAClBT,uCACAf,kBAAkB,OAAO,cAAc,GAAG6B,aAAa,GAAG,CAAC,GAC3DC;IAGF,MAAMtB,gCAAgCU,MAAM;QAC1CH;KACD;IACD,MAAMR,qBAAqBW;AAC7B,EAAE;AAEF,eAAeD,6BAA6B"}

package/src/ts/react-website/agui/files/common/src/components/__agentNameKebabCase__-chat.tsx.template

@@ -0,0 +1,9 @@
+import { CopilotChat } from './copilot';
+import type { CopilotChatProps } from '@copilotkit/react-core/v2';
+import { <%= agentNameConstantName %>_ID } from '../hooks/useAgui<%= agentNameClassName %>';
+
+export const <%= agentNameClassName %>Chat = (
+  props: Omit<CopilotChatProps, 'agentId'>,
+) => <CopilotChat agentId={<%= agentNameConstantName %>_ID} {...props} />;
+
+export default <%= agentNameClassName %>Chat;

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

@@ -104,6 +104,9 @@ class AgentCoreHttpAgent extends HttpAgent {
 }
 <%_ } _%>
 
+// The id this agent is registered under, used as `agentId` by CopilotKit components.
+export const <%= agentNameConstantName %>_ID = '<%= agentName %>';
+
 export const useAgui<%= agentNameClassName %> = (): Record<string, AbstractAgent> => {
   const runtimeConfig = useRuntimeConfig();
 <%_ if (gatewayRoute) { _%>
@@ -138,6 +141,6 @@ export const useAgui<%= agentNameClassName %> = (): Record<string, AbstractAgent
     const agent = new AgentCoreHttpAgent({ url });
     <%_ } _%>
 
-    return { '<%= agentName %>': agent };
+    return { [<%= agentNameConstantName %>_ID]: agent };
   }, [url<% if (auth === 'iam') { %>, sigv4<% } else if (auth === 'cognito') { %>, auth<% } %>]);
 };

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

@@ -50,11 +50,7 @@ export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.
 } & {
     versionOnly: true;
 }, {
-    readonly name: "clsx";
-} & {
-    versionOnly: true;
-}, {
-    readonly name: "tailwind-merge";
+    readonly name: "cn";
 } & {
     versionOnly: true;
 }, {

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

@@ -6,7 +6,7 @@ import { addAgentRuntimeToConnectionNamespace } from "../../../connection/agent-
 import { addTsDependencies } from "../../../utils/add-dependencies.js";
 import { addDestructuredImport, addSingleImport, applyGritQL } from "../../../utils/ast.js";
 import { declareDependencies, ownedElsewhere } from "../../../utils/declared-dependencies.js";
-import { kebabCase } from "../../../utils/names.js";
+import { kebabCase, snakeCase } from "../../../utils/names.js";
 import { getNpmScopePrefix } from "../../../utils/npm-scope.js";
 import { registerPnpmBuiltDependencies } from "../../../utils/pnpm-workspace.js";
 import { addSharedShadcnDependency, SHADCN_DEPENDENCIES, sharedShadcnGenerator } from "../../../utils/shared-shadcn.js";
@@ -76,12 +76,16 @@ export const DEPENDENCIES = declareDependencies()({
         auth
     };
     const scopeAlias = getNpmScopePrefix(tree);
-    // Generates `src/components/AguiProvider.tsx` (if absent) and
+    // Generates `src/components/AguiProvider.tsx` (if absent),
+    // `src/components/<agent-name>-chat.tsx` and
     // `src/hooks/useAgui<AgentName>.tsx`. Existing files are kept so re-running
     // the generator is idempotent.
     generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', 'common'), frontendProjectConfig.root, {
         agentName,
         agentNameClassName,
+        agentNameKebabCase: kebabCase(agentNameClassName),
+        // STORY_AGENT, naming the exported `<name>_ID` constant
+        agentNameConstantName: snakeCase(agentNameClassName).toUpperCase(),
         auth,
         gatewayRoute
     }, {

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

@@ -1 +1 @@
-{"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  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type ProjectConfiguration,\n  type Tree,\n} from '@nx/devkit';\nimport { addAgentRuntimeToConnectionNamespace } from '../../../connection/agent-runtime-config.js';\nimport { addTsDependencies } from '../../../utils/add-dependencies.js';\nimport {\n  addDestructuredImport,\n  addSingleImport,\n  applyGritQL,\n} from '../../../utils/ast.js';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../../utils/declared-dependencies.js';\nimport { kebabCase } from '../../../utils/names.js';\nimport { getNpmScopePrefix } from '../../../utils/npm-scope.js';\nimport { registerPnpmBuiltDependencies } from '../../../utils/pnpm-workspace.js';\nimport {\n  addSharedShadcnDependency,\n  SHADCN_DEPENDENCIES,\n  sharedShadcnGenerator,\n} from '../../../utils/shared-shadcn.js';\nimport { runtimeConfigGenerator } from '../runtime-config/generator.js';\n\n/**\n * The values this helper's predicates read. Its callers record them, so the\n * theme and auth branches taken here can be confirmed at upgrade time.\n */\nexport interface AgUiMetadata {\n  readonly theme: string;\n  readonly auth: string;\n}\n\n// Each entry names the theme or auth branch it belongs to, so the same\n// declaration drives both adding and the version sync.\nexport const DEPENDENCIES = declareDependencies<AgUiMetadata>()({\n  ts: [\n    { name: '@copilotkit/react-core' },\n    { name: '@ag-ui/client' },\n    {\n      name: '@cloudscape-design/chat-components',\n      when: (m) => m.theme === 'cloudscape',\n    },\n    { name: 'lucide-react', when: (m) => m.theme === 'shadcn' },\n    { name: 'oidc-client-ts', when: (m) => m.auth === 'iam' },\n    { name: 'aws4fetch', when: (m) => m.auth === 'iam' },\n    {\n      name: '@aws-sdk/credential-provider-cognito-identity',\n      when: (m) => m.auth === 'iam',\n    },\n    {\n      name: 'react-oidc-context',\n      when: (m) => m.auth === 'iam' || m.auth === 'cognito',\n    },\n    { name: '@smithy/types', when: (m) => m.auth === 'iam', dev: true },\n    // `sharedShadcnGenerator` adds these to the shared shadcn project it\n    // creates, not to the website.\n    ...ownedElsewhere(SHADCN_DEPENDENCIES),\n  ],\n});\n\nexport type AgUiAuth = 'iam' | 'cognito' | 'none';\n\nexport type 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   * When set, the generated hook reads the gateway's URL from\n   * runtimeConfig.gateways and routes to the agent via the gateway's\n   * path-based `/<targetName>/invocations` route instead of invoking the\n   * runtime directly. The gateway publishes its own URL to the website, so\n   * the agent's construct is left unpatched.\n   */\n  gatewayRoute?: {\n    gatewayClassName: string;\n    targetName: string;\n  };\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 {\n    frontendProjectConfig,\n    agentName,\n    agentNameClassName,\n    auth,\n    gatewayRoute,\n  } = options;\n\n  const theme = resolveAgUiTheme(frontendProjectConfig);\n  // The values the declaration's predicates read. Callers record the same pair,\n  // so what is added here is exactly what the version sync owns.\n  const metadata: AgUiMetadata = { theme, auth };\n  const scopeAlias = 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, gatewayRoute },\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, DEPENDENCIES);\n    addSharedShadcnDependency(tree, frontendProjectConfig.root);\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  addTsDependencies(tree, DEPENDENCIES, {\n    metadata,\n    projectRoot: frontendProjectConfig.root,\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. When routing via a gateway, the gateway\n  // publishes its own URL instead.\n  if (!gatewayRoute) {\n    await addAgentRuntimeToConnectionNamespace(tree, {\n      agentNameKebabCase: kebabCase(agentNameClassName),\n      agentNameClassName,\n    });\n  }\n};\n\n/**\n * The theme module this helper vends, from the website's `metadata.ux`. Callers\n * record the resolved value, so the theme predicates read what the code branches\n * on.\n */\nexport const 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":["generateFiles","joinPathFragments","OverwriteStrategy","addAgentRuntimeToConnectionNamespace","addTsDependencies","addDestructuredImport","addSingleImport","applyGritQL","declareDependencies","ownedElsewhere","kebabCase","getNpmScopePrefix","registerPnpmBuiltDependencies","addSharedShadcnDependency","SHADCN_DEPENDENCIES","sharedShadcnGenerator","runtimeConfigGenerator","DEPENDENCIES","ts","name","when","m","theme","auth","dev","addAgUiReactConnection","tree","options","frontendProjectConfig","agentName","agentNameClassName","gatewayRoute","resolveAgUiTheme","metadata","scopeAlias","dirname","root","overwriteStrategy","KeepExisting","sourceRoot","project","registerAgentHookInProvider","mainTsxPath","projectRoot","agentNameKebabCase","ux","toLowerCase","providerPath","hookName","varName","charAt","slice"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,QAGZ,aAAa;AACpB,SAASC,oCAAoC,QAAQ,8CAA8C;AACnG,SAASC,iBAAiB,QAAQ,qCAAqC;AACvE,SACEC,qBAAqB,EACrBC,eAAe,EACfC,WAAW,QACN,wBAAwB;AAC/B,SACEC,mBAAmB,EACnBC,cAAc,QACT,0CAA0C;AACjD,SAASC,SAAS,QAAQ,0BAA0B;AACpD,SAASC,iBAAiB,QAAQ,8BAA8B;AAChE,SAASC,6BAA6B,QAAQ,mCAAmC;AACjF,SACEC,yBAAyB,EACzBC,mBAAmB,EACnBC,qBAAqB,QAChB,kCAAkC;AACzC,SAASC,sBAAsB,QAAQ,iCAAiC;AAWxE,uEAAuE;AACvE,uDAAuD;AACvD,OAAO,MAAMC,eAAeT,sBAAoC;IAC9DU,IAAI;QACF;YAAEC,MAAM;QAAyB;QACjC;YAAEA,MAAM;QAAgB;QACxB;YACEA,MAAM;YACNC,MAAM,CAACC,IAAMA,EAAEC,KAAK,KAAK;QAC3B;QACA;YAAEH,MAAM;YAAgBC,MAAM,CAACC,IAAMA,EAAEC,KAAK,KAAK;QAAS;QAC1D;YAAEH,MAAM;YAAkBC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;QAAM;QACxD;YAAEJ,MAAM;YAAaC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;QAAM;QACnD;YACEJ,MAAM;YACNC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;QAC1B;QACA;YACEJ,MAAM;YACNC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK,SAASF,EAAEE,IAAI,KAAK;QAC9C;QACA;YAAEJ,MAAM;YAAiBC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;YAAOC,KAAK;QAAK;QAClE,qEAAqE;QACrE,+BAA+B;WAC5Bf,eAAeK;KACnB;AACH,GAAG;AA4BH;;;;;;;;;;;CAWC,GACD,OAAO,MAAMW,yBAAyB,OACpCC,MACAC;IAEA,MAAM,EACJC,qBAAqB,EACrBC,SAAS,EACTC,kBAAkB,EAClBP,IAAI,EACJQ,YAAY,EACb,GAAGJ;IAEJ,MAAML,QAAQU,iBAAiBJ;IAC/B,8EAA8E;IAC9E,+DAA+D;IAC/D,MAAMK,WAAyB;QAAEX;QAAOC;IAAK;IAC7C,MAAMW,aAAavB,kBAAkBe;IAErC,8DAA8D;IAC9D,4EAA4E;IAC5E,+BAA+B;IAC/B1B,cACE0B,MACAzB,kBAAkB,YAAYkC,OAAO,EAAE,SAAS,WAChDP,sBAAsBQ,IAAI,EAC1B;QAAEP;QAAWC;QAAoBP;QAAMQ;IAAa,GACpD;QAAEM,mBAAmBnC,kBAAkBoC,YAAY;IAAC;IAGtD,yEAAyE;IACzE,IAAIhB,UAAU,UAAU;QACtB,MAAMP,sBAAsBW,MAAMT;QAClCJ,0BAA0Ba,MAAME,sBAAsBQ,IAAI;IAC5D;IACApC,cACE0B,MACAzB,kBAAkB,YAAYkC,OAAO,EAAE,SAASb,QAChDM,sBAAsBQ,IAAI,EAC1B;QAAEF;IAAW,GACb;QAAEG,mBAAmBnC,kBAAkBoC,YAAY;IAAC;IAGtD,IAAIf,SAAS,OAAO;QAClBvB,cACE0B,MACAzB,kBACE,YAAYkC,OAAO,EACnB,6CAEFlC,kBAAkB2B,sBAAsBW,UAAU,EAAE,UACpD,CAAC,GACD;YAAEF,mBAAmBnC,kBAAkBoC,YAAY;QAAC;IAExD;IAEA,MAAMtB,uBAAuBU,MAAM;QACjCc,SAASZ,sBAAsBT,IAAI;IACrC;IAEA,uDAAuD;IACvD,MAAMsB,4BACJf,MACAE,uBACAE;IAGF,0DAA0D;IAC1D,MAAMY,cAAczC,kBAClB2B,sBAAsBW,UAAU,EAChC;IAEF,MAAMjC,gBACJoB,MACAgB,aACA,gBACA;IAEF,MAAMnC,YACJmB,MACAgB,aACA,CAAC,8HAA8H,CAAC;IAGlI,yEAAyE;IACzE,0DAA0D;IAC1D,2EAA2E;IAC3E,yEAAyE;IACzE,4DAA4D;IAC5D9B,8BAA8Bc,MAAM;QAAE,gBAAgB;IAAM;IAE5DtB,kBAAkBsB,MAAMT,cAAc;QACpCgB;QACAU,aAAaf,sBAAsBQ,IAAI;IACzC;IAEA,wEAAwE;IACxE,wEAAwE;IACxE,0EAA0E;IAC1E,oEAAoE;IACpE,iCAAiC;IACjC,IAAI,CAACL,cAAc;QACjB,MAAM5B,qCAAqCuB,MAAM;YAC/CkB,oBAAoBlC,UAAUoB;YAC9BA;QACF;IACF;AACF,EAAE;AAEF;;;;CAIC,GACD,OAAO,MAAME,mBAAmB,CAC9BJ;IAEA,MAAMiB,KACHjB,sBAAsBK,QAAQ,EAAUY,IACxCC;IACH,OAAQD;QACN,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;QACT;YACE,OAAO;IACX;AACF,EAAE;AAEF;;;;;;;;;;CAUC,GACD,MAAMJ,8BAA8B,OAClCf,MACAE,uBACAE;IAEA,MAAMiB,eAAe9C,kBACnB2B,sBAAsBW,UAAU,EAChC,cACA;IAEF,MAAMS,WAAW,CAAC,OAAO,EAAElB,oBAAoB;IAC/C,wCAAwC;IACxC,MAAMmB,UAAU,GAAGnB,mBAAmBoB,MAAM,CAAC,GAAGJ,WAAW,KAAKhB,mBAAmBqB,KAAK,CAAC,GAAG,MAAM,CAAC;IAEnG,yBAAyB;IACzB,MAAM9C,sBACJqB,MACAqB,cACA;QAACC;KAAS,EACV,CAAC,SAAS,EAAEA,UAAU;IAGxB,6EAA6E;IAC7E,MAAMzC,YACJmB,MACAqB,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,MAAMzC,YACJmB,MACAqB,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"}
+{"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  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type ProjectConfiguration,\n  type Tree,\n} from '@nx/devkit';\nimport { addAgentRuntimeToConnectionNamespace } from '../../../connection/agent-runtime-config.js';\nimport { addTsDependencies } from '../../../utils/add-dependencies.js';\nimport {\n  addDestructuredImport,\n  addSingleImport,\n  applyGritQL,\n} from '../../../utils/ast.js';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../../utils/declared-dependencies.js';\nimport { kebabCase, snakeCase } from '../../../utils/names.js';\nimport { getNpmScopePrefix } from '../../../utils/npm-scope.js';\nimport { registerPnpmBuiltDependencies } from '../../../utils/pnpm-workspace.js';\nimport {\n  addSharedShadcnDependency,\n  SHADCN_DEPENDENCIES,\n  sharedShadcnGenerator,\n} from '../../../utils/shared-shadcn.js';\nimport { runtimeConfigGenerator } from '../runtime-config/generator.js';\n\n/**\n * The values this helper's predicates read. Its callers record them, so the\n * theme and auth branches taken here can be confirmed at upgrade time.\n */\nexport interface AgUiMetadata {\n  readonly theme: string;\n  readonly auth: string;\n}\n\n// Each entry names the theme or auth branch it belongs to, so the same\n// declaration drives both adding and the version sync.\nexport const DEPENDENCIES = declareDependencies<AgUiMetadata>()({\n  ts: [\n    { name: '@copilotkit/react-core' },\n    { name: '@ag-ui/client' },\n    {\n      name: '@cloudscape-design/chat-components',\n      when: (m) => m.theme === 'cloudscape',\n    },\n    { name: 'lucide-react', when: (m) => m.theme === 'shadcn' },\n    { name: 'oidc-client-ts', when: (m) => m.auth === 'iam' },\n    { name: 'aws4fetch', when: (m) => m.auth === 'iam' },\n    {\n      name: '@aws-sdk/credential-provider-cognito-identity',\n      when: (m) => m.auth === 'iam',\n    },\n    {\n      name: 'react-oidc-context',\n      when: (m) => m.auth === 'iam' || m.auth === 'cognito',\n    },\n    { name: '@smithy/types', when: (m) => m.auth === 'iam', dev: true },\n    // `sharedShadcnGenerator` adds these to the shared shadcn project it\n    // creates, not to the website.\n    ...ownedElsewhere(SHADCN_DEPENDENCIES),\n  ],\n});\n\nexport type AgUiAuth = 'iam' | 'cognito' | 'none';\n\nexport type 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   * When set, the generated hook reads the gateway's URL from\n   * runtimeConfig.gateways and routes to the agent via the gateway's\n   * path-based `/<targetName>/invocations` route instead of invoking the\n   * runtime directly. The gateway publishes its own URL to the website, so\n   * the agent's construct is left unpatched.\n   */\n  gatewayRoute?: {\n    gatewayClassName: string;\n    targetName: string;\n  };\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 {\n    frontendProjectConfig,\n    agentName,\n    agentNameClassName,\n    auth,\n    gatewayRoute,\n  } = options;\n\n  const theme = resolveAgUiTheme(frontendProjectConfig);\n  // The values the declaration's predicates read. Callers record the same pair,\n  // so what is added here is exactly what the version sync owns.\n  const metadata: AgUiMetadata = { theme, auth };\n  const scopeAlias = getNpmScopePrefix(tree);\n\n  // Generates `src/components/AguiProvider.tsx` (if absent),\n  // `src/components/<agent-name>-chat.tsx` 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    {\n      agentName,\n      agentNameClassName,\n      agentNameKebabCase: kebabCase(agentNameClassName),\n      // STORY_AGENT, naming the exported `<name>_ID` constant\n      agentNameConstantName: snakeCase(agentNameClassName).toUpperCase(),\n      auth,\n      gatewayRoute,\n    },\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, DEPENDENCIES);\n    addSharedShadcnDependency(tree, frontendProjectConfig.root);\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  addTsDependencies(tree, DEPENDENCIES, {\n    metadata,\n    projectRoot: frontendProjectConfig.root,\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. When routing via a gateway, the gateway\n  // publishes its own URL instead.\n  if (!gatewayRoute) {\n    await addAgentRuntimeToConnectionNamespace(tree, {\n      agentNameKebabCase: kebabCase(agentNameClassName),\n      agentNameClassName,\n    });\n  }\n};\n\n/**\n * The theme module this helper vends, from the website's `metadata.ux`. Callers\n * record the resolved value, so the theme predicates read what the code branches\n * on.\n */\nexport const 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":["generateFiles","joinPathFragments","OverwriteStrategy","addAgentRuntimeToConnectionNamespace","addTsDependencies","addDestructuredImport","addSingleImport","applyGritQL","declareDependencies","ownedElsewhere","kebabCase","snakeCase","getNpmScopePrefix","registerPnpmBuiltDependencies","addSharedShadcnDependency","SHADCN_DEPENDENCIES","sharedShadcnGenerator","runtimeConfigGenerator","DEPENDENCIES","ts","name","when","m","theme","auth","dev","addAgUiReactConnection","tree","options","frontendProjectConfig","agentName","agentNameClassName","gatewayRoute","resolveAgUiTheme","metadata","scopeAlias","dirname","root","agentNameKebabCase","agentNameConstantName","toUpperCase","overwriteStrategy","KeepExisting","sourceRoot","project","registerAgentHookInProvider","mainTsxPath","projectRoot","ux","toLowerCase","providerPath","hookName","varName","charAt","slice"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,QAGZ,aAAa;AACpB,SAASC,oCAAoC,QAAQ,8CAA8C;AACnG,SAASC,iBAAiB,QAAQ,qCAAqC;AACvE,SACEC,qBAAqB,EACrBC,eAAe,EACfC,WAAW,QACN,wBAAwB;AAC/B,SACEC,mBAAmB,EACnBC,cAAc,QACT,0CAA0C;AACjD,SAASC,SAAS,EAAEC,SAAS,QAAQ,0BAA0B;AAC/D,SAASC,iBAAiB,QAAQ,8BAA8B;AAChE,SAASC,6BAA6B,QAAQ,mCAAmC;AACjF,SACEC,yBAAyB,EACzBC,mBAAmB,EACnBC,qBAAqB,QAChB,kCAAkC;AACzC,SAASC,sBAAsB,QAAQ,iCAAiC;AAWxE,uEAAuE;AACvE,uDAAuD;AACvD,OAAO,MAAMC,eAAeV,sBAAoC;IAC9DW,IAAI;QACF;YAAEC,MAAM;QAAyB;QACjC;YAAEA,MAAM;QAAgB;QACxB;YACEA,MAAM;YACNC,MAAM,CAACC,IAAMA,EAAEC,KAAK,KAAK;QAC3B;QACA;YAAEH,MAAM;YAAgBC,MAAM,CAACC,IAAMA,EAAEC,KAAK,KAAK;QAAS;QAC1D;YAAEH,MAAM;YAAkBC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;QAAM;QACxD;YAAEJ,MAAM;YAAaC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;QAAM;QACnD;YACEJ,MAAM;YACNC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;QAC1B;QACA;YACEJ,MAAM;YACNC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK,SAASF,EAAEE,IAAI,KAAK;QAC9C;QACA;YAAEJ,MAAM;YAAiBC,MAAM,CAACC,IAAMA,EAAEE,IAAI,KAAK;YAAOC,KAAK;QAAK;QAClE,qEAAqE;QACrE,+BAA+B;WAC5BhB,eAAeM;KACnB;AACH,GAAG;AA4BH;;;;;;;;;;;CAWC,GACD,OAAO,MAAMW,yBAAyB,OACpCC,MACAC;IAEA,MAAM,EACJC,qBAAqB,EACrBC,SAAS,EACTC,kBAAkB,EAClBP,IAAI,EACJQ,YAAY,EACb,GAAGJ;IAEJ,MAAML,QAAQU,iBAAiBJ;IAC/B,8EAA8E;IAC9E,+DAA+D;IAC/D,MAAMK,WAAyB;QAAEX;QAAOC;IAAK;IAC7C,MAAMW,aAAavB,kBAAkBe;IAErC,2DAA2D;IAC3D,6CAA6C;IAC7C,4EAA4E;IAC5E,+BAA+B;IAC/B3B,cACE2B,MACA1B,kBAAkB,YAAYmC,OAAO,EAAE,SAAS,WAChDP,sBAAsBQ,IAAI,EAC1B;QACEP;QACAC;QACAO,oBAAoB5B,UAAUqB;QAC9B,wDAAwD;QACxDQ,uBAAuB5B,UAAUoB,oBAAoBS,WAAW;QAChEhB;QACAQ;IACF,GACA;QAAES,mBAAmBvC,kBAAkBwC,YAAY;IAAC;IAGtD,yEAAyE;IACzE,IAAInB,UAAU,UAAU;QACtB,MAAMP,sBAAsBW,MAAMT;QAClCJ,0BAA0Ba,MAAME,sBAAsBQ,IAAI;IAC5D;IACArC,cACE2B,MACA1B,kBAAkB,YAAYmC,OAAO,EAAE,SAASb,QAChDM,sBAAsBQ,IAAI,EAC1B;QAAEF;IAAW,GACb;QAAEM,mBAAmBvC,kBAAkBwC,YAAY;IAAC;IAGtD,IAAIlB,SAAS,OAAO;QAClBxB,cACE2B,MACA1B,kBACE,YAAYmC,OAAO,EACnB,6CAEFnC,kBAAkB4B,sBAAsBc,UAAU,EAAE,UACpD,CAAC,GACD;YAAEF,mBAAmBvC,kBAAkBwC,YAAY;QAAC;IAExD;IAEA,MAAMzB,uBAAuBU,MAAM;QACjCiB,SAASf,sBAAsBT,IAAI;IACrC;IAEA,uDAAuD;IACvD,MAAMyB,4BACJlB,MACAE,uBACAE;IAGF,0DAA0D;IAC1D,MAAMe,cAAc7C,kBAClB4B,sBAAsBc,UAAU,EAChC;IAEF,MAAMrC,gBACJqB,MACAmB,aACA,gBACA;IAEF,MAAMvC,YACJoB,MACAmB,aACA,CAAC,8HAA8H,CAAC;IAGlI,yEAAyE;IACzE,0DAA0D;IAC1D,2EAA2E;IAC3E,yEAAyE;IACzE,4DAA4D;IAC5DjC,8BAA8Bc,MAAM;QAAE,gBAAgB;IAAM;IAE5DvB,kBAAkBuB,MAAMT,cAAc;QACpCgB;QACAa,aAAalB,sBAAsBQ,IAAI;IACzC;IAEA,wEAAwE;IACxE,wEAAwE;IACxE,0EAA0E;IAC1E,oEAAoE;IACpE,iCAAiC;IACjC,IAAI,CAACL,cAAc;QACjB,MAAM7B,qCAAqCwB,MAAM;YAC/CW,oBAAoB5B,UAAUqB;YAC9BA;QACF;IACF;AACF,EAAE;AAEF;;;;CAIC,GACD,OAAO,MAAME,mBAAmB,CAC9BJ;IAEA,MAAMmB,KACHnB,sBAAsBK,QAAQ,EAAUc,IACxCC;IACH,OAAQD;QACN,KAAK;YACH,OAAO;QACT,KAAK;YACH,OAAO;QACT;YACE,OAAO;IACX;AACF,EAAE;AAEF;;;;;;;;;;CAUC,GACD,MAAMH,8BAA8B,OAClClB,MACAE,uBACAE;IAEA,MAAMmB,eAAejD,kBACnB4B,sBAAsBc,UAAU,EAChC,cACA;IAEF,MAAMQ,WAAW,CAAC,OAAO,EAAEpB,oBAAoB;IAC/C,wCAAwC;IACxC,MAAMqB,UAAU,GAAGrB,mBAAmBsB,MAAM,CAAC,GAAGJ,WAAW,KAAKlB,mBAAmBuB,KAAK,CAAC,GAAG,MAAM,CAAC;IAEnG,yBAAyB;IACzB,MAAMjD,sBACJsB,MACAuB,cACA;QAACC;KAAS,EACV,CAAC,SAAS,EAAEA,UAAU;IAGxB,6EAA6E;IAC7E,MAAM5C,YACJoB,MACAuB,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,MAAM5C,YACJoB,MACAuB,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"}