improved pack feature

Author: dvishal485

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "leetcode-runner-cli"
-version = "0.2.2"
+version = "0.2.5"
 edition = "2021"
 
 [dependencies]

src/main.rs

@@ -1,6 +1,6 @@
 use crate::args::Cli;
 use crate::file_parser::codefile::CodeFile;
-use crate::utils::{execute_testcases, submit};
+use crate::utils::{execute_testcases, pack, submit};
 
 use args::Commands;
 use clap::Parser;
@@ -90,47 +90,8 @@ fn main() -> Result<()> {
                 bail!("Aborting submission due to failed testcase(s)".red().bold());
             }
         }
-        Some(Commands::Pack { file }) => {
-            let code_file = if let Some(path) = file {
-                CodeFile::from_file(&path)?
-            } else {
-                CodeFile::from_dir(".")?
-            };
-            let question = lc.question_content(&code_file.question_title)?;
+        Some(Commands::Pack { file }) => pack(&lc, file)?,
 
-            // create a directory if it doesn't exists with name of question
-            // create a README.md file with the question description
-            // create a file with the code
-            std::fs::create_dir_all(&code_file.question_title.replace(' ', ""))?;
-
-            std::fs::write(
-                format!(
-                    "{}/main.{}",
-                    &code_file.question_title,
-                    code_file.language.extension()
-                ),
-                code_file.code,
-            )?;
-
-            // dont create readme if it exists
-            if let Ok(mut readme_file) = std::fs::OpenOptions::new()
-                .create_new(true)
-                .write(true)
-                .open(format!("{}/{}", &code_file.question_title, GIT_README))
-            {
-                println!(
-                    "You can write your notes about question in {}/README.md",
-                    &code_file.question_title
-                );
-                std::io::Write::write_all(
-                    &mut readme_file,
-                    format!("# {}\n", code_file.question_title).as_bytes(),
-                )?;
-                std::io::Write::write_all(&mut readme_file, question.content.as_bytes())?;
-            } else {
-                println!("{} already exists, skipping creation.", GIT_README);
-            };
-        }
         None => {}
     };
 

src/utils.rs

@@ -1,5 +1,6 @@
 use std::path::Path;
 
+use crate::GIT_README;
 use crate::file_parser::codefile::CodeFile;
 use crate::handlers::leetcode::{Authorized, LeetCode};
 use crate::handlers::utils::{ExecutionResult, SubmissionResult};
@@ -52,3 +53,78 @@ pub(crate) fn submit(lc: &LeetCode<Authorized>, code_file: CodeFile) -> Result<(
 
     Ok(())
 }
+
+pub(crate) fn pack(lc: &LeetCode<Authorized>, file: Option<std::path::PathBuf>) -> Result<()> {
+    let code_file = if let Some(path) = file {
+        CodeFile::from_file(&path)?
+    } else {
+        CodeFile::from_dir(".")?
+    };
+    let question = lc.question_content(&code_file.question_title)?;
+
+    // create a directory if it doesn't exists with name of question
+    // create a README.md file with the question description
+    // create a file with the code
+    std::fs::create_dir_all(&code_file.question_title.replace(' ', ""))?;
+
+    std::fs::write(
+        format!(
+            "{}/main.{}",
+            &code_file.question_title,
+            code_file.language.extension()
+        ),
+        code_file.code,
+    )?;
+
+    // dont create readme if it exists
+    if let Ok(mut readme_file) = std::fs::OpenOptions::new()
+        .create_new(true)
+        .write(true)
+        .open(format!("{}/{}", &code_file.question_title, GIT_README))
+    {
+        println!(
+            "You can write your notes about question in {}/README.md",
+            &code_file.question_title
+        );
+        std::io::Write::write_all(
+            &mut readme_file,
+            format!("# {}\n", code_file.question_title).as_bytes(),
+        )?;
+        std::io::Write::write_all(&mut readme_file, question.content.as_bytes())?;
+    } else {
+        println!("{} already exists, skipping creation.", GIT_README);
+    };
+
+    let mut root_readme_file = std::fs::OpenOptions::new()
+        .create(true)
+        .read(true)
+        .write(true)
+        .open(GIT_README)?;
+    if root_readme_file.metadata()?.len() == 0 {
+        std::io::Write::write_all(&mut root_readme_file, "# LeetCode Solutions\n\n".as_bytes())?;
+        std::io::Write::write_all(
+            &mut root_readme_file,
+            format!(
+                "- [{title}]({title}/main.{ex})\n",
+                ex = code_file.language.extension(),
+                title = code_file.question_title,
+            )
+            .as_bytes(),
+        )?;
+    } else {
+        let mut contents = String::new();
+        std::io::Read::read_to_string(&mut root_readme_file, &mut contents)?;
+        if !contents.contains(&code_file.question_title) {
+            std::io::Write::write_all(
+                &mut root_readme_file,
+                format!(
+                    "- [{title}]({title}/main.{ex})\n",
+                    ex = code_file.language.extension(),
+                    title = code_file.question_title,
+                )
+                .as_bytes(),
+            )?;
+        }
+    }
+    Ok(())
+}