Format pseudo-class functions with complex arguments

Author: flavorjones

lib/syntax_tree/css/format.rb

@@ -100,6 +100,16 @@ def visit_pseudo_class_selector(node)
         node.value.format(q)
       end
 
+      # Visit a Selectors::PseudoClassFunction node.
+      def visit_pseudo_class_function(node)
+        q.text(node.name)
+        q.text("(")
+        q.seplist(node.arguments, -> { q.text(", ") }) do |selector|
+          selector.format(q)
+        end
+        q.text(")")
+      end
+
       # Visit a Selectors::PseudoElementSelector node.
       def visit_pseudo_element_selector(node)
         q.text(":")
@@ -127,7 +137,6 @@ def visit_compound_selector(node)
           node.child_nodes.each do |node_|
             node_.format(q)
           end
-          # TODO: pseudo-elements
         end
       end
 

lib/syntax_tree/css/pretty_print.rb

@@ -380,10 +380,19 @@ def visit_pseudo_class_function(node)
           q.breakable
           q.pp(node.name)
 
+          q.breakable
+          q.text("(arguments")
+
           if node.arguments.any?
-            q.breakable
-            q.seplist(node.arguments) { |argument| q.pp(argument) }
+            q.nest(2) do
+              q.breakable
+              q.seplist(node.arguments) { |argument| q.pp(argument) }
+            end
+
+            q.breakable("")
           end
+
+          q.text(")")
         end
       end
 
@@ -443,7 +452,10 @@ def visit_complex_selector(node)
       def visit_compound_selector(node)
         token("compound-selector") do
           q.breakable
-          q.pp(node.type)
+          token("type") do
+            q.breakable
+            q.pp(node.type)
+          end
 
           q.breakable
           q.text("(subclasses")

test/selectors_test.rb

@@ -253,6 +253,13 @@ class SelectorsTest < Minitest::Spec
             )
           end
 
+          it "with a pseudo-class function" do
+            assert_selector_format(
+              ".flex:not(div, span.wide, .hidden)",
+              ".flex:not(div, span.wide, .hidden)",
+            )
+          end
+
           it "with class selectors" do
             assert_selector_format(
               "div.flex.text-xl",