snarkvm_console_program/id/
mod.rs1mod bytes;
17mod parse;
18mod serialize;
19mod to_address;
20mod to_bits;
21mod to_fields;
22
23use crate::Identifier;
24use snarkvm_console_network::prelude::*;
25use snarkvm_console_types::{Address, Boolean, Field};
26
27fn is_lowercase_alphanumeric(s: &str) -> bool {
29 s.chars().all(|c| matches!(c, '0'..='9' | 'a'..='z' | '_'))
30}
31
32#[derive(Copy, Clone, PartialEq, Eq, Hash)]
34pub struct ProgramID<N: Network> {
35 name: Identifier<N>,
37 network: Identifier<N>,
39}
40
41impl<N: Network> From<&ProgramID<N>> for ProgramID<N> {
42 fn from(program_id: &ProgramID<N>) -> Self {
44 *program_id
45 }
46}
47
48impl<N: Network> TryFrom<(Identifier<N>, Identifier<N>)> for ProgramID<N> {
49 type Error = Error;
50
51 fn try_from((name, network): (Identifier<N>, Identifier<N>)) -> Result<Self> {
53 ensure!(is_lowercase_alphanumeric(&name.to_string()), "Program name is invalid: {name}");
55 let id = Self { name, network };
57 ensure!(id.is_aleo(), "Program network is invalid: {network}");
59 Ok(id)
61 }
62}
63
64impl<N: Network> TryFrom<String> for ProgramID<N> {
65 type Error = Error;
66
67 fn try_from(program_id: String) -> Result<Self> {
69 Self::from_str(&program_id)
70 }
71}
72
73impl<N: Network> TryFrom<&String> for ProgramID<N> {
74 type Error = Error;
75
76 fn try_from(program_id: &String) -> Result<Self> {
78 Self::from_str(program_id)
79 }
80}
81
82impl<N: Network> TryFrom<&str> for ProgramID<N> {
83 type Error = Error;
84
85 fn try_from(program_id: &str) -> Result<Self> {
87 let mut split = program_id.split('.');
89 if let (Some(name), Some(network), None) = (split.next(), split.next(), split.next()) {
91 ensure!(is_lowercase_alphanumeric(name), "Program name is invalid: {name}");
93 Self::try_from((Identifier::from_str(name)?, Identifier::from_str(network)?))
95 } else {
96 bail!("Invalid program ID '{program_id}'")
97 }
98 }
99}
100
101impl<N: Network> ProgramID<N> {
102 #[inline]
104 pub const fn name(&self) -> &Identifier<N> {
105 &self.name
106 }
107
108 #[inline]
110 pub const fn network(&self) -> &Identifier<N> {
111 &self.network
112 }
113
114 #[inline]
116 pub fn is_aleo(&self) -> bool {
117 self.network() == &Identifier::from_str("aleo").expect("Failed to parse Aleo domain")
118 }
119
120 pub fn credits() -> Self {
122 Self::from_str("credits.aleo").expect("Failed to parse credits.aleo as a program ID")
123 }
124}
125
126impl<N: Network> Ord for ProgramID<N> {
127 fn cmp(&self, other: &Self) -> Ordering {
129 match self.network == other.network {
130 true => self.name.to_string().cmp(&other.name.to_string()),
131 false => self.network.to_string().cmp(&other.network.to_string()),
132 }
133 }
134}
135
136impl<N: Network> PartialOrd for ProgramID<N> {
137 fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
139 Some(self.cmp(other))
140 }
141}
142
143impl<N: Network> Equal<Self> for ProgramID<N> {
144 type Output = Boolean<N>;
145
146 fn is_equal(&self, other: &Self) -> Self::Output {
148 Boolean::new(self == other)
149 }
150
151 fn is_not_equal(&self, other: &Self) -> Self::Output {
153 Boolean::new(self != other)
154 }
155}
156
157#[cfg(test)]
158mod tests {
159 use super::*;
160 use snarkvm_console_network::MainnetV0;
161
162 type CurrentNetwork = MainnetV0;
163
164 #[test]
165 fn test_partial_ord() -> Result<()> {
166 let import1 = ProgramID::<CurrentNetwork>::from_str("bar.aleo")?;
167 let import2 = ProgramID::<CurrentNetwork>::from_str("foo.aleo")?;
168
169 let import3 = ProgramID::<CurrentNetwork>::from_str("bar.aleo")?;
170 let import4 = ProgramID::<CurrentNetwork>::from_str("foo.aleo")?;
171
172 assert_eq!(import1.partial_cmp(&import1), Some(Ordering::Equal));
173 assert_eq!(import1.partial_cmp(&import2), Some(Ordering::Less));
174 assert_eq!(import1.partial_cmp(&import3), Some(Ordering::Equal));
175 assert_eq!(import1.partial_cmp(&import4), Some(Ordering::Less));
176
177 assert_eq!(import2.partial_cmp(&import1), Some(Ordering::Greater));
178 assert_eq!(import2.partial_cmp(&import2), Some(Ordering::Equal));
179 assert_eq!(import2.partial_cmp(&import3), Some(Ordering::Greater));
180 assert_eq!(import2.partial_cmp(&import4), Some(Ordering::Equal));
181
182 assert_eq!(import3.partial_cmp(&import1), Some(Ordering::Equal));
183 assert_eq!(import3.partial_cmp(&import2), Some(Ordering::Less));
184 assert_eq!(import3.partial_cmp(&import3), Some(Ordering::Equal));
185 assert_eq!(import3.partial_cmp(&import4), Some(Ordering::Less));
186
187 assert_eq!(import4.partial_cmp(&import1), Some(Ordering::Greater));
188 assert_eq!(import4.partial_cmp(&import2), Some(Ordering::Equal));
189 assert_eq!(import4.partial_cmp(&import3), Some(Ordering::Greater));
190 assert_eq!(import4.partial_cmp(&import4), Some(Ordering::Equal));
191
192 Ok(())
193 }
194}