feat: language mapping and result handling

yuvrajsinh5252 · yuvrajsinh5252 · commit e61a2be9744d · 2025-03-13T05:00:57.000Z
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,5 @@
 TODO.md
 __pycache__
 .venv
-leetcode_cli.egg-info
+leetcode_cli.egg-info
+*.cpp
diff --git a/src/commands/edit.py b/src/commands/edit.py
@@ -2,7 +2,7 @@
 
 from ..server.solution_manager import SolutionManager
 from ..server.auth import Auth
-from .test import map_lang
+from ..server.config import LANGUAGE_MAP
 
 solution_manager = SolutionManager(Auth().get_session())
 
@@ -26,7 +26,7 @@ def create_file_with_template(lang: str):
     filename = f"{filename_prefix}.{lang}"
     with open(filename, "w") as f:
       for snippet in question_data.get('codeSnippets', []):
-          if snippet.get('langSlug').lower() == map_lang.get(lang):
+          if snippet.get('langSlug').lower() == LANGUAGE_MAP.get(lang):
             f.write(snippet.get('code'))
             return
       typer.echo(f"No template found for language {lang}")
diff --git a/src/commands/submit.py b/src/commands/submit.py
@@ -4,6 +4,7 @@
 from typing import Optional
 from ..server.auth import Auth
 from ..server.solution_manager import SolutionManager
+from ..server.config import LANGUAGE_MAP
 from ..lib.submission_ui import (
     display_auth_error, display_file_not_found_error, display_language_detection_message,
     display_language_detection_error, display_problem_not_found_error, display_submission_details,
@@ -14,21 +15,6 @@
 auth_manager = Auth()
 solution_manager = SolutionManager(auth_manager.get_session())
 
-LANGUAGE_MAP = {
-    '.py': 'python3',
-    '.cpp': 'cpp',
-    '.c': 'c',
-    '.java': 'java',
-    '.js': 'javascript',
-    '.ts': 'typescript',
-    '.go': 'golang',
-    '.rs': 'rust',
-    '.rb': 'ruby',
-    '.cs': 'csharp',
-    '.swift': 'swift',
-    '.php': 'php',
-}
-
 def submit(
     problem: str = typer.Argument(..., help="Problem slug or number (e.g., 'two-sum' or '1')"),
     file: Path = typer.Argument(..., help="Path to solution file"),
diff --git a/src/commands/test.py b/src/commands/test.py
@@ -2,68 +2,50 @@
 from pathlib import Path
 from ..server.auth import Auth
 from ..server.solution_manager import SolutionManager
+from ..server.config import LANGUAGE_MAP
+from ..lib.submission_ui import (
+    display_auth_error, display_file_not_found_error,
+    display_language_detection_error, display_submission_results, display_exception_error,
+    create_submission_progress, display_language_detection_message
+)
 
 auth_manager = Auth()
 solution_manager = SolutionManager(auth_manager.get_session())
 
-map_lang = {
-    "py": "python3",
-    "java": "java",
-    "js": "javascript",
-    "ts": "typescript",
-    "c": "c",
-    "cpp": "cpp",
-    "cs": "csharp",
-    "go": "golang",
-    "rb": "ruby",
-    "swift": "swift",
-    "rs": "rust",
-    "php": "php"
-}
+FILE_EXT_TO_LANG = {ext.lstrip('.'): lang for ext, lang in LANGUAGE_MAP.items()}
 
 def test(
     problem: str = typer.Argument(..., help="Problem slug (e.g., 'two-sum')"),
     file: Path = typer.Argument(..., help="Path to solution file"),
 ):
     """Test a solution with LeetCode's test cases"""
     if not auth_manager.is_authenticated:
-        typer.echo(typer.style("❌ Please login first using the login command", fg=typer.colors.RED))
-        raise typer.Exit(1)
+        display_auth_error()
 
     if not file.exists():
-        typer.echo(typer.style(f"❌ File not found: {file}", fg=typer.colors.RED))
-        raise typer.Exit(1)
+        display_file_not_found_error(file)
 
     with open(file, 'r') as f:
         code = f.read()
 
-    lang = map_lang.get(file.suffix[1:])
+    file_ext = file.suffix[1:] if file.suffix else ""
+    lang = FILE_EXT_TO_LANG.get(file_ext)
     if not lang:
-        typer.echo(typer.style(f"❌ Unsupported file extension: {file.suffix}", fg=typer.colors.RED))
-        typer.echo(f"Supported extensions: {', '.join(map_lang.keys())}")
-        raise typer.Exit(1)
+        display_language_detection_error(file.suffix)
+        return
+
+    display_language_detection_message(lang)
 
     typer.echo(typer.style("🧪 Testing solution with LeetCode test cases...", fg=typer.colors.YELLOW))
+
     try:
-        result = solution_manager.test_solution(problem, code, lang)
-    except Exception as e:
-        typer.echo(typer.style(f"❌ Error connecting to LeetCode: {str(e)}", fg=typer.colors.RED))
-        raise typer.Exit(1)
+        with create_submission_progress() as progress:
+            test_task = progress.add_task("Testing...", total=1)
+            progress.update(test_task, advance=0.5)
+            result = solution_manager.test_solution(problem, code, lang)
+            progress.update(test_task, advance=0.5)
 
-    if result["success"]:
-        status_color = typer.colors.GREEN if result["status"] == "Accepted" else typer.colors.RED
-        status_prefix = "✨" if result['status'] == "Accepted" else "❌"
-        typer.echo(typer.style(f"\n{status_prefix} Status: {result['status']}", fg=status_color))
-        if "runtime" in result:
-            typer.echo(f"⏱️Runtime: {result['runtime']}")
-        if "memory" in result:
-            typer.echo(f"💾 Memory: {result['memory']}")
-        typer.echo("\nTest Case Results:\n")
-        typer.echo(f"📤 Your Output:\n{result['output']}")
-        typer.echo(f"✅ Expected:\n{result['expected']}")
-    else:
-        typer.echo(typer.style(f"\n❌ {result['status']}", fg=typer.colors.BRIGHT_RED))
-        error_message = f"\n{result['error']}"
-        if result.get('full_error'):
-            error_message += f"\n\nFull error:\n{result['full_error']}"
-        typer.echo(typer.style(error_message, fg=typer.colors.RED))
+        display_submission_results(result)
+
+    except Exception as e:
+        display_exception_error(e)
diff --git a/src/lib/submission_ui.py b/src/lib/submission_ui.py
@@ -1,9 +1,7 @@
 from rich.console import Console
 from rich.panel import Panel
-from rich.table import Table
 from rich.progress import Progress, BarColumn, TextColumn
 from rich import box
-from rich.text import Text
 from rich.columns import Columns
 import typer
 
@@ -37,7 +35,6 @@ def display_problem_not_found_error(problem):
 
 def display_submission_details(problem, problem_name, lang, file):
     """Display submission details and confirmation prompt using a clean layout"""
-    # Create formatted text lines instead of a nested table
     content = [
         f"[cyan]Problem:[/] {problem} - {problem_name}",
         f"[cyan]Language:[/] {lang}",
@@ -71,7 +68,6 @@ def display_submission_results(result):
     if result["success"]:
         status = result['status']
 
-        # Set colors based on status
         if status == "Accepted":
             status_style = "bold green"
             border_style = "green"
@@ -85,19 +81,27 @@ def display_submission_results(result):
             border_style = "red"
             emoji = "❌"
 
-        # Get metrics
         runtime = result.get('runtime', 'N/A')
         memory = result.get('memory', 'N/A')
+
+        memory_warning = result.get('memory_warning')
+        if memory_warning:
+            memory = f"{memory} [bold yellow](!)[/]"
+
         passed = result.get('passed_testcases', 0)
         total = result.get('total_testcases', 0)
         test_case_str = f"{passed}/{total} ({passed/total*100:.1f}%)" if total > 0 else "N/A"
 
         content = [
             f"[cyan]⏱️ Runtime:[/] {runtime}",
-            f"[cyan]💾 Memory:[/] {memory}",
-            f"[cyan]🧪 Test Cases:[/] {test_case_str}"
+            f"[cyan]💾 Memory:[/] {memory}"
         ]
 
+        if result.get('elapsed_time'):
+            content.append(f"[cyan]⏲️ Elapsed Time:[/] {result.get('elapsed_time')} ms")
+
+        content.append(f"[cyan]🧪 Test Cases:[/] {test_case_str}")
+
         title = f"{emoji} Submission Result: [{status_style}]{status}[/]"
         console.print(Panel(
             "\n".join(content),
@@ -106,9 +110,48 @@ def display_submission_results(result):
             box=box.ROUNDED
         ))
 
-        if status != "Accepted" and result.get('error_message'):
-            error_msg = result.get('error_message', 'No details available')
+        if memory_warning:
+            console.print(Panel(
+                memory_warning,
+                title="⚠️ Memory Usage Warning",
+                border_style="yellow",
+                box=box.ROUNDED
+            ))
+
+        if result.get('stdout'):
+            console.print(Panel(
+                result.get('stdout'),
+                title="📝 Standard Output",
+                border_style="blue",
+                box=box.ROUNDED
+            ))
+
+        if result.get('output') and result.get('expected'):
+            is_wrong_answer = status == "Wrong Answer"
+
+            output_panel = Panel(
+                result.get('output', ''),
+                title="Your Output",
+                border_style="red" if is_wrong_answer else "blue"
+            )
+            expected_panel = Panel(
+                result.get('expected', ''),
+                title="Expected Output",
+                border_style="green"
+            )
+            console.print(Columns([output_panel, expected_panel]))
+
+        if status != "Accepted" and result.get('error'):
+            error_msg = result.get('error', 'No details available')
             console.print(Panel(error_msg, title="Error Details", border_style="red"))
+
+            if result.get('full_error'):
+                console.print(Panel(
+                    result.get('full_error', ''),
+                    title="Full Error Trace",
+                    border_style="red",
+                    box=box.ROUNDED
+                ))
     else:
         error_panel = Panel(
             f"{result.get('error', 'Unknown error')}",
@@ -117,6 +160,22 @@ def display_submission_results(result):
         )
         console.print(error_panel)
 
+        if result.get('stdout'):
+            console.print(Panel(
+                result.get('stdout'),
+                title="📝 Standard Output",
+                border_style="blue",
+                box=box.ROUNDED
+            ))
+
+        if result.get('full_error'):
+            console.print(Panel(
+                result.get('full_error', ''),
+                title="Full Error Trace",
+                border_style="red",
+                box=box.ROUNDED
+            ))
+
 def display_exception_error(e):
     """Display exception error message"""
     console.print(Panel(f"❌ Error: {str(e)}", style="bold red", border_style="red"))
diff --git a/src/server/config.py b/src/server/config.py
@@ -1 +1,28 @@
 BASE_URL = "https://leetcode.com/graphql"
+LEETCODE_BASE_URL = "https://leetcode.com"
+STATUS_CODES = {
+    10: "Accepted",
+    14: "Time Limit Exceeded",
+    15: "Runtime Error",
+    16: "Memory Limit Exceeded",
+    20: "Compile Error",
+    30: "Wrong Answer"
+}
+LANGUAGE_MAP = {
+    '.py': 'python3',
+    '.cpp': 'cpp',
+    '.c': 'c',
+    '.java': 'java',
+    '.js': 'javascript',
+    '.ts': 'typescript',
+    '.go': 'golang',
+    '.rs': 'rust',
+    '.rb': 'ruby',
+    '.cs': 'csharp',
+    '.swift': 'swift',
+    '.php': 'php',
+}
+MEMORY_WARNING_THRESHOLD = 100000000  # 100MB
+MEMORY_LIMIT_THRESHOLD = 450000000    # 450MB
+TEST_RESULT_TIMEOUT = 30
+SUBMISSION_RESULT_TIMEOUT = 20
diff --git a/src/server/solution_manager.py b/src/server/solution_manager.py
