sqlite_graphrag/spawn/
executor_version.rs1use crate::errors::AppError;
4use semver::Version;
5use serde::{Deserialize, Serialize};
6use std::cmp::Ordering;
7use std::str::FromStr;
8
9#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
11pub struct ExecutorVersion {
12 pub raw: String,
14 pub semver: Option<Version>,
16 pub channel: Option<String>,
18}
19
20impl ExecutorVersion {
21 pub fn unknown() -> Self {
23 Self {
24 raw: "unknown".to_string(),
25 semver: None,
26 channel: None,
27 }
28 }
29
30 pub fn parse(s: &str) -> Result<Self, AppError> {
32 let trimmed = s.trim();
33 if trimmed.is_empty() {
34 return Err(AppError::Validation(crate::i18n::validation::empty_version_string()));
35 }
36 let channel_start = trimmed.find('-');
37 let (numeric_part, channel_part) = match channel_start {
38 Some(idx) => (
39 trimmed[..idx].to_string(),
40 Some(trimmed[idx + 1..].to_string()),
41 ),
42 None => (trimmed.to_string(), None),
43 };
44 let semver = if numeric_part
45 .chars()
46 .next()
47 .map(|c| c.is_ascii_digit())
48 .unwrap_or(false)
49 {
50 Version::from_str(&numeric_part).ok()
51 } else {
52 None
53 };
54 Ok(Self {
55 raw: trimmed.to_string(),
56 semver,
57 channel: channel_part,
58 })
59 }
60
61 pub fn major(&self) -> u64 {
63 self.semver.as_ref().map(|v| v.major).unwrap_or(0)
64 }
65
66 pub fn minor(&self) -> u64 {
68 self.semver.as_ref().map(|v| v.minor).unwrap_or(0)
69 }
70
71 pub fn patch(&self) -> u64 {
73 self.semver.as_ref().map(|v| v.patch).unwrap_or(0)
74 }
75
76 pub fn is_at_least(&self, major: u64, minor: u64, patch: u64) -> bool {
78 match &self.semver {
79 Some(v) => (v.major, v.minor, v.patch) >= (major, minor, patch),
80 None => false,
81 }
82 }
83
84 pub fn in_range(&self, min: (u64, u64, u64), max: (u64, u64, u64)) -> bool {
86 self.is_at_least(min.0, min.1, min.2) && {
87 let current = (self.major(), self.minor(), self.patch());
88 current <= max
89 }
90 }
91
92 pub fn compare(&self, other: &ExecutorVersion) -> Ordering {
94 match (&self.semver, &other.semver) {
95 (Some(a), Some(b)) => a.cmp(b),
96 (Some(_), None) => Ordering::Greater,
97 (None, Some(_)) => Ordering::Less,
98 (None, None) => Ordering::Equal,
99 }
100 }
101}
102
103impl std::fmt::Display for ExecutorVersion {
104 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
105 write!(f, "{}", self.raw)
106 }
107}
108
109impl Default for ExecutorVersion {
110 fn default() -> Self {
111 Self::unknown()
112 }
113}
114
115#[cfg(test)]
116mod tests {
117 use super::*;
118
119 #[test]
120 fn parse_simple_semver() {
121 let v = ExecutorVersion::parse("0.137.0").unwrap();
122 assert_eq!(v.major(), 0);
123 assert_eq!(v.minor(), 137);
124 assert_eq!(v.patch(), 0);
125 assert!(v.is_at_least(0, 130, 0));
126 assert!(!v.is_at_least(1, 0, 0));
127 }
128
129 #[test]
130 fn parse_with_channel() {
131 let v = ExecutorVersion::parse("2.1.0-beta.1").unwrap();
132 assert_eq!(v.major(), 2);
133 assert_eq!(v.minor(), 1);
134 assert_eq!(v.channel.as_deref(), Some("beta.1"));
135 }
136
137 #[test]
138 fn parse_unknown() {
139 let v = ExecutorVersion::parse("n/a").unwrap();
140 assert_eq!(v.semver, None);
141 assert!(!v.is_at_least(0, 0, 0));
142 }
143
144 #[test]
145 fn compare_ordering() {
146 let a = ExecutorVersion::parse("0.137.0").unwrap();
147 let b = ExecutorVersion::parse("0.138.0").unwrap();
148 assert_eq!(a.compare(&b), Ordering::Less);
149 }
150
151 #[test]
152 fn empty_string_is_error() {
153 assert!(ExecutorVersion::parse("").is_err());
154 }
155}