class ActionText::Fragment

def find_all(selector)
  source.css(selector)
end

def from_html(html)
  new(ActionText::HtmlConversion.fragment_for_html(html.to_s.strip))
end

def initialize(source)
  @source = source
end

def replace(selector)
  update do |source|
    source.css(selector).each do |node|
      replacement_node = yield(node)
      node.replace(replacement_node.to_s) if node != replacement_node
    end
  end
end

def to_html
  @html ||= HtmlConversion.node_to_html(source)
end

def to_plain_text
  @plain_text ||= PlainTextConversion.node_to_plain_text(source)
end

def to_s
  to_html
end

def update
  yield source = self.source.dup
  self.class.new(source)
end

def wrap(fragment_or_html)
  case fragment_or_html
  when self
    fragment_or_html
  when Nokogiri::XML::DocumentFragment # base class for all fragments
    new(fragment_or_html)
  else
    from_html(fragment_or_html)
  end
end