Anything

src/modules/comic/ComicReader.tsx

@@ -46,10 +46,14 @@ export function ComicReader() {
   const password = location.state?.password;
   const originalSessionId = location.state?.sessionId;
 
+  // WebSocket setup
   useEffect(() => {
     if (!password || !originalSessionId) return;
 
-    const wsUrl = import.meta.env.VITE_WS_URL || 'wss://ditzzy-yuki.hf.space/ws';
+    const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
+    const wsUrl = `${protocol}//${window.location.host}/ws`;
+    console.log('[Reader WS] Connecting to:', wsUrl);
+    
     const ws = new WebSocket(wsUrl);
     wsRef.current = ws;
 
@@ -91,10 +95,13 @@ export function ComicReader() {
 
     ws.onclose = () => {
       console.log('[Reader WS] Disconnected');
+      wsRef.current = null;
     };
 
     return () => {
-      ws.close();
+      if (ws.readyState === WebSocket.OPEN) {
+        ws.close();
+      }
     };
   }, [password, originalSessionId]);
 
@@ -214,26 +221,47 @@ export function ComicReader() {
   };
   
   const handleChapterNavigation = (chapterSlug: string) => {
-    if (!wsRef.current || wsRef.current.readyState !== WebSocket.OPEN) {
-      setError('Connection lost. Please refresh the page.');
-      return;
-    }
+    console.log('[Reader] Attempting chapter navigation to:', chapterSlug);
+    console.log('[Reader] WebSocket state:', wsRef.current?.readyState);
+    console.log('[Reader] Has password:', !!password);
+    console.log('[Reader] Has sessionId:', !!originalSessionId);
 
     if (!password || !originalSessionId) {
       setError('Missing authentication data');
       return;
     }
 
+    if (!wsRef.current || wsRef.current.readyState !== WebSocket.OPEN) {
+      if (wsRef.current?.readyState === WebSocket.CONNECTING) {
+        console.log('[Reader] WebSocket is connecting, waiting...');
+        setTimeout(() => {
+          if (wsRef.current?.readyState === WebSocket.OPEN) {
+            handleChapterNavigation(chapterSlug);
+          } else {
+            setError('Connection not ready. Please try again.');
+          }
+        }, 1000);
+        return;
+      }
+      
+      setError('Connection lost. Please refresh the page.');
+      console.error('[Reader] WebSocket not connected. State:', wsRef.current?.readyState);
+      return;
+    }
+
     setLoadingChapter(true);
     setShowMenu(false);
 
-    wsRef.current.send(JSON.stringify({
+    const requestData = {
       type: 'request_chapter_data',
       sessionId: originalSessionId,
       password: password,
       chapterSlug: chapterSlug,
       allChapters: data?.allChapters || []
-    }));
+    };
+
+    console.log('[Reader] Sending chapter request:', requestData);
+    wsRef.current.send(JSON.stringify(requestData));
   };
 
   if (loading) {