1 files changed, 24 insertions, 0 deletions

diff --git a/bin/src/utils.rs b/bin/src/utils.rs
new file mode 100644
index 0000000..747a761
--- /dev/null
+++ b/bin/src/utils.rs
@@ -0,0 +1,24 @@
+use std::collections::HashMap;
+
+use lib::{Lint, LINTS};
+use rnix::SyntaxKind;
+
+pub fn lint_map_of(
+    lints: &[&'static Box<dyn Lint>],
+) -> HashMap<SyntaxKind, Vec<&'static Box<dyn Lint>>> {
+    let mut map = HashMap::new();
+    for lint in lints.iter() {
+        let lint = *lint;
+        let matches = lint.match_kind();
+        for m in matches {
+            map.entry(m)
+                .and_modify(|v: &mut Vec<_>| v.push(lint))
+                .or_insert_with(|| vec![lint]);
+        }
+    }
+    map
+}
+
+pub fn lint_map() -> HashMap<SyntaxKind, Vec<&'static Box<dyn Lint>>> {
+    lint_map_of(&*LINTS)
+}