import { useEffect } from 'react';
import { webSocketService } from '@/services/websocket';
import type { WebSocketEventType } from '@/services/types';

export function useWebSocket() {
  useEffect(() => {
    webSocketService.connect();
    return () => webSocketService.disconnect();
  }, []);

  const subscribe = <T = unknown>(
    eventType: WebSocketEventType,
    handler: (data: T) => void
  ): (() => void) => {
    return webSocketService.subscribe<T>(eventType, handler);
  };

  const isConnected = webSocketService.getConnectionStatus();

  return { subscribe, isConnected };
}