Compare commits
5 commits
90cd9826b7
...
8d97b887cc
Author | SHA1 | Date | |
---|---|---|---|
8d97b887cc | |||
4bcff4fee1 | |||
d329b145a0 | |||
9445dafdf8 | |||
3dec6b952a |
5 changed files with 210 additions and 27 deletions
6
.gitignore
vendored
6
.gitignore
vendored
|
@ -14,3 +14,9 @@ Cargo.lock
|
|||
# MSVC Windows builds of rustc generate these, which store debugging information
|
||||
*.pdb
|
||||
|
||||
|
||||
#Tokefile
|
||||
tokefile.toml
|
||||
tokefile
|
||||
Tokefile
|
||||
Tokefile.toml
|
||||
|
|
2
Cargo.lock
generated
2
Cargo.lock
generated
|
@ -215,7 +215,7 @@ dependencies = [
|
|||
|
||||
[[package]]
|
||||
name = "toke-runner"
|
||||
version = "0.1.0"
|
||||
version = "0.1.2"
|
||||
dependencies = [
|
||||
"clap",
|
||||
"regex",
|
||||
|
|
|
@ -1,11 +1,11 @@
|
|||
[package]
|
||||
name = "toke-runner"
|
||||
description = "A command runner for toml files"
|
||||
repository = "https://git.olympuslab.net/afonso/toke"
|
||||
repository = "https://github.com/afonsofrancof/toke"
|
||||
readme = "README.md"
|
||||
keywords = ["cli", "toml", "command", "runner", "toke"]
|
||||
categories = ["command-line-utilities", "development-tools"]
|
||||
version = "0.1.0"
|
||||
version = "0.1.2"
|
||||
edition = "2021"
|
||||
license = "MIT"
|
||||
|
||||
|
|
11
examples/wildcards.toml
Normal file
11
examples/wildcards.toml
Normal file
|
@ -0,0 +1,11 @@
|
|||
[vars]
|
||||
CC="!which gcc"
|
||||
CFLAGS="-Wall -O2 -I/opt/homebrew/opt/openssl@3.2/include -L/opt/homebrew/opt/openssl@3.2/lib"
|
||||
LDFLAGS="-lssl -lcrypto"
|
||||
|
||||
[targets.build]
|
||||
# This makes use of the wildcards feature to generate a list of source files and output files
|
||||
# and compiles each source file to an output file
|
||||
wildcards=["!ls -1 src/*.c","!ls -1 src/*.c | sed -e 's/\\.c$//' | sed -e 's/^src/bin/'"]
|
||||
cmd="${CC} ${CFLAGS} ${LDFLAGS} @@ -o @@"
|
||||
|
214
src/bin/toke.rs
214
src/bin/toke.rs
|
@ -114,8 +114,32 @@ fn main() {
|
|||
exit(1);
|
||||
}
|
||||
|
||||
// Execute the target command
|
||||
run_command(&parsed_toml, target.to_string());
|
||||
//Run deps
|
||||
run_deps(&parsed_toml, target.to_string());
|
||||
|
||||
if parsed_toml
|
||||
.get("targets")
|
||||
.and_then(|targets| targets.get(target))
|
||||
.and_then(|target| target.as_table())
|
||||
.and_then(|target| target.get("wildcards"))
|
||||
.is_some()
|
||||
{
|
||||
let command_list = wildcard_iterations(&parsed_toml, target.to_string());
|
||||
for cmd in command_list {
|
||||
run_command(&cmd);
|
||||
}
|
||||
exit(0);
|
||||
} else {
|
||||
run_command(
|
||||
parsed_toml
|
||||
.get("targets")
|
||||
.and_then(|targets| targets.get(target))
|
||||
.and_then(|target| target.as_table())
|
||||
.and_then(|target| target.get("cmd"))
|
||||
.and_then(|cmd| cmd.as_str())
|
||||
.unwrap_or(""),
|
||||
);
|
||||
}
|
||||
}
|
||||
fn detect_cycle(parsed_toml: &toml::Value) {
|
||||
let mut visited_targets = HashSet::new();
|
||||
|
@ -182,7 +206,7 @@ fn replace_variables(parsed_toml: &mut toml::Value, cli_vars: HashMap<String, St
|
|||
};
|
||||
|
||||
// Iterate over each target
|
||||
for (_, target_value) in targets.iter_mut() {
|
||||
for (target_name, target_value) in targets.iter_mut() {
|
||||
if let Some(target_table) = target_value.as_table_mut() {
|
||||
// Parse local variables for the target
|
||||
let map = toml::value::Table::new();
|
||||
|
@ -199,6 +223,51 @@ fn replace_variables(parsed_toml: &mut toml::Value, cli_vars: HashMap<String, St
|
|||
merged_vars.insert(key.clone(), toml::Value::String(value.clone()));
|
||||
}
|
||||
|
||||
//Replace variables in the target's wildcards value
|
||||
if let Some(wildcards_value) = target_table.get_mut("wildcards") {
|
||||
if let Some(wildcards_array) = wildcards_value.as_array_mut() {
|
||||
for wildcard in wildcards_array {
|
||||
//Check if the value is a string
|
||||
//If it is, then it must be a command, so we need to check if it starts with the designated prefix
|
||||
if let Some(wildcards_str) = wildcard.as_str() {
|
||||
if let Some(stripped_command) = wildcards_str.strip_prefix('!') {
|
||||
let replaced_command =
|
||||
replace_variables_in_cmd(stripped_command, &merged_vars);
|
||||
// Execute shell command and capture its output
|
||||
let output = Command::new("sh")
|
||||
.arg("-c")
|
||||
.arg(replaced_command)
|
||||
.output()
|
||||
.expect("Failed to execute shell command");
|
||||
// Use the command output as the variable value
|
||||
*wildcard = toml::Value::Array(
|
||||
String::from_utf8_lossy(&output.stdout)
|
||||
.trim()
|
||||
.split('\n')
|
||||
.map(|w| toml::Value::String(w.to_string()))
|
||||
.collect::<Vec<toml::Value>>(),
|
||||
);
|
||||
} else {
|
||||
eprintln!("Invalid wildcard format on target {} : {} , it must either be a command (string starting with !) or an array", target_name,wildcards_str);
|
||||
}
|
||||
} else if let Some(wildcards_array) = wildcard.as_array_mut() {
|
||||
for wildcard in wildcards_array {
|
||||
if let Some(wildcard_str) = wildcard.as_str() {
|
||||
if wildcard_str.starts_with('!') {
|
||||
eprintln!("Invalid wildcard format on target {} : {} , it must either be a command (string starting with !) or an array", target_name,wildcard_str);
|
||||
} else {
|
||||
*wildcard = toml::Value::String(replace_variables_in_cmd(
|
||||
wildcard_str,
|
||||
&merged_vars,
|
||||
));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Replace variables in the target's cmd value
|
||||
if let Some(cmd_value) = target_table.get_mut("cmd") {
|
||||
if let Some(cmd_str) = cmd_value.as_str() {
|
||||
|
@ -230,15 +299,111 @@ fn replace_variables_in_cmd(cmd: &str, vars: &toml::value::Table) -> String {
|
|||
for capture in re.captures_iter(cmd) {
|
||||
if let Some(var_name) = capture.get(1).map(|m| m.as_str()) {
|
||||
if let Some(var_value) = vars.get(var_name).and_then(|v| v.as_str()) {
|
||||
replaced_cmd = replaced_cmd.replace(&format!("${{{}}}", var_name), var_value);
|
||||
// Check if the value starts with the designated prefix
|
||||
if let Some(stripped_command) = var_value.strip_prefix('!') {
|
||||
// Execute shell command and capture its output
|
||||
let output = Command::new("sh")
|
||||
.arg("-c")
|
||||
.arg(stripped_command)
|
||||
.output()
|
||||
.expect("Failed to execute shell command");
|
||||
// Use the command output as the variable value
|
||||
replaced_cmd = replaced_cmd.replace(
|
||||
&format!("${{{}}}", var_name),
|
||||
String::from_utf8_lossy(&output.stdout).trim(),
|
||||
);
|
||||
} else {
|
||||
// Replace variable with its value
|
||||
replaced_cmd = replaced_cmd.replace(&format!("${{{}}}", var_name), var_value);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
replaced_cmd
|
||||
}
|
||||
fn wildcard_iterations(parsed_toml: &toml::Value, target: String) -> Vec<String> {
|
||||
// We want to loop over the target's wildcards and crate a list of all the combinations
|
||||
let empty_table = toml::value::Table::new();
|
||||
let targets = parsed_toml
|
||||
.get("targets")
|
||||
.and_then(|targets| targets.as_table())
|
||||
.unwrap_or(&empty_table);
|
||||
//Get the number of wildcards in the cmd value
|
||||
//A wildcard in cmd is represented by @@, so we want to count the number of times it appears
|
||||
let re = Regex::new(r#"@{2}"#).unwrap();
|
||||
let cmd = targets
|
||||
.get(&target)
|
||||
.and_then(|target| target.as_table())
|
||||
.and_then(|target| target.get("cmd"))
|
||||
.and_then(|cmd| cmd.as_str())
|
||||
.unwrap_or("");
|
||||
|
||||
fn run_command(parsed_toml: &toml::Value, target: String) {
|
||||
let wildcard_count = re.find_iter(cmd).count();
|
||||
//Check if the wildcard count is different that the number of wildcards in the wildcards value
|
||||
//If it is, then we have an invalid tokefile
|
||||
if let Some(target_table) = targets.get(&target) {
|
||||
if let Some(wildcards_value) = target_table.get("wildcards") {
|
||||
if let Some(wildcards_array) = wildcards_value.as_array() {
|
||||
if wildcard_count != wildcards_array.len() {
|
||||
eprintln!("Invalid tokefile, the number of wildcards in the cmd value must be the same as the number of wildcards in the wildcards value");
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
//Check if each wildcard has the same number of elements/iterations
|
||||
let num_of_iters = match targets.get(&target) {
|
||||
Some(target_table) => match target_table.get("wildcards") {
|
||||
Some(wildcards_value) => match wildcards_value.as_array() {
|
||||
Some(wildcards_array) => {
|
||||
let first_length = wildcards_array[0].as_array().unwrap().len();
|
||||
let all_same_length = wildcards_array
|
||||
.iter()
|
||||
.all(|wildcard| wildcard.as_array().unwrap().len() == first_length);
|
||||
if !all_same_length {
|
||||
eprintln!("Invalid tokefile, all wildcards must have the same number of elements/iterations");
|
||||
exit(1);
|
||||
} else {
|
||||
first_length
|
||||
}
|
||||
}
|
||||
_ => 0,
|
||||
},
|
||||
_ => 0,
|
||||
},
|
||||
_ => 0,
|
||||
};
|
||||
|
||||
//Create a list of cmds where each index is a combination of the wildcards at that index
|
||||
let mut command_list = Vec::new();
|
||||
for i in 0..num_of_iters {
|
||||
let mut cmd = cmd.to_string();
|
||||
let wildcards = match targets
|
||||
.get(&target)
|
||||
.and_then(|target| target.as_table())
|
||||
.and_then(|target| target.get("wildcards"))
|
||||
.and_then(|wildcards| wildcards.as_array())
|
||||
{
|
||||
Some(wildcards) => wildcards,
|
||||
_ => {
|
||||
//Return an empty vector if the wildcards value is not an array
|
||||
return Vec::new();
|
||||
}
|
||||
};
|
||||
for wildcard in wildcards {
|
||||
if let Some(wildcard_array) = wildcard.as_array() {
|
||||
//Apple regex to replace the first instance of @@ with the value of the wildcard at index i
|
||||
cmd = cmd.replacen("@@", wildcard_array[i].as_str().unwrap(), 1);
|
||||
}
|
||||
}
|
||||
command_list.push(cmd);
|
||||
}
|
||||
command_list
|
||||
}
|
||||
|
||||
fn run_deps(parsed_toml: &toml::Value, target: String) {
|
||||
if let Some(targets) = parsed_toml
|
||||
.get("targets")
|
||||
.and_then(|targets| targets.as_table())
|
||||
|
@ -248,29 +413,30 @@ fn run_command(parsed_toml: &toml::Value, target: String) {
|
|||
if let Some(dep_array) = dep_value.as_array() {
|
||||
for dep in dep_array {
|
||||
if let Some(dep_str) = dep.as_str() {
|
||||
run_command(parsed_toml, dep_str.to_string());
|
||||
run_deps(parsed_toml, dep_str.to_string());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if let Some(cmd_value) = target_table.get("cmd") {
|
||||
if let Some(cmd_str) = cmd_value.as_str() {
|
||||
eprintln!("{}", cmd_str);
|
||||
let status = Command::new("sh")
|
||||
.arg("-c")
|
||||
.arg(cmd_str)
|
||||
.status()
|
||||
.expect("Failed to execute command");
|
||||
if !status.success() {
|
||||
eprintln!(
|
||||
"Command '{}' failed with exit code {:?}",
|
||||
cmd_str,
|
||||
status.code()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn run_command(cmd: &str) -> String {
|
||||
eprintln!("{}", cmd);
|
||||
let mut binding = Command::new("sh");
|
||||
let command = binding.arg("-c").arg(cmd);
|
||||
|
||||
let status = command.status().expect("Failed to execute command");
|
||||
if !status.success() {
|
||||
eprintln!(
|
||||
"Command '{}' failed with exit code {:?}",
|
||||
cmd,
|
||||
status.code()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
let output = command.output().expect("Failed to execute command");
|
||||
String::from_utf8_lossy(&output.stdout).to_string()
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue