utils: introduce ContainingPath trait

kaspar030 · Dec 22, 2023 · f431bf6 · f431bf6
1 parent 25cf214
commit f431bf6
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/utils.rs b/src/utils.rs
@@ -1,6 +1,9 @@
 use std::collections::hash_map::DefaultHasher;
 use std::hash::{Hash, Hasher};
 
+use camino::Utf8PathBuf;
+use indexmap::IndexMap;
+
 pub(crate) fn calculate_hash<T: Hash>(t: &T) -> u64 {
     let mut s = DefaultHasher::new();
     t.hash(&mut s);
@@ -20,3 +23,17 @@ pub enum StringOrMapString {
     String(String),
     Map(std::collections::HashMap<String, String>),
 }
+
+pub(crate) trait ContainingPath<T: AsRef<std::path::Path>> {
+    fn get_containing_path(&self, path: &T) -> Option<&T>;
+}
+
+impl ContainingPath<Utf8PathBuf> for IndexMap<&Utf8PathBuf, Utf8PathBuf> {
+    fn get_containing_path(&self, path: &Utf8PathBuf) -> Option<&Utf8PathBuf> {
+        self.get(path).or_else(|| {
+            self.iter()
+                .find(|(k, _)| path.starts_with(k))
+                .map(|(_, v)| v)
+        })
+    }
+}