apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-19 18:45:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

Centralize chat session state sync with single subscription via atom

Browse source
This commit is contained in:
CREDO23 2026-01-22 19:04:23 +02:00
parent 4cb835f19f
commit 39d434b00c
4 changed files with 48 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

35
surfsense_web/hooks/use-chat-session-state.ts
View file

@ -1,30 +1,39 @@
"use client";
import { useShape } from "@electric-sql/react";
import { useSetAtom } from "jotai";
import { useEffect } from "react";
import { chatSessionStateAtom } from "@/atoms/chat/chat-session-state.atom";
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.
* Syncs chat session state for a thread via Electric SQL.
* Call once per thread (in page.tsx). Updates global atom.
*/
export function useChatSessionState(threadId: number | null) {
const { data, isLoading, isError, error } = useShape<ChatSessionState>({
export function useChatSessionStateSync(threadId: number | null) {
const setSessionState = useSetAtom(chatSessionStateAtom);
const { data } = useShape<ChatSessionState>({
url: `${ELECTRIC_URL}/v1/shape`,
params: {
table: "chat_session_state",
where: `thread_id = ${threadId}`,
where: `thread_id = ${threadId ?? -1}`,
},
});
const sessionState = data?.[0] ?? null;
useEffect(() => {
if (!threadId) {
setSessionState(null);
return;
}
return {
sessionState,
isAiResponding: !!sessionState?.ai_responding_to_user_id,
respondingToUserId: sessionState?.ai_responding_to_user_id ?? null,
loading: isLoading,
error: isError ? error : null,
};
const row = data?.[0];
setSessionState({
threadId,
isAiResponding: !!row?.ai_responding_to_user_id,
respondingToUserId: row?.ai_responding_to_user_id ?? null,
});
}, [threadId, data, setSessionState]);
}