module Rails::Dom::Testing::Assertions::SelectorAssertions

def assert_select_jquery(*args, &block)
  jquery_method = args.first.is_a?(Symbol) ? args.shift : nil
  jquery_opt    = args.first.is_a?(Symbol) ? args.shift : nil
  id            = args.first.is_a?(String) ? escape_id(args.shift) : nil
  target_pattern   = "['\"]#{id || '.*'}['\"]"
  method_pattern   = "#{jquery_method || '\\w+'}"
  argument_pattern = jquery_opt ? "['\"]#{jquery_opt}['\"].*" : PATTERN_HTML
  # $("#id").show('blind', 1000);
  # $("#id").html("<div>something</div>");
  # $("#id").replaceWith("<div>something</div>");
  target_as_receiver_pattern = SKELETAL_PATTERN % [target_pattern, method_pattern, argument_pattern]
  # $("<div>something</div>").appendTo("#id");
  # $("<div>something</div>").prependTo("#id");
  target_as_argument_pattern = SKELETAL_PATTERN % [argument_pattern, method_pattern, target_pattern]
  # $("#id").remove();
  # $("#id").hide();
  argumentless_pattern = SKELETAL_PATTERN % [target_pattern, method_pattern, '']
  patterns = [target_as_receiver_pattern, target_as_argument_pattern]
  patterns << argumentless_pattern unless jquery_opt
  matched_pattern = nil
  patterns.each do |pattern|
    if response.body.match(Regexp.new(pattern))
      matched_pattern = pattern
      break
    end
  end
  unless matched_pattern
    opts = [jquery_method, jquery_opt, id].compact
    flunk "No JQuery call matches #{opts.inspect}"
  end
  if block_given?
    @selected ||= nil
    fragments = Nokogiri::HTML::Document.new.fragment
    if matched_pattern
      response.body.scan(Regexp.new(matched_pattern)).each do |match|
        flunk 'This function can\'t have HTML argument' if match.is_a?(String)
        doc = Nokogiri::HTML::DocumentFragment.parse(unescape_js(match.first))
        doc.children.each do |child|
          fragments << child if child.element?
        end
      end
    end
    begin
      in_scope, @selected = @selected, fragments
      yield
    ensure
      @selected = in_scope
    end
  end
end

def escape_id(selector)
  return unless selector
  id = selector.gsub('[', '\[')
  id.gsub!(']', '\]')
  id.gsub!('*', '\*')
  id.gsub!('(', '\(')
  id.gsub!(')', '\)')
  id.gsub!('.', '\.')
  id.gsub!('|', '\|')
  id.gsub!('^', '\^')
  id.gsub!('$', '\$')
  id.gsub!('+', "\\\\+")
  id.gsub!(',', '\,')
  id
end

def unescape_js(js_string)
  # js encodes double quotes and line breaks.
  unescaped= js_string.gsub('\"', '"')
  unescaped.gsub!('\\\'', "'")
  unescaped.gsub!(/\\\//, '/')
  unescaped.gsub!('\n', "\n")
  unescaped.gsub!('\076', '>')
  unescaped.gsub!('\074', '<')
  unescaped.gsub!(/\\\$/, '$')
  unescaped.gsub!(/\\`/, '`')
  # js encodes non-ascii characters.
  unescaped.gsub!(PATTERN_UNICODE_ESCAPED_CHAR) {|u| [$1.hex].pack('U*')}
  unescaped
end