"use client";

import React, { createContext, useContext, useState, useEffect, ReactNode } from "react";
import {
  login as apiLogin,
  register as apiRegister,
  getCurrentUser,
  getAccessToken,
  clearTokens,
  User,
} from "./api";
import { connectSocket, disconnectSocket, setTokenRefreshFn } from "./socket";

interface AuthContextType {
  user: User | null;
  loading: boolean;
  isAuthenticated: boolean;
  login: (email: string, password: string) => Promise<void>;
  register: (email: string, password: string, name?: string, workspaceName?: string) => Promise<void>;
  logout: () => void;
}

const AuthContext = createContext<AuthContextType | undefined>(undefined);

export function AuthProvider({ children }: { children: ReactNode }) {
  const [user, setUser] = useState<User | null>(null);
  const [loading, setLoading] = useState(true);

  useEffect(() => {
    const token = getAccessToken();
    if (token) {
      // Connect WebSocket on initial load if token exists
      connectSocket(token);
      getCurrentUser()
        .then((u) => setUser(u))
        .catch(() => {
          clearTokens();
          disconnectSocket();
          setUser(null);
        })
        .finally(() => setLoading(false));
    } else {
      setLoading(false);
    }
  }, []);

  // Register token refresh function for socket reconnects
  useEffect(() => {
    setTokenRefreshFn(async () => {
      const token = getAccessToken();
      if (!token) return null;
      return token;
    });
  }, []);

  const login = async (email: string, password: string) => {
    const { user } = await apiLogin(email, password);
    setUser(user);
    // Connect WebSocket after successful login
    const token = getAccessToken();
    if (token) {
      connectSocket(token);
    }
  };

  const register = async (email: string, password: string, name?: string, workspaceName?: string) => {
    const { user } = await apiRegister(email, password, name, workspaceName);
    setUser(user);
    // Connect WebSocket after successful registration
    const token = getAccessToken();
    if (token) {
      connectSocket(token);
    }
  };

  const logout = () => {
    disconnectSocket();
    clearTokens();
    setUser(null);
    if (typeof window !== "undefined") {
      window.location.href = "/login";
    }
  };

  return (
    <AuthContext.Provider
      value={{
        user,
        loading,
        isAuthenticated: !!user,
        login,
        register,
        logout,
      }}
    >
      {children}
    </AuthContext.Provider>
  );
}

export function useAuth() {
  const ctx = useContext(AuthContext);
  if (!ctx) throw new Error("useAuth must be used within AuthProvider");
  return ctx;
}