SurfSense/surfsense_web/hooks/use-chat-session-state.ts

"use client";

import { useShape } from "@electric-sql/react";
import type { ChatSessionState } from "@/contracts/types/chat-session-state.types";

const ELECTRIC_URL = process.env.NEXT_PUBLIC_ELECTRIC_URL || "http://localhost:5133";

/**
 * Hook to get live chat session state for collaboration.
 * Tracks which user the AI is currently responding to.
 */
export function useChatSessionState(threadId: number | null) {
	const { data, isLoading, isError, error } = useShape<ChatSessionState>({
		url: `${ELECTRIC_URL}/v1/shape`,
		params: {
			table: "chat_session_state",
			where: `thread_id = ${threadId}`,
		},
	});

	const sessionState = data?.[0] ?? null;

	return {
		sessionState,
		isAiResponding: !!sessionState?.ai_responding_to_user_id,
		respondingToUserId: sessionState?.ai_responding_to_user_id ?? null,
		loading: isLoading,
		error: isError ? error : null,
	};
}