chore: add bucket

Author: sobolk

packages/amplify-graphql-conversation-transformer/package.json

@@ -40,6 +40,8 @@
   },
   "devDependencies": {
     "@aws-amplify/graphql-transformer-test-utils": "1.0.17",
+    "@aws-sdk/client-s3": "^3.804.0",
+    "@aws-sdk/s3-request-presigner": "^3.804.0",
     "esbuild": "^0.25.0"
   },
   "peerDependencies": {

packages/amplify-graphql-conversation-transformer/src/resolvers/templates/invoke-attachment-lambda-resolver-fn.template.js

@@ -4,7 +4,7 @@ export function request(ctx) {
   const { args } = ctx;
 
   const payload = {
-    conversationId: args.conversationId,
+    conversationId: args.input.conversationId,
   };
 
   return {

packages/amplify-graphql-conversation-transformer/src/transformer-steps/attachment-lambda/handler.ts

@@ -1,5 +1,29 @@
-export const handler = async () => {
-  return {
-    url: 'https://foo.com'
+import { S3Client, PutObjectCommand } from '@aws-sdk/client-s3';
+import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
+
+/**
+ * TODO
+ */
+export const handler = async (event: {
+  conversationId: string;
+}): Promise<{
+  url: string;
+}> => {
+  console.log(JSON.stringify(event, null, 2));
+  console.log(process.env);
+
+  if (!process.env.S3_BUCKET_NAME) {
+    throw new Error('S3_BUCKET_NAME environment variable must be defined');
   }
-}
+
+  const bucketName = process.env.S3_BUCKET_NAME;
+
+  const client = new S3Client();
+  const attachmentKey = crypto.randomUUID();
+  const command = new PutObjectCommand({ Bucket: bucketName, Key: `${event.conversationId}/${attachmentKey}` });
+  const url = await getSignedUrl(client, command, { expiresIn: 3600 });
+
+  return {
+    url,
+  };
+};

packages/amplify-graphql-conversation-transformer/src/transformer-steps/conversation-resolver-generator.ts

@@ -233,15 +233,23 @@ export class ConversationResolverGenerator {
     functionStack: cdk.Stack,
     capitalizedFieldName: string,
   ): { functionDataSourceId: string; referencedFunction: IFunction } {
+    const attachmentBucket = new cdk.aws_s3.Bucket(functionStack, `${capitalizedFieldName}ConversationAttachmentBucket`, {
+      enforceSSL: true,
+    });
     const defaultConversationHandler = new cdk.aws_lambda_nodejs.NodejsFunction(
       functionStack,
       `${capitalizedFieldName}ConversationAttachmentHandler`,
       {
         runtime: cdk.aws_lambda.Runtime.NODEJS_20_X,
         entry: path.resolve(__dirname, 'attachment-lambda', 'handler.js'),
+        environment: {
+          S3_BUCKET_NAME: attachmentBucket.bucketName
+        }
       },
     );
 
+    attachmentBucket.grantReadWrite(defaultConversationHandler);
+
     const functionDataSourceId = FunctionResourceIDs.FunctionDataSourceID(`${capitalizedFieldName}ConversationAttachmentHandler`);
     const referencedFunction = defaultConversationHandler;