module ActionDispatch::Http::URL

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

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

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

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

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

def host_or_subdomain_and_domain(options)
  return options[:host] if !named_host?(options[:host]) || (options[:subdomain].nil? && options[:domain].nil?)
  tld_length = options[:tld_length] || @@tld_length
  host = ""
  unless options[:subdomain] == false
    host << (options[:subdomain] || extract_subdomain(options[:host], tld_length)).to_param
    host << "."
  end
  host << (options[:domain] || extract_domain(options[:host], tld_length))
  host
end

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

def named_host?(host)
  !(host.nil? || /\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$/.match(host))
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"]
    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 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)
  subdomains(tld_length).join(".")
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 = {})
  unless options[:host].present? || options[:only_path].present?
    raise ArgumentError, 'Missing host to link to! Please provide the :host parameter, set default_url_options[:host], or set :only_path to true'
  end
  rewritten_url = ""
  unless options[:only_path]
    unless options[:protocol] == false
      rewritten_url << (options[:protocol] || "http")
      rewritten_url << ":" unless rewritten_url.match(%r{:|//})
    end
    rewritten_url << "//" unless rewritten_url.match("//")
    rewritten_url << rewrite_authentication(options)
    rewritten_url << host_or_subdomain_and_domain(options)
    rewritten_url << ":#{options.delete(:port)}" if options[:port]
  end
  path = options.delete(:path) || ''
  params = options[:params] || {}
  params.reject! {|k,v| v.to_param.nil? }
  if options[:trailing_slash] && !path.ends_with?('/')
    rewritten_url << path.sub(/(\?|\z)/) { "/" + $& }
  else
    rewritten_url << path
  end
  rewritten_url << "?#{params.to_query}" unless params.empty?
  rewritten_url << "##{Journey::Router::Utils.escape_fragment(options[:anchor].to_param.to_s)}" if options[:anchor]
  rewritten_url
end