Génération de composant SwiftUI: Color et Image

Sources/ResgenSwift/Fonts/Generator/FontToolContentGenerator.swift

@@ -9,6 +9,19 @@ import Foundation
 import ToolCore
 
 class FontExtensionGenerator {
+
+    private static func getFontNameEnum(fontsNames: [String]) -> String {
+        var enumDefinition = "    enum FontName: String {\n"
+        
+        fontsNames.forEach {
+            enumDefinition += "        case \($0.fontNameSanitize) = \"\($0)\"\n"
+        }
+        enumDefinition += "    }\n"
+        
+        return enumDefinition
+    }
+    
+    // MARK: - UIKit
     
     static func writeExtensionFile(fontsNames: [String], staticVar: Bool, extensionName: String, extensionFilePath: String) {
         // Create extension content
@@ -47,17 +60,6 @@ class FontExtensionGenerator {
         """
     }
     
-    private static func getFontNameEnum(fontsNames: [String]) -> String {
-        var enumDefinition = "    enum FontName: String {\n"
-        
-        fontsNames.forEach {
-            enumDefinition += "        case \($0.fontNameSanitize) = \"\($0)\"\n"
-        }
-        enumDefinition += "    }\n"
-        
-        return enumDefinition
-    }
-    
     private static func getFontMethods(fontsNames: [FontName], staticVar: Bool) -> String {
         let pragma = "    // MARK: - Getter"
             
@@ -80,4 +82,66 @@ class FontExtensionGenerator {
         }
         """
     }
+    
+    // MARK: - SwiftUI
+    
+    static func writeSUIExtensionFile(fontsNames: [String], staticVar: Bool, extensionName: String, extensionFilePath: String) {
+        // Create extension content
+        let extensionContent = Self.getSUIExtensionContent(fontsNames: fontsNames,
+                                                           staticVar: staticVar,
+                                                           extensionName: extensionName)
+        
+        // Write content
+        let extensionFilePathURL = URL(fileURLWithPath: extensionFilePath)
+        do {
+            try extensionContent.write(to: extensionFilePathURL, atomically: false, encoding: .utf8)
+        } catch (let error) {
+            let error = FontsToolError.writeExtension(extensionFilePath, error.localizedDescription)
+            print(error.localizedDescription)
+            Fonts.exit(withError: error)
+        }
+    }
+    
+    static func getSUIExtensionContent(fontsNames: [String], staticVar: Bool, extensionName: String) -> String {
+        [
+            Self.getSUIHeader(extensionClassname: extensionName),
+            Self.getFontNameEnum(fontsNames: fontsNames),
+            Self.getSUIFontMethods(fontsNames: fontsNames, staticVar: staticVar),
+            Self.getSUIFooter()
+        ]
+            .joined(separator: "\n")
+    }
+    
+    private static func getSUIHeader(extensionClassname: String) -> String {
+        """
+        // Generated by ResgenSwift.\(Fonts.toolName) \(ResgenSwiftVersion)
+        
+        import SwiftUI
+        
+        extension \(extensionClassname) {\n
+        """
+    }
+    
+    private static func getSUIFontMethods(fontsNames: [FontName], staticVar: Bool) -> String {
+        let pragma = "    // MARK: - Getter"
+            
+        var propertiesOrMethods: [String] = fontsNames
+            .unique()
+            .map {
+                if staticVar {
+                    return $0.suiStaticProperty
+                } else {
+                    return $0.suiMethod
+                }
+            }
+        
+        propertiesOrMethods.insert(pragma, at: 0)
+        return propertiesOrMethods.joined(separator: "\n\n")
+    }
+    
+    private static func getSUIFooter() -> String {
+        """
+        }
+        """
+    }
 }