Fix remove class method (#13)

If all the classes were removed also remove the attribute
BinaryBirds · Feb 18, 2022 · 9e6389f · 9e6389f
1 parent ecb4d01
commit 9e6389f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/Sources/SwiftHtml/Attributes/Global.swift b/Sources/SwiftHtml/Attributes/Global.swift
@@ -107,6 +107,9 @@ public extension Tag {
     /// Removes an array of class values if the condition is true
     func `class`(remove values: [String], _ condition: Bool = true) -> Self {
         let newClasses = classArray.filter { !values.contains($0) }
+        if newClasses.isEmpty {
+            return deleteAttribute("class")
+        }
         return `class`(newClasses, condition)
     }
 

diff --git a/Tests/SwiftHtmlTests/SwiftHtmlTests.swift b/Tests/SwiftHtmlTests/SwiftHtmlTests.swift
@@ -82,6 +82,16 @@ final class SwiftHtmlTests: XCTestCase {
         XCTAssertEqual(#"<span class="a c"></span>"#, html)
     }
 
+    func testRemoveLastClass() {
+        let doc = Document {
+            Span("")
+                .class("a")
+                .class(remove: "a")
+        }
+        let html = DocumentRenderer(minify: true).render(doc)
+        XCTAssertEqual(#"<span></span>"#, html)
+    }
+
     func testToggleAddClass() {
         let doc = Document {
             Span("")