admin-dashboard-web/src/utils.js

import { createContext, useEffect, useRef, useState } from "react";
import { useNavigate } from "react-router-dom";

/**
 * constants
 */
export const Constants = {
  COLORS: {
    PRIMARY: "#e67e22",
    SECONDARY: "#9b59b6",
    ICON_INFO: "#08c",
  },
  TEXT_EMPTY_PLACEHOLDER: "-/-",
  API_ADDRESS: "http://localhost:8080/v1",
  WS_ADDRESS: "ws://localhost:8080/ws",
  ROUTE_PATHS: {
    GROUP_TASKS: "/group-tasks",
    GROUP_TASKS_VIEW: "/group-tasks/",
  },
  GROUP_TASKS_STATUS: {
    FINISHED: 1,
    RUNNING: 2,
    CANCELED: 3,
    FAILED: 4,
    INPUT_REQUIRED: 5,
  },
  GLOBALS: {
    MIN_USERNAME_LENGTH: 2,
    MAX_USERNAME_LENGTH: 20,
    MIN_PASSWORD_LENGTH: 6,
    MAX_PASSWORD_LENGTH: 64,
  },
};

/**
 * user session
 */
export function UseUserSession() {
  const getUserSession = () => {
    return JSON.parse(localStorage.getItem("session"));
  };

  const [userSession, setUserSession] = useState(getUserSession());

  const saveUserSession = (session) => {
    setUserSession(session);

    if (session === undefined) {
      localStorage.removeItem("session");
    } else {
      localStorage.setItem("session", JSON.stringify(session));
    }
  };

  return {
    setUserSession: saveUserSession,
    userSession,
  };
}

/**
 * websocket
 */
let l = "loading...";

let webSocketContextPreview = {
  User: {
    Username: l,
    Email: l,
    Sessions: [],
  },
  ConnectionBadgeStatus: "error",
  ConnectedWebSocketUsersCount: 0,
  CategoryGroups: [],
  GroupTasks: [],
  GroupTasksSteps: [],
};

export const WebSocketContext = createContext(webSocketContextPreview);

// commands received from the backend server
const ReceivedMessagesCommands = {
  InitUserSocketConnection: 1,
  UpdateConnectedUsers: 2,
  NewGroupTaskStarted: 3,
  NewGroupTaskStep: 4,
  UpdateGroupTaskStep: 5,
  UpdateGroupTask: 6,
  ReloadingGroupTasks: 7,
  GroupTasksReloaded: 8,
  UpdateUserSessions: 9,
};

// commands sent to the backend server
export const SentMessagesCommands = {
  StartGroupTasks: 1,
  TaskFailedTryAgainRunTaskStep: 2,
  TaskContinueTaskStep: 3,
  ReloadGroupTasks: 4,
};

export function WebSocketProvider({
  children,
  userSession,
  setUserSession,
  notificationApi,
}) {
  const [isReady, setIsReady] = useState(false);
  const [connectionBadgeStatus, setConnectionBadgeStatus] = useState(
    webSocketContextPreview.ConnectionBadgeStatus
  );
  const [connectedWebSocketUsersCount, setConnectedWebSocketUsersCount] =
    useState(0);
  const [user, setUser] = useState(webSocketContextPreview.User);
  const [categoryGroups, setCategoryGroups] = useState(
    webSocketContextPreview.CategoryGroups
  );
  // these are all group tasks that are then filtered and displayed in the respective tables for the category
  const [groupTasks, setGroupTasks] = useState([]);
  const [groupTasksSteps, setGroupTasksSteps] = useState([]);
  const navigate = useNavigate();
  const StartGroupTasksOpenModalRememberIdRef = useRef(null);

  let socket = null;
  const ws = useRef(null);

  const connect = () => {
    socket = new WebSocket(Constants.WS_ADDRESS + "?auth=" + userSession);

    ws.current = socket;

    socket.onopen = () => {
      console.log("connected");
      setConnectionBadgeStatus("success");
      setIsReady(true);
    };

    socket.onmessage = (event) => {
      const data = JSON.parse(event.data);
      console.log("received message", data);

      const cmd = data.Cmd;
      const body = data.Body;

      switch (cmd) {
        case ReceivedMessagesCommands.InitUserSocketConnection:
          setUser(body.User);
          setCategoryGroups(body.CategoryGroups);
          setGroupTasks(body.GroupTasks);
          setGroupTasksSteps(body.GroupTasksSteps);
          break;
        case ReceivedMessagesCommands.UpdateConnectedUsers:
          setConnectedWebSocketUsersCount(body);
          break;
        case ReceivedMessagesCommands.NewGroupTaskStarted:
          setGroupTasks((arr) => [...arr, body]);

          if (
            body.RememberId === StartGroupTasksOpenModalRememberIdRef.current
          ) {
            navigate(`${Constants.ROUTE_PATHS.GROUP_TASKS}/${body.Id}`);
          }
          break;
        case ReceivedMessagesCommands.NewGroupTaskStep:
          setGroupTasksSteps((arr) => [...arr, body]);
          break;
        case ReceivedMessagesCommands.UpdateGroupTaskStep:
          setGroupTasksSteps((arr) => {
            const newArr = [...arr];

            newArr[arr.findIndex((arr1) => arr1.Id === body.Id)] = body;

            return newArr;
          });
          break;
        case ReceivedMessagesCommands.UpdateGroupTask:
          setGroupTasks((arr) => {
            const newArr = [...arr];

            newArr[arr.findIndex((arr1) => arr1.Id === body.Id)] = body;

            return newArr;
          });
          break;
        case ReceivedMessagesCommands.ReloadingGroupTasks:
          notificationApi["warning"]({
            message: `Group ${body} is reloading`,
            duration: 2,
          });
          break;
        case ReceivedMessagesCommands.GroupTasksReloaded:
          setCategoryGroups((arr) => {
            const newArr = [...arr];

            newArr[
              arr.findIndex((arr1) => arr1.category === body.Category)
            ].groups = body.UpdatedGroups;

            return newArr;
          });

          notificationApi["info"]({
            message: `Group ${body.Category} reloaded`,
            duration: 2,
          });
          break;
        case ReceivedMessagesCommands.UpdateUserSessions:
          setUser((arr) => ({ ...arr, Sessions: body }));
          break;
        case ReceivedMessagesCommands.default:
          console.error("unknown command", cmd);
          break;
      }
    };

    socket.onclose = (event) => {
      setIsReady(false);
      setConnectionBadgeStatus("error");
      console.log("closed", event);

      // custom code defined by the backend server
      if (event.code === 4001 || event.code === 4002) {
        //Unauthorized || SessionClosed

        setUserSession();
        window.location.href = "/";
        return;
      }

      if (event.reason.code === 1005) return;

      console.log("reconnecting...");

      setTimeout(() => connect(), 1000);
    };
  };

  useEffect(() => {
    connect();

    return () => socket.close();
  }, []);

  const SendSocketMessage = (cmd, body) => {
    if (isReady) {
      ws.current.send(JSON.stringify({ Cmd: cmd, Body: body }));
    } else {
      console.log("websocket not ready");
    }
  };

  return (
    <WebSocketContext.Provider
      value={{
        ConnectionBadgeStatus: connectionBadgeStatus,
        ConnectedWebSocketUsersCount: connectedWebSocketUsersCount,
        CategoryGroups: categoryGroups,
        GroupTasks: groupTasks,
        User: user,
        SendSocketMessage: SendSocketMessage,
        GroupTasksSteps: groupTasksSteps,
        setGroupTasksSteps: setGroupTasksSteps,
        StartGroupTasksOpenModalRememberIdRef:
          StartGroupTasksOpenModalRememberIdRef,
      }}
    >
      {children}
    </WebSocketContext.Provider>
  );
}

export function FormatDatetime(datetime) {
  if (datetime === "0001-01-01T00:00:00Z") {
    return Constants.TEXT_EMPTY_PLACEHOLDER;
  }

  return new Date(datetime).toLocaleString("de-DE");
}

export function GetDuration(startTime, endTime) {
  if (endTime === "0001-01-01T00:00:00Z") {
    return Constants.TEXT_EMPTY_PLACEHOLDER;
  }

  const diff = Math.abs(new Date(startTime) - new Date(endTime));

  if (diff === 0) {
    return "0ms";
  }

  const milliseconds = diff % 1000;
  const seconds = Math.floor(diff / 1000);
  const minutes = Math.floor(seconds / 60);
  const hours = Math.floor(minutes / 60);
  const days = Math.floor(hours / 24);

  let result = "";

  if (days > 0) {
    result += days + "d ";
  }

  if (hours > 0) {
    result += (hours % 24) + "h ";
  }

  if (minutes > 0) {
    result += (minutes % 60) + "m ";
  }

  if (seconds > 0) {
    result += (seconds % 60) + "s ";
  }

  if (milliseconds > 0) {
    result += milliseconds + "ms";
  }

  return result.trim();
}