@aws/nx-plugin 0.113.0 → 0.114.1

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

@@ -0,0 +1,286 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`ts#rdb smithy-connection generator > should add rdb serve-local dependency to smithy backend serve-local 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "name": "api",
+  "root": "packages/api",
+  "targets": {
+    "serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`ts#rdb smithy-connection generator > should be idempotent across context.ts, handler.ts, and local-server.ts 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+
+export interface ServiceContext {
+  tracer: Tracer;
+  logger: Logger;
+  metrics: Metrics;
+  db: Awaited<ReturnType<typeof getDb>>;
+}
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should be idempotent across context.ts, handler.ts, and local-server.ts 2`] = `
+"import { getPrisma as getDb } from 'db';
+import {
+  convertEvent,
+  convertVersion1Response,
+} from '@aws-smithy/server-apigateway';
+import type { APIGatewayProxyEvent, APIGatewayProxyResult } from 'aws-lambda';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Service } from './service.js';
+import { getMyApiServiceHandler } from './generated/ssdk/index.js';
+
+const tracer = new Tracer();
+const logger = new Logger();
+const metrics = new Metrics();
+
+const serviceHandler = getMyApiServiceHandler(Service);
+
+export const lambdaHandler = async (
+  event: APIGatewayProxyEvent,
+): Promise<APIGatewayProxyResult> => {
+  const httpRequest = convertEvent(event);
+  const db = await getDb();
+  const httpResponse = await serviceHandler.handle(httpRequest, {
+    tracer,
+    logger,
+    metrics,
+    db,
+  });
+  return convertVersion1Response(httpResponse);
+};
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should be idempotent across context.ts, handler.ts, and local-server.ts 3`] = `
+"import { getPrisma as getDb } from 'db';
+import { IncomingMessage, ServerResponse, createServer } from 'http';
+import { convertRequest, writeResponse } from '@aws-smithy/server-node';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Service } from './service.js';
+import { getMyApiServiceHandler } from './generated/ssdk/index.js';
+
+const tracer = new Tracer();
+const logger = new Logger();
+const metrics = new Metrics();
+
+const serviceHandler = getMyApiServiceHandler(Service);
+
+const server = createServer(async function (
+  req: IncomingMessage,
+  res: ServerResponse<IncomingMessage> & { req: IncomingMessage },
+) {
+  const httpRequest = convertRequest(req);
+  const db = await getDb();
+  const httpResponse = await serviceHandler.handle(httpRequest, {
+    tracer,
+    logger,
+    metrics,
+    db,
+  });
+  return writeResponse(httpResponse, res);
+});
+
+server.listen(3000);
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should inject into context.ts, handler.ts, and local-server.ts 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+
+export interface ServiceContext {
+  tracer: Tracer;
+  logger: Logger;
+  metrics: Metrics;
+  db: Awaited<ReturnType<typeof getDb>>;
+}
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should inject into context.ts, handler.ts, and local-server.ts 2`] = `
+"import { getPrisma as getDb } from 'db';
+import {
+  convertEvent,
+  convertVersion1Response,
+} from '@aws-smithy/server-apigateway';
+import type { APIGatewayProxyEvent, APIGatewayProxyResult } from 'aws-lambda';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Service } from './service.js';
+import { getMyApiServiceHandler } from './generated/ssdk/index.js';
+
+const tracer = new Tracer();
+const logger = new Logger();
+const metrics = new Metrics();
+
+const serviceHandler = getMyApiServiceHandler(Service);
+
+export const lambdaHandler = async (
+  event: APIGatewayProxyEvent,
+): Promise<APIGatewayProxyResult> => {
+  const httpRequest = convertEvent(event);
+  const db = await getDb();
+  const httpResponse = await serviceHandler.handle(httpRequest, {
+    tracer,
+    logger,
+    metrics,
+    db,
+  });
+  return convertVersion1Response(httpResponse);
+};
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should inject into context.ts, handler.ts, and local-server.ts 3`] = `
+"import { getPrisma as getDb } from 'db';
+import { IncomingMessage, ServerResponse, createServer } from 'http';
+import { convertRequest, writeResponse } from '@aws-smithy/server-node';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Service } from './service.js';
+import { getMyApiServiceHandler } from './generated/ssdk/index.js';
+
+const tracer = new Tracer();
+const logger = new Logger();
+const metrics = new Metrics();
+
+const serviceHandler = getMyApiServiceHandler(Service);
+
+const server = createServer(async function (
+  req: IncomingMessage,
+  res: ServerResponse<IncomingMessage> & { req: IncomingMessage },
+) {
+  const httpRequest = convertRequest(req);
+  const db = await getDb();
+  const httpResponse = await serviceHandler.handle(httpRequest, {
+    tracer,
+    logger,
+    metrics,
+    db,
+  });
+  return writeResponse(httpResponse, res);
+});
+
+server.listen(3000);
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should support multiple rdb connections 1`] = `
+"import { getPrisma as getMysqlDb } from 'mysql-db';
+import { getPrisma as getPostgresDb } from 'postgres-db';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+
+export interface ServiceContext {
+  tracer: Tracer;
+  logger: Logger;
+  metrics: Metrics;
+  postgresDb: Awaited<ReturnType<typeof getPostgresDb>>;
+  mysqlDb: Awaited<ReturnType<typeof getMysqlDb>>;
+}
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should support multiple rdb connections 2`] = `
+"import { getPrisma as getMysqlDb } from 'mysql-db';
+import { getPrisma as getPostgresDb } from 'postgres-db';
+import {
+  convertEvent,
+  convertVersion1Response,
+} from '@aws-smithy/server-apigateway';
+import type { APIGatewayProxyEvent, APIGatewayProxyResult } from 'aws-lambda';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Service } from './service.js';
+import { getMyApiServiceHandler } from './generated/ssdk/index.js';
+
+const tracer = new Tracer();
+const logger = new Logger();
+const metrics = new Metrics();
+
+const serviceHandler = getMyApiServiceHandler(Service);
+
+export const lambdaHandler = async (
+  event: APIGatewayProxyEvent,
+): Promise<APIGatewayProxyResult> => {
+  const httpRequest = convertEvent(event);
+  const postgresDb = await getPostgresDb();
+  const mysqlDb = await getMysqlDb();
+  const httpResponse = await serviceHandler.handle(httpRequest, {
+    tracer,
+    logger,
+    metrics,
+    postgresDb,
+    mysqlDb,
+  });
+  return convertVersion1Response(httpResponse);
+};
+"
+`;
+
+exports[`ts#rdb smithy-connection generator > should support multiple rdb connections 3`] = `
+"import { getPrisma as getMysqlDb } from 'mysql-db';
+import { getPrisma as getPostgresDb } from 'postgres-db';
+import { IncomingMessage, ServerResponse, createServer } from 'http';
+import { convertRequest, writeResponse } from '@aws-smithy/server-node';
+import { Tracer } from '@aws-lambda-powertools/tracer';
+import { Logger } from '@aws-lambda-powertools/logger';
+import { Metrics } from '@aws-lambda-powertools/metrics';
+import { Service } from './service.js';
+import { getMyApiServiceHandler } from './generated/ssdk/index.js';
+
+const tracer = new Tracer();
+const logger = new Logger();
+const metrics = new Metrics();
+
+const serviceHandler = getMyApiServiceHandler(Service);
+
+const server = createServer(async function (
+  req: IncomingMessage,
+  res: ServerResponse<IncomingMessage> & { req: IncomingMessage },
+) {
+  const httpRequest = convertRequest(req);
+  const postgresDb = await getPostgresDb();
+  const mysqlDb = await getMysqlDb();
+  const httpResponse = await serviceHandler.handle(httpRequest, {
+    tracer,
+    logger,
+    metrics,
+    postgresDb,
+    mysqlDb,
+  });
+  return writeResponse(httpResponse, res);
+});
+
+server.listen(3000);
+"
+`;

package/src/ts/rdb/smithy-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 { Tree } from '@nx/devkit';
+import { TsRdbSmithyConnectionGeneratorSchema } from './schema';
+import { NxGeneratorInfo } from '../../../utils/nx';
+export declare const TS_RDB_SMITHY_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
+export declare const tsRdbSmithyConnectionGenerator: (tree: Tree, options: TsRdbSmithyConnectionGeneratorSchema) => Promise<void>;
+export default tsRdbSmithyConnectionGenerator;

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

@@ -0,0 +1,72 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tsRdbSmithyConnectionGenerator = exports.TS_RDB_SMITHY_CONNECTION_GENERATOR_INFO = void 0;
+const tslib_1 = require("tslib");
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const devkit_1 = require("@nx/devkit");
+const nx_1 = require("../../../utils/nx");
+const metrics_1 = require("../../../utils/metrics");
+const format_1 = require("../../../utils/format");
+const names_1 = require("../../../utils/names");
+const lodash_camelcase_1 = tslib_1.__importDefault(require("lodash.camelcase"));
+const npm_scope_1 = require("../../../utils/npm-scope");
+const ast_1 = require("../../../utils/ast");
+exports.TS_RDB_SMITHY_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
+const tsRdbSmithyConnectionGenerator = async (tree, options) => {
+    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
+    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+    if (sourceProject.targets?.['serve-local']) {
+        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, 'serve-local', {
+            projects: [targetProject.name],
+            target: 'serve-local',
+        });
+        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+    }
+    const rdbBaseName = targetProject.name.split('/').pop();
+    const rdbNameCamel = (0, lodash_camelcase_1.default)(rdbBaseName);
+    const rdbNamePascal = (0, names_1.pascalCase)(rdbBaseName);
+    const rdbPackageAlias = (0, npm_scope_1.toScopeAlias)(targetProject.name);
+    const getterAlias = `getPrisma as get${rdbNamePascal}`;
+    const contextPath = (0, devkit_1.joinPathFragments)(sourceProject.root, 'src/context.ts');
+    const handlerPath = (0, devkit_1.joinPathFragments)(sourceProject.root, 'src/handler.ts');
+    const localServerPath = (0, devkit_1.joinPathFragments)(sourceProject.root, 'src/local-server.ts');
+    if (tree.exists(contextPath)) {
+        await (0, ast_1.addDestructuredImport)(tree, contextPath, [getterAlias], rdbPackageAlias);
+        await (0, ast_1.applyGritQL)(tree, contextPath, `\`interface ServiceContext { $members }\` where {
+        $members <: not some \`${rdbNameCamel}: $_\`,
+        $members += \`\n  ${rdbNameCamel}: Awaited<ReturnType<typeof get${rdbNamePascal}>>\`
+      }`);
+    }
+    if (tree.exists(handlerPath)) {
+        await (0, ast_1.addDestructuredImport)(tree, handlerPath, [getterAlias], rdbPackageAlias);
+        const alreadyHasDecl = await (0, ast_1.matchGritQL)(tree, handlerPath, `\`const ${rdbNameCamel} = await get${rdbNamePascal}()\``);
+        if (!alreadyHasDecl) {
+            await (0, ast_1.applyGritQL)(tree, handlerPath, `\`const httpResponse = await serviceHandler.handle($req, $ctx)\` => \`const ${rdbNameCamel} = await get${rdbNamePascal}();\n  const httpResponse = await serviceHandler.handle($req, $ctx)\``);
+        }
+        await (0, ast_1.applyGritQL)(tree, handlerPath, `\`serviceHandler.handle($req, { $props })\` where {
+        $props <: not some \`${rdbNameCamel}\`,
+        $props += \`\n    ${rdbNameCamel}\`
+      }`);
+    }
+    if (tree.exists(localServerPath)) {
+        await (0, ast_1.addDestructuredImport)(tree, localServerPath, [getterAlias], rdbPackageAlias);
+        const alreadyHasDeclLocal = await (0, ast_1.matchGritQL)(tree, localServerPath, `\`const ${rdbNameCamel} = await get${rdbNamePascal}()\``);
+        if (!alreadyHasDeclLocal) {
+            await (0, ast_1.applyGritQL)(tree, localServerPath, `\`const httpResponse = await serviceHandler.handle($req, $ctx)\` => \`const ${rdbNameCamel} = await get${rdbNamePascal}();\n  const httpResponse = await serviceHandler.handle($req, $ctx)\``);
+        }
+        await (0, ast_1.applyGritQL)(tree, localServerPath, `\`serviceHandler.handle($req, { $props })\` where {
+        $props <: not some \`${rdbNameCamel}\`,
+        $props += \`\n    ${rdbNameCamel}\`
+      }`);
+    }
+    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
+        exports.TS_RDB_SMITHY_CONNECTION_GENERATOR_INFO,
+    ]);
+    await (0, format_1.formatFilesInSubtree)(tree);
+};
+exports.tsRdbSmithyConnectionGenerator = tsRdbSmithyConnectionGenerator;
+exports.default = exports.tsRdbSmithyConnectionGenerator;
+//# sourceMappingURL=generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/ts/rdb/smithy-connection/generator.ts"],"names":[],"mappings":";;;;AAAA;;;GAGG;AACH,uCAIoB;AAEpB,0CAK2B;AAC3B,oDAAyE;AACzE,kDAA6D;AAC7D,gDAAkD;AAClD,gFAAyC;AACzC,wDAAwD;AACxD,4CAI4B;AAEf,QAAA,uCAAuC,GAClD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,8BAA8B,GAAG,KAAK,EACjD,IAAU,EACV,OAA6C,EAC9B,EAAE;IACjB,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC;QAC3C,IAAA,sCAAiC,EAAC,aAAa,EAAE,aAAa,EAAE;YAC9D,QAAQ,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC;YAC9B,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,IAAA,mCAA0B,EAAC,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,WAAW,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IACxD,MAAM,YAAY,GAAG,IAAA,0BAAS,EAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,aAAa,GAAG,IAAA,kBAAU,EAAC,WAAW,CAAC,CAAC;IAC9C,MAAM,eAAe,GAAG,IAAA,wBAAY,EAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IACzD,MAAM,WAAW,GAAG,mBAAmB,aAAa,EAAE,CAAC;IAEvD,MAAM,WAAW,GAAG,IAAA,0BAAiB,EAAC,aAAa,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;IAC5E,MAAM,WAAW,GAAG,IAAA,0BAAiB,EAAC,aAAa,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;IAC5E,MAAM,eAAe,GAAG,IAAA,0BAAiB,EACvC,aAAa,CAAC,IAAI,EAClB,qBAAqB,CACtB,CAAC;IAEF,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAA,2BAAqB,EACzB,IAAI,EACJ,WAAW,EACX,CAAC,WAAW,CAAC,EACb,eAAe,CAChB,CAAC;QAEF,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,WAAW,EACX;iCAC2B,YAAY;4BACjB,YAAY,kCAAkC,aAAa;QAC/E,CACH,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAA,2BAAqB,EACzB,IAAI,EACJ,WAAW,EACX,CAAC,WAAW,CAAC,EACb,eAAe,CAChB,CAAC;QAEF,MAAM,cAAc,GAAG,MAAM,IAAA,iBAAW,EACtC,IAAI,EACJ,WAAW,EACX,WAAW,YAAY,eAAe,aAAa,MAAM,CAC1D,CAAC;QACF,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,WAAW,EACX,+EAA+E,YAAY,eAAe,aAAa,uEAAuE,CAC/L,CAAC;QACJ,CAAC;QAED,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,WAAW,EACX;+BACyB,YAAY;4BACf,YAAY;QAChC,CACH,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE,CAAC;QACjC,MAAM,IAAA,2BAAqB,EACzB,IAAI,EACJ,eAAe,EACf,CAAC,WAAW,CAAC,EACb,eAAe,CAChB,CAAC;QAEF,MAAM,mBAAmB,GAAG,MAAM,IAAA,iBAAW,EAC3C,IAAI,EACJ,eAAe,EACf,WAAW,YAAY,eAAe,aAAa,MAAM,CAC1D,CAAC;QACF,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACzB,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,eAAe,EACf,+EAA+E,YAAY,eAAe,aAAa,uEAAuE,CAC/L,CAAC;QACJ,CAAC;QAED,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,eAAe,EACf;+BACyB,YAAY;4BACf,YAAY;QAChC,CACH,CAAC;IACJ,CAAC;IAED,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,+CAAuC;KACxC,CAAC,CAAC;IACH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;AACnC,CAAC,CAAC;AAtHW,QAAA,8BAA8B,kCAsHzC;AAEF,kBAAe,sCAA8B,CAAC"}

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

@@ -0,0 +1,12 @@
+/**
+ * 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 interface TsRdbSmithyConnectionGeneratorSchema {
+  sourceProject: string;
+  targetProject: string;
+}

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

@@ -0,0 +1,18 @@
+{
+  "$schema": "https://json-schema.org/schema",
+  "$id": "ts#rdb#smithy-connection",
+  "title": "ts#rdb#smithy-connection",
+  "description": "Connect a Smithy backend to a ts#rdb project",
+  "type": "object",
+  "properties": {
+    "sourceProject": {
+      "type": "string",
+      "description": "The TypeScript project to connect from"
+    },
+    "targetProject": {
+      "type": "string",
+      "description": "The ts#rdb project to connect to"
+    }
+  },
+  "required": ["sourceProject", "targetProject"]
+}

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

@@ -0,0 +1,183 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`ts#rdb strands-agent-connection generator > should add rdb serve-local dependency to the agent prefixed serve-local 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "ts#strands-agent",
+        "name": "my-agent",
+        "path": "src/my-agent",
+        "protocol": "HTTP",
+      },
+    ],
+  },
+  "name": "my-service",
+  "root": "packages/my-service",
+  "targets": {
+    "my-agent-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`ts#rdb strands-agent-connection generator > should be idempotent for A2A agent.ts injection 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { Agent, tool } from '@strands-agents/sdk';
+import { z } from 'zod';
+
+const multiply = tool({
+  name: 'Multiply',
+  description: 'Multiply two numbers',
+  inputSchema: z.object({ a: z.number(), b: z.number() }),
+  callback: ({ a, b }) => a * b,
+});
+
+export const getAgent = async () => {
+  const db = await getDb();
+  db.$on('error', console.error);
+  return new Agent({
+    systemPrompt: 'You are a helpful assistant.',
+    tools: [multiply],
+  });
+};
+"
+`;
+
+exports[`ts#rdb strands-agent-connection generator > should be idempotent for agent.ts injection 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { Agent, tool } from '@strands-agents/sdk';
+import { z } from 'zod';
+
+const multiply = tool({
+  name: 'Multiply',
+  description: 'Multiply two numbers',
+  inputSchema: z.object({ a: z.number(), b: z.number() }),
+  callback: ({ a, b }) => a * b,
+});
+
+export const getAgent = async () => {
+  const db = await getDb();
+  db.$on('error', console.error);
+  return new Agent({
+    systemPrompt: 'You are a helpful assistant.',
+    tools: [multiply],
+  });
+};
+"
+`;
+
+exports[`ts#rdb strands-agent-connection generator > should fall back to agent-serve-local when no component name 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "ts#strands-agent",
+        "name": "agent",
+        "path": "src/agent",
+        "protocol": "HTTP",
+      },
+    ],
+  },
+  "name": "my-service",
+  "root": "packages/my-service",
+  "targets": {
+    "agent-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`ts#rdb strands-agent-connection generator > should inject db into agent.ts 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { Agent, tool } from '@strands-agents/sdk';
+import { z } from 'zod';
+
+const multiply = tool({
+  name: 'Multiply',
+  description: 'Multiply two numbers',
+  inputSchema: z.object({ a: z.number(), b: z.number() }),
+  callback: ({ a, b }) => a * b,
+});
+
+export const getAgent = async () => {
+  const db = await getDb();
+  db.$on('error', console.error);
+  return new Agent({
+    systemPrompt: 'You are a helpful assistant.',
+    tools: [multiply],
+  });
+};
+"
+`;
+
+exports[`ts#rdb strands-agent-connection generator > should inject db into agent.ts for A2A 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { Agent, tool } from '@strands-agents/sdk';
+import { z } from 'zod';
+
+const multiply = tool({
+  name: 'Multiply',
+  description: 'Multiply two numbers',
+  inputSchema: z.object({ a: z.number(), b: z.number() }),
+  callback: ({ a, b }) => a * b,
+});
+
+export const getAgent = async () => {
+  const db = await getDb();
+  db.$on('error', console.error);
+  return new Agent({
+    systemPrompt: 'You are a helpful assistant.',
+    tools: [multiply],
+  });
+};
+"
+`;
+
+exports[`ts#rdb strands-agent-connection generator > should support multiple rdb connections in agent.ts 1`] = `
+"import { getPrisma as getMysqlDb } from 'mysql-db';
+import { getPrisma as getPostgresDb } from 'postgres-db';
+import { Agent, tool } from '@strands-agents/sdk';
+import { z } from 'zod';
+
+const multiply = tool({
+  name: 'Multiply',
+  description: 'Multiply two numbers',
+  inputSchema: z.object({ a: z.number(), b: z.number() }),
+  callback: ({ a, b }) => a * b,
+});
+
+export const getAgent = async () => {
+  const mysqlDb = await getMysqlDb();
+  mysqlDb.$on('error', console.error);
+  const postgresDb = await getPostgresDb();
+  postgresDb.$on('error', console.error);
+  return new Agent({
+    systemPrompt: 'You are a helpful assistant.',
+    tools: [multiply],
+  });
+};
+"
+`;

package/src/ts/rdb/strands-agent-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 { Tree } from '@nx/devkit';
+import { TsRdbStrandsAgentConnectionGeneratorSchema } from './schema';
+import { NxGeneratorInfo } from '../../../utils/nx';
+export declare const TS_RDB_STRANDS_AGENT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
+export declare const tsRdbStrandsAgentConnectionGenerator: (tree: Tree, options: TsRdbStrandsAgentConnectionGeneratorSchema) => Promise<void>;
+export default tsRdbStrandsAgentConnectionGenerator;

package/src/ts/rdb/strands-agent-connection/generator.js

@@ -0,0 +1,65 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tsRdbStrandsAgentConnectionGenerator = exports.TS_RDB_STRANDS_AGENT_CONNECTION_GENERATOR_INFO = void 0;
+const tslib_1 = require("tslib");
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const devkit_1 = require("@nx/devkit");
+const nx_1 = require("../../../utils/nx");
+const metrics_1 = require("../../../utils/metrics");
+const format_1 = require("../../../utils/format");
+const names_1 = require("../../../utils/names");
+const lodash_camelcase_1 = tslib_1.__importDefault(require("lodash.camelcase"));
+const npm_scope_1 = require("../../../utils/npm-scope");
+const ast_1 = require("../../../utils/ast");
+exports.TS_RDB_STRANDS_AGENT_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
+const tsRdbStrandsAgentConnectionGenerator = async (tree, options) => {
+    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
+    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+    const agentName = options.sourceComponent?.name ?? 'agent';
+    const serveLocalTarget = `${agentName}-serve-local`;
+    if (sourceProject.targets?.[serveLocalTarget]) {
+        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, serveLocalTarget, {
+            projects: [targetProject.name],
+            target: 'serve-local',
+        });
+        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+    }
+    const rdbBaseName = targetProject.name.split('/').pop();
+    const rdbNameCamel = (0, lodash_camelcase_1.default)(rdbBaseName);
+    const rdbNamePascal = (0, names_1.pascalCase)(rdbBaseName);
+    const rdbPackageAlias = (0, npm_scope_1.toScopeAlias)(targetProject.name);
+    const getterAlias = `getPrisma as get${rdbNamePascal}`;
+    const agentSourceDir = (0, devkit_1.joinPathFragments)(sourceProject.root, 'src', agentName);
+    const agentPath = (0, devkit_1.joinPathFragments)(agentSourceDir, 'agent.ts');
+    if (tree.exists(agentPath)) {
+        await (0, ast_1.addDestructuredImport)(tree, agentPath, [getterAlias], rdbPackageAlias);
+        // Block body form: export const getAgent = async () => { ... }
+        await (0, ast_1.applyGritQL)(tree, agentPath, `\`export const getAgent = async () => { $body }\` => raw\`export const getAgent = async () => {
+  const ${rdbNameCamel} = await get${rdbNamePascal}();
+  $body
+}\` where { $body <: not some \`const ${rdbNameCamel} = await get${rdbNamePascal}()\` }`);
+        // Expression body form: export const getAgent = async () => expr
+        await (0, ast_1.applyGritQL)(tree, agentPath, `\`export const getAgent = async () => $expr\` => raw\`export const getAgent = async () => {
+  const ${rdbNameCamel} = await get${rdbNamePascal}();
+  return $expr;
+}\` where { $expr <: not \`{ $_ }\` }`);
+        // Add $on error handler after the db declaration.
+        // Done via string replacement because GritQL treats $on as a metavariable.
+        const dbDecl = `const ${rdbNameCamel} = await get${rdbNamePascal}();`;
+        const onCall = `${rdbNameCamel}.$on('error', console.error);`;
+        const content = tree.read(agentPath, 'utf-8');
+        if (content.includes(dbDecl) && !content.includes(`${rdbNameCamel}.$on`)) {
+            tree.write(agentPath, content.replace(dbDecl, `${dbDecl}\n  ${onCall}`));
+        }
+    }
+    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
+        exports.TS_RDB_STRANDS_AGENT_CONNECTION_GENERATOR_INFO,
+    ]);
+    await (0, format_1.formatFilesInSubtree)(tree);
+};
+exports.tsRdbStrandsAgentConnectionGenerator = tsRdbStrandsAgentConnectionGenerator;
+exports.default = exports.tsRdbStrandsAgentConnectionGenerator;
+//# sourceMappingURL=generator.js.map

package/src/ts/rdb/strands-agent-connection/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/ts/rdb/strands-agent-connection/generator.ts"],"names":[],"mappings":";;;;AAAA;;;GAGG;AACH,uCAIoB;AAEpB,0CAK2B;AAC3B,oDAAyE;AACzE,kDAA6D;AAC7D,gDAAkD;AAClD,gFAAyC;AACzC,wDAAwD;AACxD,4CAAwE;AAE3D,QAAA,8CAA8C,GACzD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,oCAAoC,GAAG,KAAK,EACvD,IAAU,EACV,OAAmD,EACpC,EAAE;IACjB,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,MAAM,SAAS,GAAG,OAAO,CAAC,eAAe,EAAE,IAAI,IAAI,OAAO,CAAC;IAC3D,MAAM,gBAAgB,GAAG,GAAG,SAAS,cAAc,CAAC;IAEpD,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC9C,IAAA,sCAAiC,EAAC,aAAa,EAAE,gBAAgB,EAAE;YACjE,QAAQ,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC;YAC9B,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,IAAA,mCAA0B,EAAC,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,WAAW,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IACxD,MAAM,YAAY,GAAG,IAAA,0BAAS,EAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,aAAa,GAAG,IAAA,kBAAU,EAAC,WAAW,CAAC,CAAC;IAC9C,MAAM,eAAe,GAAG,IAAA,wBAAY,EAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IACzD,MAAM,WAAW,GAAG,mBAAmB,aAAa,EAAE,CAAC;IAEvD,MAAM,cAAc,GAAG,IAAA,0BAAiB,EACtC,aAAa,CAAC,IAAI,EAClB,KAAK,EACL,SAAS,CACV,CAAC;IACF,MAAM,SAAS,GAAG,IAAA,0BAAiB,EAAC,cAAc,EAAE,UAAU,CAAC,CAAC;IAEhE,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAA,2BAAqB,EACzB,IAAI,EACJ,SAAS,EACT,CAAC,WAAW,CAAC,EACb,eAAe,CAChB,CAAC;QAEF,+DAA+D;QAC/D,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,SAAS,EACT;UACI,YAAY,eAAe,aAAa;;wCAEV,YAAY,eAAe,aAAa,QAAQ,CACnF,CAAC;QAEF,iEAAiE;QACjE,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,SAAS,EACT;UACI,YAAY,eAAe,aAAa;;sCAEZ,CACjC,CAAC;QAEF,kDAAkD;QAClD,2EAA2E;QAC3E,MAAM,MAAM,GAAG,SAAS,YAAY,eAAe,aAAa,KAAK,CAAC;QACtE,MAAM,MAAM,GAAG,GAAG,YAAY,+BAA+B,CAAC;QAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAE,CAAC;QAC/C,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,YAAY,MAAM,CAAC,EAAE,CAAC;YACzE,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,MAAM,OAAO,MAAM,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,sDAA8C;KAC/C,CAAC,CAAC;IACH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;AACnC,CAAC,CAAC;AA/EW,QAAA,oCAAoC,wCA+E/C;AAEF,kBAAe,4CAAoC,CAAC"}