Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/openstatusHQ/openstatus. 🫖 Status page with uptime monitoring & API monitoring as code 🫖 openstatus.dev
bun drizzle-orm monitoring monitoring-as-code nextjs observability on-call open-source shadcn-ui status-page statuspage synthetic-monitoring tinybird turso uptime uptime-checker uptime-monitor
Something went wrong. Try again.
4.0 kB · 124 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125"use client";
import { useChat } from "@ai-sdk/react";import { type ChatStoredMessage, chatStoredMessagesSchema,} from "@openstatus/db/src/schema/chat_sessions/messages";import { skipToken, useQuery, useQueryClient } from "@tanstack/react-query";import { DefaultChatTransport, type UIMessage, lastAssistantMessageIsCompleteWithApprovalResponses,} from "ai";import { useRouter } from "next/navigation";import { useEffect, useMemo, useRef } from "react";
import { useTRPC } from "@/lib/trpc/client";
// `storedMessageSchema` is a structural subset of `UIMessage` — the SDK reads `.id`, `.role`, `.parts`.// `safeParse` surfaces drift as a console warning instead of a silent render failure.function asUIMessages( rows: ChatStoredMessage[] | undefined,): UIMessage[] | undefined { if (rows === undefined) return undefined; const parsed = chatStoredMessagesSchema.safeParse(rows); if (!parsed.success) { console.warn( "chat: stored messages failed schema validation", parsed.error.issues, ); } return rows as unknown as UIMessage[];}
export function useChatSession({ sessionId }: { sessionId?: number }) { const trpc = useTRPC(); const queryClient = useQueryClient(); const router = useRouter();
const activeSession = useQuery( trpc.chatSession.get.queryOptions( sessionId !== undefined ? { sessionId } : skipToken, ), );
// Ref so the transport closure sees mid-stream id changes without recreating `useChat`. const sessionIdRef = useRef<number | undefined>(sessionId); useEffect(() => { sessionIdRef.current = sessionId; }, [sessionId]);
const transport = useMemo( () => new DefaultChatTransport({ api: "/api/chat", prepareSendMessagesRequest({ messages, body }) { return { body: { ...body, messages, sessionId: sessionIdRef.current }, }; }, }), [], );
const chat = useChat({ transport, id: sessionId !== undefined ? `chat-${sessionId}` : "chat-new", // Synchronous seed from the prefetched cache — without it the first render // is empty and the cache-miss effect below produces a visible flash. messages: sessionId !== undefined ? asUIMessages(activeSession.data?.messages) : undefined, sendAutomaticallyWhen: lastAssistantMessageIsCompleteWithApprovalResponses, onFinish: () => { queryClient.invalidateQueries({ queryKey: trpc.chatSession.list.queryKey(), }); const id = sessionIdRef.current; if (id == null) return; queryClient.invalidateQueries({ queryKey: trpc.chatSession.get.queryKey({ sessionId: id }), }); if (sessionId === undefined) { router.replace(`/chat/${id}`); } }, });
const { messages, setMessages, stop } = chat;
// Cache-miss fallback for soft nav (no prefetch). No-op when the seed above hit. useEffect(() => { if (sessionId === undefined) return; if (!activeSession.data?.messages) return; if (messages.length > 0) return; const next = asUIMessages(activeSession.data.messages); if (next) setMessages(next); }, [sessionId, activeSession.data, messages.length, setMessages]);
// Abort in-flight stream on unmount — otherwise `onFinish` still writes to the DB after navigation. useEffect( () => () => { void stop(); }, [stop], );
// Server stamps the new id on the first message's metadata; mirror it so a // follow-up message sent before `onFinish` navigates still carries the id. useEffect(() => { if (sessionId !== undefined) return; if (sessionIdRef.current !== undefined) return; for (const m of messages) { const meta = (m as { metadata?: { sessionId?: number } }).metadata; if (meta?.sessionId != null) { sessionIdRef.current = meta.sessionId; return; } } }, [messages, sessionId]);
return chat;}