nectar/kinode/build.rs

102 lines
3.5 KiB
Rust
Raw Normal View History

2024-02-13 07:51:02 +03:00
use rayon::prelude::*;
2024-02-02 08:18:09 +03:00
use std::{
2024-02-13 07:51:02 +03:00
fs::{self, File},
2024-02-05 23:39:44 +03:00
io::{Cursor, Read, Write},
2024-02-13 07:51:02 +03:00
path::{Path, PathBuf},
2024-02-02 08:18:09 +03:00
};
2024-02-13 07:51:02 +03:00
use zip::write::FileOptions;
2024-02-13 07:51:53 +03:00
fn build_and_zip_package(
entry_path: PathBuf,
parent_pkg_path: &str,
) -> anyhow::Result<(String, String, Vec<u8>)> {
2024-02-13 07:51:02 +03:00
let rt = tokio::runtime::Runtime::new().unwrap();
rt.block_on(async {
kit::build::execute(&entry_path, false, false, false, true).await?;
2024-02-05 23:39:44 +03:00
let mut writer = Cursor::new(Vec::new());
2024-02-13 07:51:02 +03:00
let options = FileOptions::default()
.compression_method(zip::CompressionMethod::Stored)
.unix_permissions(0o755);
2024-02-05 23:39:44 +03:00
{
let mut zip = zip::ZipWriter::new(&mut writer);
2024-02-13 07:51:02 +03:00
for sub_entry in walkdir::WalkDir::new(parent_pkg_path) {
let sub_entry = sub_entry?;
2024-02-05 23:39:44 +03:00
let path = sub_entry.path();
2024-02-13 07:51:02 +03:00
let name = path.strip_prefix(Path::new(parent_pkg_path))?;
2024-02-05 23:39:44 +03:00
if path.is_file() {
2024-02-13 07:51:02 +03:00
zip.start_file(name.to_string_lossy(), options)?;
let mut file = File::open(path)?;
2024-02-05 23:39:44 +03:00
let mut buffer = Vec::new();
2024-02-13 07:51:02 +03:00
file.read_to_end(&mut buffer)?;
zip.write_all(&buffer)?;
2024-02-05 23:39:44 +03:00
} else if !name.as_os_str().is_empty() {
2024-02-13 07:51:02 +03:00
zip.add_directory(name.to_string_lossy(), options)?;
2024-02-05 23:39:44 +03:00
}
}
2024-02-13 07:51:02 +03:00
zip.finish()?;
2024-02-05 23:39:44 +03:00
}
2024-02-13 07:51:02 +03:00
2024-02-05 23:39:44 +03:00
let zip_contents = writer.into_inner();
2024-02-13 07:51:02 +03:00
let zip_filename = format!("{}.zip", entry_path.file_name().unwrap().to_str().unwrap());
Ok((entry_path.display().to_string(), zip_filename, zip_contents))
2024-02-13 07:51:02 +03:00
})
}
2024-02-13 07:51:02 +03:00
fn main() -> anyhow::Result<()> {
if std::env::var("SKIP_BUILD_SCRIPT").is_ok() {
println!("Skipping build script");
return Ok(());
}
2024-02-13 07:51:02 +03:00
let pwd = std::env::current_dir()?;
let parent_dir = pwd.parent().unwrap();
let packages_dir = pwd.join("packages");
let entries: Vec<_> = fs::read_dir(packages_dir)?
.map(|entry| entry.unwrap().path())
.collect();
let results: Vec<anyhow::Result<(String, String, Vec<u8>)>> = entries
2024-02-13 07:51:53 +03:00
.par_iter()
2024-02-13 07:51:02 +03:00
.map(|entry_path| {
let parent_pkg_path = entry_path.join("pkg");
build_and_zip_package(entry_path.clone(), parent_pkg_path.to_str().unwrap())
})
.collect();
// Process results, e.g., write to `bootstrapped_processes.rs`
// This part remains sequential
let mut bootstrapped_processes = vec![];
2024-02-13 07:51:53 +03:00
writeln!(
bootstrapped_processes,
"pub static BOOTSTRAPPED_PROCESSES: &[(&str, &[u8], &[u8])] = &["
2024-02-13 07:51:53 +03:00
)?;
2024-02-13 07:51:02 +03:00
for result in results {
match result {
Ok((entry_path, zip_filename, zip_contents)) => {
2024-02-13 07:51:02 +03:00
// Further processing, like saving ZIP files and updating bootstrapped_processes
let metadata_path = format!("{}/metadata.json", entry_path);
2024-02-13 07:51:02 +03:00
let zip_path = format!("{}/target/{}", parent_dir.display(), zip_filename);
fs::write(&zip_path, &zip_contents)?;
writeln!(
bootstrapped_processes,
" (\"{}\", include_bytes!(\"{}\"), include_bytes!(\"{}\")),",
zip_filename, metadata_path, zip_path,
2024-02-13 07:51:02 +03:00
)?;
2024-02-13 07:51:53 +03:00
}
2024-02-13 07:51:02 +03:00
Err(e) => return Err(e),
2024-02-05 23:39:44 +03:00
}
}
2024-02-13 07:51:02 +03:00
writeln!(bootstrapped_processes, "];")?;
let bootstrapped_processes_path = pwd.join("src/bootstrapped_processes.rs");
2024-02-05 23:39:44 +03:00
fs::write(&bootstrapped_processes_path, bootstrapped_processes)?;
2024-02-13 07:51:02 +03:00
Ok(())
}