From 486d42d000b26a2d8413f31730ce06c4853379e5 Mon Sep 17 00:00:00 2001 From: Alexander Tarasov Date: Tue, 18 Mar 2025 17:06:53 +0300 Subject: [PATCH] main.rs --- Cargo.lock | 237 ++++++++++++++++++++++++++++++++++++++++++++++++++++ Cargo.toml | 1 + src/main.rs | 168 ++++++++++++++++++++++++++++++++++++- 3 files changed, 404 insertions(+), 2 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index 8543bed..e31a297 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2,6 +2,243 @@ # It is not intended for manual editing. version = 4 +[[package]] +name = "anstream" +version = "0.6.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8acc5369981196006228e28809f761875c0327210a891e941f4c683b3a99529b" +dependencies = [ + "anstyle", + "anstyle-parse", + "anstyle-query", + "anstyle-wincon", + "colorchoice", + "is_terminal_polyfill", + "utf8parse", +] + +[[package]] +name = "anstyle" +version = "1.0.10" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "55cc3b69f167a1ef2e161439aa98aed94e6028e5f9a59be9a6ffb47aef1651f9" + +[[package]] +name = "anstyle-parse" +version = "0.2.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3b2d16507662817a6a20a9ea92df6652ee4f94f914589377d69f3b21bc5798a9" +dependencies = [ + "utf8parse", +] + +[[package]] +name = "anstyle-query" +version = "1.1.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "79947af37f4177cfead1110013d678905c37501914fba0efea834c3fe9a8d60c" +dependencies = [ + "windows-sys", +] + +[[package]] +name = "anstyle-wincon" +version = "3.0.7" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ca3534e77181a9cc07539ad51f2141fe32f6c3ffd4df76db8ad92346b003ae4e" +dependencies = [ + "anstyle", + "once_cell", + "windows-sys", +] + [[package]] name = "build_msp" version = "0.1.0" +dependencies = [ + "clap", +] + +[[package]] +name = "clap" +version = "4.5.32" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "6088f3ae8c3608d19260cd7445411865a485688711b78b5be70d78cd96136f83" +dependencies = [ + "clap_builder", + "clap_derive", +] + +[[package]] +name = "clap_builder" +version = "4.5.32" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "22a7ef7f676155edfb82daa97f99441f3ebf4a58d5e32f295a56259f1b6facc8" +dependencies = [ + "anstream", + "anstyle", + "clap_lex", + "strsim", +] + +[[package]] +name = "clap_derive" +version = "4.5.32" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09176aae279615badda0765c0c0b3f6ed53f4709118af73cf4655d85d1530cd7" +dependencies = [ + "heck", + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "clap_lex" +version = "0.7.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "f46ad14479a25103f283c0f10005961cf086d8dc42205bb44c46ac563475dca6" + +[[package]] +name = "colorchoice" +version = "1.0.3" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5b63caa9aa9397e2d9480a9b13673856c78d8ac123288526c37d7839f2a86990" + +[[package]] +name = "heck" +version = "0.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea" + +[[package]] +name = "is_terminal_polyfill" +version = "1.70.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf" + +[[package]] +name = "once_cell" +version = "1.21.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "d75b0bedcc4fe52caa0e03d9f1151a323e4aa5e2d78ba3580400cd3c9e2bc4bc" + +[[package]] +name = "proc-macro2" +version = "1.0.94" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "a31971752e70b8b2686d7e46ec17fb38dad4051d94024c88df49b667caea9c84" +dependencies = [ + "unicode-ident", +] + +[[package]] +name = "quote" +version = "1.0.40" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1885c039570dc00dcb4ff087a89e185fd56bae234ddc7f056a945bf36467248d" +dependencies = [ + "proc-macro2", +] + +[[package]] +name = "strsim" +version = "0.11.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f" + +[[package]] +name = "syn" +version = "2.0.100" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b09a44accad81e1ba1cd74a32461ba89dee89095ba17b32f5d03683b1b1fc2a0" +dependencies = [ + "proc-macro2", + "quote", + "unicode-ident", +] + +[[package]] +name = "unicode-ident" +version = "1.0.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5a5f39404a5da50712a4c1eecf25e90dd62b613502b7e925fd4e4d19b5c96512" + +[[package]] +name = "utf8parse" +version = "0.2.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821" + +[[package]] +name = "windows-sys" +version = "0.59.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b" +dependencies = [ + "windows-targets", +] + +[[package]] +name = "windows-targets" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973" +dependencies = [ + "windows_aarch64_gnullvm", + "windows_aarch64_msvc", + "windows_i686_gnu", + "windows_i686_gnullvm", + "windows_i686_msvc", + "windows_x86_64_gnu", + "windows_x86_64_gnullvm", + "windows_x86_64_msvc", +] + +[[package]] +name = "windows_aarch64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3" + +[[package]] +name = "windows_aarch64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469" + +[[package]] +name = "windows_i686_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b" + +[[package]] +name = "windows_i686_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66" + +[[package]] +name = "windows_i686_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66" + +[[package]] +name = "windows_x86_64_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78" + +[[package]] +name = "windows_x86_64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d" + +[[package]] +name = "windows_x86_64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec" diff --git a/Cargo.toml b/Cargo.toml index 37bb584..e69dcd6 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -10,6 +10,7 @@ keywords = ["switch", "nintendo", "mod", "mods", "msp"] build = "build.rs" [dependencies] +"clap" = {version = "4.5.32", features = ["derive"]} [profile.release] strip = true diff --git a/src/main.rs b/src/main.rs index e7a11a9..d272aa5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,3 +1,167 @@ -fn main() { - println!("Hello, world!"); +use clap::{Error, Parser}; +use std::collections::HashSet; +use std::fs::{File, ReadDir}; +use std::fs::{copy, create_dir, read_dir, remove_dir_all, remove_file, set_permissions}; +use std::io::Write; +use std::path::absolute; +use std::path::{Path, PathBuf}; + +#[cfg(target_os = "windows")] +const BUILD_ROMFS_BIN: &[u8] = include_bytes!("../switch-tools/build_romfs.exe"); +#[cfg(target_os = "windows")] +const BUILD_PFS0_BIN: &[u8] = include_bytes!("../switch-tools/build_pfs0.exe"); + +#[cfg(not(target_os = "windows"))] +const BUILD_ROMFS_BIN: &[u8] = include_bytes!("../switch-tools/build_romfs"); +#[cfg(not(target_os = "windows"))] +const BUILD_PFS0_BIN: &[u8] = include_bytes!("../switch-tools/build_pfs0"); + +const FILES_TO_COPY: [&str; 28] = [ + "romfs.bin", + "exefs.nsp", + "rtld", + "main", + "main.npdm", + "compat0", + "compat1", + "compat2", + "compat3", + "compat4", + "compat5", + "compat6", + "compat7", + "compat8", + "compat9", + "subsdk0", + "subsdk1", + "subsdk2", + "subsdk3", + "subsdk4", + "subsdk5", + "subsdk6", + "subsdk7", + "subsdk8", + "subsdk9", + "sdk", + "config.ini", + "icon.jpg", +]; + +#[derive(Parser, Debug)] +#[command(version, about, long_about = None)] +struct Args { + #[arg(short = 'i', long = "input", default_value = ".")] + input: PathBuf, + + #[arg(short = 'o', long = "output", default_value = "mod.msp")] + output: PathBuf, + + #[arg(short = 'm', long = "manifest", default_value = "./manifest")] + manifest: PathBuf, +} + +fn list_items(path: &Path) -> Result, Error> { + let mut items: Vec = Vec::new(); + let entries: ReadDir = read_dir(path)?; + + for entry in entries { + let path: PathBuf = entry?.path(); + if path.is_dir() { + items.extend(list_items(&path)?); + items.push(path); + } else if path.is_file() { + items.push(path); + } + } + Ok(items) +} + +fn main() -> Result<(), Error> { + let args = Args::parse(); + + let input_path: PathBuf = absolute(args.input)?; + let output_path: PathBuf = absolute(args.output)?; + let manifest_path: PathBuf = absolute(args.manifest)?; + + assert!(input_path.exists(), "Input path does not exist"); + assert!(!output_path.exists(), "Output path already exists"); + assert!(manifest_path.exists(), "Manifest path does not exist"); + + let current_dir: PathBuf = absolute(Path::new("."))?; + let temp_dir: PathBuf = current_dir.join("temp"); + + if temp_dir.exists() { + remove_dir_all(&temp_dir)?; + } + create_dir(&temp_dir)?; + + copy(&manifest_path, temp_dir.join("manifest"))?; + + #[cfg(target_os = "windows")] + let build_romfs_path = temp_dir.join("build_romfs.exe"); + #[cfg(target_os = "windows")] + let build_pfs0_path = temp_dir.join("build_pfs0.exe"); + + #[cfg(not(target_os = "windows"))] + let build_romfs_path: PathBuf = temp_dir.join("build_romfs"); + #[cfg(not(target_os = "windows"))] + let build_pfs0_path: PathBuf = temp_dir.join("build_pfs0"); + + let mut build_romfs_file: File = File::create(&build_romfs_path)?; + build_romfs_file.write_all(BUILD_ROMFS_BIN)?; + #[cfg(not(target_os = "windows"))] + set_permissions( + &build_romfs_path, + std::os::unix::fs::PermissionsExt::from_mode(0o755), + )?; + + let mut build_pfs0_file: File = File::create(&build_pfs0_path)?; + build_pfs0_file.write_all(BUILD_PFS0_BIN)?; + #[cfg(not(target_os = "windows"))] + set_permissions( + &build_pfs0_path, + std::os::unix::fs::PermissionsExt::from_mode(0o755), + )?; + + let all_items: Vec = list_items(&input_path)?; + let files_to_copy: HashSet<&&str> = FILES_TO_COPY.iter().collect(); + for item in all_items { + let name: &str = item.file_name().unwrap().to_str().unwrap(); + + if item.is_dir() && name == "romfs" { + println!("Found romfs directory. Building romfs.bin..."); + std::process::Command::new(&build_romfs_path) + .current_dir(&temp_dir) + .arg(&item) + .arg(&temp_dir.join("romfs.bin")) + .status()?; + continue; + } + + if files_to_copy.contains(&&name) { + println!("Found {}, copying...", name); + copy(&item, &temp_dir.join(name))?; + continue; + } + + if name.ends_with(".ips") { + println!("Found {}, copying...", name); + copy(&item, &temp_dir.join(name))?; + continue; + } + } + + remove_file(&build_romfs_path)?; + remove_file(&build_pfs0_path)?; + + println!("Building {}...", output_path.display()); + std::process::Command::new(&build_pfs0_path) + .current_dir(&temp_dir) + .arg(&temp_dir) + .arg(&output_path) + .status()?; + + remove_dir_all(&temp_dir)?; + + Ok(()) }