mirror of https://github.com/Tim-Paik/neutauri.git
97 lines
3.2 KiB
Rust
97 lines
3.2 KiB
Rust
use anyhow::Context;
|
|
use neutauri_data as data;
|
|
#[cfg(windows)]
|
|
use std::{
|
|
env,
|
|
hash::{Hash, Hasher},
|
|
};
|
|
use std::{fs, io::Write};
|
|
|
|
fn options() -> fs::OpenOptions {
|
|
#[cfg(not(windows))]
|
|
use std::os::unix::prelude::OpenOptionsExt;
|
|
let mut options = fs::OpenOptions::new();
|
|
options.write(true);
|
|
options.create(true);
|
|
options.truncate(true);
|
|
#[cfg(not(windows))]
|
|
options.mode(0o755);
|
|
options
|
|
}
|
|
|
|
#[cfg(not(windows))]
|
|
fn get_runtime_data(
|
|
_icon_path: Option<std::path::PathBuf>,
|
|
_manifest_path: Option<std::path::PathBuf>,
|
|
) -> anyhow::Result<Vec<u8>> {
|
|
Ok(include_bytes!("../../target/release/neutauri_runtime").to_vec())
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
fn get_runtime_data(
|
|
icon_path: Option<std::path::PathBuf>,
|
|
manifest_path: Option<std::path::PathBuf>,
|
|
) -> anyhow::Result<Vec<u8>> {
|
|
let mut hasher = std::collections::hash_map::DefaultHasher::new();
|
|
hasher.write(b"neutauri_runtime");
|
|
std::time::SystemTime::now().hash(&mut hasher);
|
|
let temp_path = env::temp_dir().join(format!("{:x}.exe", hasher.finish()));
|
|
fs::write(
|
|
&temp_path,
|
|
include_bytes!("../../target/release/neutauri_runtime.exe"),
|
|
)?;
|
|
let mut updater = rcedit::ResourceUpdater::new();
|
|
updater.load(&temp_path)?;
|
|
if let Some(icon_path) = icon_path {
|
|
println!("{:?}", fs::canonicalize(&icon_path)?);
|
|
updater.set_icon(&fs::canonicalize(icon_path)?)?;
|
|
}
|
|
if let Some(manifest_path) = manifest_path {
|
|
updater.set_application_manifest(&fs::canonicalize(manifest_path)?)?;
|
|
}
|
|
updater.commit()?;
|
|
drop(updater);
|
|
let runtime_data =
|
|
fs::read(&temp_path).with_context(|| format!("Failed to read {}", temp_path.display()))?;
|
|
fs::remove_file(&temp_path)?;
|
|
Ok(runtime_data)
|
|
}
|
|
|
|
pub(crate) fn bundle(config_path: String) -> anyhow::Result<()> {
|
|
let config_path = std::path::Path::new(&config_path)
|
|
.canonicalize()
|
|
.with_context(|| {
|
|
format!(
|
|
"Error reading config file from {}\n\n{}",
|
|
&config_path, "You may want to create a neutauri.toml via the init subcommand?"
|
|
)
|
|
})?;
|
|
let config: data::Config = toml::from_str(fs::read_to_string(&config_path)?.as_str())
|
|
.with_context(|| "toml parsing error")?;
|
|
let source = match config_path.parent() {
|
|
Some(path) => path.join(&config.source).canonicalize()?,
|
|
None => config.source.canonicalize()?,
|
|
};
|
|
let target = match config_path.parent() {
|
|
Some(path) => data::normalize_path(&path.join(&config.target)),
|
|
None => data::normalize_path(&config.target),
|
|
};
|
|
fs::create_dir_all(target.parent().unwrap_or_else(|| std::path::Path::new("/")))?;
|
|
let target = if target.extension() == None && cfg!(windows) {
|
|
target.with_extension("exe")
|
|
} else {
|
|
target
|
|
};
|
|
if target.extension() == Some(std::ffi::OsStr::new("neu")) {
|
|
data::pack(config_path)?;
|
|
return Ok(());
|
|
}
|
|
let data = data::Data::build_from_dir(source, config.window_attr()?, config.webview_attr()?)?;
|
|
let mut f = options().open(&target)?;
|
|
f.write_all(&get_runtime_data(config.icon, config.manifest)?)?;
|
|
f.write_all(&data)?;
|
|
f.sync_all()?;
|
|
f.flush()?;
|
|
Ok(())
|
|
}
|