karton/src/args.rs

use std::convert::Infallible;
use std::fmt;
use std::net::IpAddr;
use std::str::FromStr;
use clap::Parser;
use lazy_static::lazy_static;

lazy_static! {
    pub static ref ARGS: Args = Args::parse();
}

#[derive(Parser, Debug, Clone)]
#[clap(author, version, about, long_about = None)]
pub struct Args {
    #[clap(long, env="MICROBIN_AUTH_USERNAME")]
    pub auth_username: Option<String>,

    #[clap(long, env="MICROBIN_AUTH_PASSWORD")]
    pub auth_password: Option<String>,

    #[clap(long, env="MICROBIN_EDITABLE")]
    pub editable: bool,

    #[clap(long, env="MICROBIN_FOOTER_TEXT")]
    pub footer_text: Option<String>,

    #[clap(long, env="MICROBIN_HIDE_FOOTER")]
    pub hide_footer: bool,

    #[clap(long, env="MICROBIN_HIDE_HEADER")]
    pub hide_header: bool,

    #[clap(long, env="MICROBIN_HIDE_LOGO")]
    pub hide_logo: bool,

    #[clap(long, env="MICROBIN_NO_LISTING")]
    pub no_listing: bool,

    #[clap(long, env="MICROBIN_HIGHLIGHTSYNTAX")]
    pub highlightsyntax: bool,

    #[clap(short, long, env="MICROBIN_PORT", default_value_t = 8080)]
    pub port: u16,

    #[clap(short, long, env="MICROBIN_BIND", default_value_t = IpAddr::from([0, 0, 0, 0]))]
    pub bind: IpAddr,

    #[clap(long, env="MICROBIN_PRIVATE")]
    pub private: bool,

    #[clap(long, env="MICROBIN_PURE_HTML")]
    pub pure_html: bool,

    #[clap(long, env="MICROBIN_PUBLIC_PATH", default_value_t = PublicUrl(String::from("/")))]
    pub public_path: PublicUrl,

    #[clap(long, env="MICROBIN_READONLY")]
    pub readonly: bool,

    #[clap(long, env="MICROBIN_TITLE")]
    pub title: Option<String>,

    #[clap(short, long, env="MICROBIN_THREADS", default_value_t = 1)]
    pub threads: u8,

    #[clap(long, env="MICROBIN_WIDE")]
    pub wide: bool,
}

#[derive(Debug, Clone)]
pub struct PublicUrl(pub String);

impl fmt::Display for PublicUrl {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.0)
    }
}

impl FromStr for PublicUrl {
    type Err = Infallible;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let uri = s.strip_suffix('/').unwrap_or(s).to_owned();
        Ok(PublicUrl(uri))
    }
}