1use std::cmp::Ordering;
13
14#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
15pub enum Stability {
16 Dev,
17 Alpha,
18 Beta,
19 Rc,
20 Stable,
21 Patch,
22}
23
24#[derive(Debug, Clone, PartialEq, Eq)]
25pub struct Version {
26 pub parts: [u64; 4],
27 pub stability: Stability,
28 pub pre_number: u64,
30}
31
32#[derive(Debug, thiserror::Error, PartialEq, Eq)]
33#[error("version outside the supported subset: {0:?}")]
34pub struct UnsupportedVersion(pub String);
35
36impl Version {
37 pub fn parse(input: &str) -> Result<Self, UnsupportedVersion> {
38 let s = input.trim();
39 let s = s
40 .strip_prefix('v')
41 .or_else(|| s.strip_prefix('V'))
42 .unwrap_or(s);
43 if s.is_empty() {
44 return Err(UnsupportedVersion(input.to_owned()));
45 }
46
47 let (num, suffix) = split_stability(s);
49 let (stability, pre_number) = parse_stability(suffix, input)?;
50
51 let mut parts = [0u64; 4];
52 let mut n = 0usize;
53 for piece in num.split('.') {
54 if n >= 4 || piece.is_empty() || !piece.bytes().all(|b| b.is_ascii_digit()) {
55 return Err(UnsupportedVersion(input.to_owned()));
56 }
57 parts[n] = piece
58 .parse()
59 .map_err(|_| UnsupportedVersion(input.to_owned()))?;
60 n += 1;
61 }
62 if n == 0 {
63 return Err(UnsupportedVersion(input.to_owned()));
64 }
65 Ok(Version {
66 parts,
67 stability,
68 pre_number,
69 })
70 }
71}
72
73fn split_stability(s: &str) -> (&str, &str) {
75 match s.find(|c: char| !(c.is_ascii_digit() || c == '.')) {
76 Some(i) => {
77 let suffix = &s[i..];
78 (&s[..i], suffix.trim_start_matches(['-', '_', '.']))
79 }
80 None => (s, ""),
81 }
82}
83
84fn parse_stability(suffix: &str, original: &str) -> Result<(Stability, u64), UnsupportedVersion> {
85 if suffix.is_empty() {
86 return Ok((Stability::Stable, 0));
87 }
88 let lower = suffix.to_ascii_lowercase();
89 let (word, digits) = match lower.find(|c: char| c.is_ascii_digit()) {
90 Some(i) => (&lower[..i], &lower[i..]),
91 None => (lower.as_str(), ""),
92 };
93 let word = word.trim_end_matches(['-', '_', '.']);
94 let stability = match word {
95 "dev" => Stability::Dev,
96 "alpha" | "a" => Stability::Alpha,
97 "beta" | "b" => Stability::Beta,
98 "rc" => Stability::Rc,
99 "patch" | "pl" | "p" => Stability::Patch,
100 "stable" | "" => Stability::Stable,
101 _ => return Err(UnsupportedVersion(original.to_owned())),
102 };
103 let pre_number = if digits.is_empty() {
104 0
105 } else {
106 digits
107 .parse()
108 .map_err(|_| UnsupportedVersion(original.to_owned()))?
109 };
110 Ok((stability, pre_number))
111}
112
113impl PartialOrd for Version {
114 fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
115 Some(self.cmp(other))
116 }
117}
118
119impl Ord for Version {
120 fn cmp(&self, other: &Self) -> Ordering {
121 self.parts
122 .cmp(&other.parts)
123 .then(self.stability.cmp(&other.stability))
124 .then(self.pre_number.cmp(&other.pre_number))
125 }
126}
127
128pub fn normalize_pretty(input: &str) -> Result<String, UnsupportedVersion> {
134 let s = input.trim();
135 if s.len() >= 4 && s[..4].eq_ignore_ascii_case("dev-") {
140 let rest = &s[4..];
141 if rest.is_empty() {
142 return Err(UnsupportedVersion(input.to_owned()));
143 }
144 return Ok(format!("dev-{rest}"));
145 }
146 let stripped = s
147 .strip_prefix('v')
148 .or_else(|| s.strip_prefix('V'))
149 .unwrap_or(s);
150
151 if let Some(stem) = stripped
153 .strip_suffix(".x-dev")
154 .or_else(|| stripped.strip_suffix(".X-dev"))
155 {
156 let mut out: Vec<String> = Vec::new();
160 for piece in stem.split('.') {
161 if piece.is_empty() || !piece.bytes().all(|b| b.is_ascii_digit()) || out.len() >= 3 {
162 return Err(UnsupportedVersion(input.to_owned()));
163 }
164 out.push(piece.to_owned());
165 }
166 while out.len() < 4 {
167 out.push("9999999".to_owned());
168 }
169 return Ok(format!("{}-dev", out.join(".")));
170 }
171
172 let (num, suffix) = split_stability(stripped);
173 let (stability, _) = parse_stability(suffix, input)?;
174 let mut parts: Vec<&str> = Vec::new();
176 for piece in num.split('.') {
177 if parts.len() >= 4 || piece.is_empty() || !piece.bytes().all(|b| b.is_ascii_digit()) {
178 return Err(UnsupportedVersion(input.to_owned()));
179 }
180 parts.push(piece);
181 }
182 if parts.is_empty() {
183 return Err(UnsupportedVersion(input.to_owned()));
184 }
185 while parts.len() < 4 {
186 parts.push("0");
187 }
188 let base = parts.join(".");
189 let word = match stability {
190 Stability::Stable => return Ok(base),
191 Stability::Dev => "dev",
192 Stability::Alpha => "alpha",
193 Stability::Beta => "beta",
194 Stability::Rc => "RC",
195 Stability::Patch => "patch",
196 };
197 let digits: String = suffix.chars().filter(char::is_ascii_digit).collect();
200 if digits.is_empty() {
201 Ok(format!("{base}-{word}"))
202 } else {
203 Ok(format!("{base}-{word}{digits}"))
204 }
205}
206
207#[cfg(test)]
208mod tests {
209 use super::*;
210
211 fn v(s: &str) -> Version {
212 Version::parse(s).expect(s)
213 }
214
215 #[test]
216 fn parses_and_orders() {
217 assert_eq!(v("8.5.10").parts, [8, 5, 10, 0]);
218 assert_eq!(v("v1.2").parts, [1, 2, 0, 0]);
219 assert!(v("8.1") < v("8.1.1"));
220 assert!(v("7.4.33") < v("8.0.0"));
221 assert!(v("1.0.0-dev") < v("1.0.0-alpha1"));
222 assert!(v("1.0.0-alpha2") < v("1.0.0-beta1"));
223 assert!(v("1.0.0-RC1") < v("1.0.0"));
224 assert!(v("1.0.0") < v("1.0.0-patch1"));
225 assert!(v("1.0.0-beta1") < v("1.0.0-beta2"));
226 assert_eq!(v("1.0"), v("1.0.0.0"));
227 }
228
229 #[test]
230 fn rejects_out_of_subset() {
231 for s in ["dev-master", "1.x-dev", "abc", "", "1.2.3.4.5", "1.2-foo"] {
232 assert!(Version::parse(s).is_err(), "{s} should have been rejected");
233 }
234 }
235}