39 lines
1.1 KiB
Rust
39 lines
1.1 KiB
Rust
|
use anyhow::{ensure, Result};
|
||
|
|
use std::fs::File;
|
||
|
|
use std::io::{self, Write};
|
||
|
|
use std::path::{Path, PathBuf};
|
||
|
|
use std::process::Command;
|
||
|
|
use zip::ZipWriter;
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
let out_dir = PathBuf::from(std::env::var_os("OUT_DIR").unwrap());
|
||
|
|
let mut rs_file = File::create(out_dir.join("zip.rs")).unwrap();
|
||
|
|
match attempt(&out_dir) {
|
||
|
|
Ok(()) => writeln!(rs_file, r#"Some(include_bytes!("source.zip"))"#).unwrap(),
|
||
|
|
Err(_) => writeln!(rs_file, "None").unwrap(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn attempt(out_dir: &Path) -> Result<()> {
|
||
|
|
let output = Command::new("cargo")
|
||
|
|
.args(["package", "--list", "--allow-dirty"])
|
||
|
|
.output()?;
|
||
|
|
ensure!(output.status.success(), "unsuccessful");
|
||
|
|
|
||
|
|
let mut writer = ZipWriter::new(File::create(out_dir.join("source.zip"))?);
|
||
|
|
for path in String::from_utf8(output.stdout)?.lines() {
|
||
|
|
if path == "Cargo.toml.orig" {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
writer.start_file(path, Default::default())?;
|
||
|
|
io::copy(
|
||
|
|
&mut File::open(Path::new(env!("CARGO_MANIFEST_DIR")).join(path))?,
|
||
|
|
&mut writer,
|
||
|
|
)?;
|
||
|
|
}
|
||
|
|
|
||
|
|
writer.finish()?;
|
||
|
|
Ok(())
|
||
|
|
}
|