index : gmss.git

afb6525ce67e5541255fdd256b48a5ce17b224a4 [patch]

31437887f3303029180ef6178abef015ac362afa

0823ae2a16c42434ee2e9401049dd3fb52eb7d1e

afb6525ce67e5541255fdd256b48a5ce17b224a4.tar.gz

feat: add cli args

--filename-format formats the filename
--out is the output directory or file

Diff

 Cargo.toml  |  2 ++
 src/cli.rs  | 15 +++++++++++++++
 src/main.rs | 45 ++++++++++++++++++++++++++++++++++++++-------
 3 files changed, 55 insertions(+), 7 deletions(-)

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
}