Add subcommands

src/main.rs

@@ -1,7 +1,7 @@
 use std::time::{SystemTime, UNIX_EPOCH};
 
 use base32::Alphabet;
-use clap::Parser;
+use clap::{Parser, Subcommand};
 use totp_rfc6238::{HashAlgorithm, TotpGenerator};
 
 /// Get the current time since epoch
@@ -14,6 +14,15 @@ pub fn time() -> u64 {
 #[derive(Parser, Debug)]
 #[clap(author, version, about, long_about = None)]
 struct Args {
+    #[clap(subcommand)]
+    command: SubCommands
+}
+
+#[derive(Subcommand, Debug)]
+enum SubCommands {
+    /// Get a code
+    #[clap(arg_required_else_help = true)]
+    Code {
         /// The secret to use
         #[clap(short, long)]
         secret: String,
@@ -25,20 +34,23 @@ struct Args {
         /// Size of generated secret
         #[clap(short, long, default_value_t = 6)]
         len: usize,
+    }
 }
 
 fn main() {
     let args: Args = Args::parse();
 
+    match args.command {
+        SubCommands::Code { secret, topt_step, len } => {
             let totp_generator = TotpGenerator::new()
-        .set_digit(args.len).unwrap()
-        .set_step(args.topt_step).unwrap()
+                .set_digit(len).unwrap()
+                .set_step(topt_step).unwrap()
                 .set_hash_algorithm(HashAlgorithm::SHA1)
                 .build();
 
             let key = base32::decode(
                 Alphabet::RFC4648 { padding: true },
-        &args.secret,
+                &secret,
             ).unwrap();
 
             let code = totp_generator.get_code(&key);
@@ -46,4 +58,6 @@ fn main() {
 
             println!("Secret = {}", code);
             println!("Next update = {} seconds", next_update - time());
+        }
+    }
 }