cache leetcode cookie jar into tornado application setting

Author: Sorosliu1029

jupyterlab_leetcode/handlers/cookie_handler.py

@@ -48,19 +48,18 @@ def get(self):
             cast(Cookie, cookie_session).is_expired()
             or cast(Cookie, cookie_csrf).is_expired()
         )
+        checked = exist and not expired
 
-        resp = {"exist": exist, "expired": expired}
+        resp = {"exist": exist, "expired": expired, "checked": checked}
 
-        if exist and not expired:
+        if checked:
             cookie_session_expires = cast(Cookie, cookie_session).expires
             max_age = (
                 cookie_session_expires - int(time.time())
                 if cookie_session_expires is not None
                 else 3600 * 24 * 14
             )
-            self.add_header(
-                "Set-Cookie",
-                f"leetcode_browser={browser}; Path=/; Max-Age={max_age}",
-            )
+            self.set_cookie("leetcode_browser", browser, max_age=max_age)
+            self.settings.update(leetcode_browser=browser, leetcode_cookiejar=cj)
 
         self.finish(json.dumps(resp))

src/components/BrowserCookie.tsx

@@ -61,8 +61,8 @@ const BrowserCookie = ({
       return;
     }
 
-    getCookie(browser).then(cookies => {
-      setChecked(!!cookies['exist']);
+    getCookie(browser).then(resp => {
+      setChecked(resp['checked']);
     });
   };
 

src/widget.tsx

@@ -2,6 +2,7 @@ import { ReactWidget } from '@jupyterlab/ui-components';
 import React, { StrictMode, useEffect, useState } from 'react';
 import LandingPage from './components/LandingPage';
 import LeetCode from './components/LeetCode';
+import { getCookie } from './services/cookie';
 
 const LeetCodeComponent = (): JSX.Element => {
   const [cookieLoggedIn, setCookieLoggedIn] = useState('');
@@ -12,9 +13,13 @@ const LeetCodeComponent = (): JSX.Element => {
       .find(cookie => cookie.startsWith('leetcode_browser='))
       ?.split('=')[1];
     if (leetcode_browser) {
-      setCookieLoggedIn(leetcode_browser);
+      getCookie(leetcode_browser).then(resp => {
+        if (resp['checked']) {
+          setCookieLoggedIn(leetcode_browser);
+        }
+      });
     }
-  }, []);
+  });
 
   return cookieLoggedIn ? (
     <LeetCode />