🔧 Add .idea/ directory to .gitignore to exclude IDE configuration files
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
.idea/
|
||||
# ---> Rust
|
||||
# Generated by Cargo
|
||||
# will have compiled files and executables
|
||||
|
||||
Generated
+256
@@ -0,0 +1,256 @@
|
||||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 4
|
||||
|
||||
[[package]]
|
||||
name = "anstream"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "824a212faf96e9acacdbd09febd34438f8f711fb84e09a8916013cd7815ca28d"
|
||||
dependencies = [
|
||||
"anstyle",
|
||||
"anstyle-parse",
|
||||
"anstyle-query",
|
||||
"anstyle-wincon",
|
||||
"colorchoice",
|
||||
"is_terminal_polyfill",
|
||||
"utf8parse",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anstyle"
|
||||
version = "1.0.14"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "940b3a0ca603d1eade50a4846a2afffd5ef57a9feac2c0e2ec2e14f9ead76000"
|
||||
|
||||
[[package]]
|
||||
name = "anstyle-parse"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "52ce7f38b242319f7cabaa6813055467063ecdc9d355bbb4ce0c68908cd8130e"
|
||||
dependencies = [
|
||||
"utf8parse",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anstyle-query"
|
||||
version = "1.1.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "40c48f72fd53cd289104fc64099abca73db4166ad86ea0b4341abe65af83dadc"
|
||||
dependencies = [
|
||||
"windows-sys",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anstyle-wincon"
|
||||
version = "3.0.11"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "291e6a250ff86cd4a820112fb8898808a366d8f9f58ce16d1f538353ad55747d"
|
||||
dependencies = [
|
||||
"anstyle",
|
||||
"once_cell_polyfill",
|
||||
"windows-sys",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anyhow"
|
||||
version = "1.0.102"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "7f202df86484c868dbad7eaa557ef785d5c66295e41b460ef922eca0723b842c"
|
||||
|
||||
[[package]]
|
||||
name = "clap"
|
||||
version = "4.6.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1ddb117e43bbf7dacf0a4190fef4d345b9bad68dfc649cb349e7d17d28428e51"
|
||||
dependencies = [
|
||||
"clap_builder",
|
||||
"clap_derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "clap_builder"
|
||||
version = "4.6.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "714a53001bf66416adb0e2ef5ac857140e7dc3a0c48fb28b2f10762fc4b5069f"
|
||||
dependencies = [
|
||||
"anstream",
|
||||
"anstyle",
|
||||
"clap_lex",
|
||||
"strsim",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "clap_derive"
|
||||
version = "4.6.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f2ce8604710f6733aa641a2b3731eaa1e8b3d9973d5e3565da11800813f997a9"
|
||||
dependencies = [
|
||||
"heck",
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "clap_lex"
|
||||
version = "1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c8d4a3bb8b1e0c1050499d1815f5ab16d04f0959b233085fb31653fbfc9d98f9"
|
||||
|
||||
[[package]]
|
||||
name = "colorchoice"
|
||||
version = "1.0.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1d07550c9036bf2ae0c684c4297d503f838287c83c53686d05370d0e139ae570"
|
||||
|
||||
[[package]]
|
||||
name = "environment"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"clap",
|
||||
"serde",
|
||||
"serde_json",
|
||||
]
|
||||
|
||||
[[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.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a6cb138bb79a146c1bd460005623e142ef0181e3d0219cb493e02f7d08a35695"
|
||||
|
||||
[[package]]
|
||||
name = "itoa"
|
||||
version = "1.0.18"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.8.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "88904434abc2901f197fe8cc55f0445e7ded921dba5911dad2e2b39b48e663c4"
|
||||
|
||||
[[package]]
|
||||
name = "once_cell_polyfill"
|
||||
version = "1.70.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "384b8ab6d37215f3c5301a95a4accb5d64aa607f1fcb26a11b5303878451b4fe"
|
||||
|
||||
[[package]]
|
||||
name = "proc-macro2"
|
||||
version = "1.0.106"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8fd00f0bb2e90d81d1044c2b32617f68fcb9fa3bb7640c23e9c748e53fb30934"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "quote"
|
||||
version = "1.0.45"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "41f2619966050689382d2b44f664f4bc593e129785a36d6ee376ddf37259b924"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "serde"
|
||||
version = "1.0.228"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9a8e94ea7f378bd32cbbd37198a4a91436180c5bb472411e48b5ec2e2124ae9e"
|
||||
dependencies = [
|
||||
"serde_core",
|
||||
"serde_derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "serde_core"
|
||||
version = "1.0.228"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "41d385c7d4ca58e59fc732af25c3983b67ac852c1a25000afe1175de458b67ad"
|
||||
dependencies = [
|
||||
"serde_derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "serde_derive"
|
||||
version = "1.0.228"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d540f220d3187173da220f885ab66608367b6574e925011a9353e4badda91d79"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "serde_json"
|
||||
version = "1.0.150"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
|
||||
dependencies = [
|
||||
"itoa",
|
||||
"memchr",
|
||||
"serde",
|
||||
"serde_core",
|
||||
"zmij",
|
||||
]
|
||||
|
||||
[[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.118"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1b9ae57f904213ebb649ce6895b8a66c66f0203b9319718f69a5612a065b1422"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "unicode-ident"
|
||||
version = "1.0.24"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e6e4313cd5fcd3dad5cafa179702e2b244f760991f45397d14d4ebf38247da75"
|
||||
|
||||
[[package]]
|
||||
name = "utf8parse"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821"
|
||||
|
||||
[[package]]
|
||||
name = "windows-link"
|
||||
version = "0.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f0805222e57f7521d6a62e36fa9163bc891acd422f971defe97d64e70d0a4fe5"
|
||||
|
||||
[[package]]
|
||||
name = "windows-sys"
|
||||
version = "0.61.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ae137229bcbd6cdf0f7b80a31df61766145077ddf49416a728b02cb3921ff3fc"
|
||||
dependencies = [
|
||||
"windows-link",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zmij"
|
||||
version = "1.0.21"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b8848ee67ecc8aedbaf3e4122217aff892639231befc6a1b58d29fff4c2cabaa"
|
||||
+10
@@ -0,0 +1,10 @@
|
||||
[package]
|
||||
name = "environment"
|
||||
version = "0.1.0"
|
||||
edition = "2024"
|
||||
|
||||
[dependencies]
|
||||
anyhow = "1.0"
|
||||
clap = { version = "4.6", features = ["derive"] }
|
||||
serde = { version = "1.0", features = ["derive"] }
|
||||
serde_json = "1.0"
|
||||
+189
@@ -0,0 +1,189 @@
|
||||
use anyhow::{Context, Result, anyhow};
|
||||
use clap::Parser;
|
||||
use serde_json::Value;
|
||||
use std::collections::HashMap;
|
||||
use std::env;
|
||||
use std::ffi::OsStr;
|
||||
use std::fs;
|
||||
use std::io::{self, Read};
|
||||
use std::process::{Command, Stdio, exit};
|
||||
|
||||
#[derive(Parser)]
|
||||
#[command(name = env!("CARGO_PKG_NAME"))]
|
||||
#[command(version = env!("CARGO_PKG_VERSION"))]
|
||||
#[command(about = env!("CARGO_PKG_DESCRIPTION"))]
|
||||
struct Args {
|
||||
/// Shell, default load SHELL or /bin/sh if SHELL is not set
|
||||
#[arg(short, long)]
|
||||
shell: Option<String>,
|
||||
|
||||
/// Uppercase
|
||||
#[arg(short = 'U', long)]
|
||||
uppercase: bool,
|
||||
|
||||
/// Debug
|
||||
#[arg(long)]
|
||||
debug: bool,
|
||||
|
||||
/// Mute
|
||||
#[arg(long)]
|
||||
mute: bool,
|
||||
|
||||
/// Input, default `-`(stdin)
|
||||
#[arg(long)]
|
||||
input: Option<String>,
|
||||
|
||||
/// Arguments
|
||||
args: Vec<String>,
|
||||
}
|
||||
|
||||
fn main() -> Result<()> {
|
||||
let args = Args::parse();
|
||||
if args.mute && args.debug {
|
||||
return Err(anyhow!("--mute and --debug cannot bot set to true"));
|
||||
}
|
||||
|
||||
let input_buf = read_input(&args.input.unwrap_or_else(|| "-".to_string()))?;
|
||||
|
||||
let input_json = serde_json::from_str::<Value>(&input_buf)
|
||||
.with_context(|| format!("Failed to parse JSON in {}", input_buf))?;
|
||||
|
||||
let environment = json_object_to_env(input_json)?;
|
||||
|
||||
let mut cmd = if args.args.is_empty() {
|
||||
let shell = args
|
||||
.shell
|
||||
.unwrap_or_else(|| env::var("SHELL").unwrap_or_else(|_| String::from("/bin/sh")));
|
||||
if args.debug {
|
||||
eprintln!("run with shell: {shell}")
|
||||
}
|
||||
let mut cmd = Command::new(shell);
|
||||
cmd.arg("-i");
|
||||
cmd
|
||||
} else {
|
||||
if args.debug {
|
||||
eprintln!(
|
||||
"run command:\n-----BEGIN COMMAND-----\n{}\n-----END COMMAND-----",
|
||||
args.args.join(" ")
|
||||
);
|
||||
}
|
||||
let c = &args.args[0];
|
||||
let mut cmd = Command::new(c);
|
||||
cmd.args(&args.args[1..]);
|
||||
cmd
|
||||
};
|
||||
|
||||
cmd.stdin(Stdio::inherit());
|
||||
cmd.stdout(Stdio::inherit());
|
||||
cmd.stderr(Stdio::inherit());
|
||||
for (key, value) in environment.iter() {
|
||||
if args.uppercase {
|
||||
cmd.env(key.to_uppercase(), value);
|
||||
} else {
|
||||
cmd.env(key, value);
|
||||
}
|
||||
}
|
||||
|
||||
if args.debug {
|
||||
let full_envs: Vec<(&OsStr, Option<&OsStr>)> = cmd.get_envs().collect();
|
||||
eprintln!("FULL ENVS:");
|
||||
for (key, value) in full_envs.iter() {
|
||||
eprintln!(
|
||||
" {}={}",
|
||||
key.display(),
|
||||
value.unwrap_or_default().display()
|
||||
);
|
||||
}
|
||||
}
|
||||
if args.debug {
|
||||
eprintln!("cmd: {:?}", cmd);
|
||||
}
|
||||
|
||||
match cmd.status() {
|
||||
Ok(exit_status) => {
|
||||
if args.debug {
|
||||
eprintln!("exit status: {}", exit_status);
|
||||
}
|
||||
let exit_code = exit_status.code().unwrap_or(-1);
|
||||
if !args.mute {
|
||||
print!("ended with exit code: {exit_code}");
|
||||
}
|
||||
exit(exit_code);
|
||||
}
|
||||
Err(err) => {
|
||||
eprintln!("{}", err);
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn json_object_to_env(input_json: Value) -> Result<HashMap<String, String>> {
|
||||
let input_object = match input_json {
|
||||
Value::Null => return Err(anyhow!("JSON cannot be null")),
|
||||
Value::Bool(_) => return Err(anyhow!("JSON cannot be bool")),
|
||||
Value::Number(_) => return Err(anyhow!("JSON cannot be number")),
|
||||
Value::String(_) => return Err(anyhow!("JSON cannot be string")),
|
||||
Value::Array(_) => return Err(anyhow!("JSON cannot be array")),
|
||||
Value::Object(obj) => obj,
|
||||
};
|
||||
|
||||
let mut environment = HashMap::<String, String>::new();
|
||||
for (key, value) in input_object.iter() {
|
||||
match value {
|
||||
Value::Null => {
|
||||
eprintln!("[WARN] Key: {key} ignored, value cannot be null");
|
||||
}
|
||||
Value::Bool(b) => {
|
||||
environment.insert(key.clone(), bool_to_str(*b));
|
||||
}
|
||||
Value::Number(n) => {
|
||||
environment.insert(key.clone(), n.to_string());
|
||||
}
|
||||
Value::String(s) => {
|
||||
environment.insert(key.clone(), s.to_string());
|
||||
}
|
||||
Value::Array(arr) => {
|
||||
let arr_first_value_opt = arr.iter().next();
|
||||
if let Some(arr_first_value) = arr_first_value_opt {
|
||||
match arr_first_value {
|
||||
Value::Null => { /*ignore*/ }
|
||||
Value::Bool(b) => {
|
||||
environment.insert(key.clone(), bool_to_str(*b));
|
||||
}
|
||||
Value::Number(n) => {
|
||||
environment.insert(key.clone(), n.to_string());
|
||||
}
|
||||
Value::String(s) => {
|
||||
environment.insert(key.clone(), s.to_string());
|
||||
}
|
||||
Value::Array(_) => { /*ignore*/ }
|
||||
Value::Object(_) => { /*ignore*/ }
|
||||
}
|
||||
}
|
||||
}
|
||||
Value::Object(_) => {
|
||||
eprintln!("[WARN] Key: {key} ignored, value cannot be object")
|
||||
}
|
||||
}
|
||||
}
|
||||
Ok(environment)
|
||||
}
|
||||
|
||||
fn bool_to_str(b: bool) -> String {
|
||||
(if b { "true" } else { "false" }).to_string()
|
||||
}
|
||||
|
||||
fn read_input(input: &str) -> Result<String> {
|
||||
let input_buf = match input {
|
||||
"-" => {
|
||||
let mut input_buf = String::new();
|
||||
io::stdin()
|
||||
.read_to_string(&mut input_buf)
|
||||
.with_context(|| "failed to collect stdin")?;
|
||||
input_buf
|
||||
}
|
||||
input => fs::read_to_string(input)
|
||||
.with_context(|| format!("failed to collect from input: {input}"))?,
|
||||
};
|
||||
Ok(input_buf)
|
||||
}
|
||||
Reference in New Issue
Block a user