update ruby dependencies and object classes

Author: dorianmariecom

Gemfile.lock

@@ -10,6 +10,7 @@ PATH
       json
       language-ruby
       net-http
+      nokogiri
       uri
       zeitwerk
 
@@ -101,13 +102,21 @@ GEM
     libv8-node (18.19.0.0-arm64-darwin)
     libv8-node (18.19.0.0-x86_64-linux)
     logger (1.6.4)
+    mini_portile2 (2.8.8)
     mini_racer (0.16.0)
       libv8-node (~> 18.19.0.0)
     minitest (5.25.4)
     mustermann (3.0.3)
       ruby2_keywords (~> 0.0.1)
     net-http (0.6.0)
       uri
+    nokogiri (1.18.1)
+      mini_portile2 (~> 2.8.2)
+      racc (~> 1.4)
+    nokogiri (1.18.1-arm64-darwin)
+      racc (~> 1.4)
+    nokogiri (1.18.1-x86_64-linux-gnu)
+      racc (~> 1.4)
     openssl (3.2.1)
     ostruct (0.6.1)
     parallel (1.26.3)

code-ruby.gemspec

@@ -21,6 +21,7 @@ Gem::Specification.new do |s|
   s.add_dependency "json"
   s.add_dependency "language-ruby"
   s.add_dependency "net-http"
+  s.add_dependency "nokogiri"
   s.add_dependency "uri"
   s.add_dependency "zeitwerk"
 

lib/code-ruby.rb

@@ -8,6 +8,7 @@
 require "json"
 require "language-ruby"
 require "net/http"
+require "nokogiri"
 require "stringio"
 require "timeout"
 require "uri"
@@ -136,3 +137,9 @@ def to_code
     Code::Object::Dictionary.new(self)
   end
 end
+
+class Nokogiri::XML::Element
+  def to_code
+    Code::Object::Html.new(self)
+  end
+end

lib/code/object/dictionary.rb

@@ -396,7 +396,7 @@ def code_fetch_values(*arguments)
 
       def code_flatten(level = nil)
         code_level = level.to_code
-        code_level = Integer.new(-1) if level.nothing?
+        code_level = Integer.new(-1) if code_level.nothing?
         code_to_list.code_flatten(code_level)
       end
 

lib/code/object/html.rb

@@ -3,6 +3,18 @@
 class Code
   class Object
     class Html < Object
+      def initialize(*args, **_kargs, &)
+        if args.first.is_an?(Html)
+          @raw = args.first.raw
+        elsif args.first.is_a?(::Nokogiri::XML::NodeSet)
+          @raw = args.first
+        elsif args.first.is_a?(Nokogiri::XML::Node)
+          @raw = args.first
+        else
+          @raw = Nokogiri::HTML(args.first.to_s)
+        end
+      end
+
       def self.call(**args)
         code_operator = args.fetch(:operator, nil).to_code
         code_arguments = args.fetch(:arguments, []).to_code
@@ -35,6 +47,52 @@ def self.code_escape(string = nil)
 
         String.new(CGI.escapeHTML(string.to_s))
       end
+
+      def call(**args)
+        code_operator = args.fetch(:operator, nil).to_code
+        code_arguments = args.fetch(:arguments, []).to_code
+        code_value = code_arguments.code_first
+        globals = multi_fetch(args, *GLOBALS)
+
+        case code_operator.to_s
+        when "css"
+          sig(args) { String }
+          code_css(code_value)
+        when "map"
+          sig(args) { Function }
+          code_map(code_value, **globals)
+        when "to_string"
+          sig(args)
+          code_to_string
+        else
+          super
+        end
+      end
+
+      def code_css(query)
+        code_query = query.to_code
+
+        Html.new(raw.css(code_query.raw))
+      end
+
+      def code_map(argument, **globals)
+        code_argument = argument.to_code
+
+        List.new(
+          raw.map.with_index do |element, index|
+            code_argument.call(
+              arguments: List.new([element.to_code, Integer.new(index), self]),
+              **globals
+            )
+          rescue Error::Next => e
+            e.code_value
+          end
+        )
+      end
+
+      def code_to_string
+        String.new(raw.text)
+      end
     end
   end
 end

lib/code/object/list.rb

@@ -81,6 +81,12 @@ def call(**args)
         when "select!"
           sig(args) { Function }
           code_select!(code_value, **globals)
+        when "compact"
+          sig(args) { Function.maybe }
+          code_compact(code_value, **globals)
+        when "compact!"
+          sig(args) { Function.maybe }
+          code_compact!(code_value, **globals)
         when "reject"
           sig(args) { Function }
           code_reject(code_value, **globals)
@@ -298,6 +304,52 @@ def code_reverse
         List.new(raw.reverse)
       end
 
+      def code_compact(argument = nil, **globals)
+        code_argument = argument.to_code
+
+        index = 0
+
+        List.new(
+          raw.compact do |code_element|
+            if code_argument.nothing?
+              code_element.falsy?
+            else
+              code_argument.call(
+                arguments: List.new([code_element, Integer.new(index), self]),
+                **globals
+              ).falsy?
+            end
+          rescue Error::Next => e
+            e.code_value.falsy?
+          ensure
+            index += 1
+          end
+        )
+      end
+
+      def code_compact!(argument = nil, **globals)
+        code_argument = argument.to_code
+
+        index = 0
+
+        raw.compact! do |code_element|
+          if code_argument.nothing?
+            code_element.falsy?
+          else
+            code_argument.call(
+              arguments: List.new([code_element, Integer.new(index), self]),
+              **globals
+            ).falsy?
+          end
+        rescue Error::Next => e
+          e.code_value.falsy?
+        ensure
+          index += 1
+        end
+
+        self
+      end
+
       def code_select(argument, **globals)
         code_argument = argument.to_code
 

lib/code/object/string.rb

@@ -31,6 +31,9 @@ def call(**args)
         when "include?"
           sig(args) { String }
           code_include?(code_value)
+        when "first"
+          sig(args) { Integer.maybe }
+          code_first(code_value)
         when "reverse"
           sig(args)
           code_reverse
@@ -65,6 +68,12 @@ def code_reverse
       def code_to_function(**_globals)
         Function.new([{ name: "_" }], "_.#{raw}")
       end
+
+      def code_first(n = nil)
+        code_n = n.to_code
+        code_n = Integer.new(1) if code_n.nothing?
+        String.new(raw.first(code_n.raw))
+      end
     end
   end
 end

lib/code/object/time.rb

@@ -71,6 +71,9 @@ def call(**args)
         when "hour"
           sig(args)
           code_hour
+        when "format"
+          sig(args) { String }
+          code_format(code_value)
         else
           super
         end
@@ -101,6 +104,12 @@ def code_future?
       def code_hour
         Integer.new(raw.hour)
       end
+
+      def code_format(format)
+        code_format = format.to_code
+
+        String.new(raw.strftime(code_format.raw))
+      end
     end
   end
 end