Add update method to MessageController and update routes

langx · Apr 4, 2024 · bcde3f3 · bcde3f3
1 parent ac5e3f2
commit bcde3f3
Show file tree

Hide file tree

Showing 2 changed files with 77 additions and 3 deletions.
diff --git a/src/controllers/message.controller.ts b/src/controllers/message.controller.ts
@@ -195,4 +195,77 @@ export default class MessageController {
       });
     }
   }
+
+  async update(req: Request, res: Response) {
+    try {
+      console.log("update messaage");
+      throwIfMissing(req.headers, ["x-appwrite-user-id", "x-appwrite-jwt"]);
+      if (!req.body || Object.keys(req.body).length === 0) {
+        console.log("Request body is empty.");
+        return res
+          .status(400)
+          .json({ ok: false, error: "Request body is empty." });
+      }
+      throwIfMissing(req.body, ["id", "data"]);
+
+      const sender: string = req.headers["x-appwrite-user-id"] as string;
+      const jwt: string = req.headers["x-appwrite-jwt"] as string;
+      // Set data to variables
+      const messageId: string = req.body.id;
+      const data: any = req.body.data;
+      console.log(data);
+
+      const disallowedFields = ["sender", "to", "roomId"];
+
+      for (const field of disallowedFields) {
+        if (data.hasOwnProperty(field)) {
+          console.log(`Disallowed field "${field}" found in request body.`);
+          return res
+            .status(400)
+            .json({ ok: false, error: `Field "${field}" is not allowed.` });
+        }
+      }
+
+      // Check JWT
+      const verifyUser = new Client()
+        .setEndpoint(env.APP_ENDPOINT)
+        .setProject(env.APP_PROJECT)
+        .setJWT(jwt);
+
+      const account = new Account(verifyUser);
+      const user = await account.get();
+
+      if (user.$id !== sender) {
+        return res.status(400).json({ ok: false, error: "jwt is invalid" });
+      }
+
+      const client = new Client()
+        .setEndpoint(env.APP_ENDPOINT)
+        .setProject(env.APP_PROJECT)
+        .setKey(env.API_KEY);
+
+      const database = new Databases(client);
+
+      // Update the message
+      let message = await database.updateDocument(
+        env.APP_DATABASE,
+        env.MESSAGES_COLLECTION,
+        messageId,
+        data
+      );
+
+      if (message?.$id) {
+        console.log("message updated");
+        res.status(200).json(message);
+      } else {
+        console.log("message not updated");
+        res.status(304).json({ ok: false, message: "Not Modified" });
+      }
+    } catch (err) {
+      res.status(500).json({
+        message: "Internal Server Error!",
+        err: err,
+      });
+    }
+  }
 }
diff --git a/src/routes/message.routes.ts b/src/routes/message.routes.ts
@@ -1,5 +1,5 @@
-import { Router } from 'express';
-import MessageController from '../controllers/message.controller';
+import { Router } from "express";
+import MessageController from "../controllers/message.controller";
 
 class MessageRoutes {
   router = Router();
@@ -10,7 +10,8 @@ class MessageRoutes {
   }
 
   intializeRoutes() {
-    this.router.post('/', this.controller.create);
+    this.router.post("/", this.controller.create);
+    this.router.patch("/", this.controller.update);
   }
 }