bricks/config/

lib.rs

use std::{path::PathBuf, process::Command};

use anyhow::Result;
use home::home_dir;
use serde::{Deserialize, Serialize};

use super::overrides::{OverrideDatabase, Overrides};

#[derive(Debug, Serialize, Deserialize)]
pub enum LibKind {
    #[serde(alias = "system")]
    System,
    #[serde(alias = "git")]
    Git,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct Lib {
    #[serde(default = "default_kind")]
    pub kind: LibKind,
    pub repo: Option<String>,
    pub version: Option<String>,

    pub overrides: Option<Overrides>,
    //
    // #[serde(default = "default_cflags")]
    // pub cflags: String,
    // #[serde(default = "default_ldflags")]
    // pub ldflags: String,
    //
    // macos: Option<Platform>,
    // windows: Option<Platform>,
    // linux: Option<Platform>,
}

impl Lib {
    // pub fn platform(&self) -> Option<&Platform> {
    //     if cfg!(target_os = "linux") {
    //         self.linux.as_ref()
    //     } else if cfg!(target_os = "macos") {
    //         self.macos.as_ref()
    //     } else if cfg!(target_os = "windows") {
    //         self.windows.as_ref()
    //     } else {
    //         None
    //     }
    // }
    //
    // pub fn cflags(&self) -> String {
    //     let ext_cflags = &match self.platform() {
    //         Some(platform) => match &platform.cflags {
    //             Some(v) => v,
    //             None => "",
    //         },
    //         None => "",
    //     };
    //     format!("{} {}", self.cflags, ext_cflags)
    // }
    //
    // pub fn ldflags(&self) -> String {
    //     let ext_ldflags = &match self.platform() {
    //         Some(platform) => match &platform.ldflags {
    //             Some(v) => v,
    //             None => "",
    //         },
    //         None => "",
    //     };
    //     format!("{} {}", self.cflags, ext_ldflags)
    // }

    pub fn headers(&self, name: &str, override_db: &OverrideDatabase) -> Result<String> {
        match self.kind {
            LibKind::System => {
                let output = Command::new("pkg-config")
                    .arg(name)
                    .arg("--cflags")
                    .output()?;

                Ok(String::from_utf8(output.stdout)?.trim_end().to_string())
            }
            LibKind::Git => {
                let overrides = override_db.get(name);
                let include_dir = if let Some(overrides) = overrides {
                    overrides
                        .include_dir
                        .as_ref()
                        .map(|include_dir| include_dir.to_string())
                } else {
                    None
                };
                let include_dir = match include_dir {
                    Some(v) => v,
                    None => "build/include".into(),
                };
                Ok(format!(
                    "-I{}",
                    self.pathify_repo()?.join(include_dir).display(),
                ))
            }
        }
    }

    pub fn lib_links(&self, name: &str, override_db: &OverrideDatabase) -> Result<String> {
        match self.kind {
            LibKind::System => {
                let output = Command::new("pkg-config")
                    .arg(name)
                    .args(["--libs", "--static"])
                    .output()?;

                Ok(String::from_utf8(output.stdout)?.trim_end().to_string())
            }
            LibKind::Git => {
                let overrides = override_db.get(name);
                let lib_dir = if let Some(overrides) = overrides {
                    overrides
                        .lib_dir
                        .as_ref()
                        .map(|include_dir| include_dir.to_string())
                } else {
                    None
                };
                let lib_dir = match lib_dir {
                    Some(v) => v,
                    None => "build/lib".into(),
                };
                Ok(format!(
                    "-L{} -l{}",
                    self.pathify_repo()?.join(lib_dir).display(),
                    name,
                ))
            }
        }
    }

    pub fn normalize_repo(&self) -> Option<String> {
        match &self.repo {
            Some(repo) => {
                let mut new_repo = repo.clone();
                if !repo.starts_with("https://") {
                    new_repo = format!("https://{}", new_repo);
                }
                if !new_repo.ends_with(".git") {
                    new_repo = format!("{}.git", new_repo)
                }
                Some(new_repo)
            }
            None => None,
        }
    }

    pub fn directify_repo(&self) -> Option<String> {
        match &self.repo {
            Some(repo) => {
                let new_repo = repo
                    .trim_start_matches("https://")
                    .trim_end_matches(".git")
                    .replace("/", "-")
                    .to_string();
                Some(new_repo)
            }
            None => None,
        }
    }

    pub fn pathify_repo_no_version(&self) -> Result<PathBuf, LibPathificationError> {
        let home = home_dir().ok_or(LibPathificationError::HomeDirMissing)?;
        let lib_dir = self
            .directify_repo()
            .ok_or(LibPathificationError::RepoUriMissing)?;

        Ok(home.join(".bricks").join("libs").join(lib_dir))
    }

    pub fn pathify_repo(&self) -> Result<PathBuf, LibPathificationError> {
        let home = home_dir().ok_or(LibPathificationError::HomeDirMissing)?;
        let lib_dir = self
            .directify_repo()
            .ok_or(LibPathificationError::RepoUriMissing)?;
        let version = self
            .version
            .as_ref()
            .ok_or(LibPathificationError::VersionMissing)?;

        Ok(home
            .join(".bricks")
            .join("libs")
            .join(lib_dir)
            .join(version))
    }
}

#[derive(thiserror::Error, Debug)]
pub enum LibPathificationError {
    #[error("unable to get the home directory")]
    HomeDirMissing,
    #[error("lib is missing the `repo` property")]
    RepoUriMissing,
    #[error("lib is missing the `version` property")]
    VersionMissing,
}

#[cfg(test)]
mod tests {
    use std::path::Path;

    use super::*;

    fn basic_lib() -> Lib {
        Lib {
            kind: LibKind::Git,
            repo: Some("github.com/Tesohh/strings.git".to_string()),
            version: Some("2020".to_string()),
            overrides: None,

            cflags: "".into(),
            ldflags: "".into(),
            macos: None,
            windows: None,
            linux: None,
        }
    }

    #[test]
    fn no_repo_should_return_none() {
        let mut no_repo_lib = basic_lib();
        no_repo_lib.repo = None;

        assert_eq!(no_repo_lib.normalize_repo(), None);
        assert_eq!(no_repo_lib.directify_repo(), None);
        assert!(no_repo_lib.pathify_repo().is_err());
    }

    #[test]
    fn repo_normalization() {
        let mut lib = basic_lib();
        lib.repo = Some("github.com/Tesohh/strings.git".to_string());

        assert_eq!(
            lib.normalize_repo(),
            Some("https://github.com/Tesohh/strings.git".into())
        );

        lib.repo = Some("https://github.com/Tesohh/strings".into());
        assert_eq!(
            lib.normalize_repo(),
            Some("https://github.com/Tesohh/strings.git".into())
        );

        lib.repo = Some("github.com/Tesohh/strings".into());
        assert_eq!(
            lib.normalize_repo(),
            Some("https://github.com/Tesohh/strings.git".into())
        );
    }

    #[test]
    fn repo_directification() {
        let lib = basic_lib();

        assert_eq!(
            lib.directify_repo(),
            Some("github.com-Tesohh-strings".to_string())
        );
    }

    #[test]
    fn repo_pathification() {
        let lib = basic_lib();

        let path = lib.pathify_repo().unwrap();

        assert_eq!(
            path,
            Path::new(&home_dir().unwrap())
                .join(".bricks")
                .join("libs")
                .join("github.com-Tesohh-strings")
                .join("2020")
                .to_path_buf()
        );
    }

    #[test]
    fn git_headers_and_lib_links() {
        let lib = basic_lib();
        let override_db = OverrideDatabase::new();
        assert_eq!(
            lib.headers("strings", &override_db).unwrap(),
            "-I".to_string()
                + &Path::new(&home_dir().unwrap())
                    .join(".bricks/libs")
                    .join("github.com-Tesohh-strings/2020")
                    .join("build/include")
                    .display()
                    .to_string()
        );
        assert_eq!(
            lib.lib_links("strings", &override_db).unwrap(),
            "-L".to_string()
                + &Path::new(&home_dir().unwrap())
                    .join(".bricks/libs")
                    .join("github.com-Tesohh-strings/2020")
                    .join("build/lib")
                    .display()
                    .to_string()
                + " -lstrings"
        );
    }

    #[test]
    fn overrides() {
        let lib = basic_lib();
        let mut override_db = OverrideDatabase::new();
        override_db.insert(
            "strings".into(),
            Overrides {
                build: None,
                run: None,
                include_dir: Some("evil_build/evil_include".to_string()),
                lib_dir: Some("evil_build/evil_lib".to_string()),
            },
        );

        assert_eq!(
            lib.headers("strings", &override_db).unwrap(),
            "-I".to_string()
                + &Path::new(&home_dir().unwrap())
                    .join(".bricks")
                    .join("libs")
                    .join("github.com-Tesohh-strings")
                    .join("2020")
                    .join("evil_build/evil_include")
                    .display()
                    .to_string()
        );
        assert_eq!(
            lib.lib_links("strings", &override_db).unwrap(),
            "-L".to_string()
                + &Path::new(&home_dir().unwrap())
                    .join(".bricks")
                    .join("libs")
                    .join("github.com-Tesohh-strings")
                    .join("2020")
                    .join("evil_build/evil_lib")
                    .display()
                    .to_string()
                + " -lstrings"
        );
    }
}

fn default_kind() -> LibKind {
    LibKind::Git
}