From 0c1b85af1a7f5c94c99aca2eefe546cd7affd94b Mon Sep 17 00:00:00 2001
From: Alex Mickelson <alexmickelson96@gmail.com>
Date: Fri, 15 Nov 2024 08:42:36 -0700
Subject: [PATCH] sockets now prod, handles when not available

---
 nextjs/Dockerfile                             |  3 +-
 nextjs/package.json                           |  8 +-
 .../app/realtime/ClientCacheInvalidation.tsx  | 77 ++++++++++++-------
 3 files changed, 53 insertions(+), 35 deletions(-)

diff --git a/nextjs/Dockerfile b/nextjs/Dockerfile
index b828929..badf4b5 100644
--- a/nextjs/Dockerfile
+++ b/nextjs/Dockerfile
@@ -29,5 +29,4 @@ COPY --from=builder /app/public ./public
 
 RUN mkdir -p storage && rm -rf /app/storage/*
 
-# CMD [ "pnpm", "run", "start" ]
-CMD [ "pnpm", "run", "socketProd" ]
+CMD [ "pnpm", "run", "start" ]
diff --git a/nextjs/package.json b/nextjs/package.json
index f60586b..64886ea 100644
--- a/nextjs/package.json
+++ b/nextjs/package.json
@@ -3,11 +3,11 @@
   "version": "0.1.0",
   "private": true,
   "scripts": {
-    "dev": "next dev",
-    "socket": "node src/websocket.js",
+    "devNoSocket": "next dev",
+    "dev": "node src/websocket.js",
     "build": "next build",
-    "start": "next start",
-    "socketProd": "NODE_ENV=production node src/websocket.js",
+    "startNoSocket": "next start",
+    "start": "NODE_ENV=production node src/websocket.js",
     "lint": "tsc && next lint",
     "test": "vitest"
   },
diff --git a/nextjs/src/app/realtime/ClientCacheInvalidation.tsx b/nextjs/src/app/realtime/ClientCacheInvalidation.tsx
index 1f7a17f..68ffc8b 100644
--- a/nextjs/src/app/realtime/ClientCacheInvalidation.tsx
+++ b/nextjs/src/app/realtime/ClientCacheInvalidation.tsx
@@ -1,7 +1,7 @@
 "use client";
 
 import { trpc } from "@/services/trpc/utils";
-import React, { useEffect } from "react";
+import React, { useCallback, useEffect, useState } from "react";
 import { io, Socket } from "socket.io-client";
 
 interface ServerToClientEvents {
@@ -24,18 +24,48 @@ function removeFileExtension(fileName: string): string {
 }
 
 export function ClientCacheInvalidation() {
-  const utils = trpc.useUtils();
+  const invalidateCache = useFilePathInvalidation();
+  const [connectionAttempted, setConnectionAttempted] = useState(false);
   useEffect(() => {
+    if (!connectionAttempted) {
+      socket.connect();
+      setConnectionAttempted(true);
+    }
+
+    socket.on("connect", () => {
+      console.log("Socket connected successfully.");
+    });
+
     socket.on("message", (data) => {
       console.log("Received message:", data);
     });
 
-    socket.on("fileChanged", (filePath) => {
+    socket.on("fileChanged", invalidateCache);
+
+    socket.on("connect_error", (error) => {
+      console.error("Connection error:", error);
+      console.error("File system real time updates disabled");
+      socket.disconnect();
+    });
+
+    return () => {
+      socket.off("message");
+      socket.off("fileChanged");
+      socket.off("connect_error");
+    };
+  }, [connectionAttempted, invalidateCache]);
+
+  return <></>;
+}
+
+const useFilePathInvalidation = () => {
+  const utils = trpc.useUtils();
+  return useCallback(
+    (filePath: string) => {
       const [courseName, moduleOrLectures, itemType, itemFile] =
         filePath.split("/");
 
       const itemName = itemFile ? removeFileExtension(itemFile) : undefined;
-
       const allParts = [courseName, moduleOrLectures, itemType, itemName];
 
       if (moduleOrLectures === "settings.yml") {
@@ -91,28 +121,17 @@ export function ClientCacheInvalidation() {
         });
         return;
       }
-    });
-
-    socket.on("connect_error", (error) => {
-      console.error("Connection error:", error);
-    });
-
-    return () => {
-      socket.off("message");
-      socket.off("fileChanged");
-      socket.off("connect_error");
-    };
-  }, [
-    utils.assignment.getAllAssignments,
-    utils.assignment.getAssignment,
-    utils.lectures.getLectures,
-    utils.page.getAllPages,
-    utils.page.getPage,
-    utils.quiz.getAllQuizzes,
-    utils.quiz.getQuiz,
-    utils.settings.allCoursesSettings,
-    utils.settings.courseSettings,
-  ]);
-
-  return <></>;
-}
+    },
+    [
+      utils.assignment.getAllAssignments,
+      utils.assignment.getAssignment,
+      utils.lectures.getLectures,
+      utils.page.getAllPages,
+      utils.page.getPage,
+      utils.quiz.getAllQuizzes,
+      utils.quiz.getQuiz,
+      utils.settings.allCoursesSettings,
+      utils.settings.courseSettings,
+    ]
+  );
+};