Implemented file storage handling for Corporate Transfer and Comission Transfer

src/pages/api/evaluate/interactiveSpeaking.ts

@@ -4,9 +4,9 @@ import {withIronSessionApiRoute} from "iron-session/next";
 import {sessionOptions} from "@/lib/session";
 import axios from "axios";
 import formidable from "formidable-serverless";
-import {getStorage, ref, uploadBytes} from "firebase/storage";
+import {ref, uploadBytes} from "firebase/storage";
 import fs from "fs";
-import {app} from "@/firebase";
+import {storage} from "@/firebase";
 
 export default withIronSessionApiRoute(handler, sessionOptions);
 
@@ -16,8 +16,6 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 		return;
 	}
 
-	const storage = getStorage(app);
-
 	const form = formidable({keepExtensions: true});
 	await form.parse(req, async (err: any, fields: any, files: any) => {
 		if (err) console.log(err);

src/pages/api/evaluate/speaking.ts

@@ -4,9 +4,9 @@ import {withIronSessionApiRoute} from "iron-session/next";
 import {sessionOptions} from "@/lib/session";
 import axios from "axios";
 import formidable from "formidable-serverless";
-import {getStorage, ref, uploadBytes} from "firebase/storage";
+import {ref, uploadBytes} from "firebase/storage";
 import fs from "fs";
-import {app} from "@/firebase";
+import {storage} from "@/firebase";
 
 export default withIronSessionApiRoute(handler, sessionOptions);
 
@@ -16,8 +16,6 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 		return;
 	}
 
-	const storage = getStorage(app);
-
 	const form = formidable({keepExtensions: true});
 	await form.parse(req, async (err: any, fields: any, files: any) => {
 		if (err) console.log(err);

src/pages/api/payments/files/[type]/[paymentId].ts

@@ -0,0 +1,194 @@
+// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
+import type { NextApiRequest, NextApiResponse } from "next";
+import { app, storage } from "@/firebase";
+import {
+  getFirestore,
+  getDoc,
+  doc,
+  updateDoc,
+  deleteField,
+} from "firebase/firestore";
+import { withIronSessionApiRoute } from "iron-session/next";
+import { sessionOptions } from "@/lib/session";
+import { FilesStorage } from "@/interfaces/storage.files";
+
+import { Payment } from "@/interfaces/paypal";
+import fs from "fs";
+import {
+  ref,
+  uploadBytes,
+  deleteObject,
+  getDownloadURL,
+} from "firebase/storage";
+import formidable from "formidable-serverless";
+
+const db = getFirestore(app);
+
+const getPaymentField = (type: FilesStorage) => {
+  switch (type) {
+    case "commission":
+      return "commissionTransfer";
+    case "corporate":
+      return "corporateTransfer";
+    default:
+      return null;
+  }
+};
+
+const handleDelete = async (
+  paymentId: string,
+  paymentField: "commissionTransfer" | "corporateTransfer"
+) => {
+  const paymentRef = doc(db, "payments", paymentId);
+  const paymentDoc = await getDoc(paymentRef);
+  const { [paymentField]: paymentFieldPath } = paymentDoc.data() as Payment;
+  // Create a reference to the file to delete
+  const documentRef = ref(storage, paymentFieldPath);
+  await deleteObject(documentRef);
+  await updateDoc(paymentRef, {
+    [paymentField]: deleteField(),
+  });
+};
+
+const handleUpload = async (
+  req: NextApiRequest,
+  paymentId: string,
+  paymentField: "commissionTransfer" | "corporateTransfer"
+) =>
+  new Promise((resolve, reject) => {
+    const form = formidable({ keepExtensions: true });
+    form.parse(req, async (err: any, fields: any, files: any) => {
+      if (err) {
+        reject(err);
+        return;
+      }
+      try {
+        const { file } = files;
+        const fileName = Date.now() + "-" + file.name;
+        const fileRef = ref(storage, fileName);
+
+        const binary = fs.readFileSync(file.path).buffer;
+        const snapshot = await uploadBytes(fileRef, binary);
+        fs.rmSync(file.path);
+
+        const paymentRef = doc(db, "payments", paymentId);
+
+        await updateDoc(paymentRef, {
+          [paymentField]: snapshot.ref.fullPath,
+        });
+        resolve(snapshot.ref.fullPath);
+      } catch (err) {
+        reject(err);
+      }
+    });
+  });
+
+export default withIronSessionApiRoute(handler, sessionOptions);
+async function handler(req: NextApiRequest, res: NextApiResponse) {
+  if (!req.session.user) {
+    res.status(401).json({ ok: false });
+    return;
+  }
+
+  if (req.method === "GET") return await get(req, res);
+  if (req.method === "POST") return await post(req, res);
+  if (req.method === "DELETE") return await del(req, res);
+  if (req.method === "PATCH") return await patch(req, res);
+
+  res.status(404).json(undefined);
+}
+
+async function get(req: NextApiRequest, res: NextApiResponse) {
+  const { type, paymentId } = req.query as {
+    type: FilesStorage;
+    paymentId: string;
+  };
+  const paymentField = getPaymentField(type);
+
+  if (paymentField === null) {
+    res.status(500).json({ error: "Failed to identify payment field" });
+    return;
+  }
+  const paymentRef = doc(db, "payments", paymentId);
+  const { [paymentField]: paymentFieldPath } = (
+    await getDoc(paymentRef)
+  ).data() as Payment;
+
+  // Create a reference to the file to delete
+  const documentRef = ref(storage, paymentFieldPath);
+  const url = await getDownloadURL(documentRef);
+  res.status(200).json({ url, name: documentRef.name });
+}
+
+async function post(req: NextApiRequest, res: NextApiResponse) {
+  const { type, paymentId } = req.query as {
+    type: FilesStorage;
+    paymentId: string;
+  };
+  const paymentField = getPaymentField(type);
+
+  if (paymentField === null) {
+    res.status(500).json({ error: "Failed to identify payment field" });
+    return;
+  }
+
+  try {
+    const ref = await handleUpload(req, paymentId, paymentField);
+    res.status(200).json({ ref });
+  } catch (error) {
+    res.status(500).json({ error });
+  }
+}
+
+async function del(req: NextApiRequest, res: NextApiResponse) {
+  const { type, paymentId } = req.query as {
+    type: FilesStorage;
+    paymentId: string;
+  };
+  const paymentField = getPaymentField(type);
+  if (paymentField === null) {
+    res.status(500).json({ error: "Failed to identify payment field" });
+    return;
+  }
+
+  try {
+    await handleDelete(paymentId, paymentField);
+    res.status(200).json({ ok: true });
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ error: "Failed to delete file" });
+  }
+}
+
+async function patch(req: NextApiRequest, res: NextApiResponse) {
+  const { type, paymentId } = req.query as {
+    type: FilesStorage;
+    paymentId: string;
+  };
+  const paymentField = getPaymentField(type);
+  if (paymentField === null) {
+    res.status(500).json({ error: "Failed to identify payment field" });
+    return;
+  }
+
+  try {
+    await handleDelete(paymentId, paymentField);
+  } catch (err) {
+    console.error(err);
+    res.status(500).json({ error: "Failed to delete file" });
+    return;
+  }
+
+  try {
+    const ref = await handleUpload(req, paymentId, paymentField);
+    res.status(200).json({ ref });
+  } catch (err) {
+    res.status(500).json({ error: "Failed to upload file" });
+  }
+}
+
+export const config = {
+  api: {
+    bodyParser: false,
+  },
+};

src/pages/api/speaking.ts

@@ -3,7 +3,7 @@ import type {NextApiRequest, NextApiResponse} from "next";
 import {withIronSessionApiRoute} from "iron-session/next";
 import {sessionOptions} from "@/lib/session";
 import {getDownloadURL, getStorage, ref} from "firebase/storage";
-import {app} from "@/firebase";
+import {app, storage} from "@/firebase";
 import axios from "axios";
 
 export default withIronSessionApiRoute(handler, sessionOptions);
@@ -14,7 +14,6 @@ async function handler(req: NextApiRequest, res: NextApiResponse) {
 		return;
 	}
 
-	const storage = getStorage(app);
 	const {path} = req.body as {path: string};
 
 	const pathReference = ref(storage, path);

src/pages/api/users/update.ts

@@ -1,6 +1,6 @@
 // Next.js API route support: https://nextjs.org/docs/api-routes/introduction
 import type {NextApiRequest, NextApiResponse} from "next";
-import {app} from "@/firebase";
+import {app, storage} from "@/firebase";
 import {getFirestore, collection, getDocs, getDoc, doc, setDoc} from "firebase/firestore";
 import {withIronSessionApiRoute} from "iron-session/next";
 import {sessionOptions} from "@/lib/session";
@@ -10,7 +10,6 @@ import {getAuth, signInWithEmailAndPassword, updateEmail, updatePassword} from "
 import {errorMessages} from "@/constants/errors";
 
 const db = getFirestore(app);
-const storage = getStorage(app);
 const auth = getAuth(app);
 
 export default withIronSessionApiRoute(handler, sessionOptions);