Merge pull request #196 from HackGT/reese/hexathonFiles

Reese/hexathon files
HackGT · Feb 19, 2024 · 6a1f678 · 6a1f678
2 parents 1360732 + e6445bf
commit 6a1f678
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/services/files/src/common/storage.ts b/services/files/src/common/storage.ts
@@ -7,9 +7,17 @@ import { File } from "../models/file";
 
 const storage = new Storage();
 
-export const uploadFile = async (file: Express.Multer.File, bucketName: string) => {
+export const uploadFile = async (
+  file: Express.Multer.File,
+  bucketName: string,
+  folder?: string
+) => {
   const { originalname, buffer } = file;
-  const googleFileName = `${path.parse(originalname).name}_${Date.now()}`;
+  let folderName = "";
+  if (folder !== undefined) {
+    folderName = `${folder}/`;
+  }
+  const googleFileName = `${folderName}${path.parse(originalname).name}_${Date.now()}`;
   const blob = storage.bucket(bucketName).file(googleFileName);
 
   const blobStream = blob.createWriteStream({

diff --git a/services/files/src/routes/file.ts b/services/files/src/routes/file.ts
@@ -34,7 +34,7 @@ const multerMid = multer({
 
 export const fileRoutes = express.Router();
 
-fileRoutes.route("/upload").post(
+fileRoutes.route("/upload/:folder?").post(
   checkAbility("create", "File"),
   multerMid.single("file"),
   asyncHandler(async (req, res) => {
@@ -44,7 +44,8 @@ fileRoutes.route("/upload").post(
 
     const googleFileName = await uploadFile(
       req.file,
-      config.common.googleCloud.storageBuckets.default
+      config.common.googleCloud.storageBuckets.default,
+      req.params.folder
     );
 
     const file = await FileModel.create({