attributes

Author: omochi

Codegen/Package.resolved

@@ -12,12 +12,14 @@ let package = Package(
         )
     ],
     dependencies: [
+        .package(url: "https://github.com/apple/swift-collections", from: "1.1.0"),
         .package(url: "https://github.com/omochi/CodegenKit", from: "1.4.1")
     ],
     targets: [
         .target(
             name: "SRTCodegen",
             dependencies: [
+                .product(name: "Collections", package: "swift-collections"),
                 .product(name: "CodegenKit", package: "CodegenKit")
             ]
         ),

Codegen/Package.swift

@@ -1,3 +1,5 @@
+import Collections
+
 struct Def: Codable {
     var tagNames: [String]
     var voidElements: [String]
@@ -8,4 +10,18 @@ struct Def: Codable {
             voidElements.contains(tagName)
         }
     }
+
+    var allAttributes: [String] {
+        var attrs: OrderedSet<String> = []
+        attrs.formUnion(elementAttributes["*"] ?? [])
+
+        var dict = elementAttributes
+        dict["*"] = nil
+
+        for key in dict.keys.sorted() {
+            attrs.formUnion(dict[key] ?? [])
+        }
+
+        return attrs.elements
+    }
 }

Codegen/Sources/SRTCodegen/Def.swift

@@ -0,0 +1,29 @@
+import Foundation
+import CodegenKit
+
+struct AttributeRenderer: Renderer {
+    var def: Def
+
+    func isTarget(file: URL) -> Bool {
+        file.lastPathComponent == "Attributes.swift"
+    }
+
+    func render(template: inout CodeTemplateModule.CodeTemplate, file: URL, on runner: CodegenKit.CodegenRunner) throws {
+        let code = def.allAttributes.map { (attribute) in
+            renderSetter(attribute: attribute)
+        }.joined(separator: "\n")
+
+        template["setters"] = code
+    }
+
+    private func renderSetter(attribute: String) -> String {
+        let symbol = attribute.kebabToCamel()
+
+        return """
+        public func \(renderIdentifier(symbol))(_ x: String) -> Attributes {
+            set("\(attribute)", to: x)
+        }
+
+        """
+    }
+}

Codegen/Sources/SRTCodegen/Keywords.swift

@@ -39,12 +39,4 @@ struct HTMLTagRenderer: Renderer {
 
         """
     }
-
-    private func renderIdentifier(_ text: String) -> String {
-        var text = text
-        if keywords.contains(text) {
-            text = "`\(text)`"
-        }
-        return text
-    }
 }

Codegen/Sources/SRTCodegen/Renderers/AttributeRenderer.swift

@@ -21,7 +21,8 @@ public struct SRTCodegen {
         let runner = CodegenRunner(
             renderers: [
                 HTMLTagRenderer(def: def),
-                HTMLVoidTagRenderer(def: def)
+                HTMLVoidTagRenderer(def: def),
+                AttributeRenderer(def: def)
             ]
         )
 

Codegen/Sources/SRTCodegen/HTMLTagRenderer.swift renamed to Codegen/Sources/SRTCodegen/Renderers/HTMLTagRenderer.swift

@@ -0,0 +1,36 @@
+import Foundation
+
+public let keywords = [
+    "as",
+    "class",
+    "default",
+    "defer",
+    "for",
+    "is",
+    "var",
+]
+
+func renderIdentifier(_ text: String) -> String {
+    var text = text
+    if keywords.contains(text) {
+        text = "`\(text)`"
+    }
+    return text
+}
+
+extension String {
+    func kebabToCamel() -> String {
+        let strs = self.components(separatedBy: "-")
+
+        var result = ""
+        for (index, str) in strs.enumerated() {
+            if index == 0 {
+                result += str.lowercased()
+            } else {
+                result += str.capitalized
+            }
+        }
+
+        return result
+    }
+}