Optional generation of Colors extension

Sources/ResgenSwift/Colors/Colors.swift

@@ -21,8 +21,6 @@ struct Colors: ParsableCommand {
     // MARK: - Static
 
     static let toolName = "Color"
-    static let defaultExtensionName = "Color"
-    static let defaultExtensionNameUIKit = "UIColor"
     static let assetsColorsFolderName = "Colors"
 
     // MARK: - Command options
@@ -57,22 +55,28 @@ struct Colors: ParsableCommand {
         // -> Time: 3.4505380392074585 seconds
 
         // Generate extension
-        ColorExtensionGenerator.writeExtensionFile(
-            colors: parsedColors,
-            staticVar: options.staticMembers,
-            extensionName: options.extensionName,
-            extensionFilePath: options.extensionFilePath,
-            isSwiftUI: true
-        )
+        if let extensionName = options.extensionName,
+           let extensionFilePath = options.extensionFilePath {
+            ColorExtensionGenerator.writeExtensionFile(
+                colors: parsedColors,
+                staticVar: options.staticMembers,
+                extensionName: extensionName,
+                extensionFilePath: extensionFilePath,
+                isSwiftUI: true
+            )
+        }
 
         // Generate extension
-        ColorExtensionGenerator.writeExtensionFile(
-            colors: parsedColors,
-            staticVar: options.staticMembers,
-            extensionName: options.extensionNameUIKit,
-            extensionFilePath: options.extensionFilePathUIKit,
-            isSwiftUI: false
-        )
+        if let extensionNameUIKit = options.extensionNameUIKit,
+           let extensionFilePathUIKit = options.extensionFilePathUIKit {
+            ColorExtensionGenerator.writeExtensionFile(
+                colors: parsedColors,
+                staticVar: options.staticMembers,
+                extensionName: extensionNameUIKit,
+                extensionFilePath: extensionFilePathUIKit,
+                isSwiftUI: false
+            )
+        }
 
         print("[\(Self.toolName)] Colors generated")
     }
@@ -96,18 +100,29 @@ struct Colors: ParsableCommand {
             Self.exit(withError: error)
         }
 
-        // Extension for UIKit and SwiftUI should have different name
-        guard options.extensionName != options.extensionNameUIKit else {
-            let error = ColorsToolError.extensionNamesCollision(options.extensionName)
-            print(error.description)
-            Self.exit(withError: error)
+        // Extension for UIKit and SwiftUI should have different name (if both are defined)
+        if let extensionName = options.extensionName,
+           let extensionNameUIKit = options.extensionNameUIKit {
+            guard extensionName != extensionNameUIKit else {
+                let error = ColorsToolError.extensionNamesCollision(extensionName)
+                print(error.description)
+                Self.exit(withError: error)
+            }
         }
 
         // Check if needed to regenerate
+        let fileToCompareToInput: String = {
+            // If there is no extension file to compare
+            // Then check the xcassets file instead
+            if let extensionFilePath = options.extensionFilePath {
+                return extensionFilePath
+            }
+            return options.xcassetsPath
+        }()
         guard GeneratorChecker.shouldGenerate(
             force: options.forceGeneration,
             inputFilePath: options.inputFile,
-            extensionFilePath: options.extensionFilePath
+            extensionFilePath: fileToCompareToInput
         ) else {
             print("[\(Self.toolName)] Colors are already up to date :) ")
             return false

Sources/ResgenSwift/Colors/ColorsToolOptions.swift

@@ -30,11 +30,11 @@ struct ColorsToolOptions: ParsableArguments {
     @Option(help: "Tell if it will generate static properties or not")
     var staticMembers: Bool = false
 
-    @Option(help: "Extension name. If not specified, it will generate an Color extension.")
-    var extensionName: String = Colors.defaultExtensionName
+    @Option(help: "Extension name. If not specified, no extension will be generated.")
+    var extensionName: String?
 
-    @Option(help: "SwiftUI Extension name. If not specified, it will generate an UIColor extension.")
-    var extensionNameUIKit: String = Colors.defaultExtensionNameUIKit
+    @Option(help: "SwiftUI Extension name. If not specified, no extension will be generated.")
+    var extensionNameUIKit: String?
 
     @Option(help: "Extension suffix. Ex: MyApp, it will generate {extensionName}+ColorsMyApp.swift")
     var extensionSuffix: String?
@@ -46,27 +46,35 @@ extension ColorsToolOptions {
 
     // MARK: - SwiftUI
 
-    var extensionFileName: String {
+    var extensionFileName: String? {
+        guard let extensionName else { return nil }
+
         if let extensionSuffix {
             return "\(extensionName)+\(extensionSuffix).swift"
         }
         return "\(extensionName).swift"
     }
 
-    var extensionFilePath: String {
-        "\(extensionOutputPath)/\(extensionFileName)"
+    var extensionFilePath: String? {
+        guard let extensionFileName else { return nil }
+
+        return "\(extensionOutputPath)/\(extensionFileName)"
     }
 
     // MARK: - UIKit
 
-    var extensionFileNameUIKit: String {
+    var extensionFileNameUIKit: String? {
+        guard let extensionNameUIKit else { return nil }
+
         if let extensionSuffix {
             return "\(extensionNameUIKit)+\(extensionSuffix).swift"
         }
         return "\(extensionNameUIKit).swift"
     }
 
-    var extensionFilePathUIKit: String {
-        "\(extensionOutputPath)/\(extensionFileNameUIKit)"
+    var extensionFilePathUIKit: String? {
+        guard let extensionFileNameUIKit else { return nil }
+
+        return "\(extensionOutputPath)/\(extensionFileNameUIKit)"
     }
 }