wasmer_config/package/
package_ident.rs1use 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 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}