find.js

import ruleSet from './rules/ruleSet'

export default function find (component, selector) {
  let results = []
  switch (selector.type) {
    case 'ruleSet':
      if (ruleSet(component, selector)) {
        results.push(component)
      }
      break
    case 'selectors':
      let selectorsResult = new Set(selector.selectors.map(s => find(component, s)))
      results.push(...selectorsResult)
      return results
    case 'custom':
      if (selector.match(component)) {
        results.push(component)
      }
      break
  }

  if (component.children) {
    let children = Array.isArray(component.children) ? component.children : [component.children]
    children.forEach(child => {
      if (typeof child === 'object') {
        results = [...results, ...find(child, selector)]
      }
    })
  }

  return results
}