1 class ArvadosBase < ActiveRecord::Base
2 self.abstract_class = true
3 attr_accessor :attribute_sortkey
5 def self.arvados_api_client
6 ArvadosApiClient.new_or_current
10 ArvadosApiClient.new_or_current
13 def self.uuid_infix_object_kind
14 @@uuid_infix_object_kind ||=
17 arvados_api_client.discovery[:schemas].each do |name, schema|
18 if schema[:uuidPrefix]
19 infix_kind[schema[:uuidPrefix]] =
20 'arvados#' + name.to_s.camelcase(:lower)
24 # Recognize obsolete types.
26 merge('mxsvm' => 'arvados#pipelineTemplate', # Pipeline
27 'uo14g' => 'arvados#pipelineInstance', # PipelineInvocation
28 'ldvyl' => 'arvados#group') # Project
32 def initialize raw_params={}
34 super self.class.permit_attribute_params(raw_params)
36 logger.debug raw_params
37 logger.debug self.class.permit_attribute_params(raw_params).inspect
38 logger.debug self.class.attribute_info.inspect
41 @attribute_sortkey ||= {
44 'owner_uuid' => '002',
45 'event_type' => '100',
46 'link_class' => '100',
47 'group_class' => '100',
50 'object_uuid' => '102',
52 'description' => '104',
53 'properties' => '150',
55 'created_at' => '200',
56 'modified_at' => '201',
57 'modified_by_user_uuid' => '202',
58 'modified_by_client_uuid' => '203',
64 return @columns unless @columns.nil?
66 @attribute_info ||= {}
67 schema = arvados_api_client.discovery[:schemas][self.to_s.to_sym]
68 return @columns if schema.nil?
69 schema[:properties].each do |k, coldef|
74 if coldef[:type] == coldef[:type].downcase
75 # boolean, integer, etc.
76 @columns << column(k, coldef[:type].to_sym)
79 @columns << column(k, :text)
80 serialize k, coldef[:type].constantize
82 @attribute_info[k] = coldef
88 def self.column(name, sql_type = nil, default = nil, null = true)
89 ActiveRecord::ConnectionAdapters::Column.new(name.to_s, default, sql_type.to_s, null)
92 def self.attribute_info
97 def self.find(uuid, opts={})
98 if uuid.class != String or uuid.length < 27 then
99 raise 'argument to find() must be a uuid string. Acceptable formats: warehouse locator or string with format xxxxx-xxxxx-xxxxxxxxxxxxxxx'
102 if self == ArvadosBase
103 # Determine type from uuid and defer to the appropriate subclass.
104 return resource_class_for_uuid(uuid).find(uuid, opts)
107 # Only do one lookup on the API side per {class, uuid, workbench
108 # request} unless {cache: false} is given via opts.
109 cache_key = "request_#{Thread.current.object_id}_#{self.to_s}_#{uuid}"
110 if opts[:cache] == false
111 Rails.cache.write cache_key, arvados_api_client.api(self, '/' + uuid)
113 hash = Rails.cache.fetch cache_key do
114 arvados_api_client.api(self, '/' + uuid)
116 new.private_reload(hash)
119 def self.order(*args)
120 ArvadosResourceList.new(self).order(*args)
123 def self.filter(*args)
124 ArvadosResourceList.new(self).filter(*args)
127 def self.where(*args)
128 ArvadosResourceList.new(self).where(*args)
131 def self.limit(*args)
132 ArvadosResourceList.new(self).limit(*args)
135 def self.eager(*args)
136 ArvadosResourceList.new(self).eager(*args)
140 ArvadosResourceList.new(self).all(*args)
143 def self.permit_attribute_params raw_params
144 # strong_parameters does not provide security in Workbench: anyone
145 # who can get this far can just as well do a call directly to our
146 # database (Arvados) with the same credentials we use.
147 ActionController::Parameters.new(raw_params).permit!
150 def self.create raw_params={}
151 logger.error permit_attribute_params(raw_params).inspect
152 super(permit_attribute_params(raw_params))
155 def update_attributes raw_params={}
156 super(self.class.permit_attribute_params(raw_params))
161 self.class.columns.each do |col|
162 obdata[col.name.to_sym] = self.send(col.name.to_sym)
165 postdata = { self.class.to_s.underscore => obdata }
167 postdata['_method'] = 'PUT'
169 resp = arvados_api_client.api(self.class, '/' + uuid, postdata)
171 resp = arvados_api_client.api(self.class, '', postdata)
173 return false if !resp[:etag] || !resp[:uuid]
175 # set read-only non-database attributes
179 # attributes can be modified during "save" -- we should update our copies
180 resp.keys.each do |attr|
181 if self.respond_to? "#{attr}=".to_sym
182 self.send(attr.to_s + '=', resp[attr.to_sym])
192 self.save or raise Exception.new("Save failed")
197 postdata = { '_method' => 'DELETE' }
198 resp = arvados_api_client.api(self.class, '/' + uuid, postdata)
199 resp[:etag] && resp[:uuid] && resp
207 o.merge!(args.pop) if args[-1].is_a? Hash
208 o[:link_class] ||= args.shift
209 o[:name] ||= args.shift
210 o[:tail_uuid] = self.uuid
212 return all_links.select do |m|
217 if (v.respond_to?(:uuid) ? v.uuid : v.to_s) != (test_v.respond_to?(:uuid) ? test_v.uuid : test_v.to_s)
225 @links = arvados_api_client.api Link, '', { _method: 'GET', where: o, eager: true }
226 @links = arvados_api_client.unpack_api_response(@links)
230 return @all_links if @all_links
231 res = arvados_api_client.api Link, '', {
234 tail_kind: self.kind,
239 @all_links = arvados_api_client.unpack_api_response(res)
243 private_reload(self.uuid)
246 def private_reload(uuid_or_hash)
247 raise "No such object" if !uuid_or_hash
248 if uuid_or_hash.is_a? Hash
251 hash = arvados_api_client.api(self.class, '/' + uuid_or_hash)
254 if self.respond_to?(k.to_s + '=')
255 self.send(k.to_s + '=', v)
257 # When ArvadosApiClient#schema starts telling us what to expect
258 # in API responses (not just the server side database
259 # columns), this sort of awfulness can be avoided:
260 self.instance_variable_set('@' + k.to_s, v)
261 if !self.respond_to? k
262 singleton = class << self; self end
263 singleton.send :define_method, k, lambda { instance_variable_get('@' + k.to_s) }
280 def attributes_for_display
281 self.attributes.reject { |k,v|
282 attribute_sortkey.has_key?(k) and !attribute_sortkey[k]
284 attribute_sortkey[k] or k
288 def class_for_display
297 (current_user and current_user.is_active and
298 (current_user.is_admin or
299 current_user.uuid == self.owner_uuid or
303 def attribute_editable?(attr)
304 if "created_at modified_at modified_by_user_uuid modified_by_client_uuid updated_at".index(attr.to_s)
306 elsif not (current_user.andand.is_active)
308 elsif "uuid owner_uuid".index(attr.to_s) or current_user.is_admin
309 current_user.is_admin
311 current_user.uuid == self.owner_uuid or
312 current_user.uuid == self.uuid or
317 def self.resource_class_for_uuid(uuid, opts={})
318 if uuid.is_a? ArvadosBase
321 unless uuid.is_a? String
324 if opts[:class].is_a? Class
327 if uuid.match /^[0-9a-f]{32}(\+[^,]+)*(,[0-9a-f]{32}(\+[^,]+)*)*$/
331 uuid.match /^[0-9a-z]{5}-([0-9a-z]{5})-[0-9a-z]{15}$/ do |re|
332 resource_class ||= arvados_api_client.
333 kind_class(self.uuid_infix_object_kind[re[1]])
335 if opts[:referring_object] and
336 opts[:referring_attr] and
337 opts[:referring_attr].match /_uuid$/
338 resource_class ||= arvados_api_client.
339 kind_class(opts[:referring_object].
340 attributes[opts[:referring_attr].
341 sub(/_uuid$/, '_kind')])
346 def friendly_link_name
347 (name if self.respond_to? :name) || uuid
351 self.class_for_display
366 def self.current_user
367 Thread.current[:user] ||= User.current if Thread.current[:arvados_api_token]
368 Thread.current[:user]
371 self.class.current_user