Merge pull request #199 from ls1intum/feature/create-webapp-side-API-and-Websocket-managers

feat: create api and websocket managers

Author: egenerse

package-lock.json

@@ -6,6 +6,7 @@
     "dotenv": "16.4.7",
     "express": "5.1.0",
     "mongoose": "8.14.0",
+    "node-cron": "4.1.0",
     "ws": "8.18.1"
   },
   "scripts": {

standalone/server/package.json

@@ -0,0 +1,21 @@
+import cron from "node-cron"
+import Diagram from "./models/Diagram"
+
+export const startDiagramCleanupJob = (): void => {
+  // every day at 00:00 (midnight).
+  // Schedule a cron job to delete diagrams older than 60 days
+  cron.schedule("0 0 * * *", async () => {
+    try {
+      const cutoffDate = new Date()
+      cutoffDate.setDate(cutoffDate.getDate() - 60)
+
+      const result = await Diagram.deleteMany({
+        updatedAt: { $lt: cutoffDate },
+      })
+
+      console.log(`Deleted ${result.deletedCount} diagrams older than 30 days.`)
+    } catch (error) {
+      console.error("Error during scheduled diagram cleanup:", error)
+    }
+  })
+}

standalone/server/src/database/cleanupJob.ts

@@ -0,0 +1,11 @@
+import mongoose from "mongoose"
+
+export const connectToMongoDB = async (mongoURI: string): Promise<void> => {
+  try {
+    await mongoose.connect(mongoURI)
+    console.log("Connected to MongoDB")
+  } catch (error) {
+    console.error("MongoDB connection error:", error)
+    process.exit(1)
+  }
+}

standalone/server/src/database/connect.ts

@@ -1,14 +1,13 @@
-/* eslint-disable @typescript-eslint/no-explicit-any */
 import mongoose, { Document, Schema } from "mongoose"
 
 export interface IDiagram extends Document {
   _id: string // Custom ID from library
   version: string
   title: string
   type: string
-  nodes: any[] // JSON array for nodes
-  edges: any[] // JSON array for edges
-  assessments: Record<string, any> // Optional JSON object for assessments
+  nodes: Schema.Types.Mixed[] // JSON array for nodes
+  edges: Schema.Types.Mixed[] // JSON array for edges
+  assessments: Record<string, Schema.Types.Mixed> // Optional JSON object for assessments
 }
 
 const diagramSchema: Schema = new Schema(
@@ -23,6 +22,15 @@ const diagramSchema: Schema = new Schema(
   },
   {
     minimize: false, // Disable minimization of the document
+    versionKey: false,
+    timestamps: true, // Automatically manage createdAt and updatedAt fields
+    toJSON: {
+      virtuals: true, // expose `id`
+      transform: (_doc, ret) => {
+        ret.id = ret._id // copy _id to id
+        delete ret._id // optional: remove _id
+      },
+    },
   }
 )
 

standalone/server/src/models/Diagram.ts renamed to standalone/server/src/database/models/Diagram.ts

@@ -1,6 +1,6 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import { Router, Request, Response } from "express"
-import Diagram from "./models/Diagram"
+import Diagram from "./database/models/Diagram"
 
 const router = Router()
 

standalone/server/src/diagramRouter.ts

@@ -1,10 +1,11 @@
 // First load environment variables from .env file
 import "./loadEnvironment"
 import express from "express"
-import mongoose from "mongoose"
 import { configureMiddlewares } from "./middlewares"
 import diagramRouter from "./diagramRouter"
 import { startSocketServer } from "./relaySocketServer"
+import { connectToMongoDB } from "./database/connect"
+import { startDiagramCleanupJob } from "./database/cleanupJob"
 
 const PORT = process.env.PORT || 8000
 const serverHost = process.env.HOST || "localhost"
@@ -17,20 +18,11 @@ configureMiddlewares(app)
 // Mount routes
 app.use("/api", diagramRouter)
 
-console.log("Mongo URI:", mongoURI)
-mongoose
-  .connect(mongoURI)
-  .then(() => {
-    console.log("Connected to MongoDB")
-    // Start server
-
-    app.listen(PORT, () => {
-      console.log(`HTTP server running on http://${serverHost}:${PORT}`)
-    })
-    // Start WebSocket server
-    startSocketServer()
-  })
-  .catch((error) => {
-    console.error("MongoDB connection error:", error)
-    process.exit(1) // Exit if cannot connect to DB
+connectToMongoDB(mongoURI).then(() => {
+  app.listen(PORT, () => {
+    console.log(`HTTP server running on http://${serverHost}:${PORT}`)
   })
+
+  startSocketServer()
+  startDiagramCleanupJob()
+})

standalone/server/src/server.ts

@@ -5,7 +5,7 @@ import { toast } from "react-toastify"
 import { useEditorContext, useModalContext } from "@/contexts"
 import { useNavigate } from "react-router"
 import { DiagramView } from "@/types"
-import { backendURL } from "@/constants"
+import { DiagramAPIManager } from "@/services/DiagramAPIManager"
 
 export const ShareModal = () => {
   const { editor } = useEditorContext()
@@ -17,36 +17,26 @@ export const ShareModal = () => {
       toast.error("Editor instance is not available.")
       return
     }
-    const model = editor.model
 
-    await fetch(`${backendURL}/api/`, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify(model),
-    })
-      .then(async (res) => {
-        if (res.ok) {
-          const diagramID = (await res.json())._id
+    try {
+      const model = editor.model
+      const { id: diagramID } = await DiagramAPIManager.createDiagram(model)
 
-          const newurl = `${window.location.origin}/${diagramID}?view=${viewType}`
-          copyToClipboard(newurl)
-          navigate(`/${diagramID}?view=${viewType}`)
+      const newurl = `${window.location.origin}/${diagramID}?view=${viewType}`
+      copyToClipboard(newurl)
+      navigate(`/${diagramID}?view=${viewType}`)
 
-          toast.success(
-            `The link has been copied to your clipboard and can be shared to collaborate, simply by pasting the link. You can re-access the link by going to share menu.`,
-            {
-              autoClose: 10000,
-            }
-          )
-          closeModal()
-        } else {
-          throw new Error("Network response was not ok")
+      toast.success(
+        `The link has been copied to your clipboard and can be shared to collaborate, simply by pasting the link. You can re-access the link by going to share menu.`,
+        {
+          autoClose: 10000,
         }
-      })
-      .catch((err) => {
-        console.error("Error in setDiagram endpoint:", err)
-        toast.error("Error in setDiagram endpoint:", err)
-      })
+      )
+      closeModal()
+    } catch (err) {
+      console.error("Error creating diagram:", err)
+      toast.error("Could not create diagram.")
+    }
   }
 
   const copyToClipboard = (link: string) => {