@aws/nx-plugin 1.0.0-rc.9 → 1.0.0-rc.90

package/src/py/rdb/generator.js

@@ -0,0 +1,368 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ import { relative } from "node:path";
+import { generateFiles, joinPathFragments, OverwriteStrategy, readProjectConfiguration, updateProjectConfiguration, writeJson } from "@nx/devkit";
+import { addPyDependencies, addTsDependencies } from "../../utils/add-dependencies.js";
+import { addPythonReExport } from "../../utils/agent-connection/agent-connection.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, IMAGE_BUILD_CACHE } 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 { addArtifactDependencyToTargets, 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']
+    };
+    // Models, the connection module, the Alembic config and the migration env are
+    // user-editable, so a re-run (to add infrastructure, or after a failed run)
+    // must not clobber the user's schema and edits.
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files'), dir, templateOptions, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
+    // The package barrel re-exports connection.py. `pyProjectGenerator` writes a
+    // stub __init__.py when it creates the project, so the re-exports are added to
+    // whatever is there rather than replacing it — which also leaves a user's own
+    // additions to the barrel alone on a re-run.
+    const initFilePath = joinPathFragments(dir, normalizedModuleName, '__init__.py');
+    for (const importName of [
+        'get_engine',
+        'is_local_dev',
+        'session_context'
+    ]){
+        await addPythonReExport(tree, initFilePath, '.connection', importName);
+    }
+    // config.json is framework-owned: it carries values derived from the
+    // generator's options, so it converges on a re-run with changed options.
+    writeJson(tree, joinPathFragments(dir, 'config.json'), {
+        runtimeConfigKey: nameClassName,
+        localDev: {
+            containerEngine,
+            containerName,
+            image: dockerImage,
+            port: localDbPort,
+            host: localDbHost,
+            dbName: databaseName,
+            dbUser: localDbUser,
+            dbPassword: localDbPassword,
+            dbEngine: engine
+        }
+    });
+    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,
+            inputs: [
+                'default'
+            ],
+            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,
+            inputs: [
+                'default'
+            ],
+            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: IMAGE_BUILD_CACHE,
+                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'
+                ]
+            };
+            addArtifactDependencyToTargets(projectConfig, '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  OverwriteStrategy,\n  readProjectConfiguration,\n  type Tree,\n  updateProjectConfiguration,\n  writeJson,\n} from '@nx/devkit';\nimport {\n  addPyDependencies,\n  addTsDependencies,\n} from '../../utils/add-dependencies.js';\nimport { addPythonReExport } from '../../utils/agent-connection/agent-connection.js';\nimport { addPythonBundleTarget } 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} from '../../utils/docker.js';\nimport { formatFilesInSubtree } from '../../utils/format.js';\nimport { FS_DEPENDENCIES, FsCommands } from '../../utils/fs.js';\nimport { resolveIac } from '../../utils/iac.js';\nimport { installDependencies } from '../../utils/install.js';\nimport { addGeneratorMetricsIfApplicable } from '../../utils/metrics.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  projectExists,\n} from '../../utils/nx.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 { PY_VERSIONS } from '../../utils/versions.js';\nimport pyProjectGenerator, {\n  getPyProjectDetails,\n} from '../project/generator.js';\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  // Models, the connection module, the Alembic config and the migration env are\n  // user-editable, so a re-run (to add infrastructure, or after a failed run)\n  // 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 package barrel re-exports connection.py. `pyProjectGenerator` writes a\n  // stub __init__.py when it creates the project, so the re-exports are added to\n  // whatever is there rather than replacing it — which also leaves a user's own\n  // additions to the barrel alone on a re-run.\n  const initFilePath = joinPathFragments(\n    dir,\n    normalizedModuleName,\n    '__init__.py',\n  );\n  for (const importName of ['get_engine', 'is_local_dev', 'session_context']) {\n    await addPythonReExport(tree, initFilePath, '.connection', importName);\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: engine,\n    },\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      inputs: ['default'],\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      inputs: ['default'],\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: IMAGE_BUILD_CACHE,\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      addArtifactDependencyToTargets(projectConfig, '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","OverwriteStrategy","readProjectConfiguration","updateProjectConfiguration","writeJson","addPyDependencies","addTsDependencies","addPythonReExport","addPythonBundleTarget","resolveContainers","declareDependencies","ownedElsewhere","addDockerScanTarget","DOCKER_DEPENDENCIES","IMAGE_BUILD_CACHE","formatFilesInSubtree","FS_DEPENDENCIES","FsCommands","resolveIac","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","snakeCase","toClassName","getNpmScope","addArtifactDependencyToTargets","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","overwriteStrategy","KeepExisting","initFilePath","importName","localDev","image","port","host","dbName","dbUser","dbPassword","dbEngine","scriptsDir","fs","migrationBundleDir","createDbUserBundleDir","migrationDockerImageTag","createDbUserDockerImageTag","bundleTargetName","bundleOutputDir","pythonPlatform","targets","cache","inputs","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,iBAAiB,EACjBC,wBAAwB,EAExBC,0BAA0B,EAC1BC,SAAS,QACJ,aAAa;AACpB,SACEC,iBAAiB,EACjBC,iBAAiB,QACZ,kCAAkC;AACzC,SAASC,iBAAiB,QAAQ,mDAAmD;AACrF,SAASC,qBAAqB,QAAQ,+BAA+B;AACrE,SAASC,iBAAiB,QAAQ,4BAA4B;AAC9D,SACEC,mBAAmB,EACnBC,cAAc,QACT,uCAAuC;AAC9C,SACEC,mBAAmB,EACnBC,mBAAmB,EACnBC,iBAAiB,QACZ,wBAAwB;AAC/B,SAASC,oBAAoB,QAAQ,wBAAwB;AAC7D,SAASC,eAAe,EAAEC,UAAU,QAAQ,oBAAoB;AAChE,SAASC,UAAU,QAAQ,qBAAqB;AAChD,SAASC,mBAAmB,QAAQ,yBAAyB;AAC7D,SAASC,+BAA+B,QAAQ,yBAAyB;AACzE,SAASC,SAAS,EAAEC,SAAS,EAAEC,WAAW,QAAQ,uBAAuB;AACzE,SAASC,WAAW,QAAQ,2BAA2B;AACvD,SACEC,8BAA8B,EAC9BC,iCAAiC,EACjCC,oBAAoB,EACpBC,gBAAgB,EAEhBC,aAAa,QACR,oBAAoB;AAC3B,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,sBACLC,mBAAmB,QACd,0BAA0B;AAQjC,oDAAoD,GACpD,MAAMC,aAAa,CAACC,IAAqBA,EAAEC,MAAM,KAAK;AAEtD,4EAA4E;AAC5E,2CAA2C;AAC3C,OAAO,MAAMC,eAAelC,sBAAqC;IAC/DmC,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;WAC1DrC,eAAeK;WACfL,eAAeE;WACfF,eAAeqB;WACfrB,eAAeyB;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,gBAAgBjC,YAAYgC,QAAQT,IAAI;IAC9C,MAAMW,eAAeF,QAAQE,YAAY,IAAI;IAC7C,MAAMC,eAAepC,UAAUiC,QAAQG,YAAY,IAAIH,QAAQT,IAAI;IACnE,MAAMa,kBAAkB,MAAMlD,kBAAkB6C,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,gBAAgBhE,yBAAyBoD,MAAMM;IAErD,MAAM,EAAEjB,MAAM,EAAE,GAAGY;IACnB,4EAA4E;IAC5E,wEAAwE;IACxE,MAAMY,MACJZ,QAAQa,KAAK,KAAK,SAAS,MAAMlD,WAAWoC,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,GAAGnD,YAAY8B,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;IAEA,8EAA8E;IAC9E,4EAA4E;IAC5E,gDAAgD;IAChDvC,cACEuD,MACAtD,kBAAkB,YAAY2F,OAAO,EAAE,UACvC9B,KACAgB,iBACA;QAAEe,mBAAmB3F,kBAAkB4F,YAAY;IAAC;IAGtD,6EAA6E;IAC7E,+EAA+E;IAC/E,8EAA8E;IAC9E,6CAA6C;IAC7C,MAAMC,eAAe9F,kBACnB6D,KACAC,sBACA;IAEF,KAAK,MAAMiC,cAAc;QAAC;QAAc;QAAgB;KAAkB,CAAE;QAC1E,MAAMxF,kBAAkB+C,MAAMwC,cAAc,eAAeC;IAC7D;IAEA,qEAAqE;IACrE,yEAAyE;IACzE3F,UAAUkD,MAAMtD,kBAAkB6D,KAAK,gBAAgB;QACrDiB,kBAAkBtB;QAClBwC,UAAU;YACRrC;YACAgB;YACAsB,OAAOrB;YACPsB,MAAM3B;YACN4B,MAAM3B;YACN4B,QAAQ1C;YACR2C,QAAQ5B;YACR6B,YAAY5B;YACZ6B,UAAU5D;QACZ;IACF;IAEA,MAAMN,0BAA0BiB,MAAMX,QAAQC;IAC9C,MAAM4D,aAAa1G,SACjB+D,KACA7D,kBAAkBkC,cAAcC,oBAAoB,OAAO;IAE7D,MAAMsE,KAAK,IAAIxF,WAAWqC,MAAMV;IAEhC,MAAM8D,qBAAqB1G,kBACzB,QACAkE,cAAclB,IAAI,EAClB,UACA;IAEF,MAAM2D,wBAAwB3G,kBAC5B,QACAkE,cAAclB,IAAI,EAClB,UACA;IAEF,MAAM4D,0BAA0B,GAAGpF,YAAY8B,MAAM,CAAC,EAAEjC,UAAUkC,QAAQT,IAAI,EAAE,iBAAiB,CAAC;IAClG,MAAM+D,6BAA6B,GAAGrF,YAAY8B,MAAM,CAAC,EAAEjC,UACzDkC,QAAQT,IAAI,EACZ,sBAAsB,CAAC;IAEzB,IAAIS,QAAQa,KAAK,KAAK,QAAQ;QAC5B,MAAM,EAAE0C,gBAAgB,EAAEC,eAAe,EAAE,GAAGvG,sBAC5C0D,eACA;YAAE8C,gBAAgB;QAAyB;QAG7C9C,cAAc+C,OAAO,CAAC,mBAAmB,GAAG;YAC1CC,OAAO;YACPC,QAAQ;gBAAC;aAAU;YACnBC,SAAS;gBAAC;aAAsD;YAChEC,UAAU;YACV9D,SAAS;gBACP+D,UAAU;oBACRb,GAAGc,EAAE,CAACb;oBACND,GAAGe,KAAK,CAACd;oBACTD,GAAGgB,EAAE,CAACV,iBAAiBL;oBACvBD,GAAGgB,EAAE,CACHzH,kBAAkB6D,KAAK,eACvB7D,kBAAkB0G,oBAAoB;oBAExCD,GAAGgB,EAAE,CACHzH,kBAAkB6D,KAAK,gBACvB7D,kBAAkB0G,oBAAoB;oBAExCD,GAAGgB,EAAE,CACHzH,kBAAkB6D,KAAK,yBACvB7D,kBAAkB0G,oBAAoB;iBAEzC;gBACDgB,UAAU;YACZ;YACAC,WAAW;gBAACb;aAAiB;QAC/B;QACA5C,cAAc+C,OAAO,CAAC,wBAAwB,GAAG;YAC/CC,OAAO;YACPC,QAAQ;gBAAC;aAAU;YACnBC,SAAS;gBAAC;aAA2D;YACrEC,UAAU;YACV9D,SAAS;gBACP+D,UAAU;oBACRb,GAAGc,EAAE,CAACZ;oBACNF,GAAGe,KAAK,CAACb;oBACTF,GAAGgB,EAAE,CAACV,iBAAiBJ;oBACvBF,GAAGgB,EAAE,CACHzH,kBAAkB6D,KAAK,8BACvB7D,kBAAkB2G,uBAAuB;iBAE5C;gBACDe,UAAU;YACZ;YACAC,WAAW;gBAACb;aAAiB;QAC/B;QACApF,kCACEwC,eACA,UACA;QAEFxC,kCACEwC,eACA,UACA;IAEJ;IAEAA,cAAc+C,OAAO,CAAC,aAAa,GAAG;QACpCI,UAAU;QACV9D,SAAS;YACPqE,SAAS,CAAC,IAAI,EAAEpB,WAAW,cAAc,CAAC;YAC1CqB,KAAK;QACP;IACF;IACA3D,cAAc+C,OAAO,CAAC,MAAM,GAAG;QAC7BI,UAAU;QACVS,YAAY;QACZvE,SAAS;YACPqE,SAAS,CAAC,IAAI,EAAEpB,WAAW,mBAAmB,CAAC;YAC/CqB,KAAK;QACP;QACAF,WAAW;YAAC;SAAa;IAC3B;IACAzD,cAAc+C,OAAO,CAAC,cAAc,GAAG;QACrCI,UAAU;QACVM,WAAW;YAAC;SAAM;QAClBpE,SAAS;YACPqE,SAAS,CAAC,IAAI,EAAEpB,WAAW,UAAU,EAAE7D,OAAO,MAAM,CAAC;YACrDkF,KAAK;QACP;IACF;IACA3D,cAAc+C,OAAO,CAACc,OAAO,GAAG;QAC9BV,UAAU;QACVM,WAAW;YAAC;YAAO;SAAc;QACjCpE,SAAS;YACPqE,SAAS;YACTC,KAAK;YACLG,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IACA/D,cAAc+C,OAAO,CAAC9B,OAAO,GAAG;QAC9BkC,UAAU;QACVM,WAAW;YAAC;YAAO;SAAc;QACjCpE,SAAS;YACPqE,SAAS;YACTC,KAAK;YACLG,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IAEA,IAAI1E,QAAQa,KAAK,KAAK,QAAQ;QAC5B,IAAID,QAAQ,aAAa;YACvBD,cAAc+C,OAAO,CAACiB,MAAM,GAAG;gBAC7BhB,OAAOpG;gBACPuG,UAAU;gBACV9D,SAAS;oBACP+D,UAAU;wBACR,GAAG3D,gBAAgB,oDAAoD,EAAEiD,wBAAwB,CAAC,EAAEF,oBAAoB;wBACxH,GAAG/C,gBAAgB,oDAAoD,EAAEkD,2BAA2B,CAAC,EAAEF,uBAAuB;qBAC/H;oBACDe,UAAU;gBACZ;gBACAC,WAAW;oBAAC;oBAAoB;iBAAwB;YAC1D;YACAlG,+BAA+ByC,eAAe;YAE9CtD,oBACE0C,MACA;gBACE6E,SAASjE;gBACTP;gBACAyE,iBAAiB;gBACjBC,kBAAkB;gBAClBC,WAAW;oBAAC1B;oBAAyBC;iBAA2B;YAClE,GACAjE;QAEJ;QACAlB,kCACEwC,eACA,SACA;QAEFxC,kCACEwC,eACA,SACA;IAEJ;IAEA/D,2BAA2BmD,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;YACAoE,aAAa3E;YACb4E,aAAa3E;YACbL;YACAiF,eAAepH,UAAUkC,QAAQT,IAAI;YACrC4F,sBAAsB5E;YACtBJ;YACAiF,WAAWlF;YACXd;YACA+D;YACAC;YACAiC,WAAW;YACX/B;YACAD;YACAjD;QACF;IACF;IAEAtD,kBAAkBiD,MAAMV,cAAc;QAAE0B;QAAUkE,aAAa3E;IAAI;IACnEvD,kBAAkBgD,MAAMV,cAAc;QAAE0B;IAAS;IAEjD,MAAMlD,gCAAgCkC,MAAM;QAACH;KAAsB;IAEnE,MAAMpC,qBAAqBuC;IAC3B,OAAO,IACLnC,oBAAoBmC,MAAMC,QAAQsF,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAezF,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.js';
+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.js';\nimport { installDependencies } from '../../../utils/install.js';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics.js';\nimport { toSnakeCase } from '../../../utils/names.js';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx.js';\nimport { addWorkspaceDependencyToPyProject } from '../../../utils/py.js';\nimport { injectRdsCaBundleIntoDockerfile } from '../utils.js';\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,2BAA2B;AAChE,SAASC,mBAAmB,QAAQ,4BAA4B;AAChE,SAASC,+BAA+B,QAAQ,4BAA4B;AAC5E,SAASC,WAAW,QAAQ,0BAA0B;AACtD,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAuB;AAC9B,SAASC,iCAAiC,QAAQ,uBAAuB;AACzE,SAASC,+BAA+B,QAAQ,cAAc;AAG9D,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.js';\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.js';
+
+/**
+ * TypeScript types for options defined in schema.json
+ */
+export interface PyRdbMcpServerConnectionGeneratorSchema {
+  sourceProject: string;
+  targetProject: string;
+  sourceComponent?: ComponentMetadata;
+  preferInstallDependencies?: boolean;
+}