at_css and to_s for html

Author: dorianmariecom

lib/code/object/html.rb

@@ -59,6 +59,9 @@ def call(**args)
         when "css"
           sig(args) { String }
           code_css(code_value)
+        when "at_css"
+          sig(args) { String }
+          code_at_css(code_value)
         when "map"
           sig(args) { Function }
           code_map(code_value, **globals)
@@ -76,6 +79,12 @@ def code_css(query)
         Html.new(raw.css(code_query.raw))
       end
 
+      def code_at_css(query)
+        code_query = query.to_code
+
+        Html.new(raw.at_css(code_query.raw))
+      end
+
       def code_map(argument, **globals)
         code_argument = argument.to_code
 
@@ -91,6 +100,10 @@ def code_map(argument, **globals)
         )
       end
 
+      def to_s
+        raw.text
+      end
+
       def code_to_string
         String.new(raw.text)
       end