/** @format */

import Cookies from 'js-cookie';
import React, { useMemo } from 'react';
import SockJsClient from 'react-stomp';

import { API_URL } from '../../../utils/constant';

const planeTopics = ['/event/up'];

interface EventWebSocketProps {
  env: string;
  projectId: string;

  onMessage: (msg: Record<string, any>) => void;
}

const EventWebSocket = (props: EventWebSocketProps) => {
  const {
    projectId,
    env,

    onMessage,
  } = props;

  const accessToken = Cookies.get('access_token');

  function onReserveMessage(msg: any) {
    onMessage(msg);
  }

  const topics = useMemo(
    () => planeTopics.map(topic => `/project/${projectId}` + topic),
    [planeTopics, projectId],
  );

  return (
    <SockJsClient
      autoReconnect
      url={`${API_URL[env].baseWs}?access_token=${accessToken || ''}`}
      topics={topics}
      onMessage={onReserveMessage}
    />
  );
};

export default EventWebSocket;