changes to file api

src/utils/s3.ts

@@ -3,10 +3,14 @@ import {
   PutObjectCommand,
   GetObjectCommand,
   DeleteObjectCommand,
+  CreateMultipartUploadCommand,
+  UploadPartCommand,
+  CompleteMultipartUploadCommand,
 } from "@aws-sdk/client-s3";
 import { getSignedUrl } from "@aws-sdk/s3-request-presigner";
 
 const BUCKET = process.env.BUCKET || "";
+const CHUNK_SIZE = parseInt(process.env.CHUNK_SIZE || "10000000");
 
 const client = new S3Client({
   region: process.env.REGION || "",
@@ -26,11 +30,69 @@ export async function uploadFileS3(key: string, body: Buffer) {
   const response = await client.send(command);
 }
 
-export async function getFileUrlS3(key: string, name: string) {
+export async function getUploadUrl(key: string) {
+  const command = new PutObjectCommand({
+    Bucket: BUCKET,
+    Key: key,
+  });
+
+  return await getSignedUrl(client, command, { expiresIn: 300 });
+}
+
+export async function getUploadUrlMultiPart(key: string, fileSize: number) {
+  const command = new CreateMultipartUploadCommand({
+    Bucket: BUCKET,
+    Key: key,
+  });
+
+  const res = await client.send(command);
+  const uploadId = res.UploadId;
+
+  const numberOfParts = Math.ceil(fileSize / CHUNK_SIZE);
+
+  let presignedUrls: string[] = [];
+
+  for (let i = 0; i < numberOfParts; i++) {
+    const presignedUrl = await getSignedUrl(
+      client,
+      new UploadPartCommand({
+        Bucket: BUCKET,
+        Key: key,
+        UploadId: uploadId,
+        PartNumber: i + 1,
+      }),
+      {}
+    );
+
+    presignedUrls.push(presignedUrl);
+  }
+
+  return { chunkSize: CHUNK_SIZE, uploadId, presignedUrls };
+}
+
+export async function completeMultiPartUpload(
+  key: string,
+  uploadId: string,
+  parts: { ETag: string; PartNumber: number }[]
+) {
+  const command = new CompleteMultipartUploadCommand({
+    Key: key,
+    Bucket: BUCKET,
+    UploadId: uploadId,
+    MultipartUpload: {
+      Parts: parts,
+    },
+  });
+
+  await client.send(command);
+}
+
+export async function getFileUrlS3(key: string, name: string | null) {
   const command = new GetObjectCommand({
     Bucket: BUCKET,
     Key: key,
-    ResponseContentDisposition: `attachment; filename=${name}`,
+    ResponseContentDisposition:
+      name !== null ? `attachment; filename=${name}` : undefined,
   });
 
   return await getSignedUrl(client, command, { expiresIn: 300 });