feat: add solution cmd

Author: yuvrajsinh5252

src/commands/list_problems.py

@@ -3,7 +3,7 @@
 import typer
 from rich.progress import Progress, SpinnerColumn, TextColumn
 
-from ..lib.ui import display_problem_list
+from ..lib.profile_ui import display_problem_list
 from ..server.api import fetch_problem_list
 from ..server.auth import Auth
 

src/commands/profile.py

@@ -1,7 +1,7 @@
 from rich.live import Live
 from rich.spinner import Spinner
 
-from ..lib.ui import display_user_stats
+from ..lib.profile_ui import display_user_stats
 from ..server.api import fetch_user_profile
 
 

src/commands/solution.py

@@ -1,4 +1,13 @@
 import typer
+from rich.progress import Progress, SpinnerColumn, TextColumn
+
+from src.lib.solution_ui import SolutionUI
+
+from ..server.auth import Auth
+from ..server.solution_manager import SolutionManager
+
+auth = Auth()
+solution_manager = SolutionManager(auth.get_session())
 
 
 def solutions(
@@ -10,4 +19,24 @@ def solutions(
     ),
 ):
     """Fetch solution for a problem"""
-    pass
+
+    try:
+        with Progress(
+            SpinnerColumn(),
+            TextColumn("[progress.description]{task.description}"),
+            transient=True,
+        ) as progress:
+            progress.add_task("Fetching problem solutions...", total=1)
+            fetched_solution = solution_manager.get_problem_solutions(problem, best)
+
+            if not fetched_solution:
+                typer.echo("No solution found")
+            else:
+                solution_ui = SolutionUI(
+                    fetched_solution.get("data", {}).get("ugcArticleSolutionArticles")
+                )
+                solution_ui.show_solution()
+    except Exception as e:
+        typer.secho("Error fetching solution for problem ", fg="red", nl=False)
+        typer.secho(f'"{problem}"', fg="bright_red", bold=True)
+        typer.secho(f"\n{e}", fg="yellow", italic=True)

src/lib/ui.py renamed to src/lib/profile_ui.py

@@ -1,17 +1,20 @@
+from datetime import datetime
+
+from rich import box
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
-from rich import box
-from datetime import datetime
+from rich.table import Table
 
 console = Console()
 
+
 def format_timestamp(timestamp):
     dt = datetime.fromtimestamp(timestamp)
     return dt.strftime("%Y-%m-%d %H:%M")
 
+
 def create_recent_activity(recent_submissions):
-    if not recent_submissions or 'recentAcSubmissionList' not in recent_submissions:
+    if not recent_submissions or "recentAcSubmissionList" not in recent_submissions:
         return None
 
     table = Table(
@@ -20,112 +23,136 @@ def create_recent_activity(recent_submissions):
         border_style="cyan",
         pad_edge=False,
         show_edge=True,
-        width=65
+        width=65,
     )
     table.add_column("Time", style="dim", width=16)
     table.add_column("Problem", style="cyan")
 
-    for sub in recent_submissions['recentAcSubmissionList'][:5]:
+    for sub in recent_submissions["recentAcSubmissionList"][:5]:
         table.add_row(
-            format_timestamp(int(sub['timestamp'])),
-            sub['title'],
+            format_timestamp(int(sub["timestamp"])),
+            sub["title"],
         )
     return table
 
+
 def create_progress_panel(data):
-    if not data.get('progress'):
+    if not data.get("progress"):
         return None
 
-    progress = data['progress']
-    all_count = progress['allQuestionsCount']
-    submit_stats = progress['matchedUser']['submitStats']
+    progress = data["progress"]
+    all_count = progress["allQuestionsCount"]
+    submit_stats = progress["matchedUser"]["submitStats"]
     table = Table.grid(padding=(0, 1))
     table.add_column(justify="left")
 
-    for diff in ['Easy', 'Medium', 'Hard']:
-        total = next((q['count'] for q in all_count if q['difficulty'] == diff), 0)
-        solved = next((s['count'] for s in submit_stats['acSubmissionNum'] if s['difficulty'] == diff), 0)
+    for diff in ["Easy", "Medium", "Hard"]:
+        total = next((q["count"] for q in all_count if q["difficulty"] == diff), 0)
+        solved = next(
+            (
+                s["count"]
+                for s in submit_stats["acSubmissionNum"]
+                if s["difficulty"] == diff
+            ),
+            0,
+        )
         table.add_row(f"[dim]{diff}:[/dim] {solved}/{total}")
 
     return Panel(table, title="[bold green]Progress[/bold green]", border_style="green")
 
+
 def create_social_links(user, websites=None):
     links = []
-    if user['githubUrl']:
+    if user["githubUrl"]:
         links.append(f"[blue]GitHub[/blue]: {user['githubUrl']}")
-    if user['linkedinUrl']:
+    if user["linkedinUrl"]:
         links.append(f"[blue]LinkedIn[/blue]: {user['linkedinUrl']}")
-    if user['twitterUrl']:
+    if user["twitterUrl"]:
         links.append(f"[blue]Twitter[/blue]: {user['twitterUrl']}")
 
     if websites:
         for site in websites:
-            url = f"https://{site}" if not site.startswith(('http://', 'https://')) else site
+            url = (
+                f"https://{site}"
+                if not site.startswith(("http://", "https://"))
+                else site
+            )
             links.append(f"[blue]Website[/blue]: {url}")
 
     return "\n".join(links) if links else "No social links"
 
+
 def create_language_stats(data):
-    if not data or 'matchedUser' not in data:
+    if not data or "matchedUser" not in data:
         return None
 
-    lang_stats = data['matchedUser']['languageProblemCount']
+    lang_stats = data["matchedUser"]["languageProblemCount"]
     table = Table.grid(padding=(0, 1))
     table.add_column(justify="left")
     rows = []
-    sorted_stats = sorted(lang_stats, key=lambda x: x['problemsSolved'], reverse=True)
+    sorted_stats = sorted(lang_stats, key=lambda x: x["problemsSolved"], reverse=True)
 
     for lang in sorted_stats[:5]:
         rows.append(f"[dim]{lang['languageName']}:[/dim] {lang['problemsSolved']}")
 
     table.add_row("\n".join(rows))
-    return Panel(table, title="[bold blue]Languages[/bold blue]", border_style="blue", width=35, padding=(0, 1))
+    return Panel(
+        table,
+        title="[bold blue]Languages[/bold blue]",
+        border_style="blue",
+        width=35,
+        padding=(0, 1),
+    )
+
 
 def create_contest_stats(contest_info):
     stats = []
-    ranking = contest_info.get('userContestRanking', {}) if contest_info else {}
+    ranking = contest_info.get("userContestRanking", {}) if contest_info else {}
     if not ranking:
         return "No contest stats"
 
-    rating = ranking.get('rating', 0)
-    attended = ranking.get('attendedContestsCount', 0)
+    rating = ranking.get("rating", 0)
+    attended = ranking.get("attendedContestsCount", 0)
     stats.append(f"[dim]Rating:[/dim] {rating:.1f}")
     stats.append(f"[dim]Contests:[/dim] {attended}")
     return "\n".join(stats)
 
+
 def create_skill_stats(data):
-    if not data.get('skillStats'):
+    if not data.get("skillStats"):
         return None
 
-    skills = data['skillStats']['matchedUser']['tagProblemCounts']
+    skills = data["skillStats"]["matchedUser"]["tagProblemCounts"]
     table = Table(
         title="[bold magenta]Skills[/bold magenta]",
         box=box.ROUNDED,
         border_style="magenta",
         width=65,
-        pad_edge=False
+        pad_edge=False,
     )
     table.add_column("Level", style="magenta")
     table.add_column("Tag", style="cyan")
     table.add_column("Solved", justify="right")
 
-    for level in ['advanced', 'intermediate', 'fundamental']:
+    for level in ["advanced", "intermediate", "fundamental"]:
         for skill in skills[level][:2]:
-            table.add_row(level.capitalize(), skill['tagName'], str(skill['problemsSolved']))
+            table.add_row(
+                level.capitalize(), skill["tagName"], str(skill["problemsSolved"])
+            )
 
     return table
 
+
 def display_user_stats(data):
     console.clear()
-    width = min(console.width - 4, 180)
     profile_width, stats_width, progress_width = 65, 35, 30
 
-    if not data.get('userProfile'):
+    if not data.get("userProfile"):
         console.print(Panel("Could not fetch user profile", border_style="red"))
         return
 
-    user = data['userProfile']['matchedUser']
-    profile = user['profile']
+    user = data["userProfile"]["matchedUser"]
+    profile = user["profile"]
 
     profile_table = Table.grid(padding=(0, 1))
     profile_table.add_column(justify="left")
@@ -135,21 +162,30 @@ def display_user_stats(data):
         f"[dim]Ranking:[/dim] {profile.get('ranking', 'N/A')}",
     ]
 
-    for label, field in [('Company', 'company'), ('Job Title', 'jobTitle'),
-                        ('School', 'school'), ('Location', 'countryName')]:
+    for label, field in [
+        ("Company", "company"),
+        ("Job Title", "jobTitle"),
+        ("School", "school"),
+        ("Location", "countryName"),
+    ]:
         if field in profile and profile[field]:
             profile_info.append(f"[dim]{label}:[/dim] {profile[field]}")
 
-    if profile.get('skillTags'):
+    if profile.get("skillTags"):
         profile_info.append(f"[dim]Skills:[/dim] {', '.join(profile['skillTags'])}")
 
-    social_links = create_social_links(user, profile.get('websites', []))
+    social_links = create_social_links(user, profile.get("websites", []))
     if social_links != "No social links":
         profile_info.append("\n" + social_links)
 
     profile_table.add_row("\n".join(profile_info))
-    profile_panel = Panel(profile_table, title="[bold cyan]Profile Info[/bold cyan]",
-                         border_style="cyan", width=profile_width, padding=(0, 1))
+    profile_panel = Panel(
+        profile_table,
+        title="[bold cyan]Profile Info[/bold cyan]",
+        border_style="cyan",
+        width=profile_width,
+        padding=(0, 1),
+    )
 
     stats_table = Table.grid(padding=(0, 1))
     stats_table.add_column(justify="left")
@@ -158,30 +194,41 @@ def display_user_stats(data):
         f"[dim]Reputation:[/dim] {profile.get('reputation', 0)} ({profile.get('reputationDiff', 0):+d})",
         f"[dim]Views:[/dim] {profile.get('postViewCount', 0)} ({profile.get('postViewCountDiff', 0):+d})",
     ]
-    if 'categoryDiscussCount' in profile:
-        stats.append(f"[dim]Discussions:[/dim] {profile['categoryDiscussCount']} ({profile.get('categoryDiscussCountDiff', 0):+d})")
+    if "categoryDiscussCount" in profile:
+        stats.append(
+            f"[dim]Discussions:[/dim] {profile['categoryDiscussCount']} ({profile.get('categoryDiscussCountDiff', 0):+d})"
+        )
     stats_table.add_row("\n".join(stats))
 
     progress = create_progress_panel(data)
     if progress:
         progress.width = progress_width
 
     middle_column = Table.grid(padding=(0, 1))
-    if data.get('languageStats'):
-        middle_column.add_row(create_language_stats(data['languageStats']))
+    if data.get("languageStats"):
+        middle_column.add_row(create_language_stats(data["languageStats"]))
         middle_column.add_row("")
-    if "contestInfo" in data and data.get('contestInfo'):
-        contest_panel = Panel(create_contest_stats(data['contestInfo']),
-                            title="[bold yellow]Contest Stats[/bold yellow]",
-                            border_style="yellow", width=stats_width, padding=(0, 1))
+    if "contestInfo" in data and data.get("contestInfo"):
+        contest_panel = Panel(
+            create_contest_stats(data["contestInfo"]),
+            title="[bold yellow]Contest Stats[/bold yellow]",
+            border_style="yellow",
+            width=stats_width,
+            padding=(0, 1),
+        )
         middle_column.add_row(contest_panel)
 
     right_column = Table.grid()
     if progress:
         right_column.add_row(progress)
         right_column.add_row("")
-    stats_panel = Panel(stats_table, title="[bold blue]Activity Stats[/bold blue]",
-                       border_style="blue", width=progress_width, padding=(0, 1))
+    stats_panel = Panel(
+        stats_table,
+        title="[bold blue]Activity Stats[/bold blue]",
+        border_style="blue",
+        width=progress_width,
+        padding=(0, 1),
+    )
     right_column.add_row(stats_panel)
 
     top_grid = Table.grid(padding=(0, 2))
@@ -191,8 +238,10 @@ def display_user_stats(data):
     bottom_grid.add_column(width=65)
     bottom_grid.add_column(width=65)
     bottom_grid.add_row(
-        create_skill_stats(data) if data.get('skillStats') else "",
-        create_recent_activity(data['recentAcSubmissions']) if data.get('recentAcSubmissions') else ""
+        create_skill_stats(data) if data.get("skillStats") else "",
+        create_recent_activity(data["recentAcSubmissions"])
+        if data.get("recentAcSubmissions")
+        else "",
     )
 
     console.print("\n")
@@ -201,6 +250,7 @@ def display_user_stats(data):
     console.print(bottom_grid)
     console.print("\n")
 
+
 def display_problem_list(data):
     console.clear()
     console.print("\n")
@@ -210,7 +260,7 @@ def display_problem_list(data):
         box=box.ROUNDED,
         border_style="cyan",
         pad_edge=False,
-        show_edge=True
+        show_edge=True,
     )
 
     table.add_column("ID", style="dim", width=6)
@@ -219,16 +269,16 @@ def display_problem_list(data):
     table.add_column("Status", justify="center", width=8)
     table.add_column("AC Rate", justify="right", width=8)
 
-    for problem in data['problemsetQuestionList']['questions']:
-        status_icon = "[green]✓" if problem['status'] == "ac" else "[red]✗"
+    for problem in data["problemsetQuestionList"]["questions"]:
+        status_icon = "[green]✓" if problem["status"] == "ac" else "[red]✗"
         ac_rate = f"{problem['acRate']:.1f}%"
         table.add_row(
-            problem['frontendQuestionId'],
-            problem['title'],
-            problem['difficulty'],
+            problem["frontendQuestionId"],
+            problem["title"],
+            problem["difficulty"],
             status_icon,
-            ac_rate
+            ac_rate,
         )
 
     console.print(table)
-    console.print("\n")
+    console.print("\n")