import { useEffect, useState, useRef } from "react";
import SocketIO from "socket.io-client";
import { auth } from "../../firebase/firebase.utils";

const useSocket = (bodyshop) => {
  const socketRef = useRef(null);
  const [clientId, setClientId] = useState(null);

  useEffect(() => {
    const unsubscribe = auth.onIdTokenChanged(async (user) => {
      if (user) {
        const newToken = await user.getIdToken();

        if (socketRef.current) {
          // Send new token to server
          socketRef.current.emit("update-token", newToken);
        } else if (bodyshop && bodyshop.id) {
          // Initialize the socket
          const endpoint = import.meta.env.PROD ? import.meta.env.VITE_APP_AXIOS_BASE_API_URL : "";

          const socketInstance = SocketIO(endpoint, {
            path: "/wss",
            withCredentials: true,
            auth: { token: newToken },
            reconnectionAttempts: Infinity,
            reconnectionDelay: 2000,
            reconnectionDelayMax: 10000
          });

          socketRef.current = socketInstance;

          const handleBodyshopMessage = (message) => {
            if (!import.meta.env.DEV) return;
            console.log(`Received message for bodyshop ${bodyshop.id}:`, message);
          };

          const handleConnect = () => {
            console.log("Socket connected:", socketInstance.id);
            socketInstance.emit("join-bodyshop-room", bodyshop.id);
            setClientId(socketInstance.id);
          };

          const handleReconnect = (attempt) => {
            console.log(`Socket reconnected after ${attempt} attempts`);
          };

          const handleConnectionError = (err) => {
            console.error("Socket connection error:", err);
          };

          const handleDisconnect = () => {
            console.log("Socket disconnected");
          };

          socketInstance.on("connect", handleConnect);
          socketInstance.on("reconnect", handleReconnect);
          socketInstance.on("connect_error", handleConnectionError);
          socketInstance.on("disconnect", handleDisconnect);
          socketInstance.on("bodyshop-message", handleBodyshopMessage);
        }
      } else {
        // User is not authenticated
        if (socketRef.current) {
          socketRef.current.disconnect();
          socketRef.current = null;
        }
      }
    });

    // Clean up the listener on unmount
    return () => {
      unsubscribe();
      if (socketRef.current) {
        socketRef.current.disconnect();
        socketRef.current = null;
      }
    };
  }, [bodyshop]);

  return { socket: socketRef.current, clientId };
};

export default useSocket;