feat(sync): implement recent accepted submissions retrieval.

Author: arnav-on-code

config/__init__.py

@@ -7,6 +7,7 @@
 class Config:
 
     LEETCODE_SESSION = os.getenv("LEETCODE_SESSION", "")
+    LEETCODE_USERNAME = os.getenv("LEETCODE_USERNAME", "")
     CSRF_TOKEN = os.getenv("CSRF_TOKEN", "")
 
     GITHUB_USERNAME = os.getenv("GITHUB_USERNAME", "")

config/logging.py

@@ -1,11 +1,42 @@
-from requests import Session
+from config.settings import Config
+from leetcode.parser import parse_submissions
+from leetcode.queries import (
+    PROFILE_QUERY,
+    RECENT_SUBMISSIONS_QUERY,
+)
 
 
 class LeetCodeAPI:
-    def __init__(self, session: Session):
-        self.session = session
 
-    def validate_session(self) -> bool:
-        response = self.session.get("https://leetcode.com/api/problems/all/")
+    def __init__(self, client):
+        self.client = client
 
-        return response.status_code == 200
+    def get_profile(self):
+        response = self.client.post(
+            PROFILE_QUERY,
+            {
+                "username": Config.LEETCODE_USERNAME
+            }
+        )
+
+        if "errors" in response:
+            raise Exception(response["errors"])
+
+        return response["data"]["matchedUser"]
+
+    def get_recent_submissions(self, limit=15):
+
+        response = self.client.post(
+            RECENT_SUBMISSIONS_QUERY,
+            {
+                "username": Config.LEETCODE_USERNAME,
+                "limit": limit,
+            },
+        )
+
+        if "errors" in response:
+            raise Exception(response["errors"])
+
+        return parse_submissions(
+            response["data"]["recentAcSubmissionList"]
+        )

config.py config/settings.pyconfig.py renamed to config/settings.py

@@ -1,6 +1,6 @@
 from requests import Session
 
-from config import Config
+from config.settings import Config
 
 
 class LeetCodeAuth:

leetcode/api.py

@@ -0,0 +1,26 @@
+from requests import Session
+from config.settings import Config
+
+
+class LeetCodeClient:
+    def __init__(self, session: Session):
+        self.session = session
+
+    def post(self, query: str, variables: dict = None):
+        payload = {
+            "query": query,
+            "variables": variables or {}
+        }
+
+        response = self.session.post(
+            Config.GRAPHQL_URL,
+            json=payload,
+            timeout=Config.REQUEST_TIMEOUT
+        )
+
+        print("=" * 80)
+        print("Status Code:", response.status_code)
+        print(response.text)
+        print("=" * 80)
+
+        return response.json()

leetcode/auth.py

@@ -0,0 +1,14 @@
+from dataclasses import dataclass
+from datetime import datetime
+
+
+@dataclass(slots=True)
+class Submission:
+    id: str
+    title: str
+    slug: str
+    timestamp: int
+
+    @property
+    def date(self):
+        return datetime.fromtimestamp(self.timestamp)

leetcode/client.py

@@ -0,0 +1,17 @@
+from leetcode.models import Submission
+
+
+def parse_submissions(data):
+    submissions = []
+
+    for item in data:
+        submissions.append(
+            Submission(
+                id=item["id"],
+                title=item["title"],
+                slug=item["titleSlug"],
+                timestamp=int(item["timestamp"])
+            )
+        )
+
+    return submissions

leetcode/models.py

@@ -0,0 +1,29 @@
+PROFILE_QUERY = """
+query userProblemsSolved($username: String!) {
+    matchedUser(username: $username) {
+        username
+
+        submitStatsGlobal {
+            acSubmissionNum {
+                difficulty
+                count
+                submissions
+            }
+        }
+    }
+}
+"""
+
+RECENT_SUBMISSIONS_QUERY = """
+query recentAcSubmissions($username: String!, $limit: Int!) {
+    recentAcSubmissionList(
+        username: $username,
+        limit: $limit
+    ) {
+        id
+        title
+        titleSlug
+        timestamp
+    }
+}
+"""

leetcode/parser.py

@@ -1,7 +1,9 @@
 from rich.console import Console
 from rich.panel import Panel
+from rich.table import Table
 
 from leetcode.auth import LeetCodeAuth
+from leetcode.client import LeetCodeClient
 from leetcode.api import LeetCodeAPI
 
 console = Console()
@@ -12,21 +14,73 @@ def banner():
         Panel.fit(
             "[bold cyan]LeetCode Sync[/bold cyan]\n"
             "Automatic LeetCode → GitHub Synchronization",
-            title="v0.2.0",
+            title="v0.4.0",
         )
     )
 
 
+def display_statistics(stats):
+    """Display solved problem statistics."""
+
+    table = Table(title="LeetCode Statistics")
+
+    table.add_column("Difficulty", style="cyan")
+    table.add_column("Solved", justify="right", style="green")
+
+    for item in stats:
+        table.add_row(
+            item["difficulty"],
+            str(item["count"])
+        )
+
+    console.print(table)
+
+
+def display_recent_submissions(submissions):
+    """Display recent accepted submissions."""
+
+    table = Table(title="Recent Accepted Submissions")
+
+    table.add_column("ID", style="cyan")
+    table.add_column("Title")
+    table.add_column("Date")
+
+    for submission in submissions:
+        table.add_row(
+            submission.id,
+            submission.title,
+            submission.date.strftime("%Y-%m-%d %H:%M"),
+        )
+
+    console.print(table)
+
+
 def main():
     banner()
 
     auth = LeetCodeAuth()
-    api = LeetCodeAPI(auth.get_session())
+    session = auth.get_session()
+
+    client = LeetCodeClient(session)
+    api = LeetCodeAPI(client)
+
+    # Profile
+    profile = api.get_profile()
+
+    console.print(
+        f"[bold green]Logged in as:[/bold green] {profile['username']}\n"
+    )
+
+    display_statistics(
+        profile["submitStatsGlobal"]["acSubmissionNum"]
+    )
+
+    console.print()
+
+    # Recent submissions
+    submissions = api.get_recent_submissions()
 
-    if api.validate_session():
-        console.print("[green]✓ Session is valid[/green]")
-    else:
-        console.print("[red]✗ Invalid session[/red]")
+    display_recent_submissions(submissions)
 
 
 if __name__ == "__main__":