mirror of
https://gitlab.redox-os.org/redox-os/redox.git
synced 2026-06-17 23:44:17 +08:00
Add initial repo bin
This commit is contained in:
Wildan Mubarok
parent
dc04843050
commit
605bfcfbd4
242
src/bin/repo.rs
Normal file
242
src/bin/repo.rs
Normal file
@ -0,0 +1,242 @@
|
||||
use std::error::Error;
|
||||
use std::fmt::format;
|
||||
use std::path::{Path, PathBuf};
|
||||
use std::process;
|
||||
use std::{env, fs};
|
||||
|
||||
// A repo manager, to replace repo.sh
|
||||
|
||||
const REPO_HELP_STR: &str = r#"
|
||||
Usage: repo <command> [flags] <recipe1> <recipe2> ...
|
||||
|
||||
command list:
|
||||
fetch download recipe sources
|
||||
cook build recipe packages
|
||||
unfetch delete recipe sources
|
||||
clean delete recipe artifacts
|
||||
push extract package into sysroot
|
||||
|
||||
common flags:
|
||||
--cookbook=<cookbook_dir> the "recipes" folder, default to $PWD/recipes
|
||||
--repo=<repo_dir> the "repo" folder, default to $PWD/repo
|
||||
--sysroot=<sysroot_dir> the "root" folder used for "push" command
|
||||
For Redox, defaults to "/", else default to $PWD/sysroot
|
||||
|
||||
cook flags:
|
||||
--with-package-deps include package deps
|
||||
--offline prefer to not use network
|
||||
--nonstop keep running even a recipe build failed
|
||||
--all apply to all recipes in <cookbook_dir>
|
||||
-q, --quiet surpress build logs unless error
|
||||
"#;
|
||||
|
||||
struct Config {
|
||||
cookbook_dir: PathBuf,
|
||||
repo_dir: PathBuf,
|
||||
sysroot_dir: PathBuf,
|
||||
with_package_deps: bool,
|
||||
offline: bool,
|
||||
nonstop: bool,
|
||||
all: bool,
|
||||
quiet: bool,
|
||||
}
|
||||
|
||||
impl Config {
|
||||
fn new() -> Result<Self, std::io::Error> {
|
||||
let current_dir = env::current_dir()?;
|
||||
Ok(Config {
|
||||
cookbook_dir: current_dir.join("recipes"),
|
||||
repo_dir: current_dir.join("repo"),
|
||||
sysroot_dir: if cfg!(target_os = "redox") {
|
||||
PathBuf::from("/")
|
||||
} else {
|
||||
current_dir.join("sysroot")
|
||||
},
|
||||
with_package_deps: false,
|
||||
offline: false,
|
||||
nonstop: false,
|
||||
all: false,
|
||||
quiet: false,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
fn main() -> Result<(), Box<dyn Error>> {
|
||||
let args: Vec<String> = env::args().skip(1).collect();
|
||||
|
||||
if args.is_empty() || args.contains(&"--help".to_string()) || args.contains(&"-h".to_string()) {
|
||||
println!("{}", REPO_HELP_STR);
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
let mut config = Config::new()?;
|
||||
let mut command: Option<String> = None;
|
||||
let mut recipe_paths: Vec<Path> = Vec::new();
|
||||
|
||||
for arg in args {
|
||||
if arg.starts_with("--") {
|
||||
if let Some((key, value)) = arg.split_once('=') {
|
||||
match key {
|
||||
"--cookbook" => config.cookbook_dir = PathBuf::from(value),
|
||||
"--repo" => config.repo_dir = PathBuf::from(value),
|
||||
"--sysroot" => config.sysroot_dir = Some(PathBuf::from(value)),
|
||||
_ => {
|
||||
eprintln!("Error: Unknown flag with value: {}", arg);
|
||||
process::exit(1);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
match arg.as_str() {
|
||||
"--with-package-deps" => config.with_package_deps = true,
|
||||
"--offline" => config.offline = true,
|
||||
"--nonstop" => config.nonstop = true,
|
||||
"--all" => config.all = true,
|
||||
"--quiet" => config.quiet = true,
|
||||
_ => {
|
||||
eprintln!("Error: Unknown flag: {}", arg);
|
||||
process::exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
} else if arg.starts_with('-') {
|
||||
match arg.as_str() {
|
||||
"-q" => config.quiet = true,
|
||||
_ => {
|
||||
eprintln!("Error: Unknown flag: {}", arg);
|
||||
process::exit(1);
|
||||
}
|
||||
}
|
||||
} else if command.is_none() {
|
||||
// The first non-flag argument is the command
|
||||
command = Some(arg);
|
||||
} else {
|
||||
// Subsequent non-flag arguments are recipe names
|
||||
if Some(path) = pkg::recipes::find(arg) {
|
||||
recipe_paths.push(path);
|
||||
} else {
|
||||
return Err(format!("Error: recipe not found '{arg}'"));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let command = command.ok_or("Error: No command specified.")?;
|
||||
|
||||
if !config.all && recipe_paths.is_empty() {
|
||||
return Err("Error: No recipe names provided and --all flag was not used.".into());
|
||||
}
|
||||
if config.all && !recipe_paths.is_empty() {
|
||||
return Err("Error: Cannot specify recipe names when using the --all flag.".into());
|
||||
}
|
||||
|
||||
if config.all {
|
||||
recipe_paths = pkg::recipes::list("");
|
||||
}
|
||||
|
||||
for recipe_path in &recipe_paths {
|
||||
match command.as_str() {
|
||||
"fetch" => handle_fetch(recipe_path, &config)?,
|
||||
"cook" => handle_cook(recipe_path, &config)?,
|
||||
"unfetch" => handle_unfetch(recipe_path, &config)?,
|
||||
"clean" => handle_clean(recipe_path, &config)?,
|
||||
"push" => handle_push(recipe_path, &config)?,
|
||||
_ => {
|
||||
eprintln!("Error: Unknown command '{}'\n", command);
|
||||
println!("{}", REPO_HELP_STR);
|
||||
process::exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
println!(
|
||||
"\nCommand '{}' completed for all specified recipes.",
|
||||
command
|
||||
);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_fetch(recipe_path: &Path, config: &Config) -> Result<(), String> {
|
||||
let mut cmd = Command::new("cook");
|
||||
cmd.arg("--fetch-only");
|
||||
if config.with_package_deps {
|
||||
cmd.arg("--with-package-deps");
|
||||
}
|
||||
if config.offline {
|
||||
cmd.arg("--offline");
|
||||
}
|
||||
if config.quiet {
|
||||
cmd.arg("--quiet");
|
||||
}
|
||||
cmd.arg(recipe_path);
|
||||
let status = cmd
|
||||
.status()
|
||||
.map_err(|e| format!("Failed to execute cook command: {}", e))?;
|
||||
if !status.success() && !config.nonstop {
|
||||
return Err(format!(
|
||||
"Cook command failed for recipe '{}' with exit code: {}",
|
||||
recipe_name,
|
||||
status.code().unwrap_or(1)
|
||||
));
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_cook(recipe_path: &Path, config: &Config) -> Result<(), String> {
|
||||
let mut cmd = Command::new("cook");
|
||||
cmd.arg(recipe_path);
|
||||
if config.with_package_deps {
|
||||
cmd.arg("--with-package-deps");
|
||||
}
|
||||
if config.offline {
|
||||
cmd.arg("--offline");
|
||||
}
|
||||
if config.quiet {
|
||||
cmd.arg("--quiet");
|
||||
}
|
||||
let status = cmd
|
||||
.status()
|
||||
.map_err(|e| format!("Failed to execute cook command: {}", e))?;
|
||||
if !status.success() && !config.nonstop {
|
||||
return Err(format!(
|
||||
"Cook command failed for recipe '{}' with exit code: {}",
|
||||
recipe_name,
|
||||
status.code().unwrap_or(1)
|
||||
));
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_unfetch(recipe_path: &Path, config: &Config) -> Result<(), String> {
|
||||
let dir = recipe_path.join("source");
|
||||
if dir.exists() {
|
||||
fs::remove_dir_all(dir)
|
||||
.map_err(|err| format!("failed to delete '{}': {:?}", recipe_path, errF))?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_clean(recipe_path: &Path, config: &Config) -> Result<(), String> {
|
||||
let dir = recipe_path.join("target");
|
||||
if dir.exists() {
|
||||
fs::remove_dir_all(dir)
|
||||
.map_err(|err| format!("failed to delete '{}': {:?}", recipe_path, errF))?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_push(recipe_path: &Path, config: &Config) -> Result<(), String> {
|
||||
let public_path = "build/id_ed25519.pub.toml";
|
||||
pkgar::extract(
|
||||
public_path,
|
||||
config.sysroot_dir.as_path(),
|
||||
sysroot_dir_tmp.to_str().unwrap(),
|
||||
)
|
||||
.map_err(|err| {
|
||||
format!(
|
||||
"failed to install '{}' in '{}': {:?}",
|
||||
archive_path.display(),
|
||||
config.sysroot_dir.display(),
|
||||
err
|
||||
)
|
||||
})?;
|
||||
Ok(())
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user