62 lines
1.6 KiB
Rust
62 lines
1.6 KiB
Rust
use std::fs::{self, File};
|
|
use std::io;
|
|
use std::io::Write;
|
|
|
|
use shadow_rs::SdResult;
|
|
|
|
fn main() -> SdResult<()> {
|
|
shadow_rs::new_hook(gen_presets_hook)?;
|
|
|
|
#[cfg(windows)]
|
|
{
|
|
let mut res = winres::WindowsResource::new();
|
|
res.set_manifest_file("starship.exe.manifest")
|
|
.set_icon("media/icon.ico");
|
|
res.compile()?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn gen_presets_hook(mut file: &File) -> SdResult<()> {
|
|
println!("cargo:rerun-if-changed=docs/.vuepress/public/presets/toml");
|
|
let paths = fs::read_dir("docs/.vuepress/public/presets/toml")?;
|
|
let mut sortedpaths = paths.collect::<io::Result<Vec<_>>>()?;
|
|
sortedpaths.sort_by_key(std::fs::DirEntry::path);
|
|
|
|
let mut presets = String::new();
|
|
let mut match_arms = String::new();
|
|
for unwrapped in sortedpaths {
|
|
let file_name = unwrapped.file_name();
|
|
let full_path = dunce::canonicalize(unwrapped.path())?;
|
|
let full_path = full_path.to_str().expect("failed to convert to string");
|
|
let name = file_name
|
|
.to_str()
|
|
.and_then(|v| v.strip_suffix(".toml"))
|
|
.expect("Failed to process filename");
|
|
presets.push_str(format!("print::Preset(\"{name}\"),\n").as_str());
|
|
match_arms.push_str(format!(r#""{name}" => include_bytes!(r"{full_path}"),"#).as_str());
|
|
}
|
|
|
|
writeln!(
|
|
file,
|
|
r#"
|
|
use crate::print;
|
|
|
|
pub fn get_preset_list<'a>() -> &'a [print::Preset] {{
|
|
&[
|
|
{presets}
|
|
]
|
|
}}
|
|
|
|
pub fn get_preset_content(name: &str) -> &[u8] {{
|
|
match name {{
|
|
{match_arms}
|
|
_ => unreachable!(),
|
|
}}
|
|
}}
|
|
"#
|
|
)?;
|
|
Ok(())
|
|
}
|