feat: Atty input serialization

Author: tamton-aquib

Cargo.toml

@@ -1,10 +1,10 @@
 [package]
 name = "ice"
-version = "0.2.1"
+version = "0.3.0"
 edition = "2021"
 authors = ["tamton-aquib <[email protected]>"]
 license = "MIT"
-description = "A simple CTF tool set"
+description = "A simple CTF tool store"
 homepage = "https://github.com/tamton-aquib/ice"
 repository = "https://github.com/tamton-aquib/ice"
 readme = "README.md"

PKGBUILD

@@ -1,22 +1,22 @@
 # Maintainer: Aquib Javed <[email protected]>
 pkgname=ice-bin
-pkgver=0.2.1
+pkgver=0.3.0
 pkgrel=1
 epoch=
-pkgdesc="A lightweight CTF utility."
+pkgdesc="A simple CTF tool store."
 arch=('x86_64' 'i686')
 url="https://github.com/tamton-aquib/ice"
 license=('MIT')
 makedepends=()
 options=()
 install=
 changelog=
-source=("ice.zip::https://github.com/tamton-aquib/ice/releases/download/v${pkgver}/ice-x86_64-unknown-linux-gnu.tar.gz")
+source=("ice-x86_64-unknown-linux-gnu.tar.gz::https://github.com/tamton-aquib/ice/releases/download/v${pkgver}/ice-x86_64-unknown-linux-gnu.tar.gz")
 noextract=()
 noextract=()
 sha256sums=('SKIP')
 
 package() {
     tar -xzf "${srcdir}/ice-x86_64-unknown-linux-gnu.tar.gz" -C "${srcdir}"
-    install -Dm755 "${srcdir}/ice-x86_64-unknown-linux-gnu/ice" "${pkgdir}/usr/bin/ice"
+    install -Dm755 "${srcdir}/ice" "${pkgdir}/usr/bin/ice"
 }

src/main.rs

@@ -1,17 +1,30 @@
-use pico_args::Arguments;
-
+use atty::Stream;
 use ice::app::cli::{parse_args, Command};
+use pico_args::Arguments;
+use std::{
+    ffi::OsString,
+    io::{self, Read},
+};
 
 fn print_usage() {
-    println!("ICE - A command-line tool for various ciphers, encoding, and text manipulation.\n");
-    println!("USAGE:");
-    println!("    ice <SUBCOMMAND> [ARGUMENTS]");
-    println!("    ice help");
-    println!("\nFor a list of available subcommands, run `ice help`.");
+    println!("ICE - A simple CTF tool store.");
+    println!("Usage: ice <SUBCOMMAND> [ARGUMENTS]");
+    println!("Run 'ice help' for a list of available subcommands");
 }
 
 fn main() {
-    let mut args = Arguments::from_env();
+    let mut args: Vec<OsString> = std::env::args_os().skip(1).collect();
+
+    if !atty::is(Stream::Stdin) {
+        let mut input = String::new();
+        io::stdin().read_to_string(&mut input).unwrap();
+        let input = input.trim();
+        if !input.is_empty() {
+            args.push(OsString::from(input));
+        }
+    }
+
+    let mut args = Arguments::from_vec(args);
 
     let command = match parse_args(&mut args) {
         Ok(cmd) => cmd,
@@ -23,18 +36,12 @@ fn main() {
     };
 
     match command {
-        Command::Help => {
-            print_usage();
-        }
-        Command::Version => {
-            println!("ice {}", env!("CARGO_PKG_VERSION"));
-        }
+        Command::Help => print_usage(),
+        Command::Version => println!("ice {}", env!("CARGO_PKG_VERSION")),
         Command::Unknown(cmd) => {
             eprintln!("Error: Unknown subcommand '{}'", cmd);
             print_usage();
         }
-        _ => {
-            command.run();
-        }
+        _ => command.run(),
     }
 }