changes to file api

src/file/file.controller.ts

@@ -1,8 +1,16 @@
 import { FastifyReply, FastifyRequest } from "fastify";
 
 import { generateId } from "../utils/id";
-import { deleteFileS3, getFileUrlS3, uploadFileS3 } from "../utils/s3";
+import {
+  completeMultiPartUpload,
+  deleteFileS3,
+  getFileUrlS3,
+  getUploadUrl,
+  getUploadUrlMultiPart,
+  uploadFileS3,
+} from "../utils/s3";
 import { createFile, getFile } from "./file.service";
+import { UploadMultiPartCompleteRequest } from "./file.schema";
 
 export async function fileUploadHandler(
   req: FastifyRequest,
@@ -15,6 +23,7 @@ export async function fileUploadHandler(
   try {
     const chunks = [];
     for await (const chunk of file.file) {
+      // @ts-ignore
       chunks.push(Buffer.from(chunk));
     }
 
@@ -35,18 +44,75 @@ export async function fileUploadHandler(
   }
 }
 
+export async function fileUploadS3UrlHandler(
+  req: FastifyRequest,
+  res: FastifyReply
+) {
+  try {
+    const key = generateId();
+    const signedUrl = await getUploadUrl(key);
+
+    return res.code(200).send({ key, signedUrl });
+  } catch (err) {
+    return err;
+  }
+}
+
+export async function fileUploadS3UrlMultiPartHandler(
+  req: FastifyRequest,
+  res: FastifyReply
+) {
+  try {
+    const { fileSize } = req.query as { fileSize: number };
+    if (!fileSize || fileSize == 0)
+      return res.code(400).send({ error: "invalid fileSize" });
+
+    const key = generateId();
+    const multiPartRes = await getUploadUrlMultiPart(key, fileSize);
+
+    return res.code(200).send({ key, ...multiPartRes });
+  } catch (err) {
+    return err;
+  }
+}
+
+export async function finishMulitPartUploadHandler(
+  req: FastifyRequest,
+  res: FastifyReply
+) {
+  const input = req.body as UploadMultiPartCompleteRequest;
+
+  try {
+    await completeMultiPartUpload(input.key, input.uploadId, input.parts);
+  } catch (err) {
+    return err;
+  }
+}
+
 export async function fileDownloadHandler(
   req: FastifyRequest,
   res: FastifyReply
 ) {
   const { fileId } = req.params as { fileId: string };
+  const { direct } = req.query as { direct: boolean };
 
   try {
-    const file = await getFile(fileId, req.user.tenantId);
-    if (file === null)
-      return res.code(404).send({ error: "resource not found" });
+    let id: string;
+    let name: string | null;
 
-    const signedUrl = await getFileUrlS3(file.pid, file.name);
+    if (!direct) {
+      const file = await getFile(fileId, req.user.tenantId);
+      if (file === null)
+        return res.code(404).send({ error: "resource not found" });
+
+      id = file.id;
+      name = file.name;
+    } else {
+      id = fileId;
+      name = null;
+    }
+
+    const signedUrl = await getFileUrlS3(id, name);
     return res.code(200).send({ url: signedUrl });
   } catch (err) {
     return err;

src/file/file.route.ts

@@ -3,6 +3,9 @@ import {
   deleteFileHandler,
   fileDownloadHandler,
   fileUploadHandler,
+  fileUploadS3UrlHandler,
+  fileUploadS3UrlMultiPartHandler,
+  finishMulitPartUploadHandler,
 } from "./file.controller";
 import { $file } from "./file.schema";
 
@@ -47,4 +50,42 @@ export async function fileRoutes(fastify: FastifyInstance) {
     },
     deleteFileHandler
   );
+
+  fastify.get(
+    "/",
+    {
+      config: { requiredClaims: ["file:upload"] },
+      preHandler: [fastify.authorize],
+    },
+    fileUploadS3UrlHandler
+  );
+
+  fastify.get(
+    "/multipart",
+    {
+      schema: {
+        querystring: {
+          type: "object",
+          properties: {
+            fileSize: { type: "number" },
+          },
+        },
+      },
+      config: { requiredClaims: ["file:upload"] },
+      preHandler: [fastify.authorize],
+    },
+    fileUploadS3UrlMultiPartHandler
+  );
+
+  fastify.post(
+    "/multipart/complete",
+    {
+      schema: {
+        body: $file("uploadMultipartCompleteRequest"),
+      },
+      config: { requiredClaims: ["file:upload"] },
+      preHandler: [fastify.authorize],
+    },
+    finishMulitPartUploadHandler
+  );
 }

src/file/file.schema.ts

@@ -31,6 +31,17 @@ const downloadFileResponse = z.object({
   url: z.string().url(),
 });
 
+const uploadMultipartCompleteRequest = z.object({
+  key: z.string(),
+  uploadId: z.string(),
+  parts: z.array(
+    z.object({
+      ETag: z.string(),
+      PartNumber: z.number(),
+    })
+  ),
+});
+
 export const files = z
   .object({
     pid: z.string().optional(),
@@ -58,10 +69,15 @@ export const files = z
     validateRecursive(data);
   });
 
+export type UploadMultiPartCompleteRequest = z.infer<
+  typeof uploadMultipartCompleteRequest
+>;
+
 export const { schemas: fileSchemas, $ref: $file } = buildJsonSchemas(
   {
     uploadFileResponse,
     downloadFileResponse,
+    uploadMultipartCompleteRequest,
   },
   { $id: "file" }
 );