Fix type-checking for URL.

This commit is contained in:
Markus Reiter 2020-09-20 06:03:36 +02:00
parent ec5eb56a72
commit 6f7b7cc699

View File

@ -1,32 +1,65 @@
# typed: strict
# frozen_string_literal: true # frozen_string_literal: true
require "utils/sorbet"
# Class corresponding to the `url` stanza. # Class corresponding to the `url` stanza.
# #
# @api private # @api private
class URL class URL
ATTRIBUTES = [ extend T::Sig
attr_reader :uri, :specs,
:using, :using,
:tag, :branch, :revisions, :revision, :tag, :branch, :revisions, :revision,
:trust_cert, :cookies, :referer, :user_agent, :trust_cert, :cookies, :referer, :user_agent,
:data :data
].freeze
private_constant :ATTRIBUTES
attr_reader :uri, :specs, *ATTRIBUTES
extend Forwardable extend Forwardable
def_delegators :uri, :path, :scheme, :to_s def_delegators :uri, :path, :scheme, :to_s
def initialize(uri, **options) sig do
@uri = URI(uri) params(
@user_agent = :default uri: T.any(URI::Generic, String),
using: T.nilable(Symbol),
ATTRIBUTES.each do |attribute| tag: T.nilable(String),
next unless options.key?(attribute) branch: T.nilable(String),
revisions: T.nilable(T::Array[String]),
instance_variable_set("@#{attribute}", options[attribute]) revision: T.nilable(String),
trust_cert: T.nilable(T::Boolean),
cookies: T.nilable(T::Hash[String, String]),
referer: T.nilable(T.any(URI::Generic, String)),
user_agent: T.nilable(T.any(Symbol, String)),
data: T.nilable(T::Hash[String, String]),
).returns(T.untyped)
end end
def initialize(
uri,
using: nil,
tag: nil,
branch: nil,
revisions: nil,
revision: nil,
trust_cert: nil,
cookies: nil,
referer: nil,
user_agent: nil,
data: nil
)
@uri = URI(uri)
@specs = options specs = {}
specs[:using] = @using = using
specs[:tag] = @tag = tag
specs[:branch] = @branch = branch
specs[:revisions] = @revisions = revisions
specs[:revision] = @revision = revision
specs[:trust_cert] = @trust_cert = trust_cert
specs[:cookies] = @cookies = cookies
specs[:referer] = @referer = referer
specs[:user_agent] = @user_agent = user_agent || :default
specs[:data] = @data = data
@specs = specs.compact
end end
end end