Handle bundle for Image and Color

Sources/ResgenSwift/Images/Generator/ImageExtensionGenerator.swift

@@ -19,7 +19,8 @@ enum ImageExtensionGenerator {
         extensionName: String,
         extensionFilePath: String,
         isSwiftUI: Bool,
-        visibility: ExtensionVisibility
+        visibility: ExtensionVisibility,
+        assetBundle: AssetBundle
     ) {
         // Create extension conten1t
         let extensionContent = Self.getExtensionContent(
@@ -28,7 +29,8 @@ enum ImageExtensionGenerator {
             extensionName: extensionName,
             inputFilename: inputFilename,
             isSwiftUI: isSwiftUI,
-            visibility: visibility
+            visibility: visibility,
+            assetBundle: assetBundle
         )
 
         // Write content
@@ -48,7 +50,8 @@ enum ImageExtensionGenerator {
         extensionName: String,
         inputFilename: String,
         isSwiftUI: Bool,
-        visibility: ExtensionVisibility
+        visibility: ExtensionVisibility,
+        assetBundle: AssetBundle
     ) -> String {
         [
             Self.getHeader(
@@ -60,7 +63,8 @@ enum ImageExtensionGenerator {
                 images: images,
                 staticVar: staticVar,
                 isSwiftUI: isSwiftUI,
-                visibility: visibility
+                visibility: visibility,
+                assetBundle: assetBundle
             ),
             Self.getFooter()
         ]
@@ -86,11 +90,18 @@ enum ImageExtensionGenerator {
         images: [ParsedImage],
         staticVar: Bool,
         isSwiftUI: Bool,
-        visibility: ExtensionVisibility
+        visibility: ExtensionVisibility,
+        assetBundle: AssetBundle
     ) -> String {
         images
-            .map {
-                "\n\($0.getImageProperty(isStatic: staticVar, isSwiftUI: isSwiftUI, visibility: visibility))"
+            .map { parsedImage in
+                let property = parsedImage.getImageProperty(
+                    isStatic: staticVar,
+                    isSwiftUI: isSwiftUI,
+                    visibility: visibility,
+                    assetBundle: assetBundle
+                )
+                return "\n\(property)"
             }
             .joined(separator: "\n")
     }

Sources/ResgenSwift/Images/Images.swift

@@ -62,7 +62,8 @@ struct Images: ParsableCommand {
                 extensionName: extensionName,
                 extensionFilePath: extensionFilePath,
                 isSwiftUI: true,
-                visibility: options.extensionVisibility
+                visibility: options.extensionVisibility,
+                assetBundle: options.assetBundle
             )
         }
 
@@ -75,7 +76,8 @@ struct Images: ParsableCommand {
                 extensionName: extensionNameUIKit,
                 extensionFilePath: extensionFilePathUIKit,
                 isSwiftUI: false,
-                visibility: options.extensionVisibility
+                visibility: options.extensionVisibility,
+                assetBundle: options.assetBundle
             )
         }
 

Sources/ResgenSwift/Images/ImagesOptions.swift

@@ -51,6 +51,11 @@ struct ImagesOptions: ParsableArguments {
     )
     var extensionVisibility: ExtensionVisibility = .internal
 
+    @Option(
+        help: "Bundle where the asset are generated"
+    )
+    var assetBundle: AssetBundle = .main
+
     @Option(
         help: "Path where to generate the extension.",
         transform: { $0.replaceTiltWithHomeDirectoryPath() }

Sources/ResgenSwift/Images/Model/ParsedImage.swift

@@ -130,17 +130,22 @@ struct ParsedImage {
 
     // MARK: - Extension property
 
-    func getImageProperty(isStatic: Bool, isSwiftUI: Bool, visibility: ExtensionVisibility) -> String {
+    func getImageProperty(
+        isStatic: Bool,
+        isSwiftUI: Bool,
+        visibility: ExtensionVisibility,
+        assetBundle: AssetBundle
+    ) -> String {
         if isSwiftUI {
             return """
                 \(visibility) \(isStatic ? "static " : "")var \(name): Image {
-                    Image("\(name)")
+                    Image("\(name)", bundle: Bundle.\(assetBundle))
                 }
             """
         }
         return """
             \(visibility) \(isStatic ? "static " : "")var \(name): UIImage {
-                UIImage(named: "\(name)")!
+                UIImage(named: "\(name)", in: Bundle.\(assetBundle), with: nil)!
             }
         """
     }