feat: error fix, save problem to md file..

Author: yuvrajsinh5252

src/commands/daily.py

@@ -5,18 +5,35 @@
 def daily(
     lang: str = typer.Argument("py", help="Programming language to use."),
     editor: str = typer.Option("vim", '-e', help="Code editor to use."),
+    no_editor: bool = typer.Option(False, "--no-editor", help="Skip opening editor"),
 ):
-    """Check the daily problem."""
+    """
+    Get and work on today's LeetCode daily challenge.
+
+    Fetches the daily coding challenge, displays problem details,
+    and opens it in your preferred editor.
+    """
     from .show import show
 
+    typer.echo(typer.style("Welcome to LeetCode Daily!", fg=typer.colors.GREEN))
     if editor not in ['code', 'vim', 'nano']:
         typer.echo(typer.style(f"❌ Unsupported editor: {editor}", fg=typer.colors.RED))
         raise typer.Exit(1)
 
-    result = get_daily_question()
-    question = result['data']['activeDailyCodingChallengeQuestion']
+    typer.echo(typer.style("Fetching daily challenge...", fg=typer.colors.GREEN), nl=False)
+    try:
+        result = get_daily_question()
+        question = result['data']['activeDailyCodingChallengeQuestion']
+        typer.echo("\r" + " " * 30 + "\r", nl=False)
+    except Exception as e:
+        typer.echo("\n" + typer.style(f"❌ Failed to fetch daily question: {str(e)}", fg=typer.colors.RED))
+        raise typer.Exit(1)
 
     show(problem=question['question']['titleSlug'], layout=True)
 
-    if editor:
-        edit(problem=question['question']['titleSlug'], lang=lang, editor=editor)
+    if not no_editor and editor:
+        try:
+            edit(problem=question['question']['titleSlug'], lang=lang, editor=editor)
+        except Exception as e:
+            typer.echo(typer.style(f"❌ Failed to open editor: {str(e)}", fg=typer.colors.RED))
+            raise typer.Exit(1)

src/commands/edit.py

@@ -9,7 +9,7 @@
 def edit(
   problem: str = typer.Argument(..., help="Problem name or id."),
   lang: str = typer.Argument("cpp", help="Programming language to use."),
-  editor: str = typer.Option("vim", '-e', '--editor', help="vim editor to use."),
+  editor: str = typer.Option("vim", '-e', '--editor', help="Editor to use for code editing."),
 ):
   """Solves a problem by passing lang param and open it with your code editor."""
   question_data = solution_manager.get_question_data(problem).get('data', {}).get('question')

src/commands/login.py

@@ -11,26 +11,36 @@ def login():
             typer.echo(typer.style(f"Already logged in as {saved_session['user_name']}!", fg=typer.colors.GREEN))
             return
 
-    typer.echo(typer.style("To login, you'll need both CSRF and LEETCODE_SESSION tokens:", fg=typer.colors.YELLOW))
-    typer.echo("\n1. Open LeetCode in your browser")
-    typer.echo("2. Press F12 to open Developer Tools")
-    typer.echo("3. Go to Application tab > Cookies > leetcode.com")
-    typer.echo("4. Find and copy both 'csrftoken' and 'LEETCODE_SESSION' values\n")
-
-    csrf_token = typer.prompt("Please enter your CSRF token")
-    csrf_result = auth_manager.verify_csrf_token(csrf_token)
-
-    if not csrf_result["success"]:
-        typer.echo(typer.style(f"\n✗ CSRF verification failed: {csrf_result['message']}", fg=typer.colors.RED))
-        return
+    instruction = f"""
+                    📝 To get your login tokens:
+    ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+    1️  Open LeetCode in your browser
+    2️  Press F12 to open Developer Tools
+    3️  Go to Application tab → Cookies → leetcode.com
+    4️  Find and copy both 'csrftoken' and 'LEETCODE_SESSION'
+    ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+    """
+
+    typer.echo(typer.style(instruction, fg=typer.colors.BRIGHT_BLACK))
+
+    while True:
+        csrf_token = typer.prompt("Please enter your CSRF token")
+        csrf_result = auth_manager.verify_csrf_token(csrf_token)
+
+        if not csrf_result["success"]:
+            typer.echo(typer.style(f"\n✗ CSRF verification failed: {csrf_result['message']}", fg=typer.colors.RED))
+            return
 
-    leetcode_session = typer.prompt("Please enter your LEETCODE_SESSION token")
-    result = auth_manager.login_with_session(csrf_token, leetcode_session)
+        leetcode_session = typer.prompt("Please enter your LEETCODE_SESSION token")
+        result = auth_manager.login_with_session(csrf_token, leetcode_session)
 
-    if result["success"]:
-        typer.echo(typer.style(f"\n✓ Successfully logged in as {result['user_name']}!", fg=typer.colors.GREEN))
-    else:
-        typer.echo(typer.style(f"\n✗ Login failed: {result['message']}", fg=typer.colors.RED))
+        if result["success"]:
+            typer.echo(typer.style(f"\n✓ Successfully logged in as {result['user_name']}!", fg=typer.colors.GREEN))
+            break
+        else:
+            typer.echo(typer.style(f"\n✗ Login failed: {result['message']}", fg=typer.colors.RED))
+            if not typer.confirm("Do you want to try again?", abort=True):
+                break
 
 def logout():
     """Logout from LeetCode"""

src/commands/show.py

@@ -1,4 +1,5 @@
 import typer
+import os
 from rich.console import Console
 from ..lib.problem_ui import ProblemDetails
 from ..server.auth import Auth
@@ -8,19 +9,52 @@
 solution_manager = SolutionManager(auth_manager.get_session())
 console = Console()
 
-def show(problem: str = typer.Argument(..., help="Problem slug or number (e.g., 'two-sum' or '1')"), layout: bool = False):
-    """Show problem details including description and test cases"""
+def show(
+    problem: str = typer.Argument(..., help="Problem slug or number (e.g., 'two-sum' or '1')"),
+    compact: bool = typer.Option(False, "--compact", "-c", help="Display in compact layout"),
+    save: bool = typer.Option(False, "--save", "-s", help="Save problem description to a file")
+):
+    """
+    Show problem details including description and test cases
+
+    Fetches and displays the problem statement, examples, and metadata.
+    Use --compact for a condensed view or --save to export to a file.
+    """
 
     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)
 
-    data = solution_manager.get_question_data(problem)
-    if not data.get('data', {}).get('question'):
-        typer.echo(typer.style(f"❌ Problem '{problem}' not found", fg=typer.colors.RED))
+    try:
+        data = solution_manager.get_question_data(problem)
+        if not data.get('data', {}).get('question'):
+            typer.echo(typer.style(f"❌ Problem '{problem}' not found", fg=typer.colors.RED))
+            raise typer.Exit(1)
+
+        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_full()
+
+    except Exception as e:
+        typer.echo(typer.style(f"❌ Error: {str(e)}", fg=typer.colors.RED))
         raise typer.Exit(1)
 
-    if not layout:
-        ProblemDetails(data.get('data', {}).get('question')).display_full()
-    else:
-        ProblemDetails(data.get('data', {}).get('question')).display()
+def _save_problem_to_file(question_data):
+    """Save the problem statement to a markdown file"""
+    title_slug = question_data.get('titleSlug', 'problem')
+    title = question_data.get('title', 'Untitled Problem')
+
+    filename = f"{title_slug}.md"
+    with open(filename, 'w') as f:
+        f.write(f"# {title}\n\n")
+        f.write(f"**Difficulty:** {question_data.get('difficulty', 'Unknown')}\n\n")
+        f.write(question_data.get('content', ''))
+
+    typer.echo(typer.style(f"✅ Problem saved to {os.path.abspath(filename)}", fg=typer.colors.GREEN))

src/commands/submit.py

@@ -1,17 +1,40 @@
 import typer
+import os
 from pathlib import Path
+from typing import Optional
 from ..server.auth import Auth
 from ..server.solution_manager import SolutionManager
 
 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 (e.g., 'two-sum')"),
+    problem: str = typer.Argument(..., help="Problem slug or number (e.g., 'two-sum' or '1')"),
     file: Path = typer.Argument(..., help="Path to solution file"),
-    lang: str = typer.Option("python3", help="Programming language")
+    lang: Optional[str] = typer.Option(None, help="Programming language (auto-detected if not specified)"),
+    force: bool = typer.Option(False, "--force", "-f", help="Skip confirmation prompt")
 ):
-    """Submit a solution to LeetCode"""
+    """
+    Submit a solution to LeetCode
+
+    Uploads your solution file to LeetCode and returns the verdict.
+    Language is auto-detected from file extension if not specified.
+    """
     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)
@@ -20,17 +43,50 @@ def submit(
         typer.echo(typer.style(f"❌ File not found: {file}", fg=typer.colors.RED))
         raise typer.Exit(1)
 
-    with open(file, 'r') as f:
-        code = f.read()
+    try:
+        # Auto-detect language from file extension if not provided
+        if not lang:
+            extension = os.path.splitext(file)[1].lower()
+            if extension in LANGUAGE_MAP:
+                lang = LANGUAGE_MAP[extension]
+                typer.echo(typer.style(f"🔍 Auto-detected language: {lang}", fg=typer.colors.BLUE))
+            else:
+                typer.echo(typer.style(f"❌ Could not detect language for {extension} files. Please specify with --lang", fg=typer.colors.RED))
+                raise typer.Exit(1)
+
+        with open(file, 'r') as f:
+            code = f.read()
+
+        # Confirm submission unless forced
+        if not force:
+            typer.echo(typer.style(f"Problem: {problem}", fg=typer.colors.BLUE))
+            typer.echo(typer.style(f"Language: {lang}", fg=typer.colors.BLUE))
+            typer.echo(typer.style(f"File: {file}", fg=typer.colors.BLUE))
+            if not typer.confirm("Do you want to submit this solution?"):
+                typer.echo(typer.style("Submission canceled", fg=typer.colors.YELLOW))
+                raise typer.Exit(0)
+
+        typer.echo(typer.style("📤 Submitting solution...", fg=typer.colors.YELLOW))
+        result = solution_manager.submit_solution(problem, code, lang)
+
+        if result["success"]:
+            status = result['status']
+            status_color = (
+                typer.colors.GREEN if status == "Accepted"
+                else typer.colors.YELLOW if status == "Runtime Error" or status == "Time Limit Exceeded"
+                else typer.colors.RED
+            )
+
+            typer.echo(typer.style(f"\n✨ Status: {status}", fg=status_color))
+            typer.echo(f"⏱️  Runtime: {result['runtime']}")
+            typer.echo(f"💾 Memory: {result['memory']}")
+            typer.echo(f"✅ Passed: {result['passed_testcases']}/{result['total_testcases']} test cases")
 
-    typer.echo(typer.style("📤 Submitting solution...", fg=typer.colors.YELLOW))
-    result = solution_manager.submit_solution(problem, code, lang)
+            if status != "Accepted":
+                typer.echo(typer.style(f"\n❗ Error message: {result.get('error_message', 'No details available')}", fg=typer.colors.RED))
+        else:
+            typer.echo(typer.style(f"\n❌ Submission failed: {result['error']}", fg=typer.colors.RED))
 
-    if result["success"]:
-        status_color = typer.colors.GREEN if result["status"] == "Accepted" else typer.colors.RED
-        typer.echo(typer.style(f"\n✨ Status: {result['status']}", fg=status_color))
-        typer.echo(f"⏱️  Runtime: {result['runtime']}")
-        typer.echo(f"💾 Memory: {result['memory']}")
-        typer.echo(f"✅ Passed: {result['passed_testcases']}/{result['total_testcases']} test cases")
-    else:
-        typer.echo(typer.style(f"\n❌ Submission failed: {result['error']}", fg=typer.colors.RED))
+    except Exception as e:
+        typer.echo(typer.style(f"❌ Error: {str(e)}", fg=typer.colors.RED))
+        raise typer.Exit(1)

src/commands/test.py

@@ -10,8 +10,15 @@
     "py": "python3",
     "java": "java",
     "js": "javascript",
+    "ts": "typescript",
     "c": "c",
-    "cpp": "cpp"
+    "cpp": "cpp",
+    "cs": "csharp",
+    "go": "golang",
+    "rb": "ruby",
+    "swift": "swift",
+    "rs": "rust",
+    "php": "php"
 }
 
 def test(
@@ -33,17 +40,22 @@ def test(
     lang = map_lang.get(file.suffix[1:])
     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)
 
     typer.echo(typer.style("🧪 Testing solution with LeetCode test cases...", fg=typer.colors.YELLOW))
-    result = solution_manager.test_solution(problem, code, lang)
+    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)
 
     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']}")
+            typer.echo(f"⏱️Runtime: {result['runtime']}")
         if "memory" in result:
             typer.echo(f"💾 Memory: {result['memory']}")
         typer.echo("\nTest Case Results:\n")