wasmer_config/package/
package_ident.rs

1use std::str::FromStr;
2
3use super::{NamedPackageIdent, PackageHash, PackageId, PackageParseError};
4
5#[derive(PartialEq, Eq, Clone, Debug, Hash)]
6pub enum PackageIdent {
7    Named(NamedPackageIdent),
8    Hash(PackageHash),
9}
10
11impl PackageIdent {
12    pub fn as_named(&self) -> Option<&NamedPackageIdent> {
13        if let Self::Named(v) = self {
14            Some(v)
15        } else {
16            None
17        }
18    }
19
20    pub fn as_hash(&self) -> Option<&PackageHash> {
21        if let Self::Hash(v) = self {
22            Some(v)
23        } else {
24            None
25        }
26    }
27
28    /// Returns true if this ident matches the given package id.
29    ///
30    /// Semver constraints are matched against the package id's version.
31    pub fn matches_id(&self, id: &PackageId) -> bool {
32        match (self, id) {
33            (Self::Named(a), PackageId::Named(b)) => a.matches_id(b),
34            (Self::Hash(a), PackageId::Hash(b)) => a == b,
35            _ => false,
36        }
37    }
38}
39
40impl From<NamedPackageIdent> for PackageIdent {
41    fn from(value: NamedPackageIdent) -> Self {
42        Self::Named(value)
43    }
44}
45
46impl From<PackageHash> for PackageIdent {
47    fn from(value: PackageHash) -> Self {
48        Self::Hash(value)
49    }
50}
51
52impl std::str::FromStr for PackageIdent {
53    type Err = PackageParseError;
54
55    fn from_str(s: &str) -> Result<Self, Self::Err> {
56        if let Ok(hash) = PackageHash::from_str(s) {
57            Ok(Self::Hash(hash))
58        } else if let Ok(named) = NamedPackageIdent::from_str(s) {
59            Ok(Self::Named(named))
60        } else {
61            Err(PackageParseError::new(
62                s,
63                "invalid package ident: expected a hash or a valid named package identifier",
64            ))
65        }
66    }
67}
68
69impl std::fmt::Display for PackageIdent {
70    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
71        match self {
72            Self::Named(n) => n.fmt(f),
73            Self::Hash(h) => h.fmt(f),
74        }
75    }
76}
77
78impl serde::Serialize for PackageIdent {
79    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
80    where
81        S: serde::ser::Serializer,
82    {
83        self.to_string().serialize(serializer)
84    }
85}
86
87impl<'de> serde::Deserialize<'de> for PackageIdent {
88    fn deserialize<D>(deserializer: D) -> Result<PackageIdent, D::Error>
89    where
90        D: serde::de::Deserializer<'de>,
91    {
92        let s = String::deserialize(deserializer)?;
93        Self::from_str(&s).map_err(serde::de::Error::custom)
94    }
95}
96
97impl schemars::JsonSchema for PackageIdent {
98    fn schema_name() -> String {
99        "PackageIdent".to_string()
100    }
101
102    fn json_schema(gen: &mut schemars::gen::SchemaGenerator) -> schemars::schema::Schema {
103        String::json_schema(gen)
104    }
105}
106
107#[cfg(test)]
108mod tests {
109    use super::*;
110
111    #[test]
112    fn test_package_ident_matches_id() {
113        assert!(PackageIdent::from_str("ns/pkg")
114            .unwrap()
115            .matches_id(&PackageId::new_named("ns/pkg", "1.0.0".parse().unwrap())));
116
117        assert!(PackageIdent::from_str("ns/pkg@2")
118            .unwrap()
119            .matches_id(&PackageId::new_named("ns/pkg", "2.3.7".parse().unwrap())));
120
121        assert!(!PackageIdent::from_str("ns/pkg@3")
122            .unwrap()
123            .matches_id(&PackageId::new_named("ns/pkg", "2.3.7".parse().unwrap())));
124    }
125}