autocrate/src/config/mod.rs

use std::str::FromStr;
use std::{
    collections::HashMap, fmt::Debug, fs::File, io::BufReader, path::PathBuf, process::Command,
};

use git2;
use libpt::log::{debug, error, trace};
use serde::{Deserialize, Serialize};
use url::Url;

use crate::error::*;

pub mod cli;
pub mod packages;
use cli::Cli;

pub trait YamlConfigSection: Debug + Clone + for<'a> Deserialize<'a> {
    fn check(&self) -> Result<()>;
}

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct Changelog {
    pub enable: bool,
    #[serde(alias = "git-log")]
    pub git_log: bool,
}
impl YamlConfigSection for Changelog {
    fn check(&self) -> Result<()> {
        Ok(())
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct UseCargo {
    pub publish: bool,
    pub registries: Vec<String>,
}
impl YamlConfigSection for UseCargo {
    fn check(&self) -> Result<()> {
        Ok(())
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct Uses {
    cargo: UseCargo,
}
impl YamlConfigSection for Uses {
    fn check(&self) -> Result<()> {
        self.cargo.check()?;
        Ok(())
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
#[serde(rename_all = "snake_case")]
pub enum Pass {
    /// pass specified as plainext
    #[serde(alias = "pass_text")]
    Text(String),
    /// pass to be loaded from an env var
    #[serde(alias = "pass_env")]
    Env(String),
    /// pass to be loaded from a file
    #[serde(alias = "pass_file")]
    File(PathBuf),
}
impl Pass {
    /// Get the pass, extracting from the underlying source
    pub fn get_pass(&self) -> Result<String> {
        self.check()?;
        Ok(match self {
            Self::Text(pass) => pass.clone(),
            Self::Env(key) => std::env::var(key).map_err(ConfigError::from)?,
            Self::File(file) => std::fs::read_to_string(file)?,
        })
    }
}
impl YamlConfigSection for Pass {
    fn check(&self) -> Result<()> {
        match self {
            Self::Text(_) => (),
            Self::Env(envvar) => {
                if !std::env::var(envvar).map_err(ConfigError::from)?.is_empty() {
                } else {
                    return Err(ConfigError::EnvNotSet(envvar.clone()).into());
                }
            }
            Self::File(file) => {
                if !file.exists() {
                    return Err(ConfigError::PassFileDoesNotExist(file.clone()).into());
                }
            }
        };
        Ok(())
    }
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct ApiAuth {
    pub user: String,
    pub pass: Pass,
}
impl YamlConfigSection for ApiAuth {
    fn check(&self) -> Result<()> {
        self.pass.check()?;
        Ok(())
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct Api {
    #[serde(alias = "type")]
    pub server_type: ApiType,
    /// May be left empty if the [ApiType] is [Github](ApiType::Github).
    pub endpoint: Option<Url>,
    /// May be left empty if the Api does not need auth or the auth is part of the
    /// [endpoint](Api::endpoint) [Url].
    pub auth: Option<ApiAuth>,
    /// Name of the repository on the Git server, as git itself has no concept of repository name
    pub repository: String,
}
impl YamlConfigSection for Api {
    fn check(&self) -> Result<()> {
        self.server_type.check()?;
        if self.server_type != ApiType::Github {
            if self.auth.is_none() {
                return Err(ConfigError::NoEndpointSet.into());
            }
            match self.endpoint.clone().unwrap().socket_addrs(|| None) {
                Ok(_) => (),
                Err(err) => return Err(err.into()),
            }
        } else if let Some(_url) = &self.endpoint {
            return Err(ConfigError::EndpointSetButNotNeeded.into());
        }
        if self.auth.is_some() {
            self.auth.clone().unwrap().check()?;
        }
        Ok(())
    }
}

#[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq)]
pub enum ApiType {
    #[serde(alias = "gitea")]
    Gitea,
    #[serde(alias = "gitlab")]
    Gitlab,
    #[serde(alias = "github", alias = "GitHub")]
    Github,
    #[serde(alias = "forgejo")]
    Forgejo,
}
impl ApiType {
    pub fn default_endpoint(&self) -> Option<Url> {
        match self {
            Self::Github => Some(Url::from_str("https://github.com").unwrap()),
            _ => None,
        }
    }
}

impl YamlConfigSection for ApiType {
    fn check(&self) -> Result<()> {
        Ok(())
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
#[serde(rename_all = "snake_case")]
pub enum Version {
    Text(String),
    Cmd(String),
    Cargo,
}

impl Version {
    pub fn get_version(&self) -> String {
        // TODO: Error handling
        match self {
            Self::Text(ver) => ver.clone(),
            Self::Cmd(shell_command) => {
                match Command::new("/bin/bash")
                    .arg("-c")
                    .arg(shell_command)
                    .output()
                {
                    Ok(output) => {
                        // TODO: check status
                        String::from_utf8(output.stdout).unwrap()
                    }
                    Err(err) => {
                        panic!("{err:?}");
                    }
                }
            }
            Self::Cargo => {
                match Command::new("/bin/bash")
                    .arg("-c")
                    .arg(r#"cat Cargo.toml | rg '^\s*version\s*=\s*"([^"]*)"\s*$' -or '$1'"#)
                    .output()
                {
                    Ok(output) => {
                        // TODO: check status
                        String::from_utf8(output.stdout).unwrap().trim().to_owned()
                    }
                    Err(err) => {
                        panic!("{err:?}");
                    }
                }

            },
        }
    }
}

impl YamlConfigSection for Version {
    fn check(&self) -> Result<()> {
        match self {
            Self::Text(_) => (),
            Self::Cmd(_cmd) => {
                // TODO: get the version with a command
                todo!("verion from cmd not implemented")
            }
            Self::Cargo => {
                // TODO: get the version as specified in a Cargo.toml
                todo!("verion from cargo not implemented")
            }
        }
        Ok(())
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct YamlConfig {
    pub changelog: Changelog,
    pub uses: Uses,
    pub api: HashMap<String, Api>,
    pub version: Version,
}
impl YamlConfigSection for YamlConfig {
    fn check(&self) -> Result<()> {
        self.changelog.check()?;
        self.uses.check()?;
        self.version.check()?;
        for api in self.api.values() {
            api.check()?;
        }
        Ok(())
    }
}

impl YamlConfig {
    /// check if the built configuration is valid
    pub fn check(&self) -> Result<()> {
        for api in &self.api {
            api.1.check()?;
        }
        Ok(())
    }
}

pub struct Config {
    pub yaml: YamlConfig,
    pub cli: Cli,
    pub repo: git2::Repository,
    pub path: PathBuf,
}

impl Debug for Config {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{}",
            format_args!(
                "Config {{yaml: {:?}, repo_path: {:?}}}",
                self.yaml, self.path
            )
        )
    }
}

impl Config {
    pub fn load(cli: &Cli) -> Result<Self> {
        let repo = match git2::Repository::open_from_env() {
            Ok(repo) => repo,
            Err(_err) => {
                let err = ConfigError::GitRepoNotFound.into();
                error!("{err}");
                return Err(err);
            }
        };
        let mut path = repo.path().to_path_buf();
        path.pop(); // we want the real root, not the `.git` dir

        let yaml_file_name = if path.join(".autocrate.yaml").exists() {
            ".autocrate.yaml"
        } else if path.join(".autocrate.yml").exists() {
            ".autocrate.yml"
        } else {
            let err = ConfigError::NoYamlFile.into();
            error!("{err}");
            return Err(err);
        };
        let yaml_file_path = path.join(yaml_file_name);
        // we can be sure it exists from the checks above
        assert!(yaml_file_path.exists());
        if !yaml_file_path.is_file() {
            let err = ConfigError::YamlFileIsNotFile.into();
            error!("{err}");
            return Err(err);
        }

        let yaml_rd = BufReader::new(File::open(yaml_file_path)?);
        debug!("reading yaml config and building data structure");
        let yaml: YamlConfig = serde_yaml::from_reader(yaml_rd)?;
        trace!("load config:\n{:#?}", yaml);
        yaml.check()?;
        debug!("built and checked yaml config");

        Ok(Config {
            yaml,
            repo,
            path,
            cli: cli.clone(),
        })
    }
}