hjkl_theme/
captures.rs

1use std::collections::HashMap;
2
3use crate::style::StyleSpec;
4
5/// Flat map of tree-sitter capture names to resolved `StyleSpec` values.
6///
7/// Use `resolve` for fallback-chain lookup; `get` for exact-match only.
8#[derive(Clone, Default, Debug)]
9pub struct CaptureMap {
10    flat: HashMap<String, StyleSpec>,
11}
12
13impl CaptureMap {
14    pub(crate) fn from_map(flat: HashMap<String, StyleSpec>) -> Self {
15        Self { flat }
16    }
17
18    /// Exact-match lookup. No fallback.
19    pub fn get(&self, capture: &str) -> Option<&StyleSpec> {
20        self.flat.get(capture)
21    }
22
23    /// Walk the fallback chain: `a.b.c` -> `a.b` -> `a` -> `None`.
24    pub fn resolve(&self, capture: &str) -> Option<&StyleSpec> {
25        let mut key = capture;
26        loop {
27            if let Some(spec) = self.flat.get(key) {
28                return Some(spec);
29            }
30            match key.rfind('.') {
31                Some(pos) => key = &key[..pos],
32                None => return None,
33            }
34        }
35    }
36}
hjkl_theme/captures.rs

hjkl_theme/
captures.rs
