add cookie to store browser option

Author: Sorosliu1029

jupyterlab_leetcode/handlers/cookie_handler.py

@@ -1,4 +1,7 @@
 import json
+import time
+from http.cookiejar import Cookie
+from typing import cast
 
 import browser_cookie3
 import tornado
@@ -21,7 +24,7 @@
 
 
 class GetCookieHandler(BaseHandler):
-    route = r"cookies/[a-zA-Z0-9_-]+"
+    route = r"cookies"
 
     @tornado.web.authenticated
     def get(self):
@@ -38,17 +41,26 @@ def get(self):
             return
 
         cj = BROWSER_COOKIE_METHOD_MAP[browser](domain_name="leetcode.com")
+        cookie_session = next((c for c in cj if c.name == "LEETCODE_SESSION"), None)
+        cookie_csrf = next((c for c in cj if c.name == "csrftoken"), None)
+        exist = bool(cookie_session and cookie_csrf)
+        expired = exist and (
+            cast(Cookie, cookie_session).is_expired()
+            or cast(Cookie, cookie_csrf).is_expired()
+        )
+
+        resp = {"exist": exist, "expired": expired}
 
-        self.finish(
-            json.dumps(
-                dict(
-                    map(
-                        lambda c: (
-                            c.name,
-                            {"name": c.name, "domain": c.domain, "value": c.value},
-                        ),
-                        cj,
-                    )
-                )
+        if exist and not expired:
+            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=/; HttpOnly; Max-Age={max_age}",
+            )
+
+        self.finish(json.dumps(resp))

src/components/BrowserCookie.tsx

@@ -44,23 +44,21 @@ const BrowserCookie = ({ onSuccess }: { onSuccess: () => void }) => {
     }
   }, [checked, onSuccess]);
 
-  const loadCookies = () => {
+  const checkCookie = () => {
     // TODO: change alert
     if (!browser) {
       alert('Please select a browser.');
       return;
     }
     if (browser === 'safari') {
       alert(
-        'Safari does not support loading cookies from the browser. Please use another browser.'
+        'Safari does not support getting cookies from the browser. Please use another browser.'
       );
       return;
     }
 
-    getCookie('all', browser).then(cookies => {
-      if (cookies && cookies['LEETCODE_SESSION'] && cookies['csrftoken']) {
-        setChecked(true);
-      }
+    getCookie(browser).then(cookies => {
+      setChecked(!!cookies['exist']);
     });
   };
 
@@ -87,7 +85,7 @@ const BrowserCookie = ({ onSuccess }: { onSuccess: () => void }) => {
           </option>
         ))}
       </select>
-      <button onClick={loadCookies}>Load</button>
+      <button onClick={checkCookie}>Check</button>
       <p>Checked: {checked ? 'Yes' : 'No'}</p>
     </div>
   );

src/services/cookie.ts

@@ -1,10 +1,9 @@
 import { requestAPI } from './handler';
 
 export async function getCookie(
-  name: string,
   browser: string
-): Promise<{ [key: string]: any }> {
-  return requestAPI<{ [key: string]: any }>(
-    `/cookies/${name}?browser=${browser}`
+): Promise<{ [key: string]: boolean }> {
+  return requestAPI<{ [key: string]: boolean }>(
+    `/cookies?browser=${browser}`
   ).catch(() => ({}));
 }