78 lines
2.8 KiB
Rust
78 lines
2.8 KiB
Rust
|
pub mod env;
|
||
|
|
pub mod pkginfo;
|
||
|
|
pub mod repoinfo;
|
||
|
|
|
||
|
|
use std::fs;
|
||
|
|
use std::io::{self, BufRead};
|
||
|
|
use std::path::Path;
|
||
|
|
|
||
|
|
|
||
|
|
pub fn get_stage_hook(stage: &str) -> io::Result<String> {
|
||
|
|
let cfg_path = &crate::commands::get_etc_path().join("aeropkg.md");
|
||
|
|
extract_block(cfg_path, &format!("``` sh *** hook {} ***", &stage), "```")
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn get_repo_and_stage_hook(repo: &str, stage: &str) -> io::Result<String> {
|
||
|
|
let cfg_path = &crate::commands::get_etc_path().join("aeropkg.md");
|
||
|
|
extract_block(cfg_path, &format!("``` sh *** hook {} {} ***", &repo, &stage), "```")
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn get_trim_rules(file_path: &Path) -> io::Result<String> { extract_block(file_path, "``` cfg *** Trim rules ***", "```") }
|
||
|
|
|
||
|
|
pub fn get_build_deps(file_path: &Path) -> io::Result<String> { extract_block(file_path, "``` cfg *** build deps ***", "```") }
|
||
|
|
pub fn get_run_deps(file_path: &Path) -> io::Result<String> { extract_block(file_path, "``` cfg *** run deps ***", "```") }
|
||
|
|
pub fn get_build_script(file_path: &Path) -> io::Result<String> { extract_block(file_path, "``` sh *** build ***", "```") }
|
||
|
|
pub fn get_config_script(file_path: &Path) -> io::Result<String> { extract_block(file_path, "``` sh *** config ***", "```") }
|
||
|
|
pub fn get_patch_script(file_path: &Path) -> io::Result<String> { extract_block(file_path, "``` sh *** config ***", "```") }
|
||
|
|
pub fn get_custom_script(file_path: &Path, scriptname: &String) -> String { extract_block(file_path, &format!("``` sh *** {} ***", scriptname), "```").expect(&format!("Can't get custom script: {}", &scriptname)) }
|
||
|
|
|
||
|
|
fn extract_block(
|
||
|
|
file_path: &Path,
|
||
|
|
start_marker: &str,
|
||
|
|
end_marker: &str,
|
||
|
|
) -> io::Result<String> {
|
||
|
|
let lines = read_lines(file_path)?;
|
||
|
|
let mut block_started = false;
|
||
|
|
let mut result = Vec::new();
|
||
|
|
|
||
|
|
for line in lines {
|
||
|
|
if line.trim() == start_marker {
|
||
|
|
block_started = true;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
if block_started {
|
||
|
|
if line.trim() == end_marker {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
result.push(line.trim().to_string());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if result.is_empty() {
|
||
|
|
Err(io::Error::new(
|
||
|
|
io::ErrorKind::InvalidData,
|
||
|
|
format!("Block between '{}' and '{}' not found", start_marker, end_marker),
|
||
|
|
))
|
||
|
|
} else {
|
||
|
|
Ok(result.join("\n"))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn read_first_line<P: AsRef<Path>>(file_path: P) -> io::Result<String> {
|
||
|
|
let file = fs::File::open(file_path)?;
|
||
|
|
let reader = io::BufReader::new(file);
|
||
|
|
if let Some(line) = reader.lines().next() {
|
||
|
|
line
|
||
|
|
} else {
|
||
|
|
Err(io::Error::new(io::ErrorKind::InvalidData, "File is empty"))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
fn read_lines<P: AsRef<Path>>(file_path: P) -> io::Result<Vec<String>> {
|
||
|
|
let file = fs::File::open(file_path)?;
|
||
|
|
let reader = io::BufReader::new(file);
|
||
|
|
reader.lines().collect()
|
||
|
|
}
|