1use anyhow::{bail, Context, Result};
2use chrono::{DateTime, Utc};
3use indexmap::IndexMap;
4use schemars::JsonSchema;
5use serde::{Deserialize, Serialize};
6use std::collections::HashMap;
7use std::path::{Path, PathBuf};
8
9fn deserialize_id<'de, D: serde::Deserializer<'de>>(d: D) -> Result<String, D::Error> {
10 use serde::de::{self, Visitor};
11 struct IdVisitor;
12 impl<'de> Visitor<'de> for IdVisitor {
13 type Value = String;
14 fn expecting(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
15 f.write_str("an integer or hex string")
16 }
17 fn visit_u64<E: de::Error>(self, v: u64) -> Result<String, E> {
18 Ok(format!("{v:04}"))
19 }
20 fn visit_i64<E: de::Error>(self, v: i64) -> Result<String, E> {
21 Ok(format!("{v:04}"))
22 }
23 fn visit_str<E: de::Error>(self, v: &str) -> Result<String, E> {
24 Ok(v.to_string())
25 }
26 fn visit_string<E: de::Error>(self, v: String) -> Result<String, E> {
27 Ok(v)
28 }
29 }
30 d.deserialize_any(IdVisitor)
31}
32
33#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
34pub struct Frontmatter {
35 #[serde(deserialize_with = "deserialize_id")]
36 #[schemars(with = "String")]
37 pub id: String,
38 pub title: String,
39 pub state: String,
40 #[serde(default)]
41 pub priority: u8,
42 #[serde(default)]
43 pub effort: u8,
44 #[serde(default)]
45 pub risk: u8,
46 #[serde(skip_serializing_if = "Option::is_none")]
47 pub author: Option<String>,
48 #[serde(skip_serializing_if = "Option::is_none")]
49 pub owner: Option<String>,
50 #[serde(skip_serializing_if = "Option::is_none")]
51 pub branch: Option<String>,
52 #[serde(skip_serializing_if = "Option::is_none")]
53 pub created_at: Option<DateTime<Utc>>,
54 #[serde(skip_serializing_if = "Option::is_none")]
55 pub updated_at: Option<DateTime<Utc>>,
56 #[serde(skip_serializing_if = "Option::is_none")]
57 pub focus_section: Option<String>,
58 #[serde(skip_serializing_if = "Option::is_none")]
59 pub epic: Option<String>,
60 #[serde(skip_serializing_if = "Option::is_none")]
61 pub target_branch: Option<String>,
62 #[serde(skip_serializing_if = "Option::is_none")]
63 pub depends_on: Option<Vec<String>>,
64 #[serde(default, skip_serializing_if = "Option::is_none")]
65 pub agent: Option<String>,
66 #[serde(default, skip_serializing_if = "HashMap::is_empty")]
67 pub agent_overrides: HashMap<String, String>,
68}
69
70#[derive(Debug, Clone)]
71pub struct Ticket {
72 pub frontmatter: Frontmatter,
73 pub body: String,
74 pub path: PathBuf,
75 pub local_stale: bool,
78 pub local_diverged: bool,
81}
82
83impl Ticket {
84 pub fn load(path: &Path) -> Result<Self> {
85 let raw = std::fs::read_to_string(path)
86 .with_context(|| format!("cannot read {}", path.display()))?;
87 Self::parse(path, &raw)
88 }
89
90 pub fn parse(path: &Path, raw: &str) -> Result<Self> {
91 let Some(rest) = raw.strip_prefix("+++\n") else {
92 bail!("missing frontmatter in {}", path.display());
93 };
94 let Some(end) = rest.find("\n+++") else {
95 bail!("unclosed frontmatter in {}", path.display());
96 };
97 let toml_src = &rest[..end];
98 let body = rest[end + 4..].trim_start_matches('\n').to_string();
99 let frontmatter: Frontmatter = toml::from_str(toml_src)
100 .with_context(|| format!("cannot parse frontmatter in {}", path.display()))?;
101 Ok(Self { frontmatter, body, path: path.to_owned(), local_stale: false, local_diverged: false })
102 }
103
104 pub fn serialize(&self) -> Result<String> {
105 let fm = toml::to_string(&self.frontmatter)
106 .context("cannot serialize frontmatter")?;
107 Ok(format!("+++\n{}+++\n\n{}", fm, self.body))
108 }
109
110 pub fn save(&self) -> Result<()> {
111 let content = self.serialize()?;
112 std::fs::write(&self.path, content)
113 .with_context(|| format!("cannot write {}", self.path.display()))
114 }
115
116 pub fn document(&self) -> Result<TicketDocument> {
117 TicketDocument::parse(&self.body)
118 }
119}
120
121pub fn slugify(s: &str) -> String {
122 s.chars()
123 .map(|c| if c.is_alphanumeric() { c.to_ascii_lowercase() } else { '-' })
124 .collect::<String>()
125 .split('-')
126 .filter(|p| !p.is_empty())
127 .collect::<Vec<_>>()
128 .join("-")
129 .chars()
130 .take(40)
131 .collect()
132}
133
134#[derive(Debug, Clone)]
137pub struct ChecklistItem {
138 pub checked: bool,
139 pub text: String,
140}
141
142#[derive(Debug, Clone)]
143pub enum ValidationError {
144 EmptySection(String),
145 NoAcceptanceCriteria,
146}
147
148impl std::fmt::Display for ValidationError {
149 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
150 match self {
151 Self::EmptySection(s) => write!(f, "### {s} section is empty"),
152 Self::NoAcceptanceCriteria => write!(f, "### Acceptance criteria has no checklist items"),
153 }
154 }
155}
156
157#[derive(Debug, Clone)]
158pub struct TicketDocument {
159 pub sections: IndexMap<String, String>,
160 pub(crate) raw_history: String,
161}
162
163pub(crate) fn parse_checklist(text: &str) -> Vec<ChecklistItem> {
164 text.lines()
165 .filter_map(|line| {
166 let l = line.trim();
167 if let Some(s) = l.strip_prefix("- [ ] ") {
168 Some(ChecklistItem { checked: false, text: s.to_string() })
169 } else if let Some(s) = l.strip_prefix("- [x] ") {
170 Some(ChecklistItem { checked: true, text: s.to_string() })
171 } else {
172 l.strip_prefix("- [X] ").map(|s| ChecklistItem { checked: true, text: s.to_string() })
173 }
174 })
175 .collect()
176}
177
178pub(crate) fn serialize_checklist(items: &[ChecklistItem]) -> String {
179 items.iter()
180 .map(|i| format!("- [{}] {}", if i.checked { "x" } else { " " }, i.text))
181 .collect::<Vec<_>>()
182 .join("\n")
183}
184
185impl TicketDocument {
186 pub fn parse(body: &str) -> Result<Self> {
187 let (spec_part, raw_history) = if let Some(pos) = body.find("\n## History") {
188 (&body[..pos], body[pos + 1..].to_string())
189 } else {
190 (body, String::new())
191 };
192
193 let mut sections = IndexMap::new();
194 let mut current_name: Option<String> = None;
195 let mut current_lines: Vec<&str> = Vec::new();
196
197 for line in spec_part.lines() {
198 if let Some(name) = line.strip_prefix("### ") {
199 if let Some(prev) = current_name.take() {
200 sections.insert(prev, current_lines.join("\n").trim().to_string());
201 }
202 current_name = Some(name.trim().to_string());
203 current_lines.clear();
204 } else if line.starts_with("## ") {
205 if let Some(prev) = current_name.take() {
206 sections.insert(prev, current_lines.join("\n").trim().to_string());
207 }
208 current_lines.clear();
209 } else if current_name.is_some() {
210 current_lines.push(line);
211 }
212 }
213 if let Some(name) = current_name {
214 sections.insert(name, current_lines.join("\n").trim().to_string());
215 }
216
217 Ok(Self { sections, raw_history })
218 }
219
220 pub fn serialize(&self) -> String {
221 let mut out = String::from("## Spec\n");
222
223 for (name, value) in &self.sections {
224 out.push_str(&format!("\n### {}\n\n", name));
225 if !value.is_empty() {
226 out.push_str(value);
227 out.push('\n');
228 }
229 }
230
231 if !self.raw_history.is_empty() {
232 out.push('\n');
233 out.push_str(&self.raw_history);
234 }
235
236 out
237 }
238
239 pub fn validate(&self, config_sections: &[crate::config::TicketSection]) -> Vec<ValidationError> {
240 use crate::config::SectionType;
241 let mut errors = Vec::new();
242 for sec in config_sections {
243 if !sec.required {
244 continue;
245 }
246 let val = self.sections.get(&sec.name).map(|s| s.as_str()).unwrap_or("");
247 if val.is_empty() {
248 if sec.type_ == SectionType::Tasks {
249 errors.push(ValidationError::NoAcceptanceCriteria);
250 } else {
251 errors.push(ValidationError::EmptySection(sec.name.clone()));
252 }
253 continue;
254 }
255 if sec.type_ == SectionType::Tasks && parse_checklist(val).is_empty() {
256 errors.push(ValidationError::NoAcceptanceCriteria);
257 }
258 }
259 errors
260 }
261}
262
263pub fn normalize_id_arg(arg: &str) -> Result<String> {
266 if (4..=8).contains(&arg.len()) && arg.chars().all(|c| c.is_ascii_hexdigit()) {
269 return Ok(arg.to_lowercase());
270 }
271 if !arg.is_empty() && arg.chars().all(|c| c.is_ascii_digit()) {
272 let n: u64 = arg.parse().context("invalid integer ID")?;
273 return Ok(format!("{n:04}"));
274 }
275 bail!("invalid ticket ID {:?}: use 4–8 hex chars or a plain integer", arg);
276}
277
278pub fn id_arg_prefixes(arg: &str) -> Result<Vec<String>> {
284 let canonical = normalize_id_arg(arg)?;
285 if arg.chars().all(|c| c.is_ascii_digit()) && arg.len() < 4 {
286 Ok(vec![canonical, arg.to_string()])
287 } else {
288 Ok(vec![canonical])
289 }
290}
291
292pub fn resolve_id_in_slice(tickets: &[Ticket], arg: &str) -> Result<String> {
294 let prefixes = id_arg_prefixes(arg)?;
295 let mut seen = std::collections::HashSet::new();
296 let matches: Vec<&Ticket> = tickets.iter()
297 .filter(|t| {
298 let id = &t.frontmatter.id;
299 prefixes.iter().any(|p| id.starts_with(p.as_str())) && seen.insert(id.clone())
300 })
301 .collect();
302 match matches.len() {
303 0 => bail!("no ticket matches '{arg}'"),
304 1 => Ok(matches[0].frontmatter.id.clone()),
305 _ => {
306 let mut msg = format!("error: prefix '{arg}' is ambiguous");
307 for t in &matches {
308 msg.push_str(&format!("\n {} {}", t.frontmatter.id, t.frontmatter.title));
309 }
310 bail!("{msg}")
311 }
312 }
313}
314
315#[cfg(test)]
316mod tests {
317 use super::*;
318 use std::path::Path;
319
320 fn dummy_path() -> &'static Path {
321 Path::new("test.md")
322 }
323
324 fn minimal_raw(extra_fm: &str, body: &str) -> String {
325 format!(
326 "+++\nid = \"0001\"\ntitle = \"Test\"\nstate = \"new\"\n{extra_fm}+++\n\n{body}"
327 )
328 }
329
330 fn minimal_raw_int(extra_fm: &str, body: &str) -> String {
331 format!(
332 "+++\nid = 1\ntitle = \"Test\"\nstate = \"new\"\n{extra_fm}+++\n\n{body}"
333 )
334 }
335
336 #[test]
339 fn parse_well_formed() {
340 let raw = minimal_raw("priority = 5\n", "## Spec\n\nHello\n");
341 let t = Ticket::parse(dummy_path(), &raw).unwrap();
342 assert_eq!(t.frontmatter.id, "0001");
343 assert_eq!(t.frontmatter.title, "Test");
344 assert_eq!(t.frontmatter.state, "new");
345 assert_eq!(t.frontmatter.priority, 5);
346 assert_eq!(t.body, "## Spec\n\nHello\n");
347 }
348
349 #[test]
350 fn parse_integer_id_is_zero_padded() {
351 let raw = minimal_raw_int("", "");
352 let t = Ticket::parse(dummy_path(), &raw).unwrap();
353 assert_eq!(t.frontmatter.id, "0001");
354 }
355
356 #[test]
357 fn parse_optional_fields_default() {
358 let raw = minimal_raw("", "");
359 let t = Ticket::parse(dummy_path(), &raw).unwrap();
360 assert_eq!(t.frontmatter.priority, 0);
361 assert_eq!(t.frontmatter.effort, 0);
362 assert_eq!(t.frontmatter.risk, 0);
363 assert!(t.frontmatter.branch.is_none());
364 }
365
366 #[test]
367 fn parse_epic_field() {
368 let raw = minimal_raw("epic = \"ab12cd34\"\n", "");
369 let t = Ticket::parse(dummy_path(), &raw).unwrap();
370 assert_eq!(t.frontmatter.epic, Some("ab12cd34".to_string()));
371 }
372
373 #[test]
374 fn parse_target_branch_field() {
375 let raw = minimal_raw("target_branch = \"epic/ab12cd34-user-auth\"\n", "");
376 let t = Ticket::parse(dummy_path(), &raw).unwrap();
377 assert_eq!(t.frontmatter.target_branch, Some("epic/ab12cd34-user-auth".to_string()));
378 }
379
380 #[test]
381 fn parse_depends_on_field() {
382 let raw = minimal_raw("depends_on = [\"cd56ef78\", \"12ab34cd\"]\n", "");
383 let t = Ticket::parse(dummy_path(), &raw).unwrap();
384 assert_eq!(t.frontmatter.depends_on, Some(vec!["cd56ef78".to_string(), "12ab34cd".to_string()]));
385 }
386
387 #[test]
388 fn parse_omits_new_fields() {
389 let raw = minimal_raw("", "");
390 let t = Ticket::parse(dummy_path(), &raw).unwrap();
391 assert!(t.frontmatter.epic.is_none());
392 assert!(t.frontmatter.target_branch.is_none());
393 assert!(t.frontmatter.depends_on.is_none());
394 }
395
396 #[test]
397 fn serialize_omits_absent_fields() {
398 let raw = minimal_raw("", "## Spec\n\ncontent\n");
399 let t = Ticket::parse(dummy_path(), &raw).unwrap();
400 let serialized = t.serialize().unwrap();
401 assert!(!serialized.contains("epic"));
402 assert!(!serialized.contains("target_branch"));
403 assert!(!serialized.contains("depends_on"));
404 }
405
406 #[test]
407 fn parse_missing_opening_delimiter() {
408 let raw = "id = \"0001\"\ntitle = \"Test\"\nstate = \"new\"\n+++\n\nbody\n";
409 let err = Ticket::parse(dummy_path(), raw).unwrap_err();
410 assert!(err.to_string().contains("missing frontmatter"));
411 }
412
413 #[test]
414 fn parse_unclosed_frontmatter() {
415 let raw = "+++\nid = \"0001\"\ntitle = \"Test\"\nstate = \"new\"\n\nbody\n";
416 let err = Ticket::parse(dummy_path(), raw).unwrap_err();
417 assert!(err.to_string().contains("unclosed frontmatter"));
418 }
419
420 #[test]
421 fn parse_invalid_toml() {
422 let raw = "+++\nid = not_a_number\n+++\n\nbody\n";
423 let err = Ticket::parse(dummy_path(), raw).unwrap_err();
424 assert!(err.to_string().contains("cannot parse frontmatter"));
425 }
426
427 #[test]
428 fn epic_and_depends_on_round_trip() {
429 let raw = minimal_raw(
430 "epic = \"ab12cd34\"\ndepends_on = [\"cd56ef78\", \"12ab34cd\"]\n",
431 "## Spec\n\ncontent\n",
432 );
433 let t = Ticket::parse(dummy_path(), &raw).unwrap();
434 assert_eq!(t.frontmatter.epic, Some("ab12cd34".to_string()));
435 assert_eq!(
436 t.frontmatter.depends_on,
437 Some(vec!["cd56ef78".to_string(), "12ab34cd".to_string()])
438 );
439 let serialized = t.serialize().unwrap();
440 assert!(serialized.contains("epic = \"ab12cd34\""));
441 assert!(serialized.contains("depends_on = [\"cd56ef78\", \"12ab34cd\"]"));
442 let t2 = Ticket::parse(dummy_path(), &serialized).unwrap();
443 assert_eq!(t2.frontmatter.epic, Some("ab12cd34".to_string()));
444 assert_eq!(
445 t2.frontmatter.depends_on,
446 Some(vec!["cd56ef78".to_string(), "12ab34cd".to_string()])
447 );
448 }
449
450 #[test]
451 fn target_branch_round_trips() {
452 let raw = minimal_raw("target_branch = \"epic/abc\"\n", "## Spec\n\ncontent\n");
453 let t = Ticket::parse(dummy_path(), &raw).unwrap();
454 let serialized = t.serialize().unwrap();
455 assert!(serialized.contains("target_branch = \"epic/abc\""));
456 let t2 = Ticket::parse(dummy_path(), &serialized).unwrap();
457 assert_eq!(t2.frontmatter.target_branch, Some("epic/abc".to_string()));
458 }
459
460 #[test]
461 fn target_branch_absent_not_added_on_round_trip() {
462 let raw = minimal_raw("", "## Spec\n\ncontent\n");
463 let t = Ticket::parse(dummy_path(), &raw).unwrap();
464 let serialized = t.serialize().unwrap();
465 assert!(!serialized.contains("target_branch"));
466 let t2 = Ticket::parse(dummy_path(), &serialized).unwrap();
467 assert!(t2.frontmatter.target_branch.is_none());
468 }
469
470 #[test]
473 fn serialize_round_trips() {
474 let raw = minimal_raw("effort = 3\nrisk = 1\n", "## Spec\n\ncontent\n");
475 let t = Ticket::parse(dummy_path(), &raw).unwrap();
476 let serialized = t.serialize().unwrap();
477 let t2 = Ticket::parse(dummy_path(), &serialized).unwrap();
478 assert_eq!(t2.frontmatter.id, t.frontmatter.id);
479 assert_eq!(t2.frontmatter.title, t.frontmatter.title);
480 assert_eq!(t2.frontmatter.state, t.frontmatter.state);
481 assert_eq!(t2.frontmatter.effort, t.frontmatter.effort);
482 assert_eq!(t2.frontmatter.risk, t.frontmatter.risk);
483 assert_eq!(t2.body, t.body);
484 }
485
486 #[test]
489 fn slugify_basic() {
490 assert_eq!(slugify("Hello World"), "hello-world");
491 }
492
493 #[test]
494 fn slugify_special_chars() {
495 assert_eq!(slugify("Add apm init --hooks (install git hooks)"), "add-apm-init-hooks-install-git-hooks");
496 }
497
498 #[test]
499 fn slugify_truncates_at_40() {
500 let long = "a".repeat(50);
501 assert_eq!(slugify(&long).len(), 40);
502 }
503
504 #[test]
505 fn slugify_collapses_separators() {
506 assert_eq!(slugify("foo -- bar"), "foo-bar");
507 }
508
509 #[test]
512 fn normalize_integer_pads_to_four() {
513 assert_eq!(normalize_id_arg("35").unwrap(), "0035");
514 assert_eq!(normalize_id_arg("1").unwrap(), "0001");
515 assert_eq!(normalize_id_arg("9999").unwrap(), "9999");
516 }
517
518 #[test]
519 fn normalize_hex_passthrough() {
520 assert_eq!(normalize_id_arg("a3f9b2c1").unwrap(), "a3f9b2c1");
521 assert_eq!(normalize_id_arg("a3f9").unwrap(), "a3f9");
522 }
523
524 #[test]
525 fn normalize_too_short_errors() {
526 assert!(normalize_id_arg("abc").is_err());
527 }
528
529 #[test]
530 fn normalize_non_hex_errors() {
531 assert!(normalize_id_arg("gggg").is_err());
532 }
533
534 #[test]
537 fn prefixes_short_digit_returns_two() {
538 let p = id_arg_prefixes("314").unwrap();
539 assert_eq!(p, vec!["0314", "314"]);
540 }
541
542 #[test]
543 fn prefixes_four_digit_returns_one() {
544 let p = id_arg_prefixes("3142").unwrap();
545 assert_eq!(p, vec!["3142"]);
546 }
547
548 #[test]
549 fn prefixes_hex_returns_one() {
550 let p = id_arg_prefixes("a3f9").unwrap();
551 assert_eq!(p, vec!["a3f9"]);
552 }
553
554 fn make_ticket_with_title(id: &str, title: &str) -> Ticket {
557 let raw = format!(
558 "+++\nid = \"{id}\"\ntitle = \"{title}\"\nstate = \"new\"\n+++\n\nbody\n"
559 );
560 let path = std::path::PathBuf::from(format!("tickets/{id}.md"));
561 Ticket::parse(&path, &raw).unwrap()
562 }
563
564 #[test]
565 fn resolve_short_digit_prefix_unique() {
566 let tickets = vec![make_ticket_with_title("314abcde", "Alpha")];
567 assert_eq!(resolve_id_in_slice(&tickets, "314").unwrap(), "314abcde");
568 }
569
570 #[test]
571 fn resolve_integer_one_matches_0001() {
572 let tickets = vec![make_ticket_with_title("0001", "One")];
573 assert_eq!(resolve_id_in_slice(&tickets, "1").unwrap(), "0001");
574 }
575
576 #[test]
577 fn resolve_four_digit_prefix() {
578 let tickets = vec![make_ticket_with_title("3142abcd", "Beta")];
579 assert_eq!(resolve_id_in_slice(&tickets, "3142").unwrap(), "3142abcd");
580 }
581
582 #[test]
583 fn resolve_ambiguous_prefix_lists_candidates() {
584 let tickets = vec![
585 make_ticket_with_title("314abcde", "Alpha"),
586 make_ticket_with_title("3142xxxx", "Beta"),
587 ];
588 let err = resolve_id_in_slice(&tickets, "314").unwrap_err().to_string();
589 assert!(err.contains("ambiguous"), "expected 'ambiguous' in: {err}");
590 assert!(err.contains("314abcde"), "expected first id in: {err}");
591 assert!(err.contains("3142xxxx"), "expected second id in: {err}");
592 }
593
594 fn full_body(ac: &str) -> String {
597 format!(
598 "## Spec\n\n### Problem\n\nSome problem.\n\n### Acceptance criteria\n\n{ac}\n\n### Out of scope\n\nNothing.\n\n### Approach\n\nDo it.\n\n## History\n\n| When | From | To | By |\n|------|------|----|----|"
599 )
600 }
601
602 fn minimal_ticket_sections() -> Vec<crate::config::TicketSection> {
603 use crate::config::{SectionType, TicketSection};
604 vec![
605 TicketSection { name: "Problem".into(), type_: SectionType::Free, required: true, placeholder: None },
606 TicketSection { name: "Acceptance criteria".into(), type_: SectionType::Tasks, required: true, placeholder: None },
607 TicketSection { name: "Out of scope".into(), type_: SectionType::Free, required: true, placeholder: None },
608 TicketSection { name: "Approach".into(), type_: SectionType::Free, required: true, placeholder: None },
609 ]
610 }
611
612 #[test]
613 fn document_parse_required_sections() {
614 let body = full_body("- [ ] item one\n- [x] item two");
615 let doc = TicketDocument::parse(&body).unwrap();
616 assert_eq!(doc.sections.get("Problem").map(|s| s.as_str()), Some("Some problem."));
617 let ac = doc.sections.get("Acceptance criteria").unwrap();
618 assert!(ac.contains("- [ ] item one"));
619 assert!(ac.contains("- [x] item two"));
620 assert_eq!(doc.sections.get("Out of scope").map(|s| s.as_str()), Some("Nothing."));
621 assert_eq!(doc.sections.get("Approach").map(|s| s.as_str()), Some("Do it."));
622 }
623
624 #[test]
625 fn document_parse_missing_section_fails_validate() {
626 let body = "## Spec\n\n### Problem\n\nSome problem.\n\n## History\n\n";
627 let doc = TicketDocument::parse(body).unwrap();
628 let errs = doc.validate(&minimal_ticket_sections());
629 assert!(!errs.is_empty(), "expected validation errors for missing required sections");
630 }
631
632 #[test]
633 fn document_parse_unknown_section_preserved() {
634 let body = "## Spec\n\n### Problem\n\nfoo\n\n### Acceptance criteria\n\n- [x] done\n\n### Out of scope\n\nbar\n\n### Approach\n\nbaz\n\n### Foo\n\nsome custom content\n\n## History\n\n";
635 let doc = TicketDocument::parse(body).unwrap();
636 assert_eq!(doc.sections.get("Foo").map(|s| s.as_str()), Some("some custom content"));
637 let s = doc.serialize();
638 assert!(s.contains("### Foo"), "unknown section should be preserved in serialization");
639 assert!(s.contains("some custom content"));
640 }
641
642 #[test]
643 fn document_parse_code_review_preserved() {
644 let body = "## Spec\n\n### Problem\n\nfoo\n\n### Acceptance criteria\n\n- [x] done\n\n### Out of scope\n\nbar\n\n### Approach\n\nbaz\n\n### Code review\n\n- [ ] Check tests\n\n## History\n\n";
645 let doc = TicketDocument::parse(body).unwrap();
646 let s = doc.serialize();
647 assert!(s.contains("### Code review"), "Code review section should survive round-trip");
648 assert!(s.contains("- [ ] Check tests"));
649 }
650
651 #[test]
652 fn document_round_trip() {
653 let body = full_body("- [ ] criterion A\n- [x] criterion B");
654 let doc = TicketDocument::parse(&body).unwrap();
655 let serialized = doc.serialize();
656 let doc2 = TicketDocument::parse(&serialized).unwrap();
657 assert_eq!(doc2.sections.get("Problem"), doc.sections.get("Problem"));
658 assert_eq!(doc2.sections.get("Acceptance criteria"), doc.sections.get("Acceptance criteria"));
659 assert_eq!(doc2.sections.get("Out of scope"), doc.sections.get("Out of scope"));
660 assert_eq!(doc2.sections.get("Approach"), doc.sections.get("Approach"));
661 }
662
663 #[test]
664 fn document_validate_empty_sections() {
665 let body = "## Spec\n\n### Problem\n\n\n### Acceptance criteria\n\n- [ ] x\n\n### Out of scope\n\n\n### Approach\n\ncontent\n";
666 let doc = TicketDocument::parse(body).unwrap();
667 let errs = doc.validate(&minimal_ticket_sections());
668 let msgs: Vec<String> = errs.iter().map(|e| e.to_string()).collect();
669 assert!(msgs.iter().any(|m| m.contains("Problem")));
670 assert!(msgs.iter().any(|m| m.contains("Out of scope")));
671 assert!(!msgs.iter().any(|m| m.contains("Approach")));
672 }
673
674 #[test]
675 fn document_validate_no_criteria() {
676 let body = "## Spec\n\n### Problem\n\nfoo\n\n### Acceptance criteria\n\n\n### Out of scope\n\nbar\n\n### Approach\n\nbaz\n";
677 let doc = TicketDocument::parse(body).unwrap();
678 let errs = doc.validate(&minimal_ticket_sections());
679 assert!(errs.iter().any(|e| matches!(e, ValidationError::NoAcceptanceCriteria)));
680 }
681
682 #[test]
683 fn document_validate_required_from_config() {
684 use crate::config::{SectionType, TicketSection};
685 let body = "## Spec\n\n### Problem\n\nfoo\n\n";
686 let doc = TicketDocument::parse(body).unwrap();
687 let sections = vec![
688 TicketSection { name: "Problem".into(), type_: SectionType::Free, required: true, placeholder: None },
689 TicketSection { name: "Context".into(), type_: SectionType::Free, required: true, placeholder: None },
690 ];
691 let errs = doc.validate(§ions);
692 let msgs: Vec<String> = errs.iter().map(|e| e.to_string()).collect();
693 assert!(msgs.iter().any(|m| m.contains("Context")), "required config section should be validated");
694 assert!(!msgs.iter().any(|m| m.contains("Problem")), "present section should not error");
695 }
696
697 #[test]
698 fn frontmatter_agent_round_trip() {
699 let raw = minimal_raw(
700 "agent = \"mock-happy\"\n\n[agent_overrides]\nspec_agent = \"claude\"\nimpl_agent = \"mock-sad\"\n",
701 "## Spec\n\ncontent\n",
702 );
703 let t = Ticket::parse(dummy_path(), &raw).unwrap();
704 assert_eq!(t.frontmatter.agent, Some("mock-happy".to_string()));
705 assert_eq!(t.frontmatter.agent_overrides.get("spec_agent").map(|s| s.as_str()), Some("claude"));
706 assert_eq!(t.frontmatter.agent_overrides.get("impl_agent").map(|s| s.as_str()), Some("mock-sad"));
707
708 let serialized = t.serialize().unwrap();
709 let t2 = Ticket::parse(dummy_path(), &serialized).unwrap();
710 assert_eq!(t2.frontmatter.agent, Some("mock-happy".to_string()));
711 assert_eq!(t2.frontmatter.agent_overrides.get("spec_agent").map(|s| s.as_str()), Some("claude"));
712 assert_eq!(t2.frontmatter.agent_overrides.get("impl_agent").map(|s| s.as_str()), Some("mock-sad"));
713 }
714
715 #[test]
716 fn frontmatter_agent_omitted_when_unset() {
717 let raw = minimal_raw("", "## Spec\n\ncontent\n");
718 let t = Ticket::parse(dummy_path(), &raw).unwrap();
719 assert!(t.frontmatter.agent.is_none());
720 assert!(t.frontmatter.agent_overrides.is_empty());
721
722 let serialized = t.serialize().unwrap();
723 assert!(!serialized.contains("agent"), "agent field must not appear in serialized output");
724 assert!(!serialized.contains("agent_overrides"), "agent_overrides must not appear in serialized output");
725 }
726
727 #[test]
728 fn document_history_preserved() {
729 let body = full_body("- [x] done");
730 let doc = TicketDocument::parse(&body).unwrap();
731 let s = doc.serialize();
732 assert!(s.contains("## History"));
733 assert!(s.contains("| When |"));
734 }
735}
736
737pub fn gen_hex_id() -> String {
741 use std::time::{SystemTime, UNIX_EPOCH};
742 let dur = SystemTime::now().duration_since(UNIX_EPOCH).unwrap_or_default();
743 let secs = dur.as_secs();
744 let nanos = dur.subsec_nanos() as u64;
745 let pid = std::process::id() as u64;
746 let a = secs.wrapping_mul(0x9e3779b97f4a7c15).wrapping_add(nanos);
748 let b = (a ^ (a >> 30)).wrapping_mul(0xbf58476d1ce4e5b9);
749 let c = (b ^ (b >> 27)).wrapping_mul(0x94d049bb133111eb);
750 let result = (c ^ (c >> 31)) ^ pid.wrapping_mul(0x6c62272e07bb0142);
751 format!("{:016x}", result)[..8].to_string()
752}
753
754pub fn resolve_ticket_branch(branches: &[String], arg: &str) -> Result<String> {
757 let prefixes = id_arg_prefixes(arg)?;
758 let mut seen = std::collections::HashSet::new();
759 let matches: Vec<&String> = branches.iter()
760 .filter(|b| {
761 let id = b.strip_prefix("ticket/")
762 .and_then(|s| s.split('-').next())
763 .unwrap_or("");
764 prefixes.iter().any(|p| id.starts_with(p.as_str())) && seen.insert(id.to_string())
765 })
766 .collect();
767 match matches.len() {
768 0 => bail!("no ticket matches '{arg}'"),
769 1 => Ok(matches[0].clone()),
770 _ => {
771 let mut msg = format!("error: prefix '{arg}' is ambiguous");
772 for b in &matches {
773 let id = b.strip_prefix("ticket/")
774 .and_then(|s| s.split('-').next())
775 .unwrap_or(b.as_str());
776 msg.push_str(&format!("\n {id} ({})", b));
777 }
778 bail!("{msg}")
779 }
780 }
781}
782
783pub fn branch_name_from_path(path: &Path) -> Option<String> {
786 let stem = path.file_stem()?.to_str()?;
787 Some(format!("ticket/{stem}"))
788}