Add generate and refresh arguments

Author: Eric Miller

Sources/UBKit/Models/UBKitError.swift

@@ -24,6 +24,8 @@
 import Foundation
 
 enum UBKitError: Error, CustomStringConvertible {
+    case invalidNumberOfArguments
+    case invalidArguments(String)
     case error(Error)
     case unableToCreateFile(String)
     case invalidFolder(String)
@@ -38,6 +40,10 @@ enum UBKitError: Error, CustomStringConvertible {
 
     var description: String {
         switch self {
+        case .invalidNumberOfArguments:
+            return "Invalid number of command line arguments"
+        case .invalidArguments(let str):
+            return "Invalid argument: \(str)"
         case .error(let err):
             return err.localizedDescription
         case .unableToCreateFile(let str):

Sources/UBKit/UBKit.swift

@@ -30,11 +30,46 @@ public final class UBKit {
 
     private let fileManager = FileManager.default
 
-    public init(arguments: [String] = CommandLine.arguments) {
+    public init(arguments: [String] = CommandLine.arguments) throws {
+        guard arguments.count == 2 else {
+            throw UBKitError.invalidNumberOfArguments
+        }
+
         self.arguments = arguments
     }
 
     public func run(_ completion: @escaping ((Error?) -> Void)) {
+        guard let arg = getArgument(for: arguments[1]) else {
+            completion(UBKitError.invalidArguments(arguments[1]))
+            return
+        }
+
+        switch arg {
+        case .generate:
+            generate(completion)
+        case .refresh:
+            refresh(completion)
+        }
+    }
+
+    class func validatePath(_ path: String, isDirectory: Bool) -> Bool {
+        if isDirectory {
+            var directory = ObjCBool(false)
+            let folderExists = FileManager.default.fileExists(atPath: path, isDirectory: &directory)
+            return folderExists && directory.boolValue
+        } else {
+            return FileManager.default.fileExists(atPath: path)
+        }
+    }
+}
+
+private extension UBKit {
+
+    enum Argument: String {
+        case generate, refresh
+    }
+
+    func generate(_ completion: @escaping ((Error?) -> Void)) {
         let workingPath = fileManager.currentDirectoryPath
         do {
             guard let fileData = fileManager.contents(atPath: workingPath.appending("/ubconfig.json")) else {
@@ -60,16 +95,20 @@ public final class UBKit {
             return
         }
 
+        print("\n----------")
+        print("👍 Successfully created your iOS and Unity projects")
+        print("The iOS project is located under iOS/")
+        print("The Unity project is located under Unity/")
         completion(nil)
     }
 
-    class func validatePath(_ path: String, isDirectory: Bool) -> Bool {
-        if isDirectory {
-            var directory = ObjCBool(false)
-            let folderExists = FileManager.default.fileExists(atPath: path, isDirectory: &directory)
-            return folderExists && directory.boolValue
-        } else {
-            return FileManager.default.fileExists(atPath: path)
-        }
+    func refresh(_ completion: @escaping ((Error?) -> Void)) {
+        print("\n----------")
+        print("👍 Successfully refreshed your iOS project")
+        completion(nil)
+    }
+
+    func getArgument(for arg: String) -> Argument? {
+        return Argument(rawValue: arg)
     }
 }

Sources/UnityBuildKit/main.swift

@@ -24,7 +24,17 @@
 import Foundation
 import UBKit
 
-let kit = UBKit()
+var kit: UBKit
+do {
+    kit = try UBKit()
+} catch {
+    print("\n----------")
+    print("💥 An error was encountered while creating your projects")
+    print(error.localizedDescription)
+    exit(1)
+}
+
+
 kit.run { (error) in
     guard error == nil else {
         print("\n----------")
@@ -33,10 +43,6 @@ kit.run { (error) in
         exit(1)
     }
 
-    print("\n----------")
-    print("👍 Successfully created your iOS and Unity projects!")
-    print("The iOS project is located under iOS/")
-    print("The Unity project is located under Unity/")
     print("For more information, visit <https://github.com/handsomecode/UnityBuildKit>\n\n")
     exit(0)
 }