module Hpricot::Container::Trav

# File lib/hpricot/traverse.rb, line 518
def classes
  get_attribute('class').to_s.strip.split(/\s+/)
end

# File lib/hpricot/traverse.rb, line 404
def containers
  children.grep(Container::Trav)
end

# File lib/hpricot/traverse.rb, line 498
def each_child(&block) # :yields: child_node
  children.each(&block) if children
  nil
end

# File lib/hpricot/traverse.rb, line 504
def each_child_with_index(&block) # :yields: child_node, index
  children.each_with_index(&block) if children
  nil
end

# File lib/hpricot/traverse.rb, line 614
def each_hyperlink # :yields: text
  links = []
  each_hyperlink_attribute {|elem, attr, hyperlink|
    yield hyperlink
  }
end

# File lib/hpricot/traverse.rb, line 591
def each_hyperlink_uri(base_uri=nil) # :yields: hyperlink, uri
  base_uri = URI.parse(base_uri) if String === base_uri
  links = []
  each_hyperlink_attribute {|elem, attr, hyperlink|
    if %r{\{http://www.w3.org/1999/xhtml\}(?:base)\z}i =~ elem.name
      base_uri = URI.parse(hyperlink.to_s)
    else
      links << hyperlink
    end
  }
  if base_uri
    links.each {|hyperlink| yield hyperlink, base_uri + hyperlink.to_s }
  else
    links.each {|hyperlink| yield hyperlink, URI.parse(hyperlink.to_s) }
  end
end

# File lib/hpricot/traverse.rb, line 628
def each_uri(base_uri=nil) # :yields: URI
  each_hyperlink_uri(base_uri) {|hyperlink, uri| yield uri }
end

# File lib/hpricot/traverse.rb, line 719
def filter(&block)
  subst = {}
  each_child_with_index {|descendant, i|
    if yield descendant
      if descendant.elem?
        subst[i] = descendant.filter(&block)
      else
        subst[i] = descendant
      end
    else
      subst[i] = nil
    end
  }
  to_node.subst_subnode(subst)
end

# File lib/hpricot/traverse.rb, line 512
def find_element(*names)
  traverse_element(*names) {|e| return e }
  nil
end

# File lib/hpricot/traverse.rb, line 435
def following_siblings()
  sibs = parent.containers
  si = sibs.index(self) + 1
  return Elements[*sibs[si...sibs.length]]
end

# File lib/hpricot/traverse.rb, line 522
def get_element_by_id(id)
  traverse_all_element do |ele|
      if ele.elem? and eid = ele.get_attribute('id')
          return ele if eid.to_s == id
      end
  end
  nil
end

# File lib/hpricot/traverse.rb, line 531
def get_elements_by_tag_name(*a)
  list = Elements[]
  a.delete("*")
  traverse_element(*a.map { |tag| [tag, "{http://www.w3.org/1999/xhtml}#{tag}"] }.flatten) do |e|
    list << e if e.elem?
  end
  list
end

# File lib/hpricot/traverse.rb, line 486
def insert_after(nodes, ele)
  case nodes
  when Array
    nodes.reverse_each { |n| insert_after(n, ele) }
  else
    reparent nodes
    idx = children.index(ele)
    children[idx ? idx + 1 : children.length, 0] = nodes
  end
end

# File lib/hpricot/traverse.rb, line 474
def insert_before(nodes, ele)
  case nodes
  when Array
    nodes.each { |n| insert_before(n, ele) }
  else
    reparent nodes
    children[children.index(ele) || 0, 0] = nodes
  end
end

# File lib/hpricot/traverse.rb, line 411
def next_sibling
  sib = parent.containers
  sib[sib.index(self) + 1] if parent
end

# File lib/hpricot/traverse.rb, line 427
def preceding_siblings()
  sibs = parent.containers
  si = sibs.index(self)
  return Elements[*sibs[0...si]]
end

# File lib/hpricot/traverse.rb, line 419
def previous_sibling
  sib = parent.containers
  x = sib.index(self) - 1
  sib[x] if sib and x >= 0
end

# File lib/hpricot/traverse.rb, line 467
def replace_child(old, new)
  reparent new
  children[children.index(old), 1] = [*new]
end

# File lib/hpricot/traverse.rb, line 452
def siblings_at(*pos)
  sib = parent.containers
  i, si = 0, sib.index(self)
  Elements[*
    sib.select do |x|
      sel = case i - si when *pos
              true
            end
      i += 1
      sel
    end
  ]
end

# File lib/hpricot/traverse.rb, line 688
def traverse_text_internal(&block)
  each_child {|c| c.traverse_text_internal(&block) }
end