module ActionDispatch::Http::URL

def add_anchor(path, anchor)
  if anchor
    path << "##{Journey::Router::Utils.escape_fragment(anchor.to_param)}"
  end
end

def add_params(path, params)
  params = { params: params } unless params.is_a?(Hash)
  params.reject! { |_, v| v.to_param.nil? }
  query = params.to_query
  path << "?#{query}" unless query.empty?
end

def build_host_url(host, port, protocol, options, path)
  if match = host.match(HOST_REGEXP)
    protocol ||= match[1] unless protocol == false
    host       = match[2]
    port       = match[3] unless options.key? :port
  end
  protocol = normalize_protocol protocol
  host     = normalize_host(host, options)
  result = protocol.dup
  if options[:user] && options[:password]
    result << "#{Rack::Utils.escape(options[:user])}:#{Rack::Utils.escape(options[:password])}@"
  end
  result << host
  normalize_port(port, protocol) { |normalized_port|
    result << ":#{normalized_port}"
  }
  result.concat path
end

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

def extract_domain(host, tld_length)
  extract_domain_from(host, tld_length) if named_host?(host)
end

def extract_domain_from(host, tld_length)
  host.split(".").last(1 + tld_length).join(".")
end

def extract_subdomain(host, tld_length)
  extract_subdomains(host, tld_length).join(".")
end

def extract_subdomains(host, tld_length)
  if named_host?(host)
    extract_subdomains_from(host, tld_length)
  else
    []
  end
end

def extract_subdomains_from(host, tld_length)
  parts = host.split(".")
  parts[0..-(tld_length + 2)]
end

def full_url_for(options)
  host     = options[:host]
  protocol = options[:protocol]
  port     = options[:port]
  unless host
    raise ArgumentError, "Missing host to link to! Please provide the :host parameter, set default_url_options[:host], or set :only_path to true"
  end
  build_host_url(host, port, protocol, options, path_for(options))
end

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

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

def initialize
  super
  @protocol = nil
  @port     = nil
end

def named_host?(host)
  !IP_HOST_REGEXP.match?(host)
end

def normalize_host(_host, options)
  return _host unless named_host?(_host)
  tld_length = options[:tld_length] || @@tld_length
  subdomain  = options.fetch :subdomain, true
  domain     = options[:domain]
  host = +""
  if subdomain == true
    return _host if domain.nil?
    host << extract_subdomains_from(_host, tld_length).join(".")
  elsif subdomain
    host << subdomain.to_param
  end
  host << "." unless host.empty?
  host << (domain || extract_domain_from(_host, tld_length))
  host
end

def normalize_port(port, protocol)
  return unless port
  case protocol
  when "//" then yield port
  when "https://"
    yield port unless port.to_i == 443
  else
    yield port unless port.to_i == 80
  end
end

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

def optional_port
  standard_port? ? nil : port
end

def path_for(options)
  path = options[:script_name].to_s.chomp("/")
  path << options[:path] if options.key?(:path)
  path = "/" if options[:trailing_slash] && path.blank?
  add_params(path, options[:params]) if options.key?(:params)
  add_anchor(path, options[:anchor]) if options.key?(:anchor)
  path
end

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

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

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

def raw_host_with_port
  if forwarded = x_forwarded_host.presence
    forwarded.split(/,\s?/).last
  else
    get_header("HTTP_HOST") || "#{server_name}:#{get_header('SERVER_PORT')}"
  end
end

def server_port
  get_header("SERVER_PORT").to_i
end

def standard_port
  if "https://" == protocol
    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)
  if options[:only_path]
    path_for options
  else
    full_url_for options
  end
end