From 5e1fcafe4134b948ec8e860332d448e75fa90d44 Mon Sep 17 00:00:00 2001
From: Rabindra Dhakal <contact@qaidvoid.dev>
Date: Tue, 12 Nov 2024 20:59:32 +0545
Subject: [PATCH] feat(cli): allow stdin input as args
---
src/lib.rs | 18 ++++++++++++++++--
src/misc/download.rs | 2 +-
2 files changed, 17 insertions(+), 3 deletions(-)
diff --git a/src/lib.rs b/src/lib.rs
index a680037..bd76077 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -12,7 +12,7 @@ use core::{
log::setup_logging,
util::{cleanup, print_env, setup_required_paths},
};
-use std::{env, path::Path};
+use std::{env, io::Read, path::Path};
mod cli;
pub mod core;
@@ -21,7 +21,21 @@ mod package;
mod registry;
async fn handle_cli() -> Result<()> {
- let args = Args::parse();
+ let mut args = env::args().collect::<Vec<_>>();
+
+ if let Some(command) = args.last() {
+ if command == "-" {
+ args.pop();
+ let mut stdin = std::io::stdin();
+ let mut buffer = String::new();
+ if stdin.read_to_string(&mut buffer).is_ok() {
+ let stdin_args = buffer.split_whitespace().collect::<Vec<&str>>();
+ args.extend(stdin_args.into_iter().map(String::from));
+ }
+ }
+ }
+
+ let args = Args::try_parse_from(args)?;
setup_logging(&args);
diff --git a/src/misc/download.rs b/src/misc/download.rs
index 9c477b0..a5cbfeb 100644
--- a/src/misc/download.rs
+++ b/src/misc/download.rs
@@ -287,7 +287,7 @@ pub async fn download_and_save(
}
);
- let asset_idx = select_asset_idx(&assets, releases.len())?;
+ let asset_idx = select_asset_idx(&assets, assets.len())?;
assets[asset_idx]
}
}