1use objects::object::StateId;
3use serde::{Deserialize, Serialize};
4
5mod kind;
6
7pub use kind::{AdvertisedRef, AdvertisedRefError, RefKind};
8
9#[derive(Debug, Clone, Serialize, Deserialize)]
14pub struct RefFilter {
15 #[serde(default)]
16 pub names: Vec<String>,
17 #[serde(default)]
18 pub patterns: Vec<String>,
19 #[serde(default = "default_true")]
20 pub include_threads: bool,
21 #[serde(default = "default_true")]
22 pub include_markers: bool,
23 #[serde(default)]
26 pub include_synthetic: bool,
27 #[serde(default)]
28 pub limit: Option<usize>,
29}
30
31fn default_true() -> bool {
32 true
33}
34
35impl Default for RefFilter {
36 fn default() -> Self {
37 Self {
38 names: Vec::new(),
39 patterns: Vec::new(),
40 include_threads: true,
41 include_markers: true,
42 include_synthetic: false,
43 limit: None,
44 }
45 }
46}
47
48impl RefFilter {
49 pub fn matches(&self, name: &str) -> bool {
50 if !self.names.is_empty() && self.names.iter().any(|candidate| candidate == name) {
51 return true;
52 }
53
54 if self.patterns.is_empty() {
55 return self.names.is_empty();
56 }
57
58 self.patterns
59 .iter()
60 .any(|pattern| Self::matches_pattern(name, pattern))
61 }
62
63 pub fn includes_kind(&self, kind: RefKind) -> bool {
64 match kind {
65 RefKind::Thread => self.include_threads,
66 RefKind::Marker => self.include_markers,
67 RefKind::SyntheticFrontierRoot => self.include_synthetic,
68 }
69 }
70
71 fn matches_pattern(name: &str, pattern: &str) -> bool {
72 if pattern == "*" {
73 return true;
74 }
75 if pattern.starts_with('*') && pattern.ends_with('*') && pattern.len() >= 2 {
76 return name.contains(&pattern[1..pattern.len() - 1]);
77 }
78 if let Some(suffix) = pattern.strip_prefix('*') {
79 return name.ends_with(suffix);
80 }
81 if let Some(prefix) = pattern.strip_suffix('*') {
82 return name.starts_with(prefix);
83 }
84 name == pattern
85 }
86}
87
88#[derive(Debug, Clone, Serialize, Deserialize)]
90pub enum HeadInfo {
91 Attached { thread: String },
92 Detached { state: StateId },
93}
94
95#[derive(Debug, Clone, Serialize, Deserialize)]
97pub struct RefEntry {
98 pub name: String,
99 pub state_id: StateId,
100 pub kind: RefKind,
101}
102
103#[derive(Debug, Clone, Serialize, Deserialize)]
104pub struct RefUpdated {
105 pub success: bool,
106 pub old_value: Option<StateId>,
107 pub error: Option<String>,
108}
109
110#[cfg(test)]
111mod tests {
112 use super::*;
113
114 #[test]
115 fn ref_filter_matches_union_of_names_and_patterns() {
116 let filter = RefFilter {
117 names: vec!["refs/heads/main".to_string()],
118 patterns: vec!["refs/tags/v*".to_string()],
119 ..RefFilter::default()
120 };
121
122 assert!(filter.matches("refs/heads/main"));
123 assert!(filter.matches("refs/tags/v1.0.0"));
124 assert!(!filter.matches("refs/heads/feature"));
125 assert!(!filter.matches("refs/tags/nightly"));
126 }
127
128 #[test]
129 fn ref_filter_without_names_or_patterns_matches_everything() {
130 let filter = RefFilter::default();
131
132 assert!(filter.matches("refs/heads/main"));
133 assert!(filter.matches("refs/tags/v1.0.0"));
134 assert!(filter.matches("threads/alice"));
135 assert!(!filter.includes_kind(RefKind::SyntheticFrontierRoot));
136 assert!(filter.includes_kind(RefKind::Thread));
137 assert!(filter.includes_kind(RefKind::Marker));
138 }
139
140 #[test]
141 fn ref_filter_exact_names_do_not_expand_without_patterns() {
142 let filter = RefFilter {
143 names: vec!["refs/heads/main".to_string()],
144 patterns: Vec::new(),
145 ..RefFilter::default()
146 };
147
148 assert!(filter.matches("refs/heads/main"));
149 assert!(!filter.matches("refs/heads/mainline"));
150 }
151}