load cookies from leetcode.com

Author: Sorosliu1029

jupyterlab_leetcode/handlers.py

@@ -0,0 +1,20 @@
+from jupyter_server.utils import url_path_join
+
+from .base_handler import BaseHandler
+from .cookie_handler import GetCookieHandler
+
+
+def setup_handlers(web_app):
+    host_pattern = ".*$"
+    base_url = web_app.settings["base_url"]
+    handlers: list[type[BaseHandler]] = [GetCookieHandler]
+
+    web_app.add_handlers(
+        host_pattern,
+        list(
+            map(
+                lambda h: (url_path_join(base_url, "jupyterlab-leetcode", h.route), h),
+                handlers,
+            )
+        ),
+    )

jupyterlab_leetcode/handlers/__init__.py

@@ -0,0 +1,10 @@
+from jupyter_server.base.handlers import APIHandler
+
+
+class BaseHandler(APIHandler):
+    """Base handler for JupyterLab LeetCode extension.
+    This class extends APIHandler to provide a base for other handlers.
+    It can be used to define common functionality or properties for all handlers.
+    """
+
+    route = r""

jupyterlab_leetcode/handlers/base_handler.py

@@ -0,0 +1,54 @@
+import json
+
+import browser_cookie3
+import tornado
+
+from .base_handler import BaseHandler
+
+BROWSER_COOKIE_METHOD_MAP = {
+    "chrome": browser_cookie3.chrome,
+    "chromium": browser_cookie3.chromium,
+    "opera": browser_cookie3.opera,
+    "opera_gx": browser_cookie3.opera_gx,
+    "brave": browser_cookie3.brave,
+    "edge": browser_cookie3.edge,
+    "vivaldi": browser_cookie3.vivaldi,
+    "firefox": browser_cookie3.firefox,
+    "librewolf": browser_cookie3.librewolf,
+    "safari": browser_cookie3.safari,
+    "arc": browser_cookie3.arc,
+}
+
+
+class GetCookieHandler(BaseHandler):
+    route = r"cookies/[a-zA-Z0-9_-]+"
+
+    @tornado.web.authenticated
+    def get(self):
+        self.log.info("Loading all cookies for LeetCode...")
+        browser = self.get_query_argument("browser", "", strip=True)
+        if not browser:
+            self.set_status(400)
+            self.finish(json.dumps({"message": "Browser parameter is required"}))
+            return
+
+        if browser not in BROWSER_COOKIE_METHOD_MAP:
+            self.set_status(400)
+            self.finish(json.dumps({"message": f"Unsupported browser: {browser}"}))
+            return
+
+        cj = BROWSER_COOKIE_METHOD_MAP[browser](domain_name="leetcode.com")
+
+        self.finish(
+            json.dumps(
+                dict(
+                    map(
+                        lambda c: (
+                            c.name,
+                            {"name": c.name, "domain": c.domain, "value": c.value},
+                        ),
+                        cj,
+                    )
+                )
+            )
+        )

jupyterlab_leetcode/handlers/cookie_handler.py

@@ -1,13 +1,68 @@
-import React from 'react';
+import React, { useState } from 'react';
+import { getCookie } from '../services/cookie';
 
 const BrowserCookie = () => {
+  const browsers = [
+    'Chrome',
+    'Firefox',
+    'Safari',
+    'Edge',
+    'Opera',
+    'Brave',
+    'Vivaldi',
+    'Chromium',
+    'Arc',
+    'LibreWolf',
+    'Opera GX'
+  ];
+
+  const [browser, setBrowser] = useState('');
+  const [checked, setChecked] = useState(false);
+
   const loadCookies = () => {
-    console.log('Loading cookies from leetcode.com');
+    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.'
+      );
+      return;
+    }
+
+    getCookie('all', browser).then(cookies => {
+      if (cookies && cookies['LEETCODE_SESSION'] && cookies['csrftoken']) {
+        setChecked(true);
+      }
+    });
   };
 
   return (
     <div>
+      <label htmlFor="browser-selector">
+        Choose your browser that has LeetCode logged in:
+      </label>
+      <select
+        id="browser-selector"
+        required
+        value={browser}
+        onChange={e => setBrowser(e.target.value)}
+      >
+        <option value="" disabled>
+          Select a browser
+        </option>
+        {browsers.map(browser => (
+          <option
+            key={browser.toLowerCase()}
+            value={browser.toLowerCase().replace(/\s+/g, '_')}
+          >
+            {browser}
+          </option>
+        ))}
+      </select>
       <button onClick={loadCookies}>Load</button>
+      <p>Checked: {checked ? 'Yes' : 'No'}</p>
     </div>
   );
 };

src/components/BrowserCookie.tsx

@@ -21,8 +21,6 @@ const plugin: JupyterFrontEndPlugin<void> = {
     palette: ICommandPalette,
     restorer: ILayoutRestorer | null
   ) => {
-    console.log('JupyterLab extension jupyterlab-leetcode is activated!');
-
     let leetcodeWidget: LeetCodeWidget;
 
     const command = 'leetcode-widget:open';
@@ -47,7 +45,6 @@ const plugin: JupyterFrontEndPlugin<void> = {
       namespace: 'leetcode-widget'
     });
     if (restorer) {
-      console.log('Restoring layout for jupyterlab-leetcode plugin');
       restorer.restore(tracker, { command, name: () => 'leetcode' });
     }
   }

src/index.ts

@@ -0,0 +1,10 @@
+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}`
+  ).catch(() => ({}));
+}

src/services/cookie.ts

@@ -3,10 +3,15 @@ import React from 'react';
 import BrowserCookie from './components/BrowserCookie';
 
 const LeetCodeComponent = (): JSX.Element => {
+  const options: JSX.Element[] = [<BrowserCookie />];
   return (
     <div>
       <p>Welcome to JupyterLab LeetCode Widget.</p>
-      <BrowserCookie />
+      <p>
+        For this plugin to work, you may choose one of these {options.length}{' '}
+        methods to allow this plugin to log into LeetCode.
+      </p>
+      {...options}
     </div>
   );
 };