feat(downloader): save solution source code

Author: arnav-on-code

leetcode/downloader.py

@@ -17,6 +17,18 @@ def __init__(self):
             exist_ok=True,
         )
 
+    @staticmethod
+    def get_solution_filename(language: str) -> str:
+        mapping = {
+            "python3": "solution.py",
+            "java": "Solution.java",
+            "cpp": "solution.cpp",
+            "c": "solution.c",
+            "javascript": "solution.js",
+        }
+
+        return mapping.get(language, "solution.txt")
+        
     @staticmethod
     def _format_folder_name(question_id: str, slug: str) -> str:
         """
@@ -54,4 +66,23 @@ def create_problem_directory(self, detail) -> Path:
             exist_ok=True,
         )
 
-        return problem_dir
+        return problem_dir
+    
+
+    def save_solution(self, detail) -> Path:
+        """
+        Save the source code into the appropriate solution file.
+        """
+
+        filename = self.get_solution_filename(detail.language)
+
+        problem_dir = self.create_problem_directory(detail)
+
+        solution_file = problem_dir / filename
+
+        solution_file.write_text(
+            detail.code,
+            encoding="utf-8",
+        )
+
+        return solution_file

main.py

@@ -10,7 +10,7 @@
 from leetcode.api import LeetCodeAPI
 from config.settings import Config
 
-from leetcode.downloader import SubmissionDownloader
+from leetcode.downloader import SubmissionDownloader 
 
 
 console = Console()
@@ -140,9 +140,11 @@ def main():
 
     downloader = SubmissionDownloader()
 
-    folder = downloader.create_problem_directory(detail)
+    solution_path = downloader.save_solution(detail)
 
-    console.print(f"\n[green]Directory created:[/green] {folder}")
+    console.print(
+        f"\n[bold green]Solution saved:[/bold green]\n{solution_path}"
+    )
 
 if __name__ == "__main__":
     main()