fastobo_validator/
duplicates.rs

1use std::collections::HashMap;
2use std::error::Error;
3use std::fmt::Display;
4use std::fmt::Formatter;
5use std::fmt::Result as FmtResult;
6
7use fastobo::ast::*;
8
9use fastobo::semantics::Identified;
10
11use fastobo::visit::Visit;
12
13use super::ValidationError;
14use super::Validator;
15
16#[derive(Debug)]
17pub struct DuplicateIdError {
18    id: Ident,
19    count: usize,
20}
21
22impl Display for DuplicateIdError {
23    fn fmt(&self, f: &mut Formatter) -> FmtResult {
24        write!(
25            f,
26            "id `{}` appears more than once ({} times)",
27            self.id, self.count
28        )
29    }
30}
31
32impl Error for DuplicateIdError {
33    fn description(&self) -> &str {
34        "id appears more than once"
35    }
36}
37
38#[derive(Default)]
39pub struct DuplicateIdChecker<'a> {
40    counts: HashMap<&'a Ident, usize>,
41}
42
43macro_rules! impl_visit {
44    ($name:ident, $frame:ty) => {
45        fn $name(&mut self, frame: &'a $frame) {
46            *self.counts.entry(frame.as_id()).or_default() += 1;
47        }
48    };
49}
50
51impl<'a> Visit<'a> for DuplicateIdChecker<'a> {
52    impl_visit!(visit_term_frame, TermFrame);
53    impl_visit!(visit_typedef_frame, TypedefFrame);
54    impl_visit!(visit_instance_frame, InstanceFrame);
55}
56
57impl Validator for DuplicateIdChecker<'_> {
58    fn validate(doc: &OboDoc) -> Vec<ValidationError> {
59        let mut checker = Self::default();
60        checker.visit_doc(doc);
61
62        let mut errors = Vec::new();
63        for (id, count) in checker.counts {
64            if count > 1 {
65                errors.push(ValidationError {
66                    location: String::from("complete document"),
67                    cause: Box::new(DuplicateIdError {
68                        id: id.clone(),
69                        count,
70                    }),
71                })
72            }
73        }
74
75        errors
76    }
77}