xtask/codegen/
theme_list.rs

1use std::fs;
2
3use anyhow::{anyhow, Error, Result};
4use fancy_regex::Regex;
5use once_cell::sync::Lazy;
6
7const HEADER: &str = r##"
8/////////////////////////////////////////////
9//// All following code is autogenerated ////
10//// by running `cargo xtask codegen` in ////
11//// the syntastica workspace. //////////////
12/////////////////////////////////////////////
13"##;
14
15static FUNC_REGEX: Lazy<Regex> =
16    Lazy::new(|| Regex::new(r"pub fn ([a-z_]+)\(\) -> ResolvedTheme").unwrap());
17
18pub fn write() -> Result<()> {
19    let all_themes = find_all_themes()?;
20
21    let lib_rs_path = crate::WORKSPACE_DIR.join("syntastica-themes/src/lib.rs");
22    let mut lib_rs = fs::read_to_string(&lib_rs_path)?;
23
24    if let Some((preserve, _)) = lib_rs.split_once(HEADER) {
25        lib_rs.truncate(preserve.len());
26    }
27    lib_rs += HEADER;
28
29    lib_rs += r###"
30/// Try to get a theme given its path as a string.
31///
32/// For a list of all acceptable theme names see [`THEMES`].
33///
34/// # Example
35///
36/// ```
37/// assert_eq!(
38///     syntastica_themes::from_str("one::dark"),
39///     Some(syntastica_themes::one::dark()),
40/// );
41/// ```
42pub fn from_str(theme_name: impl AsRef<str>) -> Option<ResolvedTheme> {
43    match theme_name.as_ref() {
44"###;
45    for theme in &all_themes {
46        lib_rs += &format!("        \"{theme}\" => Some({theme}()),\n");
47    }
48    lib_rs += &r###"
49        _ => None,
50    }
51}
52
53/// A list of all theme names as they are accepted by [`from_str`].
54pub const THEMES: &[&str] = &[
55"###[1..];
56    for theme in &all_themes {
57        lib_rs += &format!("    \"{theme}\",\n");
58    }
59    lib_rs += "];\n";
60
61    fs::write(&lib_rs_path, lib_rs)?;
62
63    Ok(())
64}
65
66pub fn find_all_themes() -> Result<Vec<String>> {
67    let mut all_themes = crate::WORKSPACE_DIR
68        .join("syntastica-themes/src")
69        .read_dir()?
70        .filter_map(|entry| -> Option<Result<Result<Vec<_>, _>>> {
71            let entry = match entry {
72                Ok(e) => e,
73                Err(err) => return Some(Err(Error::from(err))),
74            };
75
76            // skip the `lib.rs` file and directories
77            if entry.file_name() == "lib.rs" || entry.path().is_dir() {
78                None
79            } else {
80                let module_name = match entry.file_name().into_string() {
81                    Ok(filename) => filename.strip_suffix(".rs")?.to_owned(),
82                    Err(_) => return Some(Err(anyhow!("invalid filename encountered"))),
83                };
84
85                let contents = fs::read_to_string(entry.path()).ok()?;
86                Some(Ok(FUNC_REGEX
87                    .captures_iter(&contents)
88                    .map(|c| c.map(|captures| format!("{module_name}::{}", &captures[1])))
89                    .collect()))
90            }
91        })
92        .collect::<Result<Result<Vec<_>, _>>>()??
93        .into_iter()
94        .flatten()
95        .collect::<Vec<_>>();
96    all_themes.sort_unstable();
97    Ok(all_themes)
98}
xtask/codegen/theme_list.rs

xtask/codegen/
theme_list.rs
