Return empty list if no "links" supplied in api response. Add tests.
[arvados.git] / apps / workbench / app / models / arvados_resource_list.rb
1 class ArvadosResourceList
2   include Enumerable
3
4   def initialize resource_class=nil
5     @resource_class = resource_class
6   end
7
8   def eager(bool=true)
9     @eager = bool
10     self
11   end
12
13   def limit(max_results)
14     @limit = max_results
15     self
16   end
17
18   def offset(skip)
19     @offset = skip
20     self
21   end
22
23   def order(orderby_spec)
24     @orderby_spec = orderby_spec
25     self
26   end
27
28   def filter _filters
29     @filters ||= []
30     @filters += _filters
31     self
32   end
33
34   def where(cond)
35     cond = cond.dup
36     cond.keys.each do |uuid_key|
37       if cond[uuid_key] and (cond[uuid_key].is_a? Array or
38                              cond[uuid_key].is_a? ArvadosBase)
39         # Coerce cond[uuid_key] to an array of uuid strings.  This
40         # allows caller the convenience of passing an array of real
41         # objects and uuids in cond[uuid_key].
42         if !cond[uuid_key].is_a? Array
43           cond[uuid_key] = [cond[uuid_key]]
44         end
45         cond[uuid_key] = cond[uuid_key].collect do |item|
46           if item.is_a? ArvadosBase
47             item.uuid
48           else
49             item
50           end
51         end
52       end
53     end
54     cond.keys.select { |x| x.match /_kind$/ }.each do |kind_key|
55       if cond[kind_key].is_a? Class
56         cond = cond.merge({ kind_key => 'arvados#' + $arvados_api_client.class_kind(cond[kind_key]) })
57       end
58     end
59     api_params = {
60       _method: 'GET',
61       where: cond
62     }
63     api_params[:eager] = '1' if @eager
64     api_params[:limit] = @limit if @limit
65     api_params[:offset] = @offset if @offset
66     api_params[:order] = @orderby_spec if @orderby_spec
67     api_params[:filters] = @filters if @filters
68     res = $arvados_api_client.api @resource_class, '', api_params
69     @results = $arvados_api_client.unpack_api_response res
70     self
71   end
72
73   def results
74     self.where({}) if !@results
75     @results
76   end
77
78   def results=(r)
79     @results = r
80   end
81
82   def all
83     where({})
84   end
85
86   def each(&block)
87     results.each do |m|
88       block.call m
89     end
90     self
91   end
92
93   def first
94     results.first
95   end
96
97   def last
98     results.last
99   end
100
101   def [](*x)
102     results.send('[]', *x)
103   end
104
105   def |(x)
106     if x.is_a? Hash
107       self.to_hash | x
108     else
109       results | x.to_ary
110     end
111   end
112
113   def to_ary
114     results
115   end
116
117   def to_hash
118     Hash[results.collect { |x| [x.uuid, x] }]
119   end
120
121   def empty?
122     results.empty?
123   end
124
125   def items_available
126     results.items_available if results.respond_to? :items_available
127   end
128
129   def result_limit
130     results.limit if results.respond_to? :limit
131   end
132
133   def result_offset
134     results.offset if results.respond_to? :offset
135   end
136
137   def result_links
138     results.links if results.respond_to? :links
139   end
140
141   def links_for item_or_uuid, link_class=nil
142     return [] if !result_links
143     unless @links_for_uuid
144       @links_for_uuid = {}
145       result_links.each do |link|
146         if link.respond_to? :head_uuid
147           @links_for_uuid[link.head_uuid] ||= []
148           @links_for_uuid[link.head_uuid] << link
149         end
150       end
151     end
152     if item_or_uuid.respond_to? :uuid
153       uuid = item_or_uuid.uuid
154     else
155       uuid = item_or_uuid
156     end
157     (@links_for_uuid[uuid] || []).select do |link|
158       link.link_class == link_class
159     end
160   end
161
162   # Note: this arbitrarily chooses one of (possibly) multiple names.
163   def name_for item_or_uuid
164     links_for(item_or_uuid, 'name').first.andand.name
165   end
166
167 end