fetched mocked user info

Author: yuvrajsinh5252

src/commands/profile.py

@@ -1,12 +1,12 @@
-from server.api import fetch_user_data
-from lib.ui import display_problem_stats
+from server.api import fetch_user_profile
 from rich.spinner import Spinner
+from src.lib.ui import display_user_stats
 from rich.live import Live
 
 def profile():
     """List all available LeetCode problems."""
     spinner = Spinner('dots')
     with Live(spinner, refresh_per_second=10, transient=True) as live:
         live.console.print("[cyan]Fetching user profile...")
-        data = fetch_user_data()
-    display_problem_stats(data)
+        data = fetch_user_profile()
+    display_user_stats(data)

src/lib/ui.py

@@ -4,6 +4,8 @@
 from rich import box
 from datetime import datetime
 
+import typer
+
 console = Console()
 
 def format_timestamp(timestamp):
@@ -84,25 +86,71 @@ def create_profile_header(data):
         title="[bold cyan]LeetCode Profile[/bold cyan]"
     )
 
-def display_problem_stats(data):
+def display_user_stats(data):
     console.clear()
     console.print("\n")
 
-    # Display profile header
-    console.print(create_profile_header(data))
-    console.print("\n")
+    if not data.get('userProfile'):
+        console.print(Panel("Could not fetch user profile", border_style="red"))
+        return
+
+    user = data['userProfile']['matchedUser']
+    profile = user['profile']
+
+    # Display basic profile info
+    profile_table = Table.grid(padding=(0, 2))
+    profile_table.add_column(justify="left")
 
-    # Display recent submissions
-    recent_submissions = data.get('matchedUser', {}).get('recentSubmissionList', [])
-    if recent_submissions:
-        recent_activity = create_recent_activity(recent_submissions)
-        if recent_activity:
-            console.print(recent_activity)
-    else:
-        console.print(Panel("No recent activity", border_style="cyan"))
+    profile_info = [
+        f"[bold cyan]{user['username']}[/bold cyan]",
+        f"[dim]Ranking:[/dim] {profile['ranking']}",
+        f"[dim]Company:[/dim] {profile['company'] or 'Not specified'}",
+        f"[dim]Location:[/dim] {profile['countryName'] or 'Not specified'}"
+    ]
 
+    if profile['skillTags']:
+        skills = ", ".join(profile['skillTags'])
+        profile_info.append(f"[dim]Skills:[/dim] {skills}")
+
+    profile_table.add_row("\n".join(profile_info))
+    console.print(Panel(profile_table, title="[bold cyan]Profile Info[/bold cyan]", border_style="cyan"))
     console.print("\n")
 
+    # Display contest info if available
+    if data.get('contestInfo') and data['contestInfo'].get('userContestRanking'):
+        contest = data['contestInfo']['userContestRanking']
+        contest_table = Table.grid(padding=(0, 2))
+        contest_table.add_column(justify="left")
+        contest_table.add_row(
+            f"Rating: [yellow]{contest['rating']}[/yellow]\n"
+            f"Global Rank: {contest['globalRanking']}/{contest['totalParticipants']}\n"
+            f"Top: {contest['topPercentage']}%\n"
+            f"Contests: {contest['attendedContestsCount']}"
+        )
+        console.print(Panel(contest_table, title="[bold yellow]Contest Stats[/bold yellow]", border_style="yellow"))
+        console.print("\n")
+
+    # Display progress
+    if data.get('progress'):
+        progress = data['progress']
+        console.print(create_profile_header(progress))
+        console.print("\n")
+
+    # Display language stats
+    if data.get('languageStats'):
+        lang_stats = data['languageStats']['matchedUser']['languageProblemCount']
+        lang_table = Table(title="Languages", box=box.ROUNDED, border_style="cyan")
+        lang_table.add_column("Language", style="cyan")
+        lang_table.add_column("Problems Solved", justify="right")
+
+        for lang in sorted(lang_stats, key=lambda x: x['problemsSolved'], reverse=True)[:5]:
+            lang_table.add_row(
+                lang['languageName'],
+                str(lang['problemsSolved'])
+            )
+        console.print(lang_table)
+        console.print("\n")
+
 def display_problem_list(data):
     console.clear()
     console.print("\n")

src/server/api.py

@@ -19,38 +19,95 @@ def create_leetcode_client(csrf_token: str, session_id: str):
         fetch_schema_from_transport=False
     )
 
-def fetch_user_data(username: str = "yuvrajsinh5252"):
+def fetch_user_profile(username: str = "yuvrajsinh5252"):
   client = create_leetcode_client("csrf_token", "session_id")
-  variables = {"username": username}
-  query = gql(
-    """
-    query userProblemsSolved($username: String!) {
-      allQuestionsCount {
-        difficulty
-        count
+  queries = {
+    "userProfile": """
+      query userPublicProfile($username: String!) {
+        matchedUser(username: $username) {
+          contestBadge { name expired hoverText icon }
+          username githubUrl twitterUrl linkedinUrl
+          profile {
+            ranking userAvatar realName aboutMe school websites
+            countryName company jobTitle skillTags postViewCount
+            postViewCountDiff reputation reputationDiff solutionCount
+            solutionCountDiff categoryDiscussCount categoryDiscussCountDiff
+            certificationLevel
+          }
+        }
       }
-      matchedUser(username: $username) {
-        problemsSolvedBeatsStats {
-          difficulty
-          percentage
+    """,
+    "languageStats": """
+      query languageStats($username: String!) {
+        matchedUser(username: $username) {
+          languageProblemCount {
+            languageName
+            problemsSolved
+          }
         }
-        submitStatsGlobal {
-          acSubmissionNum {
-            difficulty
-            count
+      }
+    """,
+    "skillStats": """
+      query skillStats($username: String!) {
+        matchedUser(username: $username) {
+          tagProblemCounts {
+            advanced { tagName tagSlug problemsSolved }
+            intermediate { tagName tagSlug problemsSolved }
+            fundamental { tagName tagSlug problemsSolved }
+          }
+        }
+      }
+    """,
+    "contestInfo": """
+      query userContestRankingInfo($username: String!) {
+        userContestRanking(username: $username) {
+          attendedContestsCount rating globalRanking
+          totalParticipants topPercentage
+          badge { name }
+        }
+        userContestRankingHistory(username: $username) {
+          attended trendDirection problemsSolved totalProblems
+          finishTimeInSeconds rating ranking
+          contest { title startTime }
+        }
+      }
+    """,
+    "progress": """
+      query userSessionProgress($username: String!) {
+        allQuestionsCount { difficulty count }
+        matchedUser(username: $username) {
+          submitStats {
+            acSubmissionNum { difficulty count submissions }
+            totalSubmissionNum { difficulty count submissions }
+          }
+        }
+      }
+    """,
+    "calendar": """
+      query userProfileCalendar($username: String!, $year: Int) {
+        matchedUser(username: $username) {
+          userCalendar(year: $year) {
+            activeYears streak totalActiveDays
+            dccBadges { timestamp badge { name icon } }
+            submissionCalendar
           }
         }
       }
-    }
     """
-  )
+  }
+
+  results = {}
+  for name, query in queries.items():
+    try:
+      results[name] = client.execute(
+        gql(query),
+        variable_values={"username": username}
+      )
+    except Exception as e:
+      print(f"Error fetching {name}: {str(e)}")
+      results[name] = None
 
-  try:
-    result = client.execute(query, variable_values=variables)
-    return result
-  except Exception as e:
-    print(f"Error fetching data: {str(e)}")
-    return None
+  return results
 
 def fetch_problem_list(
     csrf_token: str,