BC-7851 - Listen for delete message (#92)

src/stores/setup.ts

@@ -6,7 +6,10 @@ import { UserPresence } from "../types/UserPresence";
 import { getConnectionOptions, getRoomId } from "../utils/connectionOptions";
 import { getEnvs } from "../utils/envConfig";
 import { getUserData } from "../utils/userData";
-import { handleRedirectIfNotValid } from "../utils/redirectUtils";
+import {
+  handleRedirectIfNotValid,
+  redirectToNotFoundErrorPage,
+} from "../utils/redirectUtils";
 import { clearErrorData } from "../utils/errorData";
 import { setDefaultState } from "../utils/userSettings";
 
@@ -34,6 +37,21 @@ const provider = new WebsocketProvider(
   },
 );
 
+provider.on("status", (event: { status: string }) => {
+  if (!provider.ws?.onmessage || event.status !== "connected") return;
+
+  const originalOnMessage = provider.ws.onmessage.bind(provider.ws);
+
+  provider.ws.onmessage = (messageEvent) => {
+    if (messageEvent.data === "action:delete") {
+      provider.disconnect();
+      redirectToNotFoundErrorPage();
+    } else {
+      originalOnMessage(messageEvent);
+    }
+  };
+});
+
 const room = new Room<UserPresence>(provider.awareness, {});
 const yShapes: Map<TDShape> = doc.getMap("shapes");
 const yBindings: Map<TDBinding> = doc.getMap("bindings");

src/utils/redirectUtils.ts

@@ -25,6 +25,11 @@ const redirectToErrorPage = () => {
   }
 };
 
+const redirectToNotFoundErrorPage = () => {
+  setErrorData(HttpStatusCode.NotFound, "error.404");
+  redirectToErrorPage();
+};
+
 const handleRedirectIfNotValid = (userResult: UserResult, envs?: Envs) => {
   if (userResult.statusCode === HttpStatusCode.Unauthorized) {
     redirectToLoginPage();
@@ -44,4 +49,9 @@ const handleRedirectIfNotValid = (userResult: UserResult, envs?: Envs) => {
   }
 };
 
-export { redirectToLoginPage, redirectToErrorPage, handleRedirectIfNotValid };
+export {
+  redirectToLoginPage,
+  redirectToErrorPage,
+  redirectToNotFoundErrorPage,
+  handleRedirectIfNotValid,
+};