Print the version with -v / --version

[rbdr/lyricli] / Sources / main.swift

diff --git a/Sources/main.swift b/Sources/main.swift
index 61f4f7154e11a8acbd658186d835a39da65274e3..b4d3ebfe4f11382d61ab1b93964160941f665cd4 100644 (file)
--- a/Sources/main.swift
+++ b/Sources/main.swift
@@ -3,13 +3,14 @@ import Foundation
 
 /// Sets up and returns a new options parser
 /// 
-/// - Returns: A new OptionParser instance
+/// - Returns: A Dictionary of Options, and a new CommandLineKit instance
 func createParser() -> ([String:Option], CommandLineKit) {
 
     let parser = CommandLineKit()
     var flags: [String:Option] = [:]
 
     flags["help"] = BoolOption(shortFlag: "h", longFlag: "help", helpMessage: "Prints a help message.")
+    flags["version"] = BoolOption(shortFlag: "v", longFlag: "version", helpMessage: "Prints the version.")
 
     parser.addOptions(Array(flags.values))
 
@@ -35,6 +36,13 @@ func main() {
             }
     }
 
+    if let versionFlag = flags["version"] as? BoolOption {
+        if versionFlag.value == true {
+                print(Lyricli.version)
+                exit(0)
+            }
+    }
+
 }
 
 main()