Add option to print version and exit

TheBicPen · Sep 8, 2024 · 557e7dd · 557e7dd
1 parent 6de5509
commit 557e7dd
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 6 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -169,7 +169,7 @@ fn real_main() -> Result<(), Box<dyn Error>> {
             &config,
         )?),
         parse_args::ParseArgsResult::Error(err) => Err(Box::new(err)),
-        parse_args::ParseArgsResult::Help => Ok(()),
+        parse_args::ParseArgsResult::PrintAndExit => Ok(()),
     }
 }
 

diff --git a/src/parse_args.rs b/src/parse_args.rs
@@ -31,12 +31,12 @@ impl Default for OutputConfig {
 pub enum ParseArgsResult {
     Config(Config),
     Error(SimpleError),
-    Help,
+    PrintAndExit,
 }
 enum ParseArgsResultInner {
     Success,
     Error(SimpleError),
-    Help,
+    PrintAndExit,
 }
 
 #[derive(Default)]
@@ -102,6 +102,10 @@ fn print_help() {
     }
 }
 
+fn print_version() {
+    println!("hunk v{}", env!("CARGO_PKG_VERSION"));
+}
+
 pub fn parse_args(args: &[&str]) -> ParseArgsResult {
     struct ParsingState {
         has_search_string: bool,
@@ -185,7 +189,11 @@ pub fn parse_args(args: &[&str]) -> ParseArgsResult {
             )),
             ["--help"] | ["-h"] => {
                 print_help();
-                ParseArgsResultInner::Help
+                ParseArgsResultInner::PrintAndExit
+            }
+            ["--version"] => {
+                print_version();
+                ParseArgsResultInner::PrintAndExit
             }
             ["--", rest @ ..] if !state.has_search_string => {
                 state.no_more_options = true;
@@ -231,7 +239,7 @@ pub fn parse_args(args: &[&str]) -> ParseArgsResult {
     match parse_slice(args, &mut parsing_state, &mut config) {
         ParseArgsResultInner::Error(err) => ParseArgsResult::Error(err),
         ParseArgsResultInner::Success => ParseArgsResult::Config(config),
-        ParseArgsResultInner::Help => ParseArgsResult::Help,
+        ParseArgsResultInner::PrintAndExit => ParseArgsResult::PrintAndExit,
     }
 }
 

diff --git a/src/test.rs b/src/test.rs
@@ -503,6 +503,11 @@ mod tests {
 
     #[test]
     fn test_parse_help() {
-        matches!(parse_args(&vec!["asd", "-h"]), ParseArgsResult::Help);
+        matches!(parse_args(&vec!["asd", "-h"]), ParseArgsResult::PrintAndExit);
+    }
+
+    #[test]
+    fn test_parse_version() {
+        matches!(parse_args(&vec!["hi", "--version"]), ParseArgsResult::PrintAndExit);
     }
 }