Anything

src/modules/comic/ComicReader.tsx

@@ -72,10 +72,12 @@ export function ComicReader() {
     console.log('[Reader WS] Connecting to:', wsUrl);
     
     const ws = new WebSocket(wsUrl);
+    let isConnected = false;
     wsRef.current = ws;
 
     ws.onopen = () => {
-      console.log('[Reader WS] Connected');
+      console.log('[Reader WS] Connected and ready');
+      isConnected = true;
     };
 
     ws.onmessage = (event) => {
@@ -91,13 +93,23 @@ export function ComicReader() {
               hasPrev: !!message.data.prevChapter,
               hasNext: !!message.data.nextChapter,
               prevChapter: message.data.prevChapter,
-              nextChapter: message.data.nextChapter
+              nextChapter: message.data.nextChapter,
+              hasAllChapters: !!message.data.allChapters,
+              allChaptersCount: message.data.allChapters?.length || 0
             });
-            setData(message.data);
+            
+            // Preserve allChapters if not in response
+            const newData = {
+              ...message.data,
+              allChapters: message.data.allChapters || data?.allChapters || []
+            };
+            
+            setData(newData);
             setCurrentPage(1);
             scrollContainerRef.current?.scrollTo({ top: 0 });
           } else {
             setError(message.error || 'Failed to load chapter');
+            setTimeout(() => setError(''), 3000);
           }
         }
       } catch (err) {
@@ -108,6 +120,9 @@ export function ComicReader() {
     ws.onerror = (err) => {
       console.error('[Reader WS] Error:', err);
       setLoadingChapter(false);
+      if (!isConnected) {
+        setError('Failed to connect to server');
+      }
     };
 
     ws.onclose = () => {