aarty/src/args.rs

pub mod args {
    use clap::{Parser, arg, ColorChoice};
    use super::enums::*;

    #[derive(Parser, Debug)]
    #[command(author, version, about, long_about=None, color=ColorChoice::Always)]
    pub struct Arguments {
        /// The art mode to use
        #[arg(short, long, default_value="normal-ascii")]
        pub mode: Mode,
        #[arg(long, default_value="stdout", alias="mo")]
        pub output_method: OutputMethod,
        /// The image to convert to ASCII art
        pub image: String,
        /// The character to use for drawing the image (lighter to darker)
        /// You can user one character if you uses the color mode
        #[arg(short, long, default_value=" .,-~!;:=*&%$@#")]
        pub characters: String,
        /// The output scale (1 is the original size)
        #[arg(short, long, default_value="4")]
        pub scale: u32,
        /// The output file to write to (if output_method is file)
        #[arg(short, long, default_value="ascii_image.txt")]
        pub output: String,
    }

    impl Arguments {
        pub fn validate(&self) -> Result<(), String> {
            if self.characters.len() == 0 {
                return Err("No characters provided".to_string());
            } else if self.characters.len() == 1 {
                if self.mode == Mode::NormalAscii {
                    return Err("One character provided but mode is normal-ascii".to_string());
                }
            } else if self.characters.len() > 32 {
                return Err("Too many characters provided, max is 32".to_string());
            }
            Ok(())
        }
    }


}

pub mod enums {
    use clap::ValueEnum;

     #[derive(Copy, Clone, ValueEnum, Debug, PartialOrd, Eq, PartialEq)]
    pub enum Mode {
        /// Normal ASCII art
        #[clap(alias = "n")]
        NormalAscii,
        /// Colored ASCII art, the colors are based on the terminal colors
        #[clap(alias = "c")]
        COLORED,
    }

    #[derive(Copy, Clone, ValueEnum, Debug, PartialOrd, Eq, PartialEq)]
    pub enum OutputMethod {
        /// Save the ascii art to a file
        #[clap(alias = "f")]
        File,
        /// Print the ascii art to the terminal
        #[clap(alias = "s")]
        Stdout,
    }
}