@aws/nx-plugin 0.113.0 → 0.114.0

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

@@ -0,0 +1,117 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`ts#rdb mcp-server-connection generator > should add rdb serve-local dependency to the mcp-server prefixed serve-local 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "ts#mcp-server",
+        "name": "my-mcp",
+        "path": "src/my-mcp",
+      },
+    ],
+  },
+  "name": "my-service",
+  "root": "packages/my-service",
+  "targets": {
+    "my-mcp-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`ts#rdb mcp-server-connection generator > should be idempotent for server.ts 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+
+export const createServer = async () => {
+  const db = await getDb();
+  db.$on('error', console.error);
+  const server = new McpServer({
+    name: 'my-service',
+    version: '1.0.0',
+  });
+
+  return server;
+};
+"
+`;
+
+exports[`ts#rdb mcp-server-connection generator > should fall back to mcp-server-serve-local when no component name 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "ts#mcp-server",
+        "name": "mcp-server",
+        "path": "src/mcp-server",
+      },
+    ],
+  },
+  "name": "my-service",
+  "root": "packages/my-service",
+  "targets": {
+    "mcp-server-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`ts#rdb mcp-server-connection generator > should inject db into server.ts 1`] = `
+"import { getPrisma as getDb } from 'db';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+
+export const createServer = async () => {
+  const db = await getDb();
+  db.$on('error', console.error);
+  const server = new McpServer({
+    name: 'my-service',
+    version: '1.0.0',
+  });
+
+  return server;
+};
+"
+`;
+
+exports[`ts#rdb mcp-server-connection generator > should support multiple rdb connections 1`] = `
+"import { getPrisma as getMysqlDb } from 'mysql-db';
+import { getPrisma as getPostgresDb } from 'postgres-db';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+
+export const createServer = async () => {
+  const mysqlDb = await getMysqlDb();
+  mysqlDb.$on('error', console.error);
+  const postgresDb = await getPostgresDb();
+  postgresDb.$on('error', console.error);
+  const server = new McpServer({
+    name: 'my-service',
+    version: '1.0.0',
+  });
+
+  return server;
+};
+"
+`;

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

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

@@ -0,0 +1,59 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tsRdbMcpServerConnectionGenerator = exports.TS_RDB_MCP_SERVER_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_MCP_SERVER_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
+const tsRdbMcpServerConnectionGenerator = async (tree, options) => {
+    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
+    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+    const mcpServerName = options.sourceComponent?.name ?? 'mcp-server';
+    const serveLocalTarget = `${mcpServerName}-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 serverPath = (0, devkit_1.joinPathFragments)(sourceProject.root, 'src', mcpServerName, 'server.ts');
+    // server.ts: inject db inside createServer body
+    if (tree.exists(serverPath)) {
+        await (0, ast_1.addDestructuredImport)(tree, serverPath, [getterAlias], rdbPackageAlias);
+        await (0, ast_1.applyGritQL)(tree, serverPath, `\`export const createServer = async () => { $body }\` => raw\`export const createServer = async () => {
+  const ${rdbNameCamel} = await get${rdbNamePascal}();
+  $body
+}\` where { $body <: not some \`const ${rdbNameCamel} = await get${rdbNamePascal}()\` }`);
+        // 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(serverPath, 'utf-8');
+        if (content.includes(dbDecl) && !content.includes(`${rdbNameCamel}.$on`)) {
+            tree.write(serverPath, content.replace(dbDecl, `${dbDecl}\n  ${onCall}`));
+        }
+    }
+    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
+        exports.TS_RDB_MCP_SERVER_CONNECTION_GENERATOR_INFO,
+    ]);
+    await (0, format_1.formatFilesInSubtree)(tree);
+};
+exports.tsRdbMcpServerConnectionGenerator = tsRdbMcpServerConnectionGenerator;
+exports.default = exports.tsRdbMcpServerConnectionGenerator;
+//# sourceMappingURL=generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/ts/rdb/mcp-server-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,2CAA2C,GACtD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,iCAAiC,GAAG,KAAK,EACpD,IAAU,EACV,OAAgD,EACjC,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,aAAa,GAAG,OAAO,CAAC,eAAe,EAAE,IAAI,IAAI,YAAY,CAAC;IACpE,MAAM,gBAAgB,GAAG,GAAG,aAAa,cAAc,CAAC;IAExD,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,UAAU,GAAG,IAAA,0BAAiB,EAClC,aAAa,CAAC,IAAI,EAClB,KAAK,EACL,aAAa,EACb,WAAW,CACZ,CAAC;IAEF,gDAAgD;IAChD,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAA,2BAAqB,EACzB,IAAI,EACJ,UAAU,EACV,CAAC,WAAW,CAAC,EACb,eAAe,CAChB,CAAC;QAEF,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,UAAU,EACV;UACI,YAAY,eAAe,aAAa;;wCAEV,YAAY,eAAe,aAAa,QAAQ,CACnF,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,UAAU,EAAE,OAAO,CAAE,CAAC;QAChD,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,YAAY,MAAM,CAAC,EAAE,CAAC;YACzE,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,MAAM,OAAO,MAAM,EAAE,CAAC,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,mDAA2C;KAC5C,CAAC,CAAC;IACH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;AACnC,CAAC,CAAC;AArEW,QAAA,iCAAiC,qCAqE5C;AAEF,kBAAe,yCAAiC,CAAC"}

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

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

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

@@ -0,0 +1,22 @@
+{
+  "$schema": "https://json-schema.org/schema",
+  "$id": "ts#rdb#mcp-server-connection",
+  "title": "ts#rdb#mcp-server-connection",
+  "description": "Connect a ts#mcp-server to a ts#rdb project",
+  "type": "object",
+  "properties": {
+    "sourceProject": {
+      "type": "string",
+      "description": "The project containing the ts#mcp-server to connect from"
+    },
+    "targetProject": {
+      "type": "string",
+      "description": "The ts#rdb project to connect to"
+    },
+    "sourceComponent": {
+      "type": "string",
+      "description": "The MCP server component name"
+    }
+  },
+  "required": ["sourceProject", "targetProject"]
+}

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"]
+}