aux-search/flake-info/src/data/flake.rs

use std::path::PathBuf;

use serde::{Deserialize, Serialize};

use super::Source;

/// Holds general infoamtion about a flake
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct Flake {
    #[serde(rename(serialize = "flake_description"))]
    pub description: Option<String>,
    #[serde(rename(serialize = "flake_path"), skip_serializing)]
    pub path: PathBuf,
    #[serde(rename(serialize = "flake_resolved"))]
    pub resolved: Repo,

    #[serde(rename(serialize = "flake_name"), skip_deserializing)]
    pub name: String,

    pub revision: String,

    #[serde(
        skip_deserializing,
        rename(serialize = "flake_source"),
        skip_serializing_if = "Option::is_none"
    )]
    pub source: Option<Source>,
}

impl Flake {
    pub(crate) fn resolve_name(mut self) -> Self {
        self.name = match &self.resolved {
            Repo::Git { .. } => Default::default(),
            Repo::GitHub { repo, .. } => repo.clone(),
            Repo::Gitlab { repo, .. } => repo.clone(),
        };
        self
    }
}

/// Information about the flake origin
/// Supports (local/raw) Git, GitHub and Gitlab repos
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(tag = "type", rename_all = "lowercase")]
pub enum Repo {
    Git { url: PathBuf },
    GitHub { owner: String, repo: String },
    Gitlab { owner: String, repo: String },
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn gitlab_flake() {
        let nix_info_out = r#"{"description":"neuropil is a secure messaging library for IoT, robotics and more.","lastModified":1616059502,"locked":{"lastModified":1616059502,"narHash":"sha256-fHB1vyjDXQq/E2/Xb6Xs3caAAc0VkUlnzu5kl/PvFW4=","owner":"pi-lar","repo":"neuropil","rev":"9e2f634ffa45da3f5feb158a12ee32e1673bfe35","type":"gitlab"},"original":{"owner":"pi-lar","repo":"neuropil","type":"gitlab"},"originalUrl":"gitlab:pi-lar/neuropil","path":"/nix/store/z4fp2fc9hca40nnvxi0116pfbrla5zgl-source","resolved":{"owner":"pi-lar","repo":"neuropil","type":"gitlab"},"resolvedUrl":"gitlab:pi-lar/neuropil","revision":"9e2f634ffa45da3f5feb158a12ee32e1673bfe35","url":"gitlab:pi-lar/neuropil/9e2f634ffa45da3f5feb158a12ee32e1673bfe35"}"#;

        assert_eq!(
            serde_json::de::from_str::<Flake>(nix_info_out).unwrap(),
            Flake {
                description: Some(
                    "neuropil is a secure messaging library for IoT, robotics and more.".into()
                ),
                path: "/nix/store/z4fp2fc9hca40nnvxi0116pfbrla5zgl-source".into(),
                resolved: Repo::Gitlab {
                    owner: "pi-lar".into(),
                    repo: "neuropil".into()
                },
                name: "".into(),
                source: None,
                revision: "9e2f634ffa45da3f5feb158a12ee32e1673bfe35".into()
            }
        );

        assert_eq!(
            serde_json::de::from_str::<Flake>(nix_info_out)
                .unwrap()
                .resolve_name()
                .name,
            "neuropil"
        );
    }
}