module ActionDispatch::Http::URL

def add_trailing_slash(path)
  # includes querysting
  if path.include?('?')
    path.sub!(/\?/, '/\&')
  # does not have a .format
  elsif !path.include?(".")
    path.sub!(/[^\/]\z|\A\z/, '\&/')
  end
  path
end

def build_host_url(options)
  if options[:host].blank? && options[:only_path].blank?
    raise ArgumentError, 'Missing host to link to! Please provide the :host parameter, set default_url_options[:host], or set :only_path to true'
  end
  result = ""
  unless options[:only_path]
    if match = options[:host].match(HOST_REGEXP)
      options[:protocol] ||= match[1] unless options[:protocol] == false
      options[:host]       = match[2]
      options[:port]       = match[3] unless options.key?(:port)
    end
    options[:protocol] = normalize_protocol(options)
    options[:host]     = normalize_host(options)
    options[:port]     = normalize_port(options)
    result << options[:protocol]
    result << rewrite_authentication(options)
    result << options[:host]
    result << ":#{options[:port]}" if options[:port]
  end
  result
end

def domain(tld_length = @@tld_length)
  ActionDispatch::Http::URL.extract_domain(host, tld_length)
end

def extract_domain(host, tld_length = @@tld_length)
  host.split('.').last(1 + tld_length).join('.') if named_host?(host)
end

def extract_subdomain(host, tld_length = @@tld_length)
  extract_subdomains(host, tld_length).join('.')
end

def extract_subdomains(host, tld_length = @@tld_length)
  if named_host?(host)
    parts = host.split('.')
    parts[0..-(tld_length + 2)]
  else
    []
  end
end

def host
  raw_host_with_port.sub(/:\d+$/, '')
end

def host_with_port
  "#{host}#{port_string}"
end

def initialize(env)
  super
  @protocol = nil
  @port     = nil
end

def named_host?(host)
  host && IP_HOST_REGEXP !~ host
end

def normalize_host(options)
  return options[:host] if !named_host?(options[:host]) || same_host?(options)
  tld_length = options[:tld_length] || @@tld_length
  host = ""
  if options[:subdomain] == true || !options.key?(:subdomain)
    host << extract_subdomain(options[:host], tld_length).to_param
  elsif options[:subdomain].present?
    host << options[:subdomain].to_param
  end
  host << "." unless host.empty?
  host << (options[:domain] || extract_domain(options[:host], tld_length))
  host
end

def normalize_port(options)
  return nil if options[:port].nil? || options[:port] == false
  case options[:protocol]
  when "//"
    options[:port]
  when "https://"
    options[:port].to_i == 443 ? nil : options[:port]
  else
    options[:port].to_i == 80 ? nil : options[:port]
  end
end

def normalize_protocol(options)
  case options[:protocol]
  when nil
    "http://"
  when false, "//"
    "//"
  when PROTOCOL_REGEXP
    "#{$1}://"
  else
    raise ArgumentError, "Invalid :protocol option: #{options[:protocol].inspect}"
  end
end

def optional_port
  standard_port? ? nil : port
end

def port
  @port ||= begin
    if raw_host_with_port =~ /:(\d+)$/
      $1.to_i
    else
      standard_port
    end
  end
end

def port_string
  standard_port? ? '' : ":#{port}"
end

def protocol
  @protocol ||= ssl? ? 'https://' : 'http://'
end

def raw_host_with_port
  if forwarded = env["HTTP_X_FORWARDED_HOST"].presence
    forwarded.split(/,\s?/).last
  else
    env['HTTP_HOST'] || "#{env['SERVER_NAME'] || env['SERVER_ADDR']}:#{env['SERVER_PORT']}"
  end
end

def rewrite_authentication(options)
  if options[:user] && options[:password]
    "#{Rack::Utils.escape(options[:user])}:#{Rack::Utils.escape(options[:password])}@"
  else
    ""
  end
end

def same_host?(options)
  (options[:subdomain] == true || !options.key?(:subdomain)) && options[:domain].nil?
end

def server_port
  @env['SERVER_PORT'].to_i
end

def standard_port
  case protocol
    when 'https://' then 443
    else 80
  end
end

def standard_port?
  port == standard_port
end

def subdomain(tld_length = @@tld_length)
  ActionDispatch::Http::URL.extract_subdomain(host, tld_length)
end

def subdomains(tld_length = @@tld_length)
  ActionDispatch::Http::URL.extract_subdomains(host, tld_length)
end

def url
  protocol + host_with_port + fullpath
end

def url_for(options = {})
  options = options.dup
  path  = options.delete(:script_name).to_s.chomp("/")
  path << options.delete(:path).to_s
  add_trailing_slash(path) if options[:trailing_slash]
  params = options[:params].is_a?(Hash) ? options[:params] : options.slice(:params)
  params.reject! { |_,v| v.to_param.nil? }
  result = build_host_url(options)
  result << path
  result << "?#{params.to_query}" unless params.empty?
  result << "##{Journey::Router::Utils.escape_fragment(options[:anchor].to_param.to_s)}" if options[:anchor]
  result
end