referencing HTML elements using the library name now works

RandomHashTags · RandomHashTags · commit 89529e779bf9 · 2024-11-24T03:20:25.000-06:00
diff --git a/Sources/HTMLKitUtilities/HTMLElementValueType.swift b/Sources/HTMLKitUtilities/HTMLElementValueType.swift
@@ -20,7 +20,15 @@ package indirect enum HTMLElementValueType {
     case array(of: HTMLElementValueType)
 
     package static func parse_element(context: some MacroExpansionContext, _ function: FunctionCallExprSyntax) -> HTMLElement? {
-        guard let key:String = function.calledExpression.declRef?.baseName.text else { return nil }
+        let called_expression:ExprSyntax = function.calledExpression
+        let key:String
+        if let member:MemberAccessExprSyntax = called_expression.memberAccess, member.base?.declRef?.baseName.text == "HTMLKit" {
+            key = member.declName.baseName.text
+        } else if let ref = called_expression.declRef {
+            key = ref.baseName.text
+        } else {
+            return nil
+        }
         let children:SyntaxChildren = function.arguments.children(viewMode: .all)
         switch key {
             case "a": return a(context: context, children)
diff --git a/Tests/HTMLKitTests/ElementTests.swift b/Tests/HTMLKitTests/ElementTests.swift
@@ -56,6 +56,12 @@ extension ElementTests {
         #expect(string == "<!DOCTYPE html><html xmlns=\"test\"></html>")
     }
 
+    // MARK: HTMLKit.element
+    @Test func with_library_decl() {
+        let string:StaticString = #html(html(HTMLKit.body()))
+        #expect(string == "<!DOCTYPE html><html><body></body></html>")
+    }
+
     // MARK: a
     // https://developer.mozilla.org/en-US/docs/Web/HTML/Element/a
     @Test func _a() {
