attributes.rs 95.7%

112 of 117 lines run, 21 of 40 functions · run #376 on main · run missed

//! `.oxyattributes` (spec §19): per-path attributes in gitattributes syntax.
//! Each line is `pattern attr...` where an attribute is `binary`, `text`
//! (or `-binary`), `eol=lf`, `eol=crlf`, `nocompress`, or `lockable`. A
//! pattern without a slash matches a basename anywhere below the file's
//! directory; one with a slash is anchored to that directory. Later rules
//! override earlier ones per attribute.

use globset::{Glob, GlobMatcher};
use std::collections::BTreeMap;

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum Eol {
    Lf,
    Crlf,
}

/// The attributes in force for one path.
#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
pub struct Attrs {
    /// `Some(true)` forces opaque, `Some(false)` forces text (absent NUL bytes).
    pub binary: Option<bool>,
    pub eol: Option<Eol>,
    pub nocompress: bool,
    pub lockable: bool,
}

struct Rule {
    matcher: GlobMatcher,
    attrs: Vec<(String, Option<String>)>,
}

#[derive(Default)]
pub struct Attributes {
    rules: Vec<Rule>,
}

impl Attributes {
    /// Add the rules of one `.oxyattributes` file living in `dir` (`""` for the root).
    pub fn add_file(&mut self, dir: &str, text: &str) {
        for line in text.lines() {
            let line = line.trim();
            if line.is_empty() || line.starts_with('#') {
                continue;
            }
            let mut parts = line.split_whitespace();
            let Some(pattern) = parts.next() else {
                continue;
            };
            let attrs: Vec<(String, Option<String>)> = parts
                .map(|a| match a.split_once('=') {
                    Some((k, v)) => (k.to_string(), Some(v.to_string())),
                    None => (a.to_string(), None),
                })
                .collect();
            let anchored = pattern.contains('/');
            let core = pattern.trim_start_matches('/');
            let full = match (dir.is_empty(), anchored) {
                (true, true) => core.to_string(),
                (true, false) => format!("**/{core}"),
                (false, true) => format!("{dir}/{core}"),
                (false, false) => format!("{dir}/**/{core}"),
            };
            if let Ok(g) = Glob::new(&full) {
                self.rules.push(Rule {
                    matcher: g.compile_matcher(),
                    attrs,
                });
            }
        }
    }

    /// Build from every `.oxyattributes` in `files` (path → content).
    pub fn from_files<'a>(files: impl IntoIterator<Item = (&'a str, &'a [u8])>) -> Attributes {
        let mut sorted: BTreeMap<&str, &[u8]> = BTreeMap::new();
        for (p, c) in files {
            if p == ".oxyattributes" || p.ends_with("/.oxyattributes") {
                sorted.insert(p, c);
            }
        }
        let mut out = Attributes::default();
        // Shallower files first so deeper ones override.
        let mut entries: Vec<(&str, &[u8])> = sorted.into_iter().collect();
        entries.sort_by_key(|(p, _)| p.matches('/').count());
        for (p, c) in entries {
            let dir = p
                .strip_suffix(".oxyattributes")
                .unwrap_or("")
                .trim_end_matches('/');
            out.add_file(dir, &String::from_utf8_lossy(c));
        }
        out
    }

    pub fn lookup(&self, path: &str) -> Attrs {
        let mut a = Attrs::default();
        for r in &self.rules {
            if !r.matcher.is_match(path) {
                continue;
            }
            for (k, v) in &r.attrs {
                match (k.as_str(), v.as_deref()) {
                    ("binary", None) => a.binary = Some(true),
                    ("-binary", None) | ("text", None) => a.binary = Some(false),
                    ("eol", Some("lf")) => a.eol = Some(Eol::Lf),
                    ("eol", Some("crlf")) => a.eol = Some(Eol::Crlf),
                    ("-eol", None) => a.eol = None,
                    ("nocompress", None) => a.nocompress = true,
                    ("-nocompress", None) => a.nocompress = false,
                    ("lockable", None) => a.lockable = true,
                    ("-lockable", None) => a.lockable = false,
                    _ => {}
                }
            }
        }
        a
    }

    pub fn is_empty(&self) -> bool {
        self.rules.is_empty()
    }
}

/// `\r\n` → `\n`, as stored for `eol=` files.
pub fn normalize_eol(bytes: &[u8]) -> Vec<u8> {
    let mut out = Vec::with_capacity(bytes.len());
    let mut i = 0;
    while i < bytes.len() {
        if bytes[i] == b'\r' && bytes.get(i + 1) == Some(&b'\n') {
            i += 1;
            continue;
        }
        out.push(bytes[i]);
        i += 1;
    }
    out
}

/// `\n` → `\r\n` (lines already ending in `\r\n` are left alone).
pub fn to_crlf(bytes: &[u8]) -> Vec<u8> {
    let mut out = Vec::with_capacity(bytes.len() + bytes.len() / 16);
    let mut prev = 0u8;
    for &b in bytes {
        if b == b'\n' && prev != b'\r' {
            out.push(b'\r');
        }
        out.push(b);
        prev = b;
    }
    out
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn parsing_and_precedence() {
        let a = Attributes::from_files([
            (".oxyattributes", &b"*.png binary nocompress\n*.txt eol=lf\ndocs/*.txt eol=crlf\nmodels/** lockable\n"[..]),
            ("src/.oxyattributes", &b"*.dat binary\ngen.txt -eol\n"[..]),
            ("README.md", &b"not attributes"[..]),
        ]);
        assert_eq!(
            a.lookup("a/b/logo.png"),
            Attrs {
                binary: Some(true),
                eol: None,
                nocompress: true,
                lockable: false
            }
        );
        assert_eq!(a.lookup("notes.txt").eol, Some(Eol::Lf));
        assert_eq!(a.lookup("docs/notes.txt").eol, Some(Eol::Crlf));
        assert!(a.lookup("models/x/y.bin").lockable);
        assert_eq!(a.lookup("src/deep/x.dat").binary, Some(true));
        assert_eq!(a.lookup("other/x.dat").binary, None);
        assert_eq!(a.lookup("src/gen.txt").eol, None, "deeper file overrides");
        assert_eq!(a.lookup("src/other.txt").eol, Some(Eol::Lf));
        assert_eq!(a.lookup("plain"), Attrs::default());
    }

    #[test]
    fn eol_conversions() {
        assert_eq!(normalize_eol(b"a\r\nb\nc\r\n"), b"a\nb\nc\n");
        assert_eq!(to_crlf(b"a\nb\r\nc\n"), b"a\r\nb\r\nc\r\n");
        assert_eq!(normalize_eol(&to_crlf(b"x\ny\n")), b"x\ny\n");
    }
}

1		//! `.oxyattributes` (spec §19): per-path attributes in gitattributes syntax.
2		//! Each line is `pattern attr...` where an attribute is `binary`, `text`
3		//! (or `-binary`), `eol=lf`, `eol=crlf`, `nocompress`, or `lockable`. A
4		//! pattern without a slash matches a basename anywhere below the file's
5		//! directory; one with a slash is anchored to that directory. Later rules
6		//! override earlier ones per attribute.
7
8		use globset::{Glob, GlobMatcher};
9		use std::collections::BTreeMap;
10
11		#[derive(Clone, Copy, Debug, PartialEq, Eq)]
12		pub enum Eol {
13		Lf,
14		Crlf,
15		}
16
17		/// The attributes in force for one path.
18		#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
19		pub struct Attrs {
20		/// `Some(true)` forces opaque, `Some(false)` forces text (absent NUL bytes).
21		pub binary: Option<bool>,
22		pub eol: Option<Eol>,
23		pub nocompress: bool,
24		pub lockable: bool,
25		}
26
27		struct Rule {
28		matcher: GlobMatcher,
29		attrs: Vec<(String, Option<String>)>,
30		}
31
32		#[derive(Default)]
33		pub struct Attributes {
34		rules: Vec<Rule>,
35		}
36
37		impl Attributes {
38		/// Add the rules of one `.oxyattributes` file living in `dir` (`""` for the root).
39	34×	pub fn add_file(&mut self, dir: &str, text: &str) {
40	68×	for line in text.lines() {
41	68×	let line = line.trim();
42	68×	if line.is_empty() \|\| line.starts_with('#') {
43	0×	continue;
44	68×	}
45	68×	let mut parts = line.split_whitespace();
46	68×	let Some(pattern) = parts.next() else {
47	0×	continue;
48		};
49	68×	let attrs: Vec<(String, Option<String>)> = parts
50	84×	.map(\|a\| match a.split_once('=') {
51	20×	Some((k, v)) => (k.to_string(), Some(v.to_string())),
52	64×	None => (a.to_string(), None),
53	84×	})
54	68×	.collect();
55	68×	let anchored = pattern.contains('/');
56	68×	let core = pattern.trim_start_matches('/');
57	68×	let full = match (dir.is_empty(), anchored) {
58	17×	(true, true) => core.to_string(),
59	49×	(true, false) => format!("**/{core}"),
60	0×	(false, true) => format!("{dir}/{core}"),
61	2×	(false, false) => format!("{dir}/**/{core}"),
62		};
63	68×	if let Ok(g) = Glob::new(&full) {
64	68×	self.rules.push(Rule {
65	68×	matcher: g.compile_matcher(),
66	68×	attrs,
67	68×	});
68	68×	}
69		}
70	34×	}
71
72		/// Build from every `.oxyattributes` in `files` (path → content).
73	3936×	pub fn from_files<'a>(files: impl IntoIterator<Item = (&'a str, &'a [u8])>) -> Attributes {
74	3936×	let mut sorted: BTreeMap<&str, &[u8]> = BTreeMap::new();
75	3936×	for (p, c) in files {
76	35×	if p == ".oxyattributes" \|\| p.ends_with("/.oxyattributes") {
77	34×	sorted.insert(p, c);
78	34×	}
79		}
80	3936×	let mut out = Attributes::default();
81		// Shallower files first so deeper ones override.
82	3936×	let mut entries: Vec<(&str, &[u8])> = sorted.into_iter().collect();
83	3936×	entries.sort_by_key(\|(p, _)\| p.matches('/').count());
84	3936×	for (p, c) in entries {
85	34×	let dir = p
86	34×	.strip_suffix(".oxyattributes")
87	34×	.unwrap_or("")
88	34×	.trim_end_matches('/');
89	34×	out.add_file(dir, &String::from_utf8_lossy(c));
90	34×	}
91	3936×	out
92	3936×	}
93
94	84157×	pub fn lookup(&self, path: &str) -> Attrs {
95	84157×	let mut a = Attrs::default();
96	84157×	for r in &self.rules {
97	486×	if !r.matcher.is_match(path) {
98	390×	continue;
99	96×	}
100	124×	for (k, v) in &r.attrs {
101	124×	match (k.as_str(), v.as_deref()) {
102	124×	("binary", None) => a.binary = Some(true),
103	95×	("-binary", None) \| ("text", None) => a.binary = Some(false),
104	95×	("eol", Some("lf")) => a.eol = Some(Eol::Lf),
105	26×	("eol", Some("crlf")) => a.eol = Some(Eol::Crlf),
106	57×	("-eol", None) => a.eol = None,
107	56×	("nocompress", None) => a.nocompress = true,
108	28×	("-nocompress", None) => a.nocompress = false,
109	28×	("lockable", None) => a.lockable = true,
110	0×	("-lockable", None) => a.lockable = false,
111	0×	_ => {}
112		}
113		}
114		}
115	84157×	a
116	84157×	}
117
118	1×	pub fn is_empty(&self) -> bool {
119	1×	self.rules.is_empty()
120	1×	}
121		}
122
123		/// `\r\n` → `\n`, as stored for `eol=` files.
124	9×	pub fn normalize_eol(bytes: &[u8]) -> Vec<u8> {
125	9×	let mut out = Vec::with_capacity(bytes.len());
126	9×	let mut i = 0;
127	85×	while i < bytes.len() {
128	76×	if bytes[i] == b'\r' && bytes.get(i + 1) == Some(&b'\n') {
129	16×	i += 1;
130	16×	continue;
131	60×	}
132	60×	out.push(bytes[i]);
133	60×	i += 1;
134		}
135	9×	out
136	9×	}
137
138		/// `\n` → `\r\n` (lines already ending in `\r\n` are left alone).
139	4×	pub fn to_crlf(bytes: &[u8]) -> Vec<u8> {
140	4×	let mut out = Vec::with_capacity(bytes.len() + bytes.len() / 16);
141	4×	let mut prev = 0u8;
142	27×	for &b in bytes {
143	27×	if b == b'\n' && prev != b'\r' {
144	8×	out.push(b'\r');
145	19×	}
146	27×	out.push(b);
147	27×	prev = b;
148		}
149	4×	out
150	4×	}
151
152		#[cfg(test)]
153		mod tests {
154		use super::*;
155
156		#[test]
157	1×	fn parsing_and_precedence() {
158	1×	let a = Attributes::from_files([
159	1×	(".oxyattributes", &b".png binary nocompress\n.txt eol=lf\ndocs/.txt eol=crlf\nmodels/* lockable\n"[..]),
160	1×	("src/.oxyattributes", &b"*.dat binary\ngen.txt -eol\n"[..]),
161	1×	("README.md", &b"not attributes"[..]),
162	1×	]);
163	1×	assert_eq!(
164	1×	a.lookup("a/b/logo.png"),
165		Attrs {
166		binary: Some(true),
167		eol: None,
168		nocompress: true,
169		lockable: false
170		}
171		);
172	1×	assert_eq!(a.lookup("notes.txt").eol, Some(Eol::Lf));
173	1×	assert_eq!(a.lookup("docs/notes.txt").eol, Some(Eol::Crlf));
174	1×	assert!(a.lookup("models/x/y.bin").lockable);
175	1×	assert_eq!(a.lookup("src/deep/x.dat").binary, Some(true));
176	1×	assert_eq!(a.lookup("other/x.dat").binary, None);
177	1×	assert_eq!(a.lookup("src/gen.txt").eol, None, "deeper file overrides");
178	1×	assert_eq!(a.lookup("src/other.txt").eol, Some(Eol::Lf));
179	1×	assert_eq!(a.lookup("plain"), Attrs::default());
180	1×	}
181
182		#[test]
183	1×	fn eol_conversions() {
184	1×	assert_eq!(normalize_eol(b"a\r\nb\nc\r\n"), b"a\nb\nc\n");
185	1×	assert_eq!(to_crlf(b"a\nb\r\nc\n"), b"a\r\nb\r\nc\r\n");
186	1×	assert_eq!(normalize_eol(&to_crlf(b"x\ny\n")), b"x\ny\n");
187	1×	}
188		}

`/`	Search
`g` `c`	Code
`g` `h`	History
`g` `b`	Bookmarks
`g` `t`	Trust
`y`	Permalink to this state (files and trees)
`?`	This sheet

Coverage / crates/oxy-core/src/attributes.rs 95.7%

Keyboard