-
Notifications
You must be signed in to change notification settings - Fork 5.3k
/
source_set.rb
48 lines (37 loc) · 898 Bytes
/
source_set.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
##
# The SourceSet chooses the best available method to query a remote index.
#
# Kind off like BestSet but filters the sources for gems
class Gem::Resolver::SourceSet < Gem::Resolver::Set
##
# Creates a SourceSet for the given +sources+ or Gem::sources if none are
# specified. +sources+ must be a Gem::SourceList.
def initialize
super()
@links = {}
@sets = {}
end
def find_all req # :nodoc:
if set = get_set(req.dependency.name)
set.find_all req
else
[]
end
end
# potentially no-op
def prefetch reqs # :nodoc:
reqs.each do |req|
if set = get_set(req.dependency.name)
set.prefetch reqs
end
end
end
def add_source_gem name, source
@links[name] = source
end
private
def get_set(name)
link = @links[name]
@sets[link] ||= Gem::Source.new(link).dependency_resolver_set if link
end
end