class Crack::JSON

def self.convert_json_to_yaml(json) #:nodoc:
  json = String.new(json) #can't modify a frozen string
  scanner, quoting, marks, pos, date_starts, date_ends = StringScanner.new(json), false, [], nil, [], []
  while scanner.scan_until(/(\\['"]|['":,\/\\]|\\.)/)
    case char = scanner[1]
    when '"', "'"
      if !quoting
        quoting = char
        pos = scanner.pos
      elsif quoting == char
        if json[pos..scanner.pos-2] =~ DATE_REGEX
          # found a date, track the exact positions of the quotes so we can remove them later.
          # oh, and increment them for each current mark, each one is an extra padded space that bumps
          # the position in the final YAML output
          total_marks = marks.size
          date_starts << pos+total_marks
          date_ends << scanner.pos+total_marks
        end
        quoting = false
      end
    when "/"
      if !quoting
        json[scanner.pos - 1] = "!ruby/regexp /"
        scanner.pos += 13
        scanner.scan_until(/\/[mix]*/)
      end
    when ":",","
      marks << scanner.pos - 1 unless quoting
    when "\\"
      scanner.skip(/\\/)
    end
  end
  if marks.empty?
    json.gsub(/\\\//, '/')
  else
    left_pos  = marks.clone.unshift(-1)
    right_pos = marks << json.length
    output    = []
    left_pos.each_with_index do |left, i|
      output << json[left.succ..right_pos[i]]
    end
    output = output * " "
    format_dates(output, date_starts, date_ends)
    output.gsub!(/\\\//, '/')
    output
  end
end

def self.format_dates(output, date_starts, date_ends)
  if YAML.constants.include?('Syck')
    (date_starts + date_ends).each { |i| output[i-1] = ' ' }
  else
    extra_chars_to_be_added = 0
    timestamp_marker = '!!timestamp '
    timestamp_marker_size = timestamp_marker.size
    date_starts.each do |i|
      output[i-2+extra_chars_to_be_added] = timestamp_marker
      extra_chars_to_be_added += timestamp_marker_size - 1
    end
  end
end

def self.parse(json)
  yaml = unescape(convert_json_to_yaml(json))
  YAML.safe_load(yaml, permitted_classes: [Regexp, Date, Time])
rescue *parser_exceptions
  raise ParseError, "Invalid JSON string"
rescue Psych::DisallowedClass
  yaml
end

def self.parse(json)
  yaml = unescape(convert_json_to_yaml(json))
  YAML.safe_load(yaml, [Regexp, Date, Time])
rescue *parser_exceptions
  raise ParseError, "Invalid JSON string"
rescue Psych::DisallowedClass
  yaml
end

def self.parser_exceptions
  @parser_exceptions ||= [ArgumentError, Psych::SyntaxError]
end

def self.unescape(str)
  # Force the encoding to be UTF-8 so we can perform regular expressions
  # on 1.9.2 without blowing up.
  # see http://stackoverflow.com/questions/1224204/ruby-mechanize-getting-force-encoding-exception for a similar issue
  str.force_encoding('UTF-8') if defined?(Encoding) && str.respond_to?(:force_encoding)
  str.gsub(/\\u0000/, "").gsub(/\\[u|U]([0-9a-fA-F]{4})/) { [$1.hex].pack("U") }
end