module ActionDispatch::Http::URL

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

def host
  raw_host_with_port.sub(/:\d+$/, '')
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 port
  if raw_host_with_port =~ /:(\d+)$/
    $1.to_i
  else
    standard_port
  end
end

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

def 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 request_uri
  ActiveSupport::Deprecation.warn "Using #request_uri is deprecated. Use fullpath instead.", caller
  fullpath
end

def scheme
  ssl? ? 'https' : 'http'
end

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

def ssl?
  @env['HTTPS'] == 'on' || @env['HTTP_X_FORWARDED_PROTO'] == 'https'
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 = 1)
  subdomains(tld_length).join('.')
end

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

def url
  protocol + host_with_port + fullpath
end