fix: parse first then add numbers from stdin #16

src/format.rs

@@ -68,10 +68,21 @@ pub struct FormatOptions {
     #[arg(short = 'z', long)]
     /// format to base32
     base32: bool,
-    #[clap(value_parser=numf_parser::<NumberType>, required=true)]
+    #[clap(value_parser=numf_parser::<NumberType>, required=false)]
     /// at least one number that should be formatted
     ///
-    /// supports either base 10 or base 16 inputs (with 0xaaaa)
+    /// Any of the [Formats](Format::format) are supported, but the prefixes are needed for formats
+    /// other than decimal.
+    ///
+    /// Formats:
+    ///
+    /// - '0x' - Hexadecimal
+    /// - '0b' - Binary
+    /// - '0o' - Octal
+    /// - '0s' - Base64
+    /// - '032s' - Base32
+    ///
+    /// The numbers may be left empty at first, if numbers are provided with the stdin.
     numbers: Vec<NumberType>,
 }
 
@@ -142,6 +153,11 @@ impl FormatOptions {
     pub fn set_prefix(&mut self, value: bool) {
         self.prefix = value;
     }
+
+    /// manually add a number
+    pub fn push_number(&mut self, value: NumberType) {
+        self.numbers.push(value)
+    }
 }
 
 impl Default for FormatOptions {

src/main.rs

@@ -6,16 +6,15 @@
 use std::io::Read;
 use std::process::exit;
 
-use clap::Parser;
+use clap::{CommandFactory, Parser};
 
 mod format;
 use format::*;
+use numf::format::numf_parser;
 
 fn main() {
     // try to read from stdin first, appending the numbers we read to the FormatOptions
-    let mut args: Vec<String> = std::env::args_os()
+    let mut options = FormatOptions::parse();
-        .map(|x| x.into_string().unwrap())
-        .collect();
     let mut stdin_nums = Vec::new();
     match std::io::stdin().lock().read_to_end(&mut stdin_nums) {
         Ok(_) => {
@@ -28,7 +27,14 @@ fn main() {
             };
             let split = whole.split_whitespace();
             for s in split {
-                args.push(s.to_string());
+                let number = match numf_parser(s) {
+                    Ok(n) => n,
+                    Err(e) => {
+                        eprintln!("could not parse number from stdin: {e:#?}");
+                        exit(2);
+                    }
+                };
+                options.push_number(number)
             }
         }
         Err(e) => {
@@ -37,7 +43,10 @@ fn main() {
         }
     };
 
-    let options = FormatOptions::parse_from(args);
+    if options.numbers().is_empty() {
+        format!("{}", FormatOptions::command().render_usage());
+        exit(1);
+    }
 
     let mut out: Vec<String> = Vec::new();