feat: problem ui done

Author: yuvrajsinh5252

src/commands/show.py

@@ -35,19 +35,19 @@ def show(
 
         if not data.get('data', {}).get('question'):
             typer.echo(typer.style(f"❌ Problem '{problem}' not found", fg=typer.colors.RED))
-            raise typer.Exit(1)
+            raise typer.Exit(data.get('errors', ['Unknown error']))
 
         question = data.get('data', {}).get('question')
         problem_details = ProblemDetails(question)
 
         if save:
             _save_problem_to_file(question)
 
-        if compact:
-            problem_details.display()
-        else:
-            problem_details.display()
-            # problem_details.display_full()
+        problem_details.display_probelm()
+
+        if not compact:
+            problem_details.display_stats()
+            problem_details.display_additional_info()
 
     except Exception as e:
         typer.echo(typer.style(f"❌ Error: {str(e)}", fg=typer.colors.RED))

src/lib/problem_ui.py

@@ -36,8 +36,7 @@ def __init__(self, problem_data: dict):
         self.difficulty = problem_data['difficulty']
         self.stats = problem_data['stats']
         self.topics = problem_data.get('topicTags', [])
-        self.companies = problem_data.get('companyTagStats', {}).get('companies', []) if 'companyTagStats' in problem_data else []
-        self.similar_questions = json.loads(problem_data.get('similarQuestions', '[]'))
+        self.similar_questions = problem_data.get('similarQuestionList', [])
 
         if isinstance(self.stats, str):
             try:
@@ -92,16 +91,26 @@ def _format_similar_questions(self) -> str:
         for q in self.similar_questions[:5]:
             title = q.get('title', '')
             difficulty = q.get('difficulty', '')
+            question_id = q.get('questionId', '') or q.get('titleSlug', '')
+            lock_symbol = " 🔒" if q.get('isPaidOnly', False) else ""
+            diff_color = COLORS['difficulty'].get(difficulty, 'white')
+
             if title and difficulty:
-                diff_color = COLORS['difficulty'].get(difficulty, 'white')
-                lines.append(f"- {title} ([{diff_color}]{difficulty}[/{diff_color}])")
+                if 'titleSlug' in q:
+                    slug = q.get('titleSlug')
+                    title_display = f"[link=https://leetcode.com/problems/{slug}/]{title}[/link]"
+                else:
+                    title_display = title
+
+                question_line = f"- [{COLORS['problem_number']}]#{question_id}[/] {title_display} {lock_symbol} ([{diff_color}]{difficulty}[/{diff_color}])"
+                lines.append(question_line)
 
         if len(lines) <= 1:
             return ""
 
         return "\n".join(lines)
 
-    def display(self):
+    def display_probelm(self):
         formatted_md = self._format_markdown(self.content)
 
         content_panel = Panel(
@@ -114,6 +123,7 @@ def display(self):
         )
         console.print(content_panel)
 
+    def display_stats(self):
         stats_panel = Panel(
             self._format_stats(),
             title=f"[{COLORS['section_title']}]Statistics[/]",
@@ -122,6 +132,8 @@ def display(self):
             padding=STYLES['panel_padding']
         )
         console.print(stats_panel)
+
+    def display_additional_info(self):
         info_content = []
 
         topics = self._format_topics()
@@ -142,19 +154,4 @@ def display(self):
                 border_style=COLORS['border'],
                 padding=STYLES['panel_padding']
             )
-            console.print(info_panel)
-
-    def display_full(self):
-        self.display()
-
-        if self.companies:
-            company_names = [company.get('name', '') for company in self.companies if 'name' in company]
-            if company_names:
-                companies_panel = Panel(
-                    Text(", ".join(company_names)),
-                    box=SIMPLE,
-                    title="Companies",
-                    border_style=COLORS['border'],
-                    padding=STYLES['panel_padding']
-                )
-                console.print(companies_panel)
+            console.print(info_panel)

src/server/config.py

@@ -2,11 +2,14 @@
 LEETCODE_BASE_URL = "https://leetcode.com"
 STATUS_CODES = {
     10: "Accepted",
+    11: "Wrong Answer",
+    12: "Memory Limit Exceeded",
+    13: "Output Limit Exceeded",
     14: "Time Limit Exceeded",
     15: "Runtime Error",
-    16: "Memory Limit Exceeded",
+    16: "Internal Error",
     20: "Compile Error",
-    11: "Wrong Answer"
+    30: "Timeout"
 }
 LANGUAGE_MAP = {
     '.py': 'python3',

src/server/solution_manager.py

@@ -73,6 +73,15 @@ def get_question_data(self, question_identifier: str) -> Dict[str, Any]:
                     questionFrontendId
                     title
                     titleSlug
+                    topicTags {
+                        name
+                    }
+                    similarQuestionList {
+                        title
+                        titleSlug
+                        difficulty
+                        isPaidOnly
+                    }
                     content
                     difficulty
                     exampleTestcaseList
@@ -88,15 +97,18 @@ def get_question_data(self, question_identifier: str) -> Dict[str, Any]:
             }
         """
 
-        response = self.session.post(
-            f"{self.BASE_URL}/graphql",
-            json={
-                "query": query,
-                "variables": {"titleSlug": title_slug}
-            }
-        )
+        try:
+            response = self.session.post(
+                f"{self.BASE_URL}/graphql",
+                json={
+                    "query": query,
+                    "variables": {"titleSlug": title_slug}
+                }
+            )
 
-        return response.json()
+            return response.json()
+        except Exception as e:
+            return {"error": str(e)}
 
     def _format_output(self, output: Union[str, List, None]) -> str:
         """Format output that could be string or list"""