get user profile from leetcode

Author: Sorosliu1029

jupyterlab_leetcode/handlers/__init__.py

@@ -2,12 +2,13 @@
 
 from .base_handler import BaseHandler
 from .cookie_handler import GetCookieHandler
+from .leetcode_handler import LeetCodeProfileHandler
 
 
 def setup_handlers(web_app):
     host_pattern = ".*$"
     base_url = web_app.settings["base_url"]
-    handlers: list[type[BaseHandler]] = [GetCookieHandler]
+    handlers: list[type[BaseHandler]] = [GetCookieHandler, LeetCodeProfileHandler]
 
     web_app.add_handlers(
         host_pattern,

jupyterlab_leetcode/handlers/cookie_handler.py

@@ -5,6 +5,8 @@
 
 import browser_cookie3
 import tornado
+from tornado.httpclient import AsyncHTTPClient
+from tornado.httputil import HTTPHeaders
 
 from .base_handler import BaseHandler
 
@@ -60,10 +62,25 @@ def get(self):
                 else 3600 * 24 * 14
             )
             self.set_cookie("leetcode_browser", browser, max_age=max_age)
+            leetcode_ua = self.request.headers.get("user-agent")
             self.settings.update(
                 leetcode_browser=browser,
                 leetcode_cookiejar=cj,
-                leetcode_ua=self.request.headers.get("user-agent"),
+                leetcode_headers=HTTPHeaders(
+                    {
+                        "Cookie": "; ".join(f"{c.name}={c.value}" for c in cj),
+                        "Content-Type": "application/json",
+                        "Origin": "https://leetcode.com",
+                        "Referer": "https://leetcode.com/",
+                        "X-CsrfToken": (
+                            cookie_csrf.value
+                            if cookie_csrf and cookie_csrf.value
+                            else ""
+                        ),
+                    }
+                ),
+                leetcode_ua=leetcode_ua,
             )
+            AsyncHTTPClient.configure(None, defaults=dict(user_agent=leetcode_ua))
 
         self.finish(json.dumps(resp))

jupyterlab_leetcode/handlers/leetcode_handler.py

@@ -0,0 +1,40 @@
+import json
+
+import tornado
+from tornado.httpclient import AsyncHTTPClient, HTTPRequest
+from tornado.httputil import HTTPHeaders
+
+from .base_handler import BaseHandler
+
+LEETCODE_GRAPHQL_URL = "https://leetcode.com/graphql"
+
+
+class LeetCodeProfileHandler(BaseHandler):
+    route = r"leetcode/profile"
+
+    @tornado.web.authenticated
+    async def get(self):
+        client = AsyncHTTPClient()
+        headers = HTTPHeaders(self.settings.get("leetcode_headers", {}))
+        req = HTTPRequest(
+            url=LEETCODE_GRAPHQL_URL,
+            method="POST",
+            headers=headers,
+            body=json.dumps(
+                {
+                    "operationName": "globalData",
+                    "variables": {},
+                    "query": "query globalData { userStatus { isSignedIn username realName avatar } }",
+                }
+            ),
+        )
+
+        try:
+            resp = await client.fetch(req)
+        except Exception as e:
+            self.log.error(f"Error fetching LeetCode profile: {e}")
+            self.set_status(500)
+            self.finish(json.dumps({"message": "Failed to fetch LeetCode profile"}))
+            return
+        else:
+            self.finish(resp.body)

src/components/LeetCode.tsx

@@ -1,13 +1,22 @@
-import React, { useState } from 'react';
+import React, { useEffect, useState } from 'react';
+import { getProfile } from '../services/leetcode';
 
 const LeetCode = () => {
-  const [_username, _setUsername] = useState('');
+  const [username, setUsername] = useState('');
 
-  // useEffect(() => {}, []);
+  useEffect(() => {
+    getProfile().then(profile => {
+      if (!profile || !profile.isSignedIn) {
+        alert('Please sign in to LeetCode.');
+        return;
+      }
+      setUsername(profile.username);
+    });
+  }, []);
 
   return (
     <div>
-      <p>Welcome username</p>
+      <p>Welcome {username}</p>
     </div>
   );
 };

src/services/leetcode.ts

@@ -0,0 +1,10 @@
+import { LeetCodeProfile } from '../types/leetcode';
+import { requestAPI } from './handler';
+
+export async function getProfile(): Promise<LeetCodeProfile | null> {
+  return requestAPI<{ data: { userStatus: LeetCodeProfile } }>(
+    '/leetcode/profile'
+  )
+    .then(d => d.data.userStatus)
+    .catch(() => null);
+}

src/types/leetcode.d.ts

@@ -0,0 +1,6 @@
+export type LeetCodeProfile = {
+  avatar: string;
+  isSignedIn: boolean;
+  realName: string;
+  username: string;
+};