actually use version-env on mumctl

2 files changed, 5 insertions, 4 deletions

diff --git a/mumctl/src/main.rs b/mumctl/src/main.rs
index 5b2bc05..fbeb0cd 100644
--- a/mumctl/src/main.rs
+++ b/mumctl/src/main.rs
@@ -223,6 +223,10 @@ impl fmt::Display for CliError {
 }
 
 fn main() {
+    if std::env::args().any(|s| s.as_str() == "--version" || s.as_str() == "-V") {
+        println!("mumctl {}", env!("VERSION"));
+        return;
+    }
     log::set_logger(&LOGGER)
         .map(|()| log::set_max_level(LevelFilter::Info))
         .unwrap();
diff --git a/mumd/src/main.rs b/mumd/src/main.rs
index bc72779..a96944c 100644
--- a/mumd/src/main.rs
+++ b/mumd/src/main.rs
@@ -22,10 +22,7 @@ use tokio_util::codec::{FramedRead, FramedWrite, LengthDelimitedCodec};
 
 #[tokio::main]
 async fn main() {
-    if std::env::args()
-        .find(|s| s.as_str() == "--version")
-        .is_some()
-    {
+    if std::env::args().any(|s| s.as_str() == "--version" || s.as_str() == "-V") {
         println!("mumd {}", env!("VERSION"));
         return;
     }