class Array

# File lib/grafter/core_ext/array.rb, line 2
def graft(args={})
  branch   = args[:branch].to_sym
  operator = args[:operator] && args[:operator].to_sym || "===".to_sym
  check_response_in_graft(branch,operator)

  base_trees = self
  base_trees.each do |base_tree|
    graftable_trees = base_trees.select{|obj| base_tree.send(operator,obj) && !base_tree.equal?(obj) }
    graftable_trees.each do |graftable_tree|
      base_tree.send(branch).concat(graftable_tree.send(branch))
      base_trees.delete(graftable_tree)
    end
  end

  base_trees.each{|base_tree| yield(base_tree.send(branch)) } if block_given?
  base_trees
end

# File lib/grafter/core_ext/array.rb, line 22
def check_response_in_graft(*methods)
  methods.each do |method|
    self.each do |obj|
      raise NoMethodError, "#{obj.class}##{method} is not defined" unless obj.respond_to?(method)
    end
  end
end