changes to file routes

src/file/file.controller.ts

@@ -3,56 +3,114 @@ import { FastifyReply, FastifyRequest } from "fastify";
 import { generateId } from "../utils/id";
 import {
   completeMultiPartUpload,
-  deleteFileS3,
   getFileUrlS3,
   getUploadUrl,
   getUploadUrlMultiPart,
-  uploadFileS3,
 } from "../utils/s3";
-import { createFile, getFile } from "./file.service";
-import { UploadMultiPartCompleteRequest } from "./file.schema";
+import {
+  createFile,
+  deleteFile,
+  getChildren,
+  getFile,
+  updateFile,
+  uploadDone,
+} from "./file.service";
+import {
+  CreateFileInput,
+  UpdateFileInput,
+  UploadMultiPartCompleteRequest,
+} from "./file.schema";
 
-export async function fileUploadHandler(
+export async function createFileHandler(
   req: FastifyRequest,
   res: FastifyReply
 ) {
-  const file = await req.file();
-
-  if (!file) return res.code(400).send({ error: "file not found in the body" });
+  const input = req.body as CreateFileInput;
 
   try {
-    const chunks = [];
-    for await (const chunk of file.file) {
-      // @ts-ignore
-      chunks.push(Buffer.from(chunk));
+    const file = await createFile(input, req.user);
+    const resObj = {
+      file,
+    };
+
+    if (file.mimeType != "folder") {
+      const signedUrl = await getUploadUrl(file.pid);
+      resObj["signedUrl"] = signedUrl;
     }
-
-    const fileData = Buffer.concat(chunks);
-    const key = generateId();
-    await uploadFileS3(key, fileData);
-
-    const fileRec = await createFile(
-      key,
-      file.filename,
-      file.mimetype,
-      req.user.tenantId
-    );
-
-    return res.code(201).send(fileRec);
+    return res.code(201).send(resObj);
   } catch (err) {
     return err;
   }
 }
 
-export async function fileUploadS3UrlHandler(
+export async function uploadDoneHandler(
   req: FastifyRequest,
   res: FastifyReply
 ) {
-  try {
-    const key = generateId();
-    const signedUrl = await getUploadUrl(key);
+  const { fileId } = req.params as { fileId: string };
 
-    return res.code(200).send({ key, signedUrl });
+  try {
+    const file = await uploadDone(fileId, req.user.tenantId);
+    return res.code(200).send();
+  } catch (err) {
+    return err;
+  }
+}
+
+export async function getFileHandler(req: FastifyRequest, res: FastifyReply) {
+  const { fileId } = req.params as { fileId: string };
+
+  try {
+    const file = await getFile(fileId, req.user.tenantId);
+    if (!file) return res.code(404).send({ error: "resource not found" });
+
+    return res.code(200).send(file);
+  } catch (err) {
+    return err;
+  }
+}
+
+export async function getChildrenHandler(
+  req: FastifyRequest,
+  res: FastifyReply
+) {
+  const { fileId } = req.params as { fileId: string };
+
+  try {
+    const files = await getChildren(fileId, req.user.tenantId);
+    return res.code(200).send(files);
+  } catch (err) {
+    return err;
+  }
+}
+
+export async function updateFileHandler(
+  req: FastifyRequest,
+  res: FastifyReply
+) {
+  const { fileId } = req.params as { fileId: string };
+  const input = req.body as UpdateFileInput;
+
+  try {
+    const updatedFile = await updateFile(fileId, input, req.user.tenantId);
+    if (!updatedFile)
+      return res.code(404).send({ error: "resource not found" });
+
+    return res.code(200).send(updatedFile);
+  } catch (err) {
+    return err;
+  }
+}
+
+export async function deleteFileHandler(
+  req: FastifyRequest,
+  res: FastifyReply
+) {
+  const { fileId } = req.params as { fileId: string };
+
+  try {
+    await deleteFile(fileId, req.user.tenantId);
+    return res.code(204).send();
   } catch (err) {
     return err;
   }
@@ -95,45 +153,17 @@ export async function fileDownloadHandler(
   res: FastifyReply
 ) {
   const { fileId } = req.params as { fileId: string };
-  const { direct } = req.query as { direct: boolean };
-
-  try {
-    let id: string;
-    let name: string | null;
-
-    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;
-  }
-}
-
-export async function deleteFileHandler(
-  req: FastifyRequest,
-  res: FastifyReply
-) {
-  const { fileId } = req.params as { fileId: string };
 
   try {
     const file = await getFile(fileId, req.user.tenantId);
     if (file === null)
       return res.code(404).send({ error: "resource not found" });
 
-    await deleteFileS3(file.pid);
-    await file.deleteOne();
-    return res.code(204).send();
+    if (file.mimeType == "folder")
+      return res.code(400).send({ error: "cannot download a folder" });
+
+    const signedUrl = await getFileUrlS3(file.pid, file.name);
+    return res.code(200).send({ url: signedUrl });
   } catch (err) {
     return err;
   }