From afb6525ce67e5541255fdd256b48a5ce17b224a4 Mon Sep 17 00:00:00 2001
From: holly sparkles <sparkles@holly.sh>
Date: Tue, 19 Dec 2023 22:09:30 +0100
Subject: [PATCH] feat: add cli args

--filename-format formats the filename
--out is the output directory or file
---
 Cargo.toml  |  2 ++
 src/cli.rs  | 15 +++++++++++++++
 src/main.rs | 45 ++++++++++++++++++++++++++++++++++++++-------
 3 files changed, 55 insertions(+), 7 deletions(-)
 create mode 100644 src/cli.rs

diff --git a/Cargo.toml b/Cargo.toml
index 6b8b219..1473e25 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -7,4 +7,6 @@ edition = "2021"
 
 [dependencies]
 chrono = "0.4.31"
+clap = { version = "4.4.11", features = ["derive"] }
 screenshots = "0.8.6"
+serde = { version = "1.0.193", features = ["derive"] }
diff --git a/src/cli.rs b/src/cli.rs
new file mode 100644
index 0000000..75d895d
--- /dev/null
+++ b/src/cli.rs
@@ -0,0 +1,15 @@
+use clap::Parser;
+
+#[derive(Parser, Debug)]
+pub struct Args {
+    /// The format to use for timestamps
+    #[arg(long, default_value = "screenshot-%Y-%m-%d_%H-%M-%S.png")]
+    pub filename_format: Option<String>,
+
+    /// The output directory to write the file to.
+    ///
+    /// If a filename is specified instead of a directory, this will be used
+    /// for the output filename in place of `filename_format`.
+    #[arg(short, long)]
+    pub out: String,
+}
diff --git a/src/main.rs b/src/main.rs
index 28d5875..fedf979 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,11 +1,18 @@
+use std::path::PathBuf;
+
 use chrono::{DateTime, Local, Utc};
+use clap::Parser;
 use screenshots::{image::ImageBuffer, Screen};
 
+mod cli;
+
 fn main() {
-    capture_all();
+    let args = cli::Args::parse();
+
+    capture_all(args.out, args.filename_format.unwrap_or_default());
 }
 
-fn capture_all() {
+fn capture_all(out_dir: String, filename_format: String) {
     let mut screens = Screen::all().unwrap_or_else(|err| {
         eprintln!("there was an error retrieving screens: {:?}", err);
         vec![]
@@ -40,12 +47,14 @@ fn capture_all() {
         });
 
     if !img.is_empty() {
-        let filename: DateTime<Local> = Utc::now().with_timezone(&Local);
+        let timestamp: DateTime<Local> = Utc::now().with_timezone(&Local);
 
-        match img.save(format!(
-            "target/{}.png",
-            filename.format("%Y-%m-%d_%H-%M-%S")
-        )) {
+        let mut path = PathBuf::from(out_dir);
+        if path.is_dir() {
+            path = path.join(timestamp.format(&filename_format).to_string());
+        }
+
+        match img.save(generate_filename(&path)) {
             Ok(_) => println!("screenshot saved successfully!"),
             Err(err) => println!("there was an error saving the image: {:?}", err),
         }
@@ -63,3 +72,25 @@ impl std::fmt::Display for ImageDimensions {
         f.write_fmt(format_args!("({}, {})", self.x, self.y))
     }
 }
+
+fn generate_filename(filename: &PathBuf) -> PathBuf {
+    let mut index = 0;
+    let mut new_filename = filename.clone();
+
+    let extension = filename
+        .extension()
+        .map(|ext| format!(".{}", ext.to_string_lossy()))
+        .unwrap_or_default();
+
+    while new_filename.exists() {
+        index += 1;
+        new_filename.set_file_name(format!(
+            "{}_{}{}",
+            filename.file_stem().unwrap().to_string_lossy(),
+            index,
+            extension
+        ));
+    }
+
+    new_filename
+}
--
libgit2 1.7.2