// /redux/websocketMiddleware.js
const websocketMiddleware = () => {
  let socket;

  return ({ dispatch }) =>
    (next) =>
    (action) => {
      if (action.type === "WS_CONNECT") {
        socket = new WebSocket(action.payload.url);

        socket.onmessage = (event) => {
          const data = JSON.parse(event.data);
          dispatch({ type: "WS_MESSAGE_RECEIVED", payload: data });
        };

        socket.onclose = () => {
          dispatch({ type: "WS_DISCONNECTED" });
        };
      }

      if (action.type === "WS_DISCONNECT" && socket) {
        socket.close();
      }

      return next(action);
    };
};

export default websocketMiddleware;