claude-agent 0.2.25

use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum SourceType {
    Builtin,
    #[default]
    User,
    Project,
    Managed,
    Plugin,
}

impl std::fmt::Display for SourceType {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Builtin => write!(f, "builtin"),
            Self::User => write!(f, "user"),
            Self::Project => write!(f, "project"),
            Self::Managed => write!(f, "managed"),
            Self::Plugin => write!(f, "plugin"),
        }
    }
}

impl std::str::FromStr for SourceType {
    type Err = std::convert::Infallible;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        Ok(match s {
            "builtin" => Self::Builtin,
            "project" => Self::Project,
            "managed" => Self::Managed,
            "plugin" => Self::Plugin,
            _ => Self::User,
        })
    }
}

impl SourceType {
    pub fn from_str_opt(s: Option<&str>) -> Self {
        s.map(|v| v.parse().unwrap_or_default()).unwrap_or_default()
    }
}

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

    #[test]
    fn test_default() {
        assert_eq!(SourceType::default(), SourceType::User);
    }

    #[test]
    fn test_display() {
        assert_eq!(SourceType::Builtin.to_string(), "builtin");
        assert_eq!(SourceType::User.to_string(), "user");
        assert_eq!(SourceType::Project.to_string(), "project");
        assert_eq!(SourceType::Managed.to_string(), "managed");
        assert_eq!(SourceType::Plugin.to_string(), "plugin");
    }

    #[test]
    fn test_from_str_opt() {
        assert_eq!(
            SourceType::from_str_opt(Some("builtin")),
            SourceType::Builtin
        );
        assert_eq!(
            SourceType::from_str_opt(Some("project")),
            SourceType::Project
        );
        assert_eq!(
            SourceType::from_str_opt(Some("managed")),
            SourceType::Managed
        );
        assert_eq!(SourceType::from_str_opt(Some("plugin")), SourceType::Plugin);
        assert_eq!(SourceType::from_str_opt(Some("user")), SourceType::User);
        assert_eq!(SourceType::from_str_opt(None), SourceType::User);
        assert_eq!(SourceType::from_str_opt(Some("unknown")), SourceType::User);
    }

    #[test]
    fn test_from_str() {
        assert_eq!(
            "builtin".parse::<SourceType>().unwrap(),
            SourceType::Builtin
        );
        assert_eq!(
            "project".parse::<SourceType>().unwrap(),
            SourceType::Project
        );
        assert_eq!(
            "managed".parse::<SourceType>().unwrap(),
            SourceType::Managed
        );
        assert_eq!("plugin".parse::<SourceType>().unwrap(), SourceType::Plugin);
        assert_eq!("user".parse::<SourceType>().unwrap(), SourceType::User);
        assert_eq!("unknown".parse::<SourceType>().unwrap(), SourceType::User);
    }

    #[test]
    fn test_serde() {
        let json = serde_json::to_string(&SourceType::Builtin).unwrap();
        assert_eq!(json, "\"builtin\"");

        let parsed: SourceType = serde_json::from_str("\"project\"").unwrap();
        assert_eq!(parsed, SourceType::Project);

        let plugin_json = serde_json::to_string(&SourceType::Plugin).unwrap();
        assert_eq!(plugin_json, "\"plugin\"");

        let parsed_plugin: SourceType = serde_json::from_str("\"plugin\"").unwrap();
        assert_eq!(parsed_plugin, SourceType::Plugin);
    }
}