hook.ts

import { useQueryClient } from '@tanstack/react-query';
import { authApi } from '@open-webui-react-native/shared/data-access/api';
import { authState$ } from '@open-webui-react-native/shared/data-access/auth';
import { cookieService } from '@open-webui-react-native/shared/data-access/cookie';
import { queryPersister } from '@open-webui-react-native/shared/data-access/persist-query-storage';

export const useLogout = (): { logout: () => Promise<void>; isLoading: boolean } => {
  const queryClient = useQueryClient();
  const { mutateAsync: signOut, isPending } = authApi.useSignOut();

  const logout = async (): Promise<void> => {
    try {
      await signOut();
    } catch {
      // Session may already be gone; still wipe client state and redirect.
    }
    authState$.logout();
    queryClient.removeQueries();
    cookieService.clearAll();
    await queryPersister.removeClient();
  };

  return { logout, isLoading: isPending };
};