module Excon::Utils

def binary_encode(string)
  if FORCE_ENC && string.encoding != Encoding::ASCII_8BIT
    string.force_encoding('BINARY')
  end
end

def connection_uri(datum = @data)
  unless datum
    raise ArgumentError, '`datum` must be given unless called on a Connection'
  end
  if datum[:scheme] == UNIX
    "#{datum[:scheme]}://#{datum[:socket]}"
  else
    "#{datum[:scheme]}://#{datum[:host]}#{port_string(datum)}"
  end
end

def escape_uri(str)
  str = str.dup
  binary_encode(str)
  str.gsub(UNESCAPED) { "%%%02X" % $1[0].ord }
end

def port_string(datum)
  if datum[:port].nil? || (datum[:omit_default_port] && ((datum[:scheme].casecmp('http') == 0 && datum[:port] == 80) || (datum[:scheme].casecmp('https') == 0 && datum[:port] == 443)))
    ''
  else
    ':' + datum[:port].to_s
  end
end

def query_string(datum)
  str = String.new
  case datum[:query]
  when String
    str << '?' << datum[:query]
  when Hash
    str << '?'
    datum[:query].sort_by {|k,_| k.to_s }.each do |key, values|
      key = CGI.escape(key.to_s)
      if values.nil?
        str << key << '&'
      else
        [values].flatten.each do |value|
          str << key << '=' << CGI.escape(value.to_s) << '&'
        end
      end
    end
    str.chop! # remove trailing '&'
  end
  str
end

def redact(datum)
  datum = datum.dup
  if datum.has_key?(:headers) && datum[:headers].has_key?('Authorization')
    datum[:headers] = datum[:headers].dup
    datum[:headers]['Authorization'] = REDACTED
  end
  if datum.has_key?(:password)
    datum[:password] = REDACTED
  end
  datum
end

def request_uri(datum)
  connection_uri(datum) + datum[:path] + query_string(datum)
end

def split_header_value(str)
  return [] if str.nil?
  str = str.dup.strip
  binary_encode(str)
  str.scan(%r'\G((?:"(?:\\.|[^"])+?"|[^",]+)+)
                (?:,\s*|\Z)'xn).flatten
end

def unescape_form(str)
  str = str.dup
  binary_encode(str)
  str.gsub!(/\+/, ' ')
  str.gsub(ESCAPED) { $1.hex.chr }
end

def unescape_uri(str)
  str = str.dup
  binary_encode(str)
  str.gsub(ESCAPED) { $1.hex.chr }
end