@aws/nx-plugin 1.0.0-rc.5 → 1.0.0-rc.50

package/src/py/rdb/files/__name__/models/example.py.template

@@ -0,0 +1,8 @@
+from sqlalchemy import Column, String
+from sqlmodel import Field, SQLModel
+
+
+class ExampleModel(SQLModel, table=True):
+    id: int | None = Field(default=None, primary_key=True)
+    name: str = Field(sa_column=Column(String(255), nullable=False))
+    description: str | None = Field(default=None, sa_column=Column(String(255), nullable=True))

package/src/py/rdb/files/__name__/utils.py.template

@@ -0,0 +1,104 @@
+import json
+import os
+import time
+from collections.abc import Callable
+from typing import Any
+
+import boto3
+from aws_lambda_powertools.utilities.parameters import AppConfigProvider
+from sqlalchemy import URL
+
+DatabaseConfig = dict[str, Any]
+DatabaseSecret = dict[str, Any]
+
+def build_database_url(
+    username: str,
+    host: str,
+    port: int,
+    database: str,
+    password: str | None = None,
+) -> URL:
+    return URL.create(
+        drivername='<%= engine === "mysql" ? "mysql+aiomysql" : "postgresql+asyncpg" %>',
+        username=username,
+        password=password,
+        host=host,
+        port=port,
+        database=database,
+    )
+
+
+def is_local_dev() -> bool:
+    return os.environ.get('LOCAL_DEV') == 'true'
+
+
+def get_local_dev_config() -> dict:
+    config_path = os.path.join(os.path.dirname(__file__), '..', 'config.json')
+    with open(config_path) as f:
+        return json.load(f)['localDev']
+
+
+def get_database_secret() -> DatabaseSecret:
+    secret_arn = os.environ.get('DATABASE_SECRET_ARN')
+    if not secret_arn:
+        raise ValueError('Missing required environment variable DATABASE_SECRET_ARN.')
+    client = boto3.client('secretsmanager')
+    response = client.get_secret_value(SecretId=secret_arn)
+    return json.loads(response['SecretString'])
+
+
+_database_config: dict[str, DatabaseConfig] = {}
+
+
+def get_database_config(runtime_config_key: str) -> DatabaseConfig:
+    if runtime_config_key not in _database_config:
+        app_id = os.environ.get('RUNTIME_CONFIG_APP_ID')
+        if not app_id:
+            raise ValueError('Missing required environment variable RUNTIME_CONFIG_APP_ID.')
+        provider = AppConfigProvider(application=app_id, environment='default')
+        config = provider.get('database', transform='json')
+        if not isinstance(config, dict):
+            raise ValueError('Unexpected RuntimeConfig format: expected JSON object.')
+        db_config = config.get(runtime_config_key)
+        if not db_config:
+            raise ValueError(f'RuntimeConfig is missing database.{runtime_config_key}.')
+        _database_config[runtime_config_key] = db_config
+    return _database_config[runtime_config_key]
+
+
+_TRANSIENT_ERROR_PATTERNS = [
+    'timeout',
+    'connection refused',
+    'could not connect',
+    'connection reset',
+    'connection timed out',
+    'operationalerror',
+    'password authentication failed',
+    'access denied',
+    'temporary failure',
+]
+
+
+def is_transient_connection_error(error: Exception) -> bool:
+    error_str = str(error).lower()
+    return any(p in error_str for p in _TRANSIENT_ERROR_PATTERNS)
+
+
+def with_connection_retry[T](
+    fn: Callable[[], T],
+    max_attempts: int = 6,
+    delay_seconds: float = 10.0,
+) -> T:
+    for attempt in range(1, max_attempts + 1):
+        try:
+            return fn()
+        except Exception as e:
+            if attempt == max_attempts or not is_transient_connection_error(e):
+                raise
+            wait = delay_seconds * attempt
+            print(
+                f'Transient connection error (attempt {attempt}/{max_attempts}), '
+                f'retrying in {wait:.0f}s: {e}'
+            )
+            time.sleep(wait)
+    raise RuntimeError('unreachable')

package/src/py/rdb/files/alembic.ini.template

@@ -0,0 +1,38 @@
+[alembic]
+script_location = migrations
+file_template = %%(year)d%%(month).2d%%(day).2d_%%(hour).2d%%(minute).2d_%%(rev)s_%%(slug)s
+prepend_sys_path = .
+
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARN
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARN
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S

package/src/py/rdb/files/config.json.template

@@ -0,0 +1,14 @@
+{
+  "runtimeConfigKey": "<%= runtimeConfigKey %>",
+  "localDev": {
+    "containerEngine": "<%= containerEngine %>",
+    "containerName": "<%= containerName %>",
+    "image": "<%= dockerImage %>",
+    "port": <%= localDbPort %>,
+    "host": "<%= localDbHost %>",
+    "dbName": "<%= localDbName %>",
+    "dbUser": "<%= localDbUser %>",
+    "dbPassword": "<%= localDbPassword %>",
+    "dbEngine": "<%= engine %>"
+  }
+}

package/src/py/rdb/files/migrations/env.py.template

@@ -0,0 +1,80 @@
+import asyncio
+import json
+import os
+from logging.config import fileConfig
+
+from alembic import context
+from sqlalchemy import URL
+from sqlalchemy.ext.asyncio import create_async_engine
+from sqlmodel import SQLModel
+
+# Import all models to register them in SQLModel.metadata
+from <%= name %>.models import *  # noqa: F401, F403
+
+config = context.config
+
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+target_metadata = SQLModel.metadata
+
+
+def _get_local_url():
+    if os.environ.get('LOCAL_DEV') == 'true':
+        config_path = os.path.join(os.path.dirname(__file__), '..', 'config.json')
+        with open(config_path) as f:
+            cfg = json.load(f)['localDev']
+        engine = cfg['dbEngine']
+        driver = 'mysql+aiomysql' if engine == 'mysql' else 'postgresql+asyncpg'
+        return URL.create(
+            driver,
+            username=cfg['dbUser'],
+            password=cfg['dbPassword'],
+            host=cfg['host'],
+            port=cfg['port'],
+            database=cfg['dbName'],
+        )
+
+    raise RuntimeError('Run alembic via nx run <project>:alembic')
+
+
+# Generates SQL script without a live DB connection (alembic upgrade head --sql)
+def run_migrations_offline() -> None:
+    context.configure(
+        url=_get_local_url(),
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={'paramstyle': 'named'},
+    )
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def _run_migrations_sync(sync_conn) -> None:
+    context.configure(connection=sync_conn, target_metadata=target_metadata)
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+# Runs migrations against a live DB connection
+def run_migrations_online() -> None:
+    supplied_connection = config.attributes.get('connection')
+    # Connection injected by migration_handler.py (Lambda)
+    if supplied_connection is not None:
+        _run_migrations_sync(supplied_connection)
+        return
+
+    # Local alembic commands run via nx run <project>:alembic
+    async def _run_async() -> None:
+        engine = create_async_engine(_get_local_url())
+        async with engine.begin() as conn:
+            await conn.run_sync(_run_migrations_sync)
+        await engine.dispose()
+
+    asyncio.run(_run_async())
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

package/src/py/rdb/files/migrations/script.py.mako.template

@@ -0,0 +1,28 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, Sequence[str], None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    ${downgrades if downgrades else "pass"}

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

@@ -0,0 +1,6 @@
+import { type GeneratorCallback, type Tree } from '@nx/devkit';
+import { type NxGeneratorInfo } from '../../utils/nx';
+import type { PyRdbGeneratorSchema } from './schema';
+export declare const PY_RDB_GENERATOR_INFO: NxGeneratorInfo;
+export declare const pyRdbGenerator: (tree: Tree, options: PyRdbGeneratorSchema) => Promise<GeneratorCallback>;
+export default pyRdbGenerator;

package/src/py/rdb/generator.js

@@ -0,0 +1,280 @@
+/**
+ * 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 { addPythonBundleTarget } from "../../utils/bundle/bundle.js";
+import { resolveContainers } from "../../utils/containers.js";
+import { addDependenciesToPackageJson } from "../../utils/dependencies.js";
+import { addDockerScanTarget } from "../../utils/docker.js";
+import { formatFilesInSubtree } from "../../utils/format.js";
+import { 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 { addDependenciesToPyProjectToml } from "../../utils/py.js";
+import { addRdbInfra } from "../../utils/rdb-constructs/rdb-constructs.js";
+import { sharedConstructsGenerator } from "../../utils/shared-constructs.js";
+import { PACKAGES_DIR, SHARED_SCRIPTS_DIR } from "../../utils/shared-constructs-constants.js";
+import { sharedRdbScriptsGenerator } from "../../utils/shared-rdb-scripts.js";
+import { PY_VERSIONS, withVersions } from "../../utils/versions.js";
+import pyProjectGenerator, { getPyProjectDetails } from "../project/generator.js";
+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;
+    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);
+    // Used by local dev script wait-for-*-db.ts
+    addDependenciesToPackageJson(tree, withVersions([
+        engine === 'mysql' ? 'mariadb' : 'pg'
+    ]), engine === 'mysql' ? {} : withVersions([
+        '@types/pg'
+    ]));
+    const scriptsDir = relative(dir, joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'rdb'));
+    const fs = new FsCommands(tree);
+    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') {
+        const iac = await resolveIac(tree, options.iac);
+        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
+                ]
+            });
+        }
+        addDependencyToTargetIfNotPresent(projectConfig, 'build', 'bundle-migration');
+        addDependencyToTargetIfNotPresent(projectConfig, 'build', 'bundle-create-db-user');
+    }
+    updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);
+    addGeneratorMetadata(tree, fullyQualifiedName, PY_RDB_GENERATOR_INFO, {
+        engine
+    });
+    if (options.infra !== 'none') {
+        const iac = await resolveIac(tree, options.iac);
+        await sharedConstructsGenerator(tree, {
+            iac
+        });
+        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
+        });
+    }
+    addDependenciesToPyProjectToml(tree, dir, [
+        'sqlmodel',
+        '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.
+        'greenlet',
+        ...engine === 'mysql' ? [
+            'aiomysql',
+            'boto3',
+            'aws-lambda-powertools'
+        ] : [
+            'asyncpg',
+            'boto3',
+            'aws-lambda-powertools'
+        ]
+    ]);
+    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 { addPythonBundleTarget } from '../../utils/bundle/bundle';\nimport { resolveContainers } from '../../utils/containers';\nimport { addDependenciesToPackageJson } from '../../utils/dependencies';\nimport { addDockerScanTarget } from '../../utils/docker';\nimport { formatFilesInSubtree } from '../../utils/format';\nimport { 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 { addDependenciesToPyProjectToml } from '../../utils/py';\nimport { addRdbInfra } from '../../utils/rdb-constructs/rdb-constructs';\nimport { sharedConstructsGenerator } from '../../utils/shared-constructs';\nimport {\n  PACKAGES_DIR,\n  SHARED_SCRIPTS_DIR,\n} from '../../utils/shared-constructs-constants';\nimport { sharedRdbScriptsGenerator } from '../../utils/shared-rdb-scripts';\nimport { PY_VERSIONS, withVersions } from '../../utils/versions';\nimport pyProjectGenerator, { getPyProjectDetails } from '../project/generator';\nimport type { PyRdbGeneratorSchema } from './schema';\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  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);\n  // Used by local dev script wait-for-*-db.ts\n  addDependenciesToPackageJson(\n    tree,\n    withVersions([engine === 'mysql' ? 'mariadb' : 'pg']),\n    engine === 'mysql' ? {} : withVersions(['@types/pg']),\n  );\n  const scriptsDir = relative(\n    dir,\n    joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'rdb'),\n  );\n  const fs = new FsCommands(tree);\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    const iac = await resolveIac(tree, options.iac);\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(tree, {\n        project: projectConfig,\n        containerEngine,\n        trivyTargetName: 'trivy',\n        dockerTargetName: 'docker',\n        imageTags: [migrationDockerImageTag, createDbUserDockerImageTag],\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(tree, fullyQualifiedName, PY_RDB_GENERATOR_INFO, {\n    engine,\n  });\n\n  if (options.infra !== 'none') {\n    const iac = await resolveIac(tree, options.iac);\n    await sharedConstructsGenerator(tree, { iac });\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  addDependenciesToPyProjectToml(tree, dir, [\n    'sqlmodel',\n    '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    'greenlet',\n    ...(engine === 'mysql'\n      ? (['aiomysql', 'boto3', 'aws-lambda-powertools'] as const)\n      : (['asyncpg', 'boto3', 'aws-lambda-powertools'] as const)),\n  ]);\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","addPythonBundleTarget","resolveContainers","addDependenciesToPackageJson","addDockerScanTarget","formatFilesInSubtree","FsCommands","resolveIac","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","snakeCase","toClassName","getNpmScope","addDependencyToTargetIfNotPresent","addGeneratorMetadata","getGeneratorInfo","projectExists","assignPort","addDependenciesToPyProjectToml","addRdbInfra","sharedConstructsGenerator","PACKAGES_DIR","SHARED_SCRIPTS_DIR","sharedRdbScriptsGenerator","PY_VERSIONS","withVersions","pyProjectGenerator","getPyProjectDetails","PY_RDB_GENERATOR_INFO","filename","pyRdbGenerator","tree","options","nameClassName","name","databaseUser","databaseName","containerEngine","fullyQualifiedName","dir","normalizedModuleName","directory","subDirectory","type","projectConfig","engine","localDbPort","localDbHost","localDbUser","localDbPassword","containerName","dockerImage","templateOptions","runtimeConfigKey","localDbName","sqlmodelVersion","sqlmodel","alembicVersion","alembic","aiomysqlVersion","aiomysql","asyncpgVersion","asyncpg","boto3Version","boto3","awsLambdaPowertoolsVersion","dirname","scriptsDir","fs","migrationBundleDir","root","createDbUserBundleDir","migrationDockerImageTag","createDbUserDockerImageTag","infra","bundleTargetName","bundleOutputDir","pythonPlatform","targets","cache","outputs","executor","commands","rm","mkdir","cp","parallel","dependsOn","command","cwd","continuous","migrate","env","LOCAL_DEV","iac","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,SAASC,qBAAqB,QAAQ,+BAA4B;AAClE,SAASC,iBAAiB,QAAQ,4BAAyB;AAC3D,SAASC,4BAA4B,QAAQ,8BAA2B;AACxE,SAASC,mBAAmB,QAAQ,wBAAqB;AACzD,SAASC,oBAAoB,QAAQ,wBAAqB;AAC1D,SAASC,UAAU,QAAQ,oBAAiB;AAC5C,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,8BAA8B,QAAQ,oBAAiB;AAChE,SAASC,WAAW,QAAQ,+CAA4C;AACxE,SAASC,yBAAyB,QAAQ,mCAAgC;AAC1E,SACEC,YAAY,EACZC,kBAAkB,QACb,6CAA0C;AACjD,SAASC,yBAAyB,QAAQ,oCAAiC;AAC3E,SAASC,WAAW,EAAEC,YAAY,QAAQ,0BAAuB;AACjE,OAAOC,sBAAsBC,mBAAmB,QAAQ,0BAAuB;AAG/E,OAAO,MAAMC,wBAAyCb,iBACpD,YAAYc,QAAQ,EACpB;AAEF,OAAO,MAAMC,iBAAiB,OAC5BC,MACAC;IAEA,MAAMC,gBAAgBtB,YAAYqB,QAAQE,IAAI;IAC9C,MAAMC,eAAeH,QAAQG,YAAY,IAAI;IAC7C,MAAMC,eAAe1B,UAAUsB,QAAQI,YAAY,IAAIJ,QAAQE,IAAI;IACnE,MAAMG,kBAAkB,MAAMpC,kBAAkB8B,MAAM;IAEtD,MAAM,EAAEO,kBAAkB,EAAEC,GAAG,EAAEC,oBAAoB,EAAE,GAAGb,oBACxDI,MACA;QACEG,MAAMF,QAAQE,IAAI;QAClBO,WAAWT,QAAQS,SAAS;QAC5BC,cAAcV,QAAQU,YAAY;IACpC;IAGF,IAAI,CAAC1B,cAAce,MAAMO,qBAAqB;QAC5C,MAAMZ,mBAAmBK,MAAM;YAC7BG,MAAMF,QAAQE,IAAI;YAClBO,WAAWT,QAAQS,SAAS;YAC5BC,cAAcV,QAAQU,YAAY;YAClCC,MAAM;QACR;IACF;IAEA,MAAMC,gBAAgB9C,yBAAyBiC,MAAMO;IAErD,MAAM,EAAEO,MAAM,EAAE,GAAGb;IACnB,MAAMc,cAAc7B,WAClBc,MACAa,eACAC,WAAW,UAAU,OAAO;IAE9B,MAAME,cAAc;IACpB,MAAMC,cAAcH,WAAW,UAAU,SAAS;IAClD,MAAMI,kBAAkB;IACxB,MAAMC,gBAAgB,GAAGtC,YAAYmB,MAAM,CAAC,EAAEK,cAAc;IAC5D,MAAMe,cACJN,WAAW,UACP,+CACA;IAEN,MAAMO,kBAAkB;QACtBlB,MAAMM;QACNa,kBAAkBpB;QAClBY;QACAC;QACAC;QACAO,aAAalB;QACbY;QACAC;QACAZ;QACAa;QACAC;QACAI,iBAAiB/B,YAAYgC,QAAQ;QACrCC,gBAAgBjC,YAAYkC,OAAO;QACnCC,iBAAiBnC,YAAYoC,QAAQ;QACrCC,gBAAgBrC,YAAYsC,OAAO;QACnCC,cAAcvC,YAAYwC,KAAK;QAC/BC,4BAA4BzC,WAAW,CAAC,wBAAwB;IAClE;IAEA5B,cACEmC,MACAlC,kBAAkB,YAAYqE,OAAO,EAAE,UACvC3B,KACAa;IAGF,MAAM7B,0BAA0BQ,MAAMc;IACtC,4CAA4C;IAC5C3C,6BACE6B,MACAN,aAAa;QAACoB,WAAW,UAAU,YAAY;KAAK,GACpDA,WAAW,UAAU,CAAC,IAAIpB,aAAa;QAAC;KAAY;IAEtD,MAAM0C,aAAaxE,SACjB4C,KACA1C,kBAAkBwB,cAAcC,oBAAoB,OAAO;IAE7D,MAAM8C,KAAK,IAAI/D,WAAW0B;IAE1B,MAAMsC,qBAAqBxE,kBACzB,QACA+C,cAAc0B,IAAI,EAClB,UACA;IAEF,MAAMC,wBAAwB1E,kBAC5B,QACA+C,cAAc0B,IAAI,EAClB,UACA;IAEF,MAAME,0BAA0B,GAAG5D,YAAYmB,MAAM,CAAC,EAAEtB,UAAUuB,QAAQE,IAAI,EAAE,iBAAiB,CAAC;IAClG,MAAMuC,6BAA6B,GAAG7D,YAAYmB,MAAM,CAAC,EAAEtB,UACzDuB,QAAQE,IAAI,EACZ,sBAAsB,CAAC;IAEzB,IAAIF,QAAQ0C,KAAK,KAAK,QAAQ;QAC5B,MAAM,EAAEC,gBAAgB,EAAEC,eAAe,EAAE,GAAG5E,sBAC5C4C,eACA;YAAEiC,gBAAgB;QAAyB;QAG7CjC,cAAckC,OAAO,CAAC,mBAAmB,GAAG;YAC1CC,OAAO;YACPC,SAAS;gBAAC;aAAsD;YAChEC,UAAU;YACVjD,SAAS;gBACPkD,UAAU;oBACRd,GAAGe,EAAE,CAACd;oBACND,GAAGgB,KAAK,CAACf;oBACTD,GAAGiB,EAAE,CAACT,iBAAiBP;oBACvBD,GAAGiB,EAAE,CACHxF,kBAAkB0C,KAAK,eACvB1C,kBAAkBwE,oBAAoB;oBAExCD,GAAGiB,EAAE,CACHxF,kBAAkB0C,KAAK,gBACvB1C,kBAAkBwE,oBAAoB;oBAExCD,GAAGiB,EAAE,CACHxF,kBAAkB0C,KAAK,yBACvB1C,kBAAkBwE,oBAAoB;iBAEzC;gBACDiB,UAAU;YACZ;YACAC,WAAW;gBAACZ;aAAiB;QAC/B;QACA/B,cAAckC,OAAO,CAAC,wBAAwB,GAAG;YAC/CC,OAAO;YACPC,SAAS;gBAAC;aAA2D;YACrEC,UAAU;YACVjD,SAAS;gBACPkD,UAAU;oBACRd,GAAGe,EAAE,CAACZ;oBACNH,GAAGgB,KAAK,CAACb;oBACTH,GAAGiB,EAAE,CAACT,iBAAiBL;oBACvBH,GAAGiB,EAAE,CACHxF,kBAAkB0C,KAAK,8BACvB1C,kBAAkB0E,uBAAuB;iBAE5C;gBACDe,UAAU;YACZ;YACAC,WAAW;gBAACZ;aAAiB;QAC/B;QACA9D,kCACE+B,eACA,UACA;QAEF/B,kCACE+B,eACA,UACA;IAEJ;IAEAA,cAAckC,OAAO,CAAC,aAAa,GAAG;QACpCG,UAAU;QACVjD,SAAS;YACPwD,SAAS,CAAC,IAAI,EAAErB,WAAW,cAAc,CAAC;YAC1CsB,KAAK;QACP;IACF;IACA7C,cAAckC,OAAO,CAAC,MAAM,GAAG;QAC7BG,UAAU;QACVS,YAAY;QACZ1D,SAAS;YACPwD,SAAS,CAAC,IAAI,EAAErB,WAAW,mBAAmB,CAAC;YAC/CsB,KAAK;QACP;QACAF,WAAW;YAAC;SAAa;IAC3B;IACA3C,cAAckC,OAAO,CAAC,cAAc,GAAG;QACrCG,UAAU;QACVM,WAAW;YAAC;SAAM;QAClBvD,SAAS;YACPwD,SAAS,CAAC,IAAI,EAAErB,WAAW,UAAU,EAAEtB,OAAO,MAAM,CAAC;YACrD4C,KAAK;QACP;IACF;IACA7C,cAAckC,OAAO,CAACa,OAAO,GAAG;QAC9BV,UAAU;QACVM,WAAW;YAAC;YAAO;SAAc;QACjCvD,SAAS;YACPwD,SAAS;YACTC,KAAK;YACLG,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IACAjD,cAAckC,OAAO,CAACpB,OAAO,GAAG;QAC9BuB,UAAU;QACVM,WAAW;YAAC;YAAO;SAAc;QACjCvD,SAAS;YACPwD,SAAS;YACTC,KAAK;YACLG,KAAK;gBACHC,WAAW;YACb;QACF;IACF;IAEA,IAAI7D,QAAQ0C,KAAK,KAAK,QAAQ;QAC5B,MAAMoB,MAAM,MAAMxF,WAAWyB,MAAMC,QAAQ8D,GAAG;QAC9C,IAAIA,QAAQ,aAAa;YACvBlD,cAAckC,OAAO,CAACiB,MAAM,GAAG;gBAC7BhB,OAAO;gBACPE,UAAU;gBACVjD,SAAS;oBACPkD,UAAU;wBACR,GAAG7C,gBAAgB,oDAAoD,EAAEmC,wBAAwB,CAAC,EAAEH,oBAAoB;wBACxH,GAAGhC,gBAAgB,oDAAoD,EAAEoC,2BAA2B,CAAC,EAAEF,uBAAuB;qBAC/H;oBACDe,UAAU;gBACZ;gBACAC,WAAW;oBAAC;oBAAoB;iBAAwB;YAC1D;YACA1E,kCAAkC+B,eAAe,SAAS;YAE1DzC,oBAAoB4B,MAAM;gBACxBiE,SAASpD;gBACTP;gBACA4D,iBAAiB;gBACjBC,kBAAkB;gBAClBC,WAAW;oBAAC3B;oBAAyBC;iBAA2B;YAClE;QACF;QACA5D,kCACE+B,eACA,SACA;QAEF/B,kCACE+B,eACA,SACA;IAEJ;IAEA7C,2BAA2BgC,MAAMO,oBAAoBM;IACrD9B,qBAAqBiB,MAAMO,oBAAoBV,uBAAuB;QACpEiB;IACF;IAEA,IAAIb,QAAQ0C,KAAK,KAAK,QAAQ;QAC5B,MAAMoB,MAAM,MAAMxF,WAAWyB,MAAMC,QAAQ8D,GAAG;QAC9C,MAAM1E,0BAA0BW,MAAM;YAAE+D;QAAI;QAC5C,MAAM3E,YAAYY,MAAM;YACtB+D;YACAM,aAAa9D;YACb+D,aAAa9D;YACbN;YACAqE,eAAe7F,UAAUuB,QAAQE,IAAI;YACrCqE,sBAAsB/D;YACtBJ;YACAoE,WAAWrE;YACXU;YACAwB;YACAE;YACAkC,WAAW;YACXhC;YACAD;YACAnC;QACF;IACF;IAEAnB,+BAA+Ba,MAAMQ,KAAK;QACxC;QACA;QACA,qEAAqE;QACrE,0EAA0E;QAC1E,mEAAmE;QACnE,6CAA6C;QAC7C;WACIM,WAAW,UACV;YAAC;YAAY;YAAS;SAAwB,GAC9C;YAAC;YAAW;YAAS;SAAwB;KACnD;IAED,MAAMrC,gCAAgCuB,MAAM;QAACH;KAAsB;IAEnE,MAAMxB,qBAAqB2B;IAC3B,OAAO,IACLxB,oBAAoBwB,MAAMC,QAAQ0E,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAe7E,eAAe"}