Add --version and --help options

Author: kylef

bin/querykit.swift

@@ -5,6 +5,8 @@ import PathKit
 import Stencil
 
 
+let version = "0.1.2"
+
 extension Path {
   static var processPath:Path {
     if Process.arguments[0].componentsSeparatedByString(Path.separator).count > 1 {
@@ -148,36 +150,48 @@ func render(model:NSManagedObjectModel, destination:Path, templatePath:Path) {
   }
 }
 
-func run() {
-  let arguments = Process.arguments
+func generate(modelPath:Path, outputPath:Path) {
+  let modelExtension = modelPath.`extension`
+  let isDataModel = modelExtension == "xcdatamodel"
+  let isDataModeld = modelExtension == "xcdatamodeld"
 
-  if arguments.count == 3 {
-    let modelPath = Path(arguments[1])
-    let outputPath = Path(arguments[2])
-    let modelExtension = modelPath.`extension`
-    let isDataModel = modelExtension == "xcdatamodel"
-    let isDataModeld = modelExtension == "xcdatamodeld"
-
-    if isDataModel || isDataModeld {
-      if modelPath.isReadable {
-        let templatePath = Path.defaultTemplatePath
-        if !templatePath.isReadable {
-          print("Template '\(templatePath)' is not readable.")
-        } else {
-          let compiledModel = compileCoreDataModel(modelPath)
-          let modelURL = NSURL(fileURLWithPath: compiledModel.description)
-          let model = NSManagedObjectModel(contentsOfURL: modelURL)!
-          render(model, destination: outputPath, templatePath: templatePath)
-        }
+  if isDataModel || isDataModeld {
+    if modelPath.isReadable {
+      let templatePath = Path.defaultTemplatePath
+      if !templatePath.isReadable {
+        print("Template '\(templatePath)' is not readable.")
       } else {
-        print("'\(modelPath)' does not exist or is not readable.")
+        let compiledModel = compileCoreDataModel(modelPath)
+        let modelURL = NSURL(fileURLWithPath: compiledModel.description)
+        let model = NSManagedObjectModel(contentsOfURL: modelURL)!
+        render(model, destination: outputPath, templatePath: templatePath)
       }
     } else {
-      print("'\(modelPath)' is not a Core Data model.")
+      print("'\(modelPath)' does not exist or is not readable.")
     }
   } else {
-    let processName = arguments[0]
-    print("Usage: \(processName) <model> <output-directory>")
+    print("'\(modelPath)' is not a Core Data model.")
+  }
+}
+
+func usage() {
+  let processName = Process.arguments.first!
+  print("Usage: \(processName) <model> <output-directory>")
+}
+
+func run() {
+  let arguments = Process.arguments
+
+  if arguments.contains("--help") {
+    usage()
+  } else if arguments.contains("--version") {
+    print(version)
+  } else if arguments.count != 3 {
+    usage()
+  } else {
+    let modelPath = Path(arguments[1])
+    let outputPath = Path(arguments[2])
+    generate(modelPath, outputPath: outputPath)
   }
 }