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

package/src/py/rdb/generator.js

@@ -0,0 +1,328 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ import { relative } from "node:path";
+import { generateFiles, joinPathFragments, readProjectConfiguration, updateProjectConfiguration } from "@nx/devkit";
+import { addPyDependencies, addTsDependencies } from "../../utils/add-dependencies.js";
+import { addPythonBundleTarget } from "../../utils/bundle/bundle.js";
+import { resolveContainers } from "../../utils/containers.js";
+import { declareDependencies, ownedElsewhere } from "../../utils/declared-dependencies.js";
+import { addDockerScanTarget, DOCKER_DEPENDENCIES } from "../../utils/docker.js";
+import { formatFilesInSubtree } from "../../utils/format.js";
+import { FS_DEPENDENCIES, FsCommands } from "../../utils/fs.js";
+import { resolveIac } from "../../utils/iac.js";
+import { installDependencies } from "../../utils/install.js";
+import { addGeneratorMetricsIfApplicable } from "../../utils/metrics.js";
+import { kebabCase, snakeCase, toClassName } from "../../utils/names.js";
+import { getNpmScope } from "../../utils/npm-scope.js";
+import { addDependencyToTargetIfNotPresent, addGeneratorMetadata, getGeneratorInfo, projectExists } from "../../utils/nx.js";
+import { assignPort } from "../../utils/port.js";
+import { addRdbInfra } from "../../utils/rdb-constructs/rdb-constructs.js";
+import { SHARED_CONSTRUCTS_DEPENDENCIES, sharedConstructsGenerator } from "../../utils/shared-constructs.js";
+import { PACKAGES_DIR, SHARED_SCRIPTS_DIR } from "../../utils/shared-constructs-constants.js";
+import { SHARED_RDB_SCRIPTS_DEPENDENCIES, sharedRdbScriptsGenerator } from "../../utils/shared-rdb-scripts.js";
+import { PY_VERSIONS } from "../../utils/versions.js";
+import pyProjectGenerator, { getPyProjectDetails } from "../project/generator.js";
+/** The Postgres engine, which MySQL takes none of. */ const isPostgres = (m)=>m.engine !== 'mysql';
+// Each entry names the engine branch it belongs to, so the same declaration
+// drives both adding and the version sync.
+export const DEPENDENCIES = declareDependencies()({
+    ts: [
+        // The engine client the local dev script wait-for-*-db.ts imports.
+        {
+            name: 'mariadb',
+            when: (m)=>m.engine === 'mysql',
+            root: true
+        },
+        {
+            name: 'pg',
+            when: isPostgres,
+            root: true
+        },
+        {
+            name: '@types/pg',
+            when: isPostgres,
+            dev: true,
+            root: true
+        },
+        // Added by the helpers that own the projects they belong to.
+        ...ownedElsewhere(FS_DEPENDENCIES),
+        ...ownedElsewhere(DOCKER_DEPENDENCIES),
+        ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES),
+        ...ownedElsewhere(SHARED_RDB_SCRIPTS_DEPENDENCIES)
+    ],
+    py: [
+        {
+            name: 'sqlmodel'
+        },
+        {
+            name: 'alembic'
+        },
+        // SQLAlchemy's async engine (used by connection.py for both engines)
+        // requires greenlet at runtime. Vend it explicitly at a pinned version so
+        // the dependency graph is fully determined and doesn't float to an
+        // unpublished-wheel release at install time.
+        {
+            name: 'greenlet'
+        },
+        {
+            name: 'aiomysql',
+            when: (m)=>m.engine === 'mysql'
+        },
+        {
+            name: 'asyncpg',
+            when: (m)=>m.engine !== 'mysql'
+        },
+        {
+            name: 'boto3'
+        },
+        {
+            name: 'aws-lambda-powertools'
+        }
+    ]
+});
+export const PY_RDB_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
+export const pyRdbGenerator = async (tree, options)=>{
+    const nameClassName = toClassName(options.name);
+    const databaseUser = options.databaseUser ?? 'dbadmin';
+    const databaseName = snakeCase(options.databaseName ?? options.name);
+    const containerEngine = await resolveContainers(tree, 'inherit');
+    const { fullyQualifiedName, dir, normalizedModuleName } = getPyProjectDetails(tree, {
+        name: options.name,
+        directory: options.directory,
+        subDirectory: options.subDirectory
+    });
+    if (!projectExists(tree, fullyQualifiedName)) {
+        await pyProjectGenerator(tree, {
+            name: options.name,
+            directory: options.directory,
+            subDirectory: options.subDirectory,
+            type: 'library'
+        });
+    }
+    const projectConfig = readProjectConfiguration(tree, fullyQualifiedName);
+    const { engine } = options;
+    // Recorded in the metadata below so the version sync can tell a CDK project
+    // from a Terraform one; undefined when no infrastructure was generated.
+    const iac = options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;
+    // Recorded below and read by the declaration's predicates, so the packages
+    // added here are exactly the ones the version sync will own.
+    const metadata = {
+        engine,
+        ...iac ? {
+            iac
+        } : {}
+    };
+    const localDbPort = assignPort(tree, projectConfig, engine === 'mysql' ? 3306 : 5432);
+    const localDbHost = 'localhost';
+    const localDbUser = engine === 'mysql' ? 'root' : 'dbadmin';
+    const localDbPassword = 'password';
+    const containerName = `${getNpmScope(tree)}-${databaseName}`;
+    const dockerImage = engine === 'mysql' ? 'public.ecr.aws/docker/library/mysql:8.0.44' : 'public.ecr.aws/docker/library/postgres:17.7';
+    const templateOptions = {
+        name: normalizedModuleName,
+        runtimeConfigKey: nameClassName,
+        engine,
+        localDbPort,
+        localDbHost,
+        localDbName: databaseName,
+        localDbUser,
+        localDbPassword,
+        containerEngine,
+        containerName,
+        dockerImage,
+        sqlmodelVersion: PY_VERSIONS.sqlmodel,
+        alembicVersion: PY_VERSIONS.alembic,
+        aiomysqlVersion: PY_VERSIONS.aiomysql,
+        asyncpgVersion: PY_VERSIONS.asyncpg,
+        boto3Version: PY_VERSIONS.boto3,
+        awsLambdaPowertoolsVersion: PY_VERSIONS['aws-lambda-powertools']
+    };
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files'), dir, templateOptions);
+    await sharedRdbScriptsGenerator(tree, engine, DEPENDENCIES);
+    const scriptsDir = relative(dir, joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'rdb'));
+    const fs = new FsCommands(tree, DEPENDENCIES);
+    const migrationBundleDir = joinPathFragments('dist', projectConfig.root, 'docker', 'migration');
+    const createDbUserBundleDir = joinPathFragments('dist', projectConfig.root, 'docker', 'create-db-user');
+    const migrationDockerImageTag = `${getNpmScope(tree)}-${kebabCase(options.name)}-migration:latest`;
+    const createDbUserDockerImageTag = `${getNpmScope(tree)}-${kebabCase(options.name)}-create-db-user:latest`;
+    if (options.infra !== 'none') {
+        const { bundleTargetName, bundleOutputDir } = addPythonBundleTarget(projectConfig, {
+            pythonPlatform: 'aarch64-manylinux_2_28'
+        });
+        projectConfig.targets['bundle-migration'] = {
+            cache: true,
+            outputs: [
+                '{workspaceRoot}/dist/{projectRoot}/docker/migration'
+            ],
+            executor: 'nx:run-commands',
+            options: {
+                commands: [
+                    fs.rm(migrationBundleDir),
+                    fs.mkdir(migrationBundleDir),
+                    fs.cp(bundleOutputDir, migrationBundleDir),
+                    fs.cp(joinPathFragments(dir, 'migrations'), joinPathFragments(migrationBundleDir, 'migrations')),
+                    fs.cp(joinPathFragments(dir, 'alembic.ini'), joinPathFragments(migrationBundleDir, 'alembic.ini')),
+                    fs.cp(joinPathFragments(dir, 'Dockerfile.migration'), joinPathFragments(migrationBundleDir, 'Dockerfile'))
+                ],
+                parallel: false
+            },
+            dependsOn: [
+                bundleTargetName
+            ]
+        };
+        projectConfig.targets['bundle-create-db-user'] = {
+            cache: true,
+            outputs: [
+                '{workspaceRoot}/dist/{projectRoot}/docker/create-db-user'
+            ],
+            executor: 'nx:run-commands',
+            options: {
+                commands: [
+                    fs.rm(createDbUserBundleDir),
+                    fs.mkdir(createDbUserBundleDir),
+                    fs.cp(bundleOutputDir, createDbUserBundleDir),
+                    fs.cp(joinPathFragments(dir, 'Dockerfile.create-db-user'), joinPathFragments(createDbUserBundleDir, 'Dockerfile'))
+                ],
+                parallel: false
+            },
+            dependsOn: [
+                bundleTargetName
+            ]
+        };
+        addDependencyToTargetIfNotPresent(projectConfig, 'bundle', 'bundle-migration');
+        addDependencyToTargetIfNotPresent(projectConfig, 'bundle', 'bundle-create-db-user');
+    }
+    projectConfig.targets['pull-image'] = {
+        executor: 'nx:run-commands',
+        options: {
+            command: `tsx ${scriptsDir}/pull-image.ts`,
+            cwd: '{projectRoot}'
+        }
+    };
+    projectConfig.targets['dev'] = {
+        executor: 'nx:run-commands',
+        continuous: true,
+        options: {
+            command: `tsx ${scriptsDir}/start-container.ts`,
+            cwd: '{projectRoot}'
+        },
+        dependsOn: [
+            'pull-image'
+        ]
+    };
+    projectConfig.targets['wait-for-db'] = {
+        executor: 'nx:run-commands',
+        dependsOn: [
+            'dev'
+        ],
+        options: {
+            command: `tsx ${scriptsDir}/wait-for-${engine}-db.ts`,
+            cwd: '{projectRoot}'
+        }
+    };
+    projectConfig.targets.migrate = {
+        executor: 'nx:run-commands',
+        dependsOn: [
+            'dev',
+            'wait-for-db'
+        ],
+        options: {
+            command: 'uv run alembic upgrade head',
+            cwd: '{projectRoot}',
+            env: {
+                LOCAL_DEV: 'true'
+            }
+        }
+    };
+    projectConfig.targets.alembic = {
+        executor: 'nx:run-commands',
+        dependsOn: [
+            'dev',
+            'wait-for-db'
+        ],
+        options: {
+            command: 'uv run alembic',
+            cwd: '{projectRoot}',
+            env: {
+                LOCAL_DEV: 'true'
+            }
+        }
+    };
+    if (options.infra !== 'none') {
+        if (iac === 'terraform') {
+            projectConfig.targets.docker = {
+                cache: true,
+                executor: 'nx:run-commands',
+                options: {
+                    commands: [
+                        `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`,
+                        `${containerEngine} build --platform linux/arm64 --provenance=false -t ${createDbUserDockerImageTag} ${createDbUserBundleDir}`
+                    ],
+                    parallel: false
+                },
+                dependsOn: [
+                    'bundle-migration',
+                    'bundle-create-db-user'
+                ]
+            };
+            addDependencyToTargetIfNotPresent(projectConfig, 'build', 'docker');
+            addDockerScanTarget(tree, {
+                project: projectConfig,
+                containerEngine,
+                trivyTargetName: 'trivy',
+                dockerTargetName: 'docker',
+                imageTags: [
+                    migrationDockerImageTag,
+                    createDbUserDockerImageTag
+                ]
+            }, DEPENDENCIES);
+        }
+        addDependencyToTargetIfNotPresent(projectConfig, 'build', 'bundle-migration');
+        addDependencyToTargetIfNotPresent(projectConfig, 'build', 'bundle-create-db-user');
+    }
+    updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);
+    addGeneratorMetadata(tree, fullyQualifiedName, PY_RDB_GENERATOR_INFO, metadata);
+    if (options.infra !== 'none') {
+        await sharedConstructsGenerator(tree, {
+            iac
+        }, DEPENDENCIES);
+        await addRdbInfra(tree, {
+            iac,
+            projectName: fullyQualifiedName,
+            projectRoot: dir,
+            nameClassName,
+            nameKebabCase: kebabCase(options.name),
+            databasePackageAlias: normalizedModuleName,
+            databaseName,
+            adminUser: databaseUser,
+            engine,
+            migrationBundleDir,
+            createDbUserBundleDir,
+            framework: 'sqlmodel',
+            createDbUserDockerImageTag,
+            migrationDockerImageTag,
+            containerEngine
+        });
+    }
+    addPyDependencies(tree, DEPENDENCIES, {
+        metadata,
+        projectRoot: dir
+    });
+    addTsDependencies(tree, DEPENDENCIES, {
+        metadata
+    });
+    await addGeneratorMetricsIfApplicable(tree, [
+        PY_RDB_GENERATOR_INFO
+    ]);
+    await formatFilesInSubtree(tree);
+    return ()=>installDependencies(tree, options.preferInstallDependencies, {
+            languages: [
+                'typescript',
+                'python'
+            ]
+        });
+};
+export default pyRdbGenerator;
+
+//# sourceMappingURL=generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/py/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  type GeneratorCallback,\n  generateFiles,\n  joinPathFragments,\n  readProjectConfiguration,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport {\n  addPyDependencies,\n  addTsDependencies,\n} from '../../utils/add-dependencies';\nimport { addPythonBundleTarget } from '../../utils/bundle/bundle';\nimport { resolveContainers } from '../../utils/containers';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../utils/declared-dependencies';\nimport { addDockerScanTarget, DOCKER_DEPENDENCIES } from '../../utils/docker';\nimport { formatFilesInSubtree } from '../../utils/format';\nimport { FS_DEPENDENCIES, FsCommands } from '../../utils/fs';\nimport { resolveIac } from '../../utils/iac';\nimport { installDependencies } from '../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../utils/metrics';\nimport { kebabCase, snakeCase, toClassName } from '../../utils/names';\nimport { getNpmScope } from '../../utils/npm-scope';\nimport {\n  addDependencyToTargetIfNotPresent,\n  addGeneratorMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  projectExists,\n} from '../../utils/nx';\nimport { assignPort } from '../../utils/port';\nimport { addRdbInfra } from '../../utils/rdb-constructs/rdb-constructs';\nimport {\n  SHARED_CONSTRUCTS_DEPENDENCIES,\n  sharedConstructsGenerator,\n} from '../../utils/shared-constructs';\nimport {\n  type IacMetadata,\n  PACKAGES_DIR,\n  SHARED_SCRIPTS_DIR,\n} from '../../utils/shared-constructs-constants';\nimport {\n  SHARED_RDB_SCRIPTS_DEPENDENCIES,\n  sharedRdbScriptsGenerator,\n} from '../../utils/shared-rdb-scripts';\nimport { PY_VERSIONS } from '../../utils/versions';\nimport pyProjectGenerator, { getPyProjectDetails } from '../project/generator';\nimport type { PyRdbGeneratorSchema } from './schema';\n\n/** The metadata this generator records, which its predicates read. */\nexport interface PyRdbMetadata extends IacMetadata {\n  readonly engine: PyRdbGeneratorSchema['engine'];\n}\n\n/** The Postgres engine, which MySQL takes none of. */\nconst isPostgres = (m: PyRdbMetadata) => m.engine !== 'mysql';\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<PyRdbMetadata>()({\n  ts: [\n    // The engine client the local dev script wait-for-*-db.ts imports.\n    { name: 'mariadb', when: (m) => m.engine === 'mysql', root: true },\n    { name: 'pg', when: isPostgres, root: true },\n    { name: '@types/pg', when: isPostgres, dev: true, root: true },\n    // Added by the helpers that own the projects they belong to.\n    ...ownedElsewhere(FS_DEPENDENCIES),\n    ...ownedElsewhere(DOCKER_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_RDB_SCRIPTS_DEPENDENCIES),\n  ],\n  py: [\n    { name: 'sqlmodel' },\n    { name: 'alembic' },\n    // SQLAlchemy's async engine (used by connection.py for both engines)\n    // requires greenlet at runtime. Vend it explicitly at a pinned version so\n    // the dependency graph is fully determined and doesn't float to an\n    // unpublished-wheel release at install time.\n    { name: 'greenlet' },\n    { name: 'aiomysql', when: (m) => m.engine === 'mysql' },\n    { name: 'asyncpg', when: (m) => m.engine !== 'mysql' },\n    { name: 'boto3' },\n    { name: 'aws-lambda-powertools' },\n  ],\n});\n\nexport const PY_RDB_GENERATOR_INFO: NxGeneratorInfo = getGeneratorInfo(\n  import.meta.filename,\n);\n\nexport const pyRdbGenerator = async (\n  tree: Tree,\n  options: PyRdbGeneratorSchema,\n): Promise<GeneratorCallback> => {\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\n  const { fullyQualifiedName, dir, normalizedModuleName } = getPyProjectDetails(\n    tree,\n    {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n    },\n  );\n\n  if (!projectExists(tree, fullyQualifiedName)) {\n    await pyProjectGenerator(tree, {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n      type: 'library',\n    });\n  }\n\n  const projectConfig = readProjectConfiguration(tree, fullyQualifiedName);\n\n  const { engine } = options;\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.\n  const iac =\n    options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;\n\n  // Recorded below and read by the declaration's predicates, so the packages\n  // added here are exactly the ones the version sync will own.\n  const metadata: PyRdbMetadata = { engine, ...(iac ? { iac } : {}) };\n  const localDbPort = assignPort(\n    tree,\n    projectConfig,\n    engine === 'mysql' ? 3306 : 5432,\n  );\n  const localDbHost = 'localhost';\n  const localDbUser = engine === 'mysql' ? 'root' : 'dbadmin';\n  const localDbPassword = 'password';\n  const containerName = `${getNpmScope(tree)}-${databaseName}`;\n  const dockerImage =\n    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    name: normalizedModuleName,\n    runtimeConfigKey: nameClassName,\n    engine,\n    localDbPort,\n    localDbHost,\n    localDbName: databaseName,\n    localDbUser,\n    localDbPassword,\n    containerEngine,\n    containerName,\n    dockerImage,\n    sqlmodelVersion: PY_VERSIONS.sqlmodel,\n    alembicVersion: PY_VERSIONS.alembic,\n    aiomysqlVersion: PY_VERSIONS.aiomysql,\n    asyncpgVersion: PY_VERSIONS.asyncpg,\n    boto3Version: PY_VERSIONS.boto3,\n    awsLambdaPowertoolsVersion: PY_VERSIONS['aws-lambda-powertools'],\n  };\n\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    dir,\n    templateOptions,\n  );\n\n  await sharedRdbScriptsGenerator(tree, engine, DEPENDENCIES);\n  const scriptsDir = relative(\n    dir,\n    joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'rdb'),\n  );\n  const fs = new FsCommands(tree, DEPENDENCIES);\n\n  const migrationBundleDir = joinPathFragments(\n    'dist',\n    projectConfig.root,\n    'docker',\n    'migration',\n  );\n  const createDbUserBundleDir = joinPathFragments(\n    'dist',\n    projectConfig.root,\n    'docker',\n    'create-db-user',\n  );\n  const migrationDockerImageTag = `${getNpmScope(tree)}-${kebabCase(options.name)}-migration:latest`;\n  const createDbUserDockerImageTag = `${getNpmScope(tree)}-${kebabCase(\n    options.name,\n  )}-create-db-user:latest`;\n\n  if (options.infra !== 'none') {\n    const { bundleTargetName, bundleOutputDir } = addPythonBundleTarget(\n      projectConfig,\n      { pythonPlatform: 'aarch64-manylinux_2_28' },\n    );\n\n    projectConfig.targets['bundle-migration'] = {\n      cache: true,\n      outputs: ['{workspaceRoot}/dist/{projectRoot}/docker/migration'],\n      executor: 'nx:run-commands',\n      options: {\n        commands: [\n          fs.rm(migrationBundleDir),\n          fs.mkdir(migrationBundleDir),\n          fs.cp(bundleOutputDir, migrationBundleDir),\n          fs.cp(\n            joinPathFragments(dir, 'migrations'),\n            joinPathFragments(migrationBundleDir, 'migrations'),\n          ),\n          fs.cp(\n            joinPathFragments(dir, 'alembic.ini'),\n            joinPathFragments(migrationBundleDir, 'alembic.ini'),\n          ),\n          fs.cp(\n            joinPathFragments(dir, 'Dockerfile.migration'),\n            joinPathFragments(migrationBundleDir, 'Dockerfile'),\n          ),\n        ],\n        parallel: false,\n      },\n      dependsOn: [bundleTargetName],\n    };\n    projectConfig.targets['bundle-create-db-user'] = {\n      cache: true,\n      outputs: ['{workspaceRoot}/dist/{projectRoot}/docker/create-db-user'],\n      executor: 'nx:run-commands',\n      options: {\n        commands: [\n          fs.rm(createDbUserBundleDir),\n          fs.mkdir(createDbUserBundleDir),\n          fs.cp(bundleOutputDir, createDbUserBundleDir),\n          fs.cp(\n            joinPathFragments(dir, 'Dockerfile.create-db-user'),\n            joinPathFragments(createDbUserBundleDir, 'Dockerfile'),\n          ),\n        ],\n        parallel: false,\n      },\n      dependsOn: [bundleTargetName],\n    };\n    addDependencyToTargetIfNotPresent(\n      projectConfig,\n      'bundle',\n      'bundle-migration',\n    );\n    addDependencyToTargetIfNotPresent(\n      projectConfig,\n      'bundle',\n      'bundle-create-db-user',\n    );\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    continuous: true,\n    options: {\n      command: `tsx ${scriptsDir}/start-container.ts`,\n      cwd: '{projectRoot}',\n    },\n    dependsOn: ['pull-image'],\n  };\n  projectConfig.targets['wait-for-db'] = {\n    executor: 'nx:run-commands',\n    dependsOn: ['dev'],\n    options: {\n      command: `tsx ${scriptsDir}/wait-for-${engine}-db.ts`,\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets.migrate = {\n    executor: 'nx:run-commands',\n    dependsOn: ['dev', 'wait-for-db'],\n    options: {\n      command: 'uv run alembic upgrade head',\n      cwd: '{projectRoot}',\n      env: {\n        LOCAL_DEV: 'true',\n      },\n    },\n  };\n  projectConfig.targets.alembic = {\n    executor: 'nx:run-commands',\n    dependsOn: ['dev', 'wait-for-db'],\n    options: {\n      command: 'uv run alembic',\n      cwd: '{projectRoot}',\n      env: {\n        LOCAL_DEV: 'true',\n      },\n    },\n  };\n\n  if (options.infra !== 'none') {\n    if (iac === 'terraform') {\n      projectConfig.targets.docker = {\n        cache: true,\n        executor: 'nx:run-commands',\n        options: {\n          commands: [\n            `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`,\n            `${containerEngine} build --platform linux/arm64 --provenance=false -t ${createDbUserDockerImageTag} ${createDbUserBundleDir}`,\n          ],\n          parallel: false,\n        },\n        dependsOn: ['bundle-migration', 'bundle-create-db-user'],\n      };\n      addDependencyToTargetIfNotPresent(projectConfig, 'build', 'docker');\n\n      addDockerScanTarget(\n        tree,\n        {\n          project: projectConfig,\n          containerEngine,\n          trivyTargetName: 'trivy',\n          dockerTargetName: 'docker',\n          imageTags: [migrationDockerImageTag, createDbUserDockerImageTag],\n        },\n        DEPENDENCIES,\n      );\n    }\n    addDependencyToTargetIfNotPresent(\n      projectConfig,\n      'build',\n      'bundle-migration',\n    );\n    addDependencyToTargetIfNotPresent(\n      projectConfig,\n      'build',\n      'bundle-create-db-user',\n    );\n  }\n\n  updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);\n  addGeneratorMetadata(\n    tree,\n    fullyQualifiedName,\n    PY_RDB_GENERATOR_INFO,\n    metadata,\n  );\n\n  if (options.infra !== 'none') {\n    await sharedConstructsGenerator(tree, { iac }, DEPENDENCIES);\n    await addRdbInfra(tree, {\n      iac,\n      projectName: fullyQualifiedName,\n      projectRoot: dir,\n      nameClassName,\n      nameKebabCase: kebabCase(options.name),\n      databasePackageAlias: normalizedModuleName,\n      databaseName,\n      adminUser: databaseUser,\n      engine,\n      migrationBundleDir,\n      createDbUserBundleDir,\n      framework: 'sqlmodel',\n      createDbUserDockerImageTag,\n      migrationDockerImageTag,\n      containerEngine,\n    });\n  }\n\n  addPyDependencies(tree, DEPENDENCIES, { metadata, projectRoot: dir });\n  addTsDependencies(tree, DEPENDENCIES, { metadata });\n\n  await addGeneratorMetricsIfApplicable(tree, [PY_RDB_GENERATOR_INFO]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyRdbGenerator;\n"],"names":["relative","generateFiles","joinPathFragments","readProjectConfiguration","updateProjectConfiguration","addPyDependencies","addTsDependencies","addPythonBundleTarget","resolveContainers","declareDependencies","ownedElsewhere","addDockerScanTarget","DOCKER_DEPENDENCIES","formatFilesInSubtree","FS_DEPENDENCIES","FsCommands","resolveIac","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","snakeCase","toClassName","getNpmScope","addDependencyToTargetIfNotPresent","addGeneratorMetadata","getGeneratorInfo","projectExists","assignPort","addRdbInfra","SHARED_CONSTRUCTS_DEPENDENCIES","sharedConstructsGenerator","PACKAGES_DIR","SHARED_SCRIPTS_DIR","SHARED_RDB_SCRIPTS_DEPENDENCIES","sharedRdbScriptsGenerator","PY_VERSIONS","pyProjectGenerator","getPyProjectDetails","isPostgres","m","engine","DEPENDENCIES","ts","name","when","root","dev","py","PY_RDB_GENERATOR_INFO","filename","pyRdbGenerator","tree","options","nameClassName","databaseUser","databaseName","containerEngine","fullyQualifiedName","dir","normalizedModuleName","directory","subDirectory","type","projectConfig","iac","infra","undefined","metadata","localDbPort","localDbHost","localDbUser","localDbPassword","containerName","dockerImage","templateOptions","runtimeConfigKey","localDbName","sqlmodelVersion","sqlmodel","alembicVersion","alembic","aiomysqlVersion","aiomysql","asyncpgVersion","asyncpg","boto3Version","boto3","awsLambdaPowertoolsVersion","dirname","scriptsDir","fs","migrationBundleDir","createDbUserBundleDir","migrationDockerImageTag","createDbUserDockerImageTag","bundleTargetName","bundleOutputDir","pythonPlatform","targets","cache","outputs","executor","commands","rm","mkdir","cp","parallel","dependsOn","command","cwd","continuous","migrate","env","LOCAL_DEV","docker","project","trivyTargetName","dockerTargetName","imageTags","projectName","projectRoot","nameKebabCase","databasePackageAlias","adminUser","framework","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAASA,QAAQ,QAAQ,YAAY;AACrC,SAEEC,aAAa,EACbC,iBAAiB,EACjBC,wBAAwB,EAExBC,0BAA0B,QACrB,aAAa;AACpB,SACEC,iBAAiB,EACjBC,iBAAiB,QACZ,kCAA+B;AACtC,SAASC,qBAAqB,QAAQ,+BAA4B;AAClE,SAASC,iBAAiB,QAAQ,4BAAyB;AAC3D,SACEC,mBAAmB,EACnBC,cAAc,QACT,uCAAoC;AAC3C,SAASC,mBAAmB,EAAEC,mBAAmB,QAAQ,wBAAqB;AAC9E,SAASC,oBAAoB,QAAQ,wBAAqB;AAC1D,SAASC,eAAe,EAAEC,UAAU,QAAQ,oBAAiB;AAC7D,SAASC,UAAU,QAAQ,qBAAkB;AAC7C,SAASC,mBAAmB,QAAQ,yBAAsB;AAC1D,SAASC,+BAA+B,QAAQ,yBAAsB;AACtE,SAASC,SAAS,EAAEC,SAAS,EAAEC,WAAW,QAAQ,uBAAoB;AACtE,SAASC,WAAW,QAAQ,2BAAwB;AACpD,SACEC,iCAAiC,EACjCC,oBAAoB,EACpBC,gBAAgB,EAEhBC,aAAa,QACR,oBAAiB;AACxB,SAASC,UAAU,QAAQ,sBAAmB;AAC9C,SAASC,WAAW,QAAQ,+CAA4C;AACxE,SACEC,8BAA8B,EAC9BC,yBAAyB,QACpB,mCAAgC;AACvC,SAEEC,YAAY,EACZC,kBAAkB,QACb,6CAA0C;AACjD,SACEC,+BAA+B,EAC/BC,yBAAyB,QACpB,oCAAiC;AACxC,SAASC,WAAW,QAAQ,0BAAuB;AACnD,OAAOC,sBAAsBC,mBAAmB,QAAQ,0BAAuB;AAQ/E,oDAAoD,GACpD,MAAMC,aAAa,CAACC,IAAqBA,EAAEC,MAAM,KAAK;AAEtD,4EAA4E;AAC5E,2CAA2C;AAC3C,OAAO,MAAMC,eAAehC,sBAAqC;IAC/DiC,IAAI;QACF,mEAAmE;QACnE;YAAEC,MAAM;YAAWC,MAAM,CAACL,IAAMA,EAAEC,MAAM,KAAK;YAASK,MAAM;QAAK;QACjE;YAAEF,MAAM;YAAMC,MAAMN;YAAYO,MAAM;QAAK;QAC3C;YAAEF,MAAM;YAAaC,MAAMN;YAAYQ,KAAK;YAAMD,MAAM;QAAK;QAC7D,6DAA6D;WAC1DnC,eAAeI;WACfJ,eAAeE;WACfF,eAAemB;WACfnB,eAAeuB;KACnB;IACDc,IAAI;QACF;YAAEJ,MAAM;QAAW;QACnB;YAAEA,MAAM;QAAU;QAClB,qEAAqE;QACrE,0EAA0E;QAC1E,mEAAmE;QACnE,6CAA6C;QAC7C;YAAEA,MAAM;QAAW;QACnB;YAAEA,MAAM;YAAYC,MAAM,CAACL,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QACtD;YAAEG,MAAM;YAAWC,MAAM,CAACL,IAAMA,EAAEC,MAAM,KAAK;QAAQ;QACrD;YAAEG,MAAM;QAAQ;QAChB;YAAEA,MAAM;QAAwB;KACjC;AACH,GAAG;AAEH,OAAO,MAAMK,wBAAyCvB,iBACpD,YAAYwB,QAAQ,EACpB;AAEF,OAAO,MAAMC,iBAAiB,OAC5BC,MACAC;IAEA,MAAMC,gBAAgBhC,YAAY+B,QAAQT,IAAI;IAC9C,MAAMW,eAAeF,QAAQE,YAAY,IAAI;IAC7C,MAAMC,eAAenC,UAAUgC,QAAQG,YAAY,IAAIH,QAAQT,IAAI;IACnE,MAAMa,kBAAkB,MAAMhD,kBAAkB2C,MAAM;IAEtD,MAAM,EAAEM,kBAAkB,EAAEC,GAAG,EAAEC,oBAAoB,EAAE,GAAGtB,oBACxDc,MACA;QACER,MAAMS,QAAQT,IAAI;QAClBiB,WAAWR,QAAQQ,SAAS;QAC5BC,cAAcT,QAAQS,YAAY;IACpC;IAGF,IAAI,CAACnC,cAAcyB,MAAMM,qBAAqB;QAC5C,MAAMrB,mBAAmBe,MAAM;YAC7BR,MAAMS,QAAQT,IAAI;YAClBiB,WAAWR,QAAQQ,SAAS;YAC5BC,cAAcT,QAAQS,YAAY;YAClCC,MAAM;QACR;IACF;IAEA,MAAMC,gBAAgB5D,yBAAyBgD,MAAMM;IAErD,MAAM,EAAEjB,MAAM,EAAE,GAAGY;IACnB,4EAA4E;IAC5E,wEAAwE;IACxE,MAAMY,MACJZ,QAAQa,KAAK,KAAK,SAAS,MAAMjD,WAAWmC,MAAMC,QAAQY,GAAG,IAAIE;IAEnE,2EAA2E;IAC3E,6DAA6D;IAC7D,MAAMC,WAA0B;QAAE3B;QAAQ,GAAIwB,MAAM;YAAEA;QAAI,IAAI,CAAC,CAAC;IAAE;IAClE,MAAMI,cAAczC,WAClBwB,MACAY,eACAvB,WAAW,UAAU,OAAO;IAE9B,MAAM6B,cAAc;IACpB,MAAMC,cAAc9B,WAAW,UAAU,SAAS;IAClD,MAAM+B,kBAAkB;IACxB,MAAMC,gBAAgB,GAAGlD,YAAY6B,MAAM,CAAC,EAAEI,cAAc;IAC5D,MAAMkB,cACJjC,WAAW,UACP,+CACA;IAEN,MAAMkC,kBAAkB;QACtB/B,MAAMgB;QACNgB,kBAAkBtB;QAClBb;QACA4B;QACAC;QACAO,aAAarB;QACbe;QACAC;QACAf;QACAgB;QACAC;QACAI,iBAAiB1C,YAAY2C,QAAQ;QACrCC,gBAAgB5C,YAAY6C,OAAO;QACnCC,iBAAiB9C,YAAY+C,QAAQ;QACrCC,gBAAgBhD,YAAYiD,OAAO;QACnCC,cAAclD,YAAYmD,KAAK;QAC/BC,4BAA4BpD,WAAW,CAAC,wBAAwB;IAClE;IAEAlC,cACEkD,MACAjD,kBAAkB,YAAYsF,OAAO,EAAE,UACvC9B,KACAgB;IAGF,MAAMxC,0BAA0BiB,MAAMX,QAAQC;IAC9C,MAAMgD,aAAazF,SACjB0D,KACAxD,kBAAkB6B,cAAcC,oBAAoB,OAAO;IAE7D,MAAM0D,KAAK,IAAI3E,WAAWoC,MAAMV;IAEhC,MAAMkD,qBAAqBzF,kBACzB,QACA6D,cAAclB,IAAI,EAClB,UACA;IAEF,MAAM+C,wBAAwB1F,kBAC5B,QACA6D,cAAclB,IAAI,EAClB,UACA;IAEF,MAAMgD,0BAA0B,GAAGvE,YAAY6B,MAAM,CAAC,EAAEhC,UAAUiC,QAAQT,IAAI,EAAE,iBAAiB,CAAC;IAClG,MAAMmD,6BAA6B,GAAGxE,YAAY6B,MAAM,CAAC,EAAEhC,UACzDiC,QAAQT,IAAI,EACZ,sBAAsB,CAAC;IAEzB,IAAIS,QAAQa,KAAK,KAAK,QAAQ;QAC5B,MAAM,EAAE8B,gBAAgB,EAAEC,eAAe,EAAE,GAAGzF,sBAC5CwD,eACA;YAAEkC,gBAAgB;QAAyB;QAG7ClC,cAAcmC,OAAO,CAAC,mBAAmB,GAAG;YAC1CC,OAAO;YACPC,SAAS;gBAAC;aAAsD;YAChEC,UAAU;YACVjD,SAAS;gBACPkD,UAAU;oBACRZ,GAAGa,EAAE,CAACZ;oBACND,GAAGc,KAAK,CAACb;oBACTD,GAAGe,EAAE,CAACT,iBAAiBL;oBACvBD,GAAGe,EAAE,CACHvG,kBAAkBwD,KAAK,eACvBxD,kBAAkByF,oBAAoB;oBAExCD,GAAGe,EAAE,CACHvG,kBAAkBwD,KAAK,gBACvBxD,kBAAkByF,oBAAoB;oBAExCD,GAAGe,EAAE,CACHvG,kBAAkBwD,KAAK,yBACvBxD,kBAAkByF,oBAAoB;iBAEzC;gBACDe,UAAU;YACZ;YACAC,WAAW;gBAACZ;aAAiB;QAC/B;QACAhC,cAAcmC,OAAO,CAAC,wBAAwB,GAAG;YAC/CC,OAAO;YACPC,SAAS;gBAAC;aAA2D;YACrEC,UAAU;YACVjD,SAAS;gBACPkD,UAAU;oBACRZ,GAAGa,EAAE,CAACX;oBACNF,GAAGc,KAAK,CAACZ;oBACTF,GAAGe,EAAE,CAACT,iBAAiBJ;oBACvBF,GAAGe,EAAE,CACHvG,kBAAkBwD,KAAK,8BACvBxD,kBAAkB0F,uBAAuB;iBAE5C;gBACDc,UAAU;YACZ;YACAC,WAAW;gBAACZ;aAAiB;QAC/B;QACAxE,kCACEwC,eACA,UACA;QAEFxC,kCACEwC,eACA,UACA;IAEJ;IAEAA,cAAcmC,OAAO,CAAC,aAAa,GAAG;QACpCG,UAAU;QACVjD,SAAS;YACPwD,SAAS,CAAC,IAAI,EAAEnB,WAAW,cAAc,CAAC;YAC1CoB,KAAK;QACP;IACF;IACA9C,cAAcmC,OAAO,CAAC,MAAM,GAAG;QAC7BG,UAAU;QACVS,YAAY;QACZ1D,SAAS;YACPwD,SAAS,CAAC,IAAI,EAAEnB,WAAW,mBAAmB,CAAC;YAC/CoB,KAAK;QACP;QACAF,WAAW;YAAC;SAAa;IAC3B;IACA5C,cAAcmC,OAAO,CAAC,cAAc,GAAG;QACrCG,UAAU;QACVM,WAAW;YAAC;SAAM;QAClBvD,SAAS;YACPwD,SAAS,CAAC,IAAI,EAAEnB,WAAW,UAAU,EAAEjD,OAAO,MAAM,CAAC;YACrDqE,KAAK;QACP;IACF;IACA9C,cAAcmC,OAAO,CAACa,OAAO,GAAG;QAC9BV,UAAU;QACVM,WAAW;YAAC;YAAO;SAAc;QACjCvD,SAAS;YACPwD,SAAS;YACTC,KAAK;YACLG,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IACAlD,cAAcmC,OAAO,CAAClB,OAAO,GAAG;QAC9BqB,UAAU;QACVM,WAAW;YAAC;YAAO;SAAc;QACjCvD,SAAS;YACPwD,SAAS;YACTC,KAAK;YACLG,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IAEA,IAAI7D,QAAQa,KAAK,KAAK,QAAQ;QAC5B,IAAID,QAAQ,aAAa;YACvBD,cAAcmC,OAAO,CAACgB,MAAM,GAAG;gBAC7Bf,OAAO;gBACPE,UAAU;gBACVjD,SAAS;oBACPkD,UAAU;wBACR,GAAG9C,gBAAgB,oDAAoD,EAAEqC,wBAAwB,CAAC,EAAEF,oBAAoB;wBACxH,GAAGnC,gBAAgB,oDAAoD,EAAEsC,2BAA2B,CAAC,EAAEF,uBAAuB;qBAC/H;oBACDc,UAAU;gBACZ;gBACAC,WAAW;oBAAC;oBAAoB;iBAAwB;YAC1D;YACApF,kCAAkCwC,eAAe,SAAS;YAE1DpD,oBACEwC,MACA;gBACEgE,SAASpD;gBACTP;gBACA4D,iBAAiB;gBACjBC,kBAAkB;gBAClBC,WAAW;oBAACzB;oBAAyBC;iBAA2B;YAClE,GACArD;QAEJ;QACAlB,kCACEwC,eACA,SACA;QAEFxC,kCACEwC,eACA,SACA;IAEJ;IAEA3D,2BAA2B+C,MAAMM,oBAAoBM;IACrDvC,qBACE2B,MACAM,oBACAT,uBACAmB;IAGF,IAAIf,QAAQa,KAAK,KAAK,QAAQ;QAC5B,MAAMnC,0BAA0BqB,MAAM;YAAEa;QAAI,GAAGvB;QAC/C,MAAMb,YAAYuB,MAAM;YACtBa;YACAuD,aAAa9D;YACb+D,aAAa9D;YACbL;YACAoE,eAAetG,UAAUiC,QAAQT,IAAI;YACrC+E,sBAAsB/D;YACtBJ;YACAoE,WAAWrE;YACXd;YACAmD;YACAC;YACAgC,WAAW;YACX9B;YACAD;YACArC;QACF;IACF;IAEAnD,kBAAkB8C,MAAMV,cAAc;QAAE0B;QAAUqD,aAAa9D;IAAI;IACnEpD,kBAAkB6C,MAAMV,cAAc;QAAE0B;IAAS;IAEjD,MAAMjD,gCAAgCiC,MAAM;QAACH;KAAsB;IAEnE,MAAMnC,qBAAqBsC;IAC3B,OAAO,IACLlC,oBAAoBkC,MAAMC,QAAQyE,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAe5E,eAAe"}

package/src/py/rdb/mcp-server-connection/__snapshots__/generator.spec.ts.snap

@@ -0,0 +1,119 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`py#rdb mcp-server-connection generator > should add RDS CA bundle to Dockerfile 1`] = `
+"FROM public.ecr.aws/docker/library/python:3.14-slim
+
+WORKDIR /app
+
+COPY . /app
+
+ADD https://truststore.pki.rds.amazonaws.com/global/global-bundle.pem /usr/local/share/ca-certificates/rds-global-bundle.crt
+RUN update-ca-certificates
+
+EXPOSE 8000
+
+ENV PYTHONPATH=/app
+
+CMD ["python", "-m", "uvicorn", "proj_my_mcp_server.mcp_server.http:app", "--host", "0.0.0.0", "--port", "8000"]
+"
+`;
+
+exports[`py#rdb mcp-server-connection generator > should add RDS CA bundle to sourceComponent Dockerfile 1`] = `
+"FROM public.ecr.aws/docker/library/python:3.14-slim
+
+WORKDIR /app
+
+COPY . /app
+
+ADD https://truststore.pki.rds.amazonaws.com/global/global-bundle.pem /usr/local/share/ca-certificates/rds-global-bundle.crt
+RUN update-ca-certificates
+
+EXPOSE 8000
+
+ENV PYTHONPATH=/app
+
+CMD ["python", "-m", "uvicorn", "proj_my_mcp_server.custom_mcp_server.http:app", "--host", "0.0.0.0", "--port", "8000"]
+"
+`;
+
+exports[`py#rdb mcp-server-connection generator > should add rdb dev dependency to mcp-server dev 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "py#rdb#mcp-server-connection",
+        "name": "mcp-server-db",
+        "path": "packages/db",
+      },
+    ],
+  },
+  "name": "my-mcp-server",
+  "root": "packages/my-mcp-server",
+  "sourceRoot": "packages/my-mcp-server/src/proj_my_mcp_server",
+  "targets": {
+    "mcp-server-dev": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "dev",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`py#rdb mcp-server-connection generator > should be idempotent for Dockerfile 1`] = `
+"FROM public.ecr.aws/docker/library/python:3.14-slim
+
+WORKDIR /app
+
+COPY . /app
+
+ADD https://truststore.pki.rds.amazonaws.com/global/global-bundle.pem /usr/local/share/ca-certificates/rds-global-bundle.crt
+RUN update-ca-certificates
+
+EXPOSE 8000
+
+ENV PYTHONPATH=/app
+
+CMD ["python", "-m", "uvicorn", "proj_my_mcp_server.mcp_server.http:app", "--host", "0.0.0.0", "--port", "8000"]
+"
+`;
+
+exports[`py#rdb mcp-server-connection generator > should use sourceComponent name when provided 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "py#rdb#mcp-server-connection",
+        "name": "custom-mcp-server-db",
+        "path": "packages/db",
+      },
+    ],
+  },
+  "name": "my-mcp-server",
+  "root": "packages/my-mcp-server",
+  "sourceRoot": "packages/my-mcp-server/src/proj_my_mcp_server",
+  "targets": {
+    "custom-mcp-server-dev": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "dev",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;

package/src/py/rdb/mcp-server-connection/generator.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type GeneratorCallback, type Tree } from '@nx/devkit';
+import { type NxGeneratorInfo } from '../../../utils/nx';
+import type { PyRdbMcpServerConnectionGeneratorSchema } from './schema';
+export declare const PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
+export declare const pyRdbMcpServerConnectionGenerator: (tree: Tree, options: PyRdbMcpServerConnectionGeneratorSchema) => Promise<GeneratorCallback>;
+export default pyRdbMcpServerConnectionGenerator;

package/src/py/rdb/mcp-server-connection/generator.js

@@ -0,0 +1,52 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ import { joinPathFragments, updateProjectConfiguration } from "@nx/devkit";
+import { formatFilesInSubtree } from "../../../utils/format.js";
+import { installDependencies } from "../../../utils/install.js";
+import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
+import { toSnakeCase } from "../../../utils/names.js";
+import { addComponentGeneratorMetadata, addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { addWorkspaceDependencyToPyProject } from "../../../utils/py.js";
+import { injectRdsCaBundleIntoDockerfile } from "../utils.js";
+export const PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
+export const pyRdbMcpServerConnectionGenerator = async (tree, options)=>{
+    const sourceProject = readProjectConfigurationUnqualified(tree, options.sourceProject);
+    const targetProject = readProjectConfigurationUnqualified(tree, options.targetProject);
+    addWorkspaceDependencyToPyProject(tree, sourceProject, targetProject);
+    const mcpServerName = options.sourceComponent?.name ?? 'mcp-server';
+    const devTarget = `${mcpServerName}-dev`;
+    const mcpServerSourceDir = options.sourceComponent?.path ? joinPathFragments(sourceProject.root, options.sourceComponent.path) : sourceProject.sourceRoot ? joinPathFragments(sourceProject.sourceRoot, toSnakeCase(mcpServerName)) : undefined;
+    const dockerfilePath = mcpServerSourceDir ? joinPathFragments(mcpServerSourceDir, 'Dockerfile') : undefined;
+    if (dockerfilePath) {
+        injectRdsCaBundleIntoDockerfile(tree, dockerfilePath);
+    }
+    if (sourceProject.targets?.[devTarget]) {
+        addDependencyToTargetIfNotPresent(sourceProject, devTarget, {
+            projects: [
+                targetProject.name
+            ],
+            target: 'dev'
+        });
+        updateProjectConfiguration(tree, sourceProject.name, sourceProject);
+    }
+    // Recorded so the version sync can identify this connection.
+    addComponentGeneratorMetadata(tree, sourceProject.name, PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO, targetProject.root, `${mcpServerName}-${targetProject.name}`, // The source component this connection is made from, so the pair is
+    // identifiable rather than just the two projects.
+    {
+        sourcePath: options.sourceComponent?.path
+    });
+    await addGeneratorMetricsIfApplicable(tree, [
+        PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO
+    ]);
+    await formatFilesInSubtree(tree);
+    return ()=>installDependencies(tree, options.preferInstallDependencies, {
+            languages: [
+                'typescript',
+                'python'
+            ]
+        });
+};
+export default pyRdbMcpServerConnectionGenerator;
+
+//# sourceMappingURL=generator.js.map

package/src/py/rdb/mcp-server-connection/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/rdb/mcp-server-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  type GeneratorCallback,\n  joinPathFragments,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { toSnakeCase } from '../../../utils/names';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { addWorkspaceDependencyToPyProject } from '../../../utils/py';\nimport { injectRdsCaBundleIntoDockerfile } from '../utils';\nimport type { PyRdbMcpServerConnectionGeneratorSchema } from './schema';\n\nexport const PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyRdbMcpServerConnectionGenerator = async (\n  tree: Tree,\n  options: PyRdbMcpServerConnectionGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const sourceProject = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const targetProject = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  addWorkspaceDependencyToPyProject(tree, sourceProject, targetProject);\n\n  const mcpServerName = options.sourceComponent?.name ?? 'mcp-server';\n  const devTarget = `${mcpServerName}-dev`;\n  const mcpServerSourceDir = options.sourceComponent?.path\n    ? joinPathFragments(sourceProject.root, options.sourceComponent.path)\n    : sourceProject.sourceRoot\n      ? joinPathFragments(sourceProject.sourceRoot, toSnakeCase(mcpServerName))\n      : undefined;\n  const dockerfilePath = mcpServerSourceDir\n    ? joinPathFragments(mcpServerSourceDir, 'Dockerfile')\n    : undefined;\n\n  if (dockerfilePath) {\n    injectRdsCaBundleIntoDockerfile(tree, dockerfilePath);\n  }\n\n  if (sourceProject.targets?.[devTarget]) {\n    addDependencyToTargetIfNotPresent(sourceProject, devTarget, {\n      projects: [targetProject.name],\n      target: 'dev',\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  // Recorded so the version sync can identify this connection.\n  addComponentGeneratorMetadata(\n    tree,\n    sourceProject.name,\n    PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO,\n    targetProject.root,\n    `${mcpServerName}-${targetProject.name}`,\n    // The source component this connection is made from, so the pair is\n    // identifiable rather than just the two projects.\n    { sourcePath: options.sourceComponent?.path },\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO,\n  ]);\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyRdbMcpServerConnectionGenerator;\n"],"names":["joinPathFragments","updateProjectConfiguration","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","toSnakeCase","addComponentGeneratorMetadata","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","addWorkspaceDependencyToPyProject","injectRdsCaBundleIntoDockerfile","PY_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO","filename","pyRdbMcpServerConnectionGenerator","tree","options","sourceProject","targetProject","mcpServerName","sourceComponent","name","devTarget","mcpServerSourceDir","path","root","sourceRoot","undefined","dockerfilePath","targets","projects","target","sourcePath","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAEEA,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,WAAW,QAAQ,0BAAuB;AACnD,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,iCAAiC,QAAQ,uBAAoB;AACtE,SAASC,+BAA+B,QAAQ,cAAW;AAG3D,OAAO,MAAMC,8CACXJ,iBAAiB,YAAYK,QAAQ,EAAE;AAEzC,OAAO,MAAMC,oCAAoC,OAC/CC,MACAC;IAEA,MAAMC,gBAAgBR,oCACpBM,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBT,oCACpBM,MACAC,QAAQE,aAAa;IAGvBR,kCAAkCK,MAAME,eAAeC;IAEvD,MAAMC,gBAAgBH,QAAQI,eAAe,EAAEC,QAAQ;IACvD,MAAMC,YAAY,GAAGH,cAAc,IAAI,CAAC;IACxC,MAAMI,qBAAqBP,QAAQI,eAAe,EAAEI,OAChDxB,kBAAkBiB,cAAcQ,IAAI,EAAET,QAAQI,eAAe,CAACI,IAAI,IAClEP,cAAcS,UAAU,GACtB1B,kBAAkBiB,cAAcS,UAAU,EAAErB,YAAYc,kBACxDQ;IACN,MAAMC,iBAAiBL,qBACnBvB,kBAAkBuB,oBAAoB,gBACtCI;IAEJ,IAAIC,gBAAgB;QAClBjB,gCAAgCI,MAAMa;IACxC;IAEA,IAAIX,cAAcY,OAAO,EAAE,CAACP,UAAU,EAAE;QACtCf,kCAAkCU,eAAeK,WAAW;YAC1DQ,UAAU;gBAACZ,cAAcG,IAAI;aAAC;YAC9BU,QAAQ;QACV;QACA9B,2BAA2Bc,MAAME,cAAcI,IAAI,EAAEJ;IACvD;IAEA,6DAA6D;IAC7DX,8BACES,MACAE,cAAcI,IAAI,EAClBT,6CACAM,cAAcO,IAAI,EAClB,GAAGN,cAAc,CAAC,EAAED,cAAcG,IAAI,EAAE,EACxC,oEAAoE;IACpE,kDAAkD;IAClD;QAAEW,YAAYhB,QAAQI,eAAe,EAAEI;IAAK;IAG9C,MAAMpB,gCAAgCW,MAAM;QAC1CH;KACD;IACD,MAAMV,qBAAqBa;IAC3B,OAAO,IACLZ,oBAAoBY,MAAMC,QAAQiB,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAepB,kCAAkC"}

package/src/py/rdb/mcp-server-connection/schema.d.js

@@ -0,0 +1,8 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ /**
+ * TypeScript types for options defined in schema.json
+ */ export { };
+
+//# sourceMappingURL=schema.d.js.map

package/src/py/rdb/mcp-server-connection/schema.d.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/rdb/mcp-server-connection/schema.d.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport { ComponentMetadata } from '../../../utils/nx';\n\n/**\n * TypeScript types for options defined in schema.json\n */\nexport interface PyRdbMcpServerConnectionGeneratorSchema {\n  sourceProject: string;\n  targetProject: string;\n  sourceComponent?: ComponentMetadata;\n  preferInstallDependencies?: boolean;\n}\n"],"names":[],"mappings":"AAAA;;;CAGC,GAGD;;CAEC,GACD,WAKC"}

package/src/py/rdb/mcp-server-connection/schema.d.ts

@@ -0,0 +1,15 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ComponentMetadata } from '../../../utils/nx';
+
+/**
+ * TypeScript types for options defined in schema.json
+ */
+export interface PyRdbMcpServerConnectionGeneratorSchema {
+  sourceProject: string;
+  targetProject: string;
+  sourceComponent?: ComponentMetadata;
+  preferInstallDependencies?: boolean;
+}

package/src/py/rdb/mcp-server-connection/schema.json

@@ -0,0 +1,27 @@
+{
+  "$schema": "https://json-schema.org/schema",
+  "$id": "py#rdb#mcp-server-connection",
+  "title": "py#rdb#mcp-server-connection",
+  "description": "Connect a py#mcp-server to a py#rdb project",
+  "type": "object",
+  "properties": {
+    "sourceProject": {
+      "type": "string",
+      "description": "The project containing the py#mcp-server to connect from"
+    },
+    "targetProject": {
+      "type": "string",
+      "description": "The py#rdb project to connect to"
+    },
+    "sourceComponent": {
+      "type": "string",
+      "description": "The MCP server component name"
+    },
+    "preferInstallDependencies": {
+      "type": "boolean",
+      "description": "Whether to prefer installing dependencies after the generator runs. Set to false to defer installing when batching multiple generators (an install still runs if needed so subsequent generators can compute the Nx project graph); install once at the end.",
+      "default": true
+    }
+  },
+  "required": ["sourceProject", "targetProject"]
+}

package/src/py/rdb/schema.d.js

@@ -0,0 +1,6 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ export { };
+
+//# sourceMappingURL=schema.d.js.map

package/src/py/rdb/schema.d.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/py/rdb/schema.d.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nexport interface PyRdbGeneratorSchema {\n  name: string;\n  directory?: string;\n  subDirectory?: string;\n  infra: 'aurora' | 'none';\n  engine: 'postgres' | 'mysql';\n  databaseUser?: string;\n  databaseName?: string;\n  framework: 'sqlmodel';\n  iac?: 'inherit' | 'cdk' | 'terraform';\n  preferInstallDependencies?: boolean;\n}\n"],"names":[],"mappings":"AAAA;;;CAGC,GACD,WAWC"}