@thrillee/aegischat 0.1.12 → 0.1.15

package/src/hooks/useAutoRead.ts

@@ -2,7 +2,7 @@
 // AegisChat React SDK - useAutoRead Hook
 // ============================================================================
 
-import { useCallback, useEffect, useState } from 'react';
+import { useCallback, useEffect, useRef } from 'react';
 import { channelsApi } from '../services/api';
 
 const SESSION_STORAGE_KEY = '@aegischat/activeChannel';
@@ -18,13 +18,11 @@ export interface UseAutoReadReturn {
 }
 
 export function useAutoRead(options: UseAutoReadOptions = {}): UseAutoReadReturn {
-  const [isFocused, setIsFocused] = useState(false);
+  const isFocusedRef = useRef(typeof document !== 'undefined' && document.hasFocus());
 
   useEffect(() => {
-    setIsFocused(typeof document !== 'undefined' && document.hasFocus());
-
-    const handleFocus = () => setIsFocused(true);
-    const handleBlur = () => setIsFocused(false);
+    const handleFocus = () => { isFocusedRef.current = true; };
+    const handleBlur = () => { isFocusedRef.current = false; };
 
     window.addEventListener('focus', handleFocus);
     window.addEventListener('blur', handleBlur);
@@ -36,17 +34,17 @@ export function useAutoRead(options: UseAutoReadOptions = {}): UseAutoReadReturn
   }, []);
 
   const markAsRead = useCallback(async (channelId: string) => {
-    if (!isFocused) return;
+    if (!isFocusedRef.current) return;
     try {
       await channelsApi.markAsRead(channelId);
       options.onMarkAsRead?.(channelId);
     } catch (error) {
       console.error('[AegisChat] useAutoRead: Failed to mark as read:', error);
     }
-  }, [isFocused, options.onMarkAsRead]);
+  }, [options.onMarkAsRead]);
 
   const markAllAsRead = useCallback(async () => {
-    if (!isFocused) return;
+    if (!isFocusedRef.current) return;
     try {
       const response = await channelsApi.list({});
       const channels = response.channels || [];
@@ -56,9 +54,9 @@ export function useAutoRead(options: UseAutoReadOptions = {}): UseAutoReadReturn
     } catch (error) {
       console.error('[AegisChat] useAutoRead: Failed to mark all as read:', error);
     }
-  }, [isFocused]);
+  }, []);
 
-  return { markAsRead, markAllAsRead, isFocused };
+  return { markAsRead, markAllAsRead, isFocused: isFocusedRef.current };
 }
 
 export default useAutoRead;

package/src/hooks/useChat.ts

@@ -84,6 +84,7 @@ export interface UseChatReturn {
   deleteFailedMessage: (tempId: string) => void;
   markAsRead: (channelId: string) => Promise<void>;
   setup: (options: UseChatOptions) => void;
+  updateChannel: (channelId: string, updates: Partial<ChannelListItem>) => void;
 }
 
 export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
@@ -101,8 +102,12 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
   const [session, setSession] = useState<ChatSession | null>(null);
   const [isConnected, setIsConnected] = useState(false);
   const [isConnecting, setIsConnecting] = useState(false);
+  const getStoredActiveChannel = (): string | null => {
+    if (typeof window === "undefined") return null;
+    return sessionStorage.getItem(SESSION_STORAGE_KEY);
+  };
   const [activeChannelId, setActiveChannelIdState] = useState<string | null>(
-    null,
+    getStoredActiveChannel,
   );
   const [channels, setChannels] = useState<ChannelListItem[]>([]);
   const [messages, setMessages] = useState<Message[]>([]);
@@ -126,9 +131,14 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
   const roleRef = useRef<string | undefined>(undefined);
   const clientIdRef = useRef<string | undefined>(undefined);
   const autoConnectRef = useRef(true);
-  const onMessageRef = useRef<(message: Message) => void | undefined>(undefined);
-  const onTypingRef = useRef<((channelId: string, user: TypingUser) => void) | undefined>(undefined);
-  const onConnectionChangeRef = useRef<((connected: boolean) => void) | undefined>(undefined);
+  const onMessageRef =
+    useRef<(message: Message) => void | undefined>(undefined);
+  const onTypingRef = useRef<
+    ((channelId: string, user: TypingUser) => void) | undefined
+  >(undefined);
+  const onConnectionChangeRef = useRef<
+    ((connected: boolean) => void) | undefined
+  >(undefined);
 
   useEffect(() => {
     activeChannelIdRef.current = activeChannelId;
@@ -140,6 +150,7 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
   }, []);
 
   const setActiveChannelId = useCallback((id: string | null) => {
+    activeChannelIdRef.current = id;
     setActiveChannelIdState(id);
     if (typeof window !== "undefined") {
       if (id) {
@@ -459,7 +470,9 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
 
     setIsLoadingChannels(true);
     try {
-      const response = await fetchFromComms<{ channels: ChannelListItem[] }>('/channels');
+      const response = await fetchFromComms<{ channels: ChannelListItem[] }>(
+        "/channels",
+      );
       setChannels(response.channels || []);
     } catch (error) {
       console.error("[AegisChat] Failed to fetch channels:", error);
@@ -503,7 +516,6 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
         if (response.oldest_id) {
           oldestMessageId.current = response.oldest_id;
         }
-
       } catch (error) {
         console.error("[AegisChat] Failed to load messages:", error);
         setMessages([]);
@@ -525,6 +537,15 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
     [fetchFromComms],
   );
 
+  const updateChannel = useCallback(
+    (channelId: string, updates: Partial<ChannelListItem>) => {
+      setChannels((prev) =>
+        prev.map((ch) => (ch.id === channelId ? { ...ch, ...updates } : ch)),
+      );
+    },
+    [],
+  );
+
   const loadMoreMessages = useCallback(async () => {
     if (!activeChannelId || !hasMoreMessages || isLoadingMessages) return;
 
@@ -961,9 +982,8 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
 
       if (!config) {
         const url = initialSession.api_url;
-        const normalizedUrl = url.includes('/api/v1') || url.includes('/v')
-          ? url
-          : `${url}/api/v1`;
+        const normalizedUrl =
+          url.includes("/api/v1") || url.includes("/v") ? url : `${url}/api/v1`;
         configureApiClient({
           baseUrl: normalizedUrl,
           getAccessToken: async () => sessionRef.current?.access_token || "",
@@ -1047,6 +1067,7 @@ export function useChat(options: Partial<UseChatOptions> = {}): UseChatReturn {
     retryMessage,
     deleteFailedMessage,
     markAsRead,
+    updateChannel,
     setup,
   };
 }