lib/scrap_kit/recipe.rb in scrap_kit-0.1.1 vs lib/scrap_kit/recipe.rb in scrap_kit-0.1.2

@@ -30,17 +30,37 @@
     end
 
     def run_step(browser, step)
     end
 
+    def elements_from_selector(browser_or_element, selector)
+      if selector.is_a?(String)
+        browser_or_element.elements(css: selector)
+      elsif selector.is_a?(Array)
+        *remainder, condition = selector
+        elements = browser_or_element
+
+        remainder.each do |item|
+          elements = elements.elements(css: item)
+        end
+
+        elements.filter do |element|
+          condition_key = condition.keys[0].to_s
+          condition_value = condition.values[0]
+          found_element = element.element(css: condition_key)
+          found_element&.text_content&.match(condition_value)
+        end
+      end
+    end
+
     def extract_attribute(browser_or_element, selector_or_hash)
       if selector_or_hash.is_a?(String)
         browser_or_element.element(css: selector_or_hash)&.text_content
       elsif selector_or_hash.is_a?(Hash)
         selector = selector_or_hash[:selector]
         selector_for_children_attributes = selector_or_hash[:children_attributes]
 
-        browser_or_element.elements(css: selector).map do |element|
+        elements_from_selector(browser_or_element, selector).map do |element|
           output = {}
 
           selector_for_children_attributes.each do |child_attribute_name, child_selector|
             output[child_attribute_name] = extract_attribute(element, child_selector)
           end