feat: Short flag for the version command

siketyan · Jan 2, 2024 · 4721f21 · 4721f21
1 parent 8b0e57a
commit 4721f21
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/cmd/version.rs b/src/cmd/version.rs
@@ -1,14 +1,23 @@
 use anyhow::Result;
 use clap::Parser;
 
-use crate::BUILD_INFO;
+use crate::{BUILD_INFO, VERSION};
 
 #[derive(Debug, Parser)]
-pub struct Cmd {}
+pub struct Cmd {
+    #[clap(short, long)]
+    short: bool,
+}
 
 impl Cmd {
     pub fn run(self) -> Result<()> {
-        println!("{}", BUILD_INFO);
+        println!(
+            "{}",
+            match self.short {
+                true => VERSION,
+                _ => BUILD_INFO,
+            },
+        );
 
         Ok(())
     }

diff --git a/src/main.rs b/src/main.rs
@@ -27,6 +27,8 @@ const BUILD_INFO: &str = build_info::format!(
     $.timestamp,
 );
 
+const VERSION: &str = build_info::format!("{}", $.crate_info.version);
+
 #[tokio::main]
 async fn main() {
     if let Err(e) = Cli::parse().run().await {