Anything

src/modules/comic/ComicReader.tsx

@@ -36,19 +36,33 @@ export function ComicReader() {
   const [showPageJump, setShowPageJump] = useState(false);
   const [jumpPage, setJumpPage] = useState('');
   const [loadingChapter, setLoadingChapter] = useState(false);
+  const [password, setPassword] = useState<string>('');
+  const [originalSessionId, setOriginalSessionId] = useState<string>('');
   
   const imageRefs = useRef<{ [key: number]: HTMLImageElement | null }>({});
   const controlsTimeout = useRef<number | null>(null);
   const scrollContainerRef = useRef<HTMLDivElement | null>(null);
   const wsRef = useRef<WebSocket | null>(null);
-
   const chapterData = location.state?.chapterData;
-  const password = location.state?.password;
-  const originalSessionId = location.state?.sessionId;
+  const statePassword = location.state?.password;
+  const stateSessionId = location.state?.sessionId;
 
-  // WebSocket setup
   useEffect(() => {
-    if (!password || !originalSessionId) return;
+    if (statePassword) {
+      console.log('[Reader] Setting password from state');
+      setPassword(statePassword);
+    }
+    if (stateSessionId) {
+      console.log('[Reader] Setting sessionId from state');
+      setOriginalSessionId(stateSessionId);
+    }
+  }, [statePassword, stateSessionId]);
+
+  useEffect(() => {
+    if (!password || !originalSessionId) {
+      console.log('[Reader] Waiting for password and sessionId...', { password: !!password, sessionId: !!originalSessionId });
+      return;
+    }
 
     const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
     const wsUrl = `${protocol}//${window.location.host}/ws`;
@@ -120,7 +134,7 @@ export function ComicReader() {
       return;
     }
 
-    if (!password || !originalSessionId) {
+    if (!statePassword || !stateSessionId) {
       setError('Unauthorized access - please open from comic page');
       setLoading(false);
       return;
@@ -128,7 +142,7 @@ export function ComicReader() {
 
     setError('Please open chapter from comic page');
     setLoading(false);
-  }, [sessionId, password, chapterData, originalSessionId]);
+  }, [sessionId, statePassword, chapterData, stateSessionId]);
   
   useEffect(() => {
     const resetTimeout = () => {