1use std::collections::{BTreeMap, HashMap};
16use std::io::Write;
17use std::path::Path;
18
19use serde_json::Value;
20
21pub const SYNCED: [&str; 6] = ["lessons.jsonl", "stats.jsonl", borg::outcome::TEAM_FILE, borg::veto::FILE, borg::msg::FILE, ABSORBED];
54
55pub const ABSORBED: &str = borg::absorb::FILE;
59
60pub const REASONING: &str = "reasoning.jsonl";
63
64pub type Hash = [u8; 32];
65
66pub fn logs(home: &Path) -> Vec<std::path::PathBuf> {
69 let mut out: Vec<std::path::PathBuf> = SYNCED.iter().map(|f| home.join(f)).collect();
70 out.push(home.join(REASONING));
71 out
72}
73
74pub fn travels(file: &str, line: &str, key: &str, now: i64) -> bool {
87 if file == ABSORBED {
88 return borg::absorb::written(line, now).is_some();
89 }
90 if file != borg::msg::FILE {
91 return true;
92 }
93 let Some(l) = borg::msg::Line::parse(line) else { return false };
94 l.node() == key && borg::msg::fresh_at(l.ms(), now) && l.message().is_none_or(|m| !m.stays_here())
95}
96
97pub fn hash(line: &str) -> Hash {
98 *blake3::hash(line.as_bytes()).as_bytes()
99}
100
101fn complete(line: &str) -> bool {
102 !line.trim().is_empty() && serde_json::from_str::<Value>(line).is_ok()
103}
104
105pub fn read(path: &Path) -> BTreeMap<Hash, String> {
107 let text = std::fs::read_to_string(path).unwrap_or_default();
108 text.lines().filter(|l| complete(l)).map(|l| (hash(l), l.to_string())).collect()
109}
110
111pub fn digest(set: &BTreeMap<Hash, String>) -> Hash {
113 let mut h = blake3::Hasher::new();
114 for k in set.keys() {
115 h.update(k);
116 }
117 *h.finalize().as_bytes()
118}
119
120pub fn merge(path: &Path, incoming: &[String]) -> Result<usize, String> {
123 let mut known = read(path);
124 let new: Vec<&String> = incoming.iter().filter(|l| complete(l) && known.insert(hash(l), (*l).clone()).is_none()).collect();
125 append(path, &new)?;
126 settle(path)?;
127 Ok(new.len())
128}
129
130fn append(path: &Path, lines: &[&String]) -> Result<(), String> {
131 if lines.is_empty() {
132 return Ok(());
133 }
134 if let Some(d) = path.parent() {
135 std::fs::create_dir_all(d).map_err(|e| e.to_string())?;
136 }
137 let ends_open = std::fs::read(path).map(|b| b.last().is_some_and(|c| *c != b'\n')).unwrap_or(false);
138 let mut out = String::from(if ends_open { "\n" } else { "" });
139 for l in lines {
140 out.push_str(l);
141 out.push('\n');
142 }
143 let mut f = std::fs::OpenOptions::new().create(true).append(true).open(path).map_err(|e| format!("{}: {e}", path.display()))?;
144 f.write_all(out.as_bytes()).map_err(|e| format!("{}: {e}", path.display()))
145}
146
147pub fn settle(path: &Path) -> Result<(), String> {
152 if path.file_name().is_some_and(|f| f == ABSORBED) {
153 return Ok(());
154 }
155 let text = std::fs::read_to_string(path).unwrap_or_default();
156 let mut winner: HashMap<String, ((u64, Hash), &str)> = HashMap::new();
157 let mut last: HashMap<String, Hash> = HashMap::new();
158 for l in text.lines() {
159 let Ok(v) = serde_json::from_str::<Value>(l) else { continue };
160 let (Some(id), Some(version)) = (v["id"].as_str(), v["version"].as_u64()) else { continue };
161 let rank = (version, hash(l));
162 last.insert(id.to_string(), rank.1);
163 if winner.get(id).is_none_or(|(r, _)| rank > *r) {
164 winner.insert(id.to_string(), (rank, l));
165 }
166 }
167 let mut stale: Vec<(&String, String)> = winner.iter().filter(|(id, (r, _))| last.get(*id) != Some(&r.1)).map(|(id, (_, l))| (id, l.to_string())).collect();
168 stale.sort();
169 append(path, &stale.iter().map(|(_, l)| l).collect::<Vec<_>>())
170}
171
172#[cfg(test)]
173mod tests {
174 use super::*;
175
176 fn lesson(id: &str, version: u32, q: &str) -> String {
177 serde_json::json!({"id": id, "origin_ms": 1, "question": q, "version": version}).to_string()
178 }
179
180 fn tmp(tag: &str) -> std::path::PathBuf {
181 let d = std::env::temp_dir().join(format!("borg-mesh-lines-{tag}-{}", std::process::id()));
182 let _ = std::fs::remove_dir_all(&d);
183 std::fs::create_dir_all(&d).unwrap();
184 d
185 }
186
187 #[test]
189 fn a_local_broadcast_never_travels() {
190 let key = "ab".repeat(32);
191 let from = borg::msg::Sender { by: "s1".into(), name: "One of Two".into(), harness: "claude".into(), node: key.clone() };
192 let line = |to: &str| serde_json::to_string(&borg::msg::Message::new(borg::msg::TELL, &from, to, None, 1_000, "freeze merges")).unwrap();
193 assert!(!travels(borg::msg::FILE, &line(&format!("{}@{key}", borg::msg::LOCAL)), &key, 1_000));
194 assert!(travels(borg::msg::FILE, &line(borg::msg::ALL), &key, 1_000));
195 assert!(travels(borg::msg::FILE, &line(&format!("s2@{}", "cd".repeat(32))), &key, 1_000));
196 }
197
198 #[test]
201 fn a_seen_line_travels_only_under_its_node() {
202 let (key, other) = ("ab".repeat(32), "cd".repeat(32));
203 let seen = |node: &str, ms: i64, ids: &[&str]| serde_json::to_string(&borg::msg::Seen::new("s2", node, ms, ids)).unwrap();
204 assert!(travels(borg::msg::FILE, &seen(&key, 1_000, &["m0123456789ab"]), &key, 1_000));
205 assert!(!travels(borg::msg::FILE, &seen(&key, 1_000, &["m0123456789ab"]), &other, 1_000), "signed by another node");
206 assert!(!travels(borg::msg::FILE, &seen(&key, 1_000, &["m0123456789ab"]), &key, 1_000 + borg::msg::TTL_MS + 1), "expired");
207 assert!(!travels(borg::msg::FILE, &seen(&key, 1_000, &["not an id"]), &key, 1_000), "malformed");
208 }
209
210 #[test]
213 fn an_absorb_travels_only_well_formed() {
214 let key = "ab".repeat(32);
215 let ok = |v: Value| travels(ABSORBED, &v.to_string(), &key, 10);
216 const ABSORB_NOTE: usize = borg::absorb::NOTE_CHARS;
217 let absorb = serde_json::json!({"id": "L1", "kind": "absorb", "ms": 5, "by": "lead@x", "note": "skills/review@2026-10-30", "version": 2});
218 assert!(ok(absorb.clone()));
219 assert!(ok(serde_json::json!({"id": "L1", "kind": "release", "ms": 6, "by": "lead@x"})));
220 assert!(ok(serde_json::json!({"id": "L1", "kind": "release", "ms": 6, "by": "lead@x", "note": null, "version": null})), "null is absent");
221 assert!(ok(serde_json::json!({"id": "L1", "kind": "absorb", "ms": 5, "by": "lead@x", "note": "é".repeat(ABSORB_NOTE)})), "characters, not bytes");
222 let with = |k: &str, v: Value| {
223 let mut a = absorb.clone();
224 a[k] = v;
225 a
226 };
227 for (what, bad) in [
228 ("an oversize note", with("note", Value::from("n".repeat(ABSORB_NOTE + 1)))),
229 ("another kind", with("kind", Value::from("delete"))),
230 ("no kind", serde_json::json!({"id": "L1", "ms": 5, "by": "lead@x"})),
231 ("a numeric id", with("id", Value::from(1))),
232 ("an empty id", with("id", Value::from(""))),
233 ("a blank id", with("id", Value::from(" "))),
234 ("a version borg cannot read", with("version", Value::from(5_000_000_000_u64))),
235 ("stamped past the clock skew", with("ms", Value::from(10 + borg::msg::SKEW_MS + 1))),
236 ("a text ms", with("ms", Value::from("5"))),
237 ("a fractional ms", with("ms", Value::from(5.5))),
238 ("no by", serde_json::json!({"id": "L1", "kind": "absorb", "ms": 5})),
239 ("a note that is no text", with("note", serde_json::json!({"x": 1}))),
240 ("a negative version", with("version", Value::from(-1))),
241 ("an unknown field", with("reach", Value::from("team"))),
242 ("no object", serde_json::json!(["L1", "absorb"])),
243 ] {
244 assert!(!ok(bad), "{what} is refused");
245 }
246 assert!(!travels(ABSORBED, "not json", &key, 1));
247 assert!(!travels(ABSORBED, r#"{"id":"A","id":"B","kind":"absorb","ms":1,"by":"a@x"}"#, &key, 10), "a duplicate key");
248 assert!(logs(Path::new("s")).contains(&Path::new("s").join(ABSORBED)), "stamped, signed and quarantined like every synced log");
249 }
250
251 #[test]
254 fn absorbs_are_not_settled_as_versions() {
255 let d = tmp("absorbed");
256 let p = d.join(ABSORBED);
257 let a = |version: u32, ms: i64| serde_json::json!({"id": "L1", "kind": "absorb", "ms": ms, "by": "lead@x", "version": version}).to_string();
258 merge(&p, &[a(3, 1)]).unwrap();
259 merge(&p, &[a(2, 2)]).unwrap();
260 assert_eq!(std::fs::read_to_string(&p).unwrap(), format!("{}\n{}\n", a(3, 1), a(2, 2)));
261 let _ = std::fs::remove_dir_all(d);
262 }
263
264 #[test]
265 fn union_skips_known_and_torn_lines() {
266 let d = tmp("union");
267 let p = d.join("lessons.jsonl");
268 std::fs::write(&p, format!("{}\n{{\"id\":\"L2\",\"ver", lesson("L1", 1, "a"))).unwrap();
269 assert_eq!(read(&p).len(), 1, "a half-written line does not count");
270 let n = merge(&p, &[lesson("L1", 1, "a"), lesson("L3", 1, "c"), "garbage".into()]).unwrap();
271 assert_eq!(n, 1, "only the unknown complete line is new");
272 assert_eq!(merge(&p, &[lesson("L3", 1, "c")]).unwrap(), 0, "idempotent");
273 let _ = std::fs::remove_dir_all(d);
274 }
275
276 #[test]
277 fn ratified_constraints_sync_and_load_as_the_same_standing_lesson_everywhere() {
278 let d = tmp("constraints");
279 assert!(logs(&d).contains(&d.join(borg::veto::FILE)), "stamped, signed and quarantined like the lessons");
280 let c = |by: &str, ms: i64| {
281 serde_json::json!({"id": "cabc", "text": "merge lookups go through the review API", "lesson": "L1", "by": by, "ms": ms, "source": "veto"})
282 .to_string()
283 };
284 let (a, b) = (d.join("a"), d.join("b"));
285 for (home, own) in [(&a, c("bob@x", 9)), (&b, c("ann@x", 5))] {
286 std::fs::create_dir_all(home).unwrap();
287 std::fs::write(home.join("lessons.jsonl"), format!("{}\n", lesson("L1", 1, "lookup?"))).unwrap();
288 merge(&home.join(borg::veto::FILE), &[own]).unwrap();
289 }
290 merge(&a.join(borg::veto::FILE), &read(&b.join(borg::veto::FILE)).into_values().collect::<Vec<_>>()).unwrap();
291 merge(&b.join(borg::veto::FILE), &read(&a.join(borg::veto::FILE)).into_values().collect::<Vec<_>>()).unwrap();
292 let standing = |h: &Path| {
293 let v: Vec<_> = borg::conscience::load(&h.join("lessons.jsonl"), None)
294 .into_iter()
295 .filter(|c| c.standing())
296 .map(|c| (c.id, c.engineers, c.consolidated))
297 .collect();
298 v
299 };
300 assert_eq!(standing(&a), [("cabc".to_string(), vec!["ann@x".to_string()], true)], "the earliest ratification, whatever the order");
301 assert_eq!(standing(&a), standing(&b));
302 let _ = std::fs::remove_dir_all(d);
303 }
304
305 #[test]
306 fn concurrent_versions_settle_on_the_same_winner_everywhere() {
307 let d = tmp("settle");
308 let (a, b) = (d.join("a/lessons.jsonl"), d.join("b/lessons.jsonl"));
309 let (v1, v2a, v2b) = (lesson("L1", 1, "orig"), lesson("L1", 2, "alice's merge"), lesson("L1", 2, "bob's merge"));
310 merge(&a, &[v1.clone(), v2a.clone()]).unwrap();
311 merge(&b, &[v1.clone(), v2b.clone()]).unwrap();
312 merge(&a, &read(&b).into_values().collect::<Vec<_>>()).unwrap();
313 merge(&b, &read(&a).into_values().collect::<Vec<_>>()).unwrap();
314 assert_eq!(digest(&read(&a)), digest(&read(&b)), "same set");
315 let q = |p: &Path| borg::conscience::load(p, None).into_iter().map(|c| c.question).collect::<Vec<_>>();
316 assert_eq!(q(&a), q(&b), "the conscience asks the same version on both nodes");
317 assert_ne!(q(&a), vec!["orig".to_string()], "an older version never wins");
318 let c = d.join("c/lessons.jsonl");
320 merge(&c, &[v2a]).unwrap();
321 merge(&c, &[v1]).unwrap();
322 assert_eq!(q(&c), vec!["alice's merge".to_string()]);
323 let _ = std::fs::remove_dir_all(d);
324 }
325}