rexml-css_selector 0.1.0

data/lib/rexml/css_selector/queries/has_query.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class HasQuery
+        def initialize(cont:, query:, needs_parent:)
+          @cont = cont
+          @query = query
+          @needs_parent = needs_parent
+        end
+
+        def call(node, context)
+          matched = false
+          context.scoped(node) do
+            base = node
+            base = context.adapter.get_parent_node(base) if @needs_parent
+            context
+              .adapter
+              .each_recursive_node(base) do |child|
+                next if node == child
+                if @query.call(child, context)
+                  matched = true
+                  break
+                end
+              end
+          end
+
+          matched && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/id_query.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class IdQuery
+        def initialize(cont:, name:)
+          @cont = cont
+          @name = name
+        end
+
+        def call(node, context)
+          context.adapter.get_id(node) == @name && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/nested_query.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NestedQuery
+        def initialize(cont:, query:)
+          @cont = cont
+          @query = query
+        end
+
+        def call(node, context)
+          @query.call(node, context) && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/not_query.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NotQuery
+        def initialize(cont:, query:)
+          @cont = cont
+          @query = query
+        end
+
+        def call(node, context)
+          !@query.call(node, context) && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/nth_child_of_query.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NthChildOfQuery
+        def initialize(cont:, a:, b:, query:)
+          @cont = cont
+          @a = a
+          @b = b
+          @query = query
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          matched_children = context.cache[[object_id, parent.object_id]]
+          unless matched_children
+            children = context.adapter.get_children_elements(parent)
+            matched_children = children.filter { @query.call(_1, context) }
+            context.cache[[object_id, parent.object_id]] = matched_children
+          end
+
+          index = matched_children.index(node)
+          return false unless index
+          index += 1
+
+          if @a.zero?
+            index == @b && @cont.call(node, context)
+          else
+            ((index - @b) % @a).zero? && (index - @b) / @a >= 0 && @cont.call(node, context)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/nth_child_query.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NthChildQuery
+        def initialize(cont:, a:, b:)
+          @cont = cont
+          @a = a
+          @b = b
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          index = context.adapter.get_element_index(parent, node)
+          return false unless index
+          index += 1
+
+          if @a.zero?
+            index == @b && @cont.call(node, context)
+          else
+            ((index - @b) % @a).zero? && (index - @b) / @a >= 0 && @cont.call(node, context)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/nth_last_child_of_query.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NthLastChildOfQuery
+        def initialize(cont:, a:, b:, query:)
+          @cont = cont
+          @a = a
+          @b = b
+          @query = query
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          matched_children = context.cache[[object_id, parent.object_id]]
+          unless matched_children
+            children = context.adapter.get_children_elements(parent)
+            matched_children = children.filter { @query.call(_1, context) }
+            context.cache[[object_id, parent.object_id]] = matched_children
+          end
+
+          index = matched_children.index(node)
+          return false unless index
+          index = matched_children.size - index
+
+          if @a.zero?
+            index == @b && @cont.call(node, context)
+          else
+            ((index - @b) % @a).zero? && (index - @b) / @a >= 0 && @cont.call(node, context)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/nth_last_child_query.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NthLastChildQuery
+        def initialize(cont:, a:, b:)
+          @cont = cont
+          @a = a
+          @b = b
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          children_size = context.adapter.get_children_elements(parent).size
+          index = context.adapter.get_element_index(parent, node)
+          return false unless index
+          index = children_size - index
+
+          if @a.zero?
+            index == @b && @cont.call(node, context)
+          else
+            ((index - @b) % @a).zero? && (index - @b) / @a >= 0 && @cont.call(node, context)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/nth_last_of_type_query.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NthLastOfTypeQuery
+        def initialize(cont:, a:, b:)
+          @cont = cont
+          @a = a
+          @b = b
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          insensitive = context.options[:tag_name_case] == :insensitive
+          tag_name = context.adapter.get_tag_name(node)
+          tag_name = tag_name.downcase(:ascii) if insensitive
+
+          children = context.adapter.get_children_elements(parent)
+          children =
+            children.filter do |child|
+              child_tag_name = context.adapter.get_tag_name(child)
+              child_tag_name = child_tag_name.downcase(:ascii) if insensitive
+              tag_name == child_tag_name
+            end
+
+          index = children.index(node)
+          return false unless index
+          index = children.size - index
+
+          if @a.zero?
+            index == @b && @cont.call(node, context)
+          else
+            ((index - @b) % @a).zero? && (index - @b) / @a >= 0 && @cont.call(node, context)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/nth_of_type_query.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class NthOfTypeQuery
+        def initialize(cont:, a:, b:)
+          @cont = cont
+          @a = a
+          @b = b
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          insensitive = context.options[:tag_name_case] == :insensitive
+          tag_name = context.adapter.get_tag_name(node)
+          tag_name = tag_name.downcase(:ascii) if insensitive
+
+          children = context.adapter.get_children_elements(parent)
+          children =
+            children.filter do |child|
+              child_tag_name = context.adapter.get_tag_name(child)
+              child_tag_name = child_tag_name.downcase(:ascii) if insensitive
+              tag_name == child_tag_name
+            end
+
+          index = children.index(node)
+          return false unless index
+          index += 1
+
+          if @a.zero?
+            index == @b && @cont.call(node, context)
+          else
+            ((index - @b) % @a).zero? && (index - @b) / @a >= 0 && @cont.call(node, context)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/one_of_query.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class OneOfQuery
+        def initialize(conts:)
+          @conts = conts
+        end
+
+        def call(node, context)
+          @conts.any? { |cont| cont.call(node, context) }
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/only_child_query.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class OnlyChildQuery
+        def initialize(cont:)
+          @cont = cont
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          children = context.adapter.get_children_elements(parent)
+          children.size == 1 && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/only_of_type_query.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class OnlyOfTypeQuery
+        def initialize(cont:)
+          @cont = cont
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          parent = context.adapter.get_parent_node(node)
+          return false unless parent
+
+          insensitive = context.options[:tag_name_case] == :insensitive
+          tag_name = context.adapter.get_tag_name(node)
+          tag_name = tag_name.downcase(:ascii) if insensitive
+
+          children = context.adapter.get_children_elements(parent)
+          children =
+            children.filter do |child|
+              child_tag_name = context.adapter.get_tag_name(child)
+              child_tag_name = child_tag_name.downcase(:ascii) if insensitive
+              tag_name == child_tag_name
+            end
+
+          children.size == 1 && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/root_query.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class RootQuery
+        def initialize(cont:)
+          @cont = cont
+        end
+
+        def call(node, context)
+          context.adapter.element?(node) && context.adapter.root?(node) && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/scope_query.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class ScopeQuery
+        def initialize(cont:)
+          @cont = cont
+        end
+
+        def call(node, context)
+          node == context.scope && @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/sibling_query.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class SiblingQuery
+        def initialize(cont:)
+          @cont = cont
+        end
+
+        def call(node, context)
+          while (node = context.adapter.get_previous_sibling_element(node))
+            return true if context.adapter.element?(node) && @cont.call(node, context)
+          end
+
+          false
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/tag_name_type_query.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class TagNameTypeQuery
+        def initialize(cont:, tag_name:, namespace:)
+          @tag_name = tag_name
+          @namespace = namespace
+          @cont = cont
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          return false if @namespace && context.adapter.get_namespace(node) != @namespace
+
+          tag_name = context.adapter.get_tag_name(node)
+          case context.options[:tag_name_case]
+          in :sensitive
+            tag_name == @tag_name && @cont.call(node, context)
+          in :insensitive
+            tag_name = tag_name.downcase(:ascii)
+            tag_name == @tag_name.downcase(:ascii) && @cont.call(node, context)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/true_query.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class TrueQuery
+        def call(_node, _context)
+          true
+        end
+
+        INSTANCE = new
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/queries/universal_type_query.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    module Queries
+      class UniversalTypeQuery
+        def initialize(cont:, namespace:)
+          @namespace = namespace
+          @cont = cont
+        end
+
+        def call(node, context)
+          return false unless context.adapter.element?(node)
+
+          return false if @namespace && context.adapter.get_namespace(node) != @namespace
+
+          @cont.call(node, context)
+        end
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/query_context.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    # QueryContext is a context on matching CSS selector.
+    class QueryContext
+      def initialize(scope:, substitutions:, adapter:, options:)
+        @scope = scope
+        @substitutions = substitutions
+        @adapter = adapter
+        @options = options
+        @cache = {}
+      end
+
+      attr_reader :scope, :substitutions, :adapter, :cache, :options
+
+      def scoped(new_scope)
+        old_scope = @scope
+        @scope = new_scope
+        yield
+        @scope = old_scope
+      end
+    end
+  end
+end

data/lib/rexml/css_selector/version.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module REXML
+  module CSSSelector
+    # A version number of this library.
+    VERSION = "0.1.0"
+  end
+end