feat(trpc): generate a client for machine to machine api calls

Generate a client for machine to machine usage as part of trpc backend.

re #49

Author: cogwirrel

packages/nx-plugin/src/trpc/backend/__snapshots__/generator.spec.ts.snap

@@ -1,8 +1,59 @@
 // Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
+exports[`trpc backend generator > should generate backend and schema projects > apps/test-api/backend/src/client/index.ts 1`] = `
+"import {
+  createTRPCClient,
+  httpBatchLink,
+  httpLink,
+  HTTPBatchLinkOptions,
+  HTTPLinkOptions,
+  splitLink,
+} from '@trpc/client';
+
+import { AppRouter } from '../router.js';
+import { sigv4Fetch } from './sigv4.js';
+
+export interface TestApiClientConfig {
+  readonly url: string;
+}
+
+export const createTestApiClient = (config: TestApiClientConfig) => {
+  const linkOptions: HTTPLinkOptions<any> & HTTPBatchLinkOptions<any> = {
+    url: config.url,
+    fetch: sigv4Fetch,
+  };
+  return createTRPCClient<AppRouter>({
+    links: [
+      splitLink({
+        condition(op) {
+          return op.context.skipBatch === true;
+        },
+        true: httpLink(linkOptions),
+        false: httpBatchLink(linkOptions),
+      }),
+    ],
+  });
+};
+"
+`;
+
+exports[`trpc backend generator > should generate backend and schema projects > apps/test-api/backend/src/client/sigv4.ts 1`] = `
+"import { AwsClient } from 'aws4fetch';
+import { fromNodeProviderChain } from '@aws-sdk/credential-providers';
+
+const credentialProvider = fromNodeProviderChain();
+
+export const sigv4Fetch = (async (...args) => {
+  const client = new AwsClient(await credentialProvider());
+  return client.fetch(...args);
+}) satisfies AwsClient['fetch'];
+"
+`;
+
 exports[`trpc backend generator > should generate backend and schema projects > apps/test-api/backend/src/index.ts 1`] = `
 "export type { AppRouter } from './router.js';
 export type { Context } from './init.js';
+export * from './client/index.js';
 "
 `;
 

packages/nx-plugin/src/trpc/backend/files/backend/src/client/index.ts.template

@@ -0,0 +1,33 @@
+import {
+  createTRPCClient,
+  httpBatchLink,
+  httpLink,
+  HTTPBatchLinkOptions,
+  HTTPLinkOptions,
+  splitLink,
+} from '@trpc/client';
+
+import { AppRouter } from "../router.js";
+import { sigv4Fetch } from './sigv4.js';
+
+export interface <%= apiNameClassName %>ClientConfig {
+  readonly url: string;
+}
+
+export const create<%= apiNameClassName %>Client = (config: <%= apiNameClassName %>ClientConfig) => {
+  const linkOptions: HTTPLinkOptions<any> & HTTPBatchLinkOptions<any> = {
+    url: config.url,
+    fetch: sigv4Fetch,
+  };
+  return createTRPCClient<AppRouter>({
+    links: [
+      splitLink({
+        condition(op) {
+          return op.context.skipBatch === true;
+        },
+        true: httpLink(linkOptions),
+        false: httpBatchLink(linkOptions),
+      }),
+    ],
+  });
+};

packages/nx-plugin/src/trpc/backend/files/backend/src/client/sigv4.ts.template

@@ -0,0 +1,9 @@
+import { AwsClient } from 'aws4fetch';
+import { fromNodeProviderChain } from '@aws-sdk/credential-providers';
+
+const credentialProvider = fromNodeProviderChain();
+
+export const sigv4Fetch = (async (...args) => {
+  const client = new AwsClient(await credentialProvider());
+  return client.fetch(...args);
+}) satisfies AwsClient['fetch'];

packages/nx-plugin/src/trpc/backend/files/backend/src/index.ts.template

@@ -1,2 +1,3 @@
 export type { AppRouter } from './router.js';
 export type { Context } from './init.js';
+export * from './client/index.js';

packages/nx-plugin/src/trpc/backend/generator.spec.ts

@@ -2,7 +2,7 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
-import { getProjects, readProjectConfiguration, Tree } from '@nx/devkit';
+import { readProjectConfiguration, Tree } from '@nx/devkit';
 import { trpcBackendGenerator } from './generator';
 import {
   createTreeUsingTsSolutionSetup,

packages/nx-plugin/src/trpc/backend/generator.ts

@@ -211,6 +211,8 @@ export async function trpcBackendGenerator(
       '@aws-lambda-powertools/metrics',
       '@aws-lambda-powertools/tracer',
       '@trpc/server',
+      'aws4fetch',
+      '@aws-sdk/credential-providers',
     ]),
     withVersions(['@types/aws-lambda', 'tsx']),
   );

packages/nx-plugin/src/trpc/react/files/src/hooks/useSigV4.tsx.template

@@ -50,6 +50,6 @@ export const useSigV4 = () => {
       const awsClient = new AwsClient(credential);
       return awsClient.fetch(input, init);
     },
-    [cognitoProps, user?.id_token]
+    [cognitoProps, user?.id_token, user.profile.sub, withCachedCredentials]
   );
 };

packages/nx-plugin/src/utils/versions.ts

@@ -6,6 +6,7 @@ export const VERSIONS = {
   '@cdklabs/cdk-validator-cfnguard': '^0.0.60',
   '@aws-cdk/aws-cognito-identitypool-alpha': '^2.166.0-alpha.0',
   '@aws-sdk/client-cognito-identity': '^3.721.0',
+  '@aws-sdk/credential-providers': '^3.721.0',
   '@aws-sdk/credential-provider-cognito-identity': '^3.721.0',
   '@aws-lambda-powertools/logger': '^2.11.0',
   '@aws-lambda-powertools/metrics': '^2.11.0',