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     unless @links_for_uuid
 143       @links_for_uuid = {}
 144       results.links.each do |link|
 145         if link.respond_to? :head_uuid
 146           @links_for_uuid[link.head_uuid] ||= []
 147           @links_for_uuid[link.head_uuid] << link
 148         end
 149       end
 150     end
 151     if item_or_uuid.respond_to? :uuid
 152       uuid = item_or_uuid.uuid
 153     else
 154       uuid = item_or_uuid
 155     end
 156     (@links_for_uuid[uuid] || []).select do |link|
 157       link.link_class == link_class
 158     end
 159   end
 160
 161   # Note: this arbitrarily chooses one of (possibly) multiple names.
 162   def name_for item_or_uuid
 163     links_for(item_or_uuid, 'name').first.andand.name
 164   end
 165
 166 end