feat: add option to overwrite files if they exist

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "aoc-cli"
 description = "Advent of Code command-line helper tool"
-version = "0.4.8"
+version = "0.5.0"
 authors = ["Sergio de Carvalho <[email protected]>"]
 categories = ["command-line-utilities"]
 edition = "2021"

README.md

@@ -111,6 +111,7 @@ Options:
   -y, --year <YEAR>          Puzzle year [default: year of current or last Advent of Code event]
   -s, --session-file <PATH>  Path to session cookie file [default: ~/.adventofcode.session]
   -w, --width <WIDTH>        Width at which to wrap output [default: terminal width]
+  -o, --overwrite            Overwrite file if it already exists
   -i, --input-file <PATH>    Path where to save puzzle input [default: input]
   -p, --puzzle-file <PATH>   Path where to save puzzle description [default: puzzle.md]
   -h, --help                 Print help information

src/aoc.rs

@@ -105,6 +105,7 @@ pub fn download_input(
     opt_year: Option<PuzzleYear>,
     opt_day: Option<PuzzleDay>,
     filename: &str,
+    overwrite: bool,
 ) -> Result<(), String> {
     let (year, day) = puzzle_year_day(opt_year, opt_day)?;
 
@@ -119,9 +120,15 @@ pub fn download_input(
         .map_err(|err| err.to_string())?;
 
     eprintln!("Saving puzzle input to \"{}\"...", filename);
-    OpenOptions::new()
-        .write(true)
-        .create_new(true)
+
+    let mut file = OpenOptions::new();
+    if overwrite {
+        file.create(true);
+    } else {
+        file.create_new(true);
+    };
+
+    file.write(true)
         .open(filename)
         .map_err(|err| format!("Failed to create file: {}", err))?
         .write(puzzle_input.as_bytes())
@@ -173,6 +180,7 @@ pub fn read_puzzle(
     opt_day: Option<PuzzleDay>,
     col_width: usize,
     filename: &str,
+    overwrite: bool,
 ) -> Result<(), String> {
     let (year, day) = puzzle_year_day(opt_year, opt_day)?;
 
@@ -196,9 +204,15 @@ pub fn read_puzzle(
     println!("\n{}", from_read(description.as_bytes(), col_width));
 
     println!("Saving puzzle description to \"{}\"...", filename);
-    OpenOptions::new()
-        .write(true)
-        .create_new(true)
+
+    let mut file = OpenOptions::new();
+    if overwrite {
+        file.create(true);
+    } else {
+        file.create_new(true);
+    };
+
+    file.write(true)
         .open(filename)
         .map_err(|err| format!("Failed to create file: {}", err))?
         .write(parse_html(description).as_bytes())

src/args.rs

@@ -25,6 +25,10 @@ pub struct Args {
     #[arg(short, long, global = true, value_parser = valid_width)]
     pub width: Option<usize>,
 
+    /// Overwrite file if it already exists
+    #[arg(short, long, global = true)]
+    pub overwrite: bool,
+
     /// Path where to save puzzle input
     #[arg(
         short,

src/main.rs

@@ -23,15 +23,24 @@ fn main() -> Result<(), String> {
         .unwrap_or(DEFAULT_COL_WIDTH);
 
     match &args.command {
-        Some(Command::Download) => {
-            download_input(&session, args.year, args.day, &args.input_file)
-        }
+        Some(Command::Download) => download_input(
+            &session,
+            args.year,
+            args.day,
+            &args.input_file,
+            args.overwrite,
+        ),
         Some(Command::Submit { part, answer }) => {
             submit_answer(&session, args.year, args.day, part, answer, width)
         }
-        _ => {
-            read_puzzle(&session, args.year, args.day, width, &args.puzzle_file)
-        }
+        _ => read_puzzle(
+            &session,
+            args.year,
+            args.day,
+            width,
+            &args.puzzle_file,
+            args.overwrite,
+        ),
     }
 }