apps/workbench/app/controllers/actions_controller.rb

   1 class ActionsController < ApplicationController
   2
   3   skip_filter :require_thread_api_token, only: [:report_issue_popup, :report_issue]
   4   skip_filter :check_user_agreements, only: [:report_issue_popup, :report_issue]
   5
   6   @@exposed_actions = {}
   7   def self.expose_action method, &block
   8     @@exposed_actions[method] = true
   9     define_method method, block
  10   end
  11
  12   def model_class
  13     ArvadosBase::resource_class_for_uuid(params[:uuid])
  14   end
  15
  16   def show
  17     @object = model_class.andand.find(params[:uuid])
  18     if @object.is_a? Link and
  19         @object.link_class == 'name' and
  20         ArvadosBase::resource_class_for_uuid(@object.head_uuid) == Collection
  21       redirect_to collection_path(id: @object.uuid)
  22     elsif @object
  23       redirect_to @object
  24     else
  25       raise ActiveRecord::RecordNotFound
  26     end
  27   end
  28
  29   def post
  30     params.keys.collect(&:to_sym).each do |param|
  31       if @@exposed_actions[param]
  32         return self.send(param)
  33       end
  34     end
  35     redirect_to :back
  36   end
  37
  38   expose_action :copy_selections_into_project do
  39     move_or_copy :copy
  40   end
  41
  42   expose_action :move_selections_into_project do
  43     move_or_copy :move
  44   end
  45
  46   def move_or_copy action
  47     uuids_to_add = params["selection"]
  48     uuids_to_add = [ uuids_to_add ] unless uuids_to_add.is_a? Array
  49     resource_classes = uuids_to_add.
  50       collect { |x| ArvadosBase::resource_class_for_uuid(x) }.
  51       uniq
  52     resource_classes.each do |resource_class|
  53       resource_class.filter([['uuid','in',uuids_to_add]]).each do |src|
  54         if resource_class == Collection and not Collection.attribute_info.include?(:name)
  55           dst = Link.new(owner_uuid: @object.uuid,
  56                          tail_uuid: @object.uuid,
  57                          head_uuid: src.uuid,
  58                          link_class: 'name',
  59                          name: src.uuid)
  60         else
  61           case action
  62           when :copy
  63             dst = src.dup
  64             if dst.respond_to? :'name='
  65               if dst.name
  66                 dst.name = "Copy of #{dst.name}"
  67               else
  68                 dst.name = "Copy of unnamed #{dst.class_for_display.downcase}"
  69               end
  70             end
  71             if resource_class == Collection
  72               dst.manifest_text = Collection.select([:manifest_text]).where(uuid: src.uuid).first.manifest_text
  73             end
  74           when :move
  75             dst = src
  76           else
  77             raise ArgumentError.new "Unsupported action #{action}"
  78           end
  79           dst.owner_uuid = @object.uuid
  80           dst.tail_uuid = @object.uuid if dst.class == Link
  81         end
  82         begin
  83           dst.save!
  84         rescue
  85           dst.name += " (#{Time.now.localtime})" if dst.respond_to? :name=
  86           dst.save!
  87         end
  88       end
  89     end
  90     if (resource_classes == [Collection] and
  91         @object.is_a? Group and
  92         @object.group_class == 'project')
  93       # In the common case where only collections are copied/moved
  94       # into a project, it's polite to land on the collections tab on
  95       # the destination project.
  96       redirect_to project_url(@object.uuid, anchor: 'Data_collections')
  97     else
  98       # Otherwise just land on the default (Description) tab.
  99       redirect_to @object
 100     end
 101   end
 102
 103   def arv_normalize mt, *opts
 104     r = ""
 105     env = Hash[ENV].
 106       merge({'ARVADOS_API_HOST' =>
 107               arvados_api_client.arvados_v1_base.
 108               sub(/\/arvados\/v1/, '').
 109               sub(/^https?:\/\//, ''),
 110               'ARVADOS_API_TOKEN' => 'x',
 111               'ARVADOS_API_HOST_INSECURE' =>
 112               Rails.configuration.arvados_insecure_https ? 'true' : 'false'
 113             })
 114     IO.popen([env, 'arv-normalize'] + opts, 'w+b') do |io|
 115       io.write mt
 116       io.close_write
 117       while buf = io.read(2**16)
 118         r += buf
 119       end
 120     end
 121     r
 122   end
 123
 124   expose_action :combine_selected_files_into_collection do
 125     uuids = []
 126     pdhs = []
 127     files = []
 128     params["selection"].each do |s|
 129       a = ArvadosBase::resource_class_for_uuid s
 130       if a == Link
 131         begin
 132           if (m = CollectionsHelper.match(Link.find(s).head_uuid))
 133             pdhs.append(m[1] + m[2])
 134             files.append(m)
 135           end
 136         rescue
 137         end
 138       elsif (m = CollectionsHelper.match(s))
 139         pdhs.append(m[1] + m[2])
 140         files.append(m)
 141       elsif (m = CollectionsHelper.match_uuid_with_optional_filepath(s))
 142         uuids.append(m[1])
 143         files.append(m)
 144       end
 145     end
 146
 147     pdhs = pdhs.uniq
 148     uuids = uuids.uniq
 149     chash = {}
 150
 151     Collection.select([:uuid, :manifest_text]).where(uuid: uuids).each do |c|
 152       chash[c.uuid] = c
 153     end
 154
 155     Collection.select([:portable_data_hash, :manifest_text]).where(portable_data_hash: pdhs).each do |c|
 156       chash[c.portable_data_hash] = c
 157     end
 158
 159     combined = ""
 160     files_in_dirs = {}
 161     files.each do |m|
 162       mt = chash[m[1]+m[2]].andand.manifest_text
 163       if not m[4].nil? and m[4].size > 1
 164         manifest_files = files_in_dirs['.']
 165         if !manifest_files
 166           manifest_files = []
 167           files_in_dirs['.'] = manifest_files
 168         end
 169         manifest_file = m[4].split('/')[-1]
 170         uniq_file = derive_unique_filename(manifest_file, manifest_files)
 171         normalized = arv_normalize mt, '--extract', ".#{m[4]}"
 172         normalized = normalized.gsub(/(\d+:\d+:)(#{Regexp.quote manifest_file})/) {|s| "#{$1}#{uniq_file}" }
 173         combined += normalized
 174         manifest_files << uniq_file
 175       else
 176         mt = arv_normalize mt
 177         manifest_streams = mt.split "\n"
 178         adjusted_streams = []
 179         manifest_streams.each do |stream|
 180           manifest_parts = stream.split
 181           adjusted_parts = []
 182           manifest_files = files_in_dirs[manifest_parts[0]]
 183           if !manifest_files
 184             manifest_files = []
 185             files_in_dirs[manifest_parts[0]] = manifest_files
 186           end
 187
 188           manifest_parts.each do |part|
 189             part_match = /(\d+:\d+:)(\S+)/.match(part)
 190             if part_match
 191               uniq_file = derive_unique_filename(part_match[2], manifest_files)
 192               adjusted_parts << "#{part_match[1]}#{uniq_file}"
 193               manifest_files << uniq_file
 194             else
 195               adjusted_parts << part
 196             end
 197           end
 198           adjusted_streams << adjusted_parts.join(' ')
 199         end
 200         adjusted_streams.each do |stream|
 201           combined += (stream + "\n")
 202         end
 203       end
 204     end
 205
 206     normalized = arv_normalize combined
 207     newc = Collection.new({:manifest_text => normalized})
 208     newc.name = newc.name || "Collection created at #{Time.now.localtime}"
 209
 210     # set owner_uuid to current project, provided it is writable
 211     current_project_writable = false
 212     action_data = JSON.parse(params['action_data']) if params['action_data']
 213     if action_data && action_data['current_project_uuid']
 214       current_project = Group.find(action_data['current_project_uuid']) rescue nil
 215       if (current_project && current_project.writable_by.andand.include?(current_user.uuid))
 216         newc.owner_uuid = action_data['current_project_uuid']
 217         current_project_writable = true
 218       end
 219     end
 220
 221     newc.save!
 222
 223     chash.each do |k,v|
 224       l = Link.new({
 225                      tail_uuid: k,
 226                      head_uuid: newc.uuid,
 227                      link_class: "provenance",
 228                      name: "provided"
 229                    })
 230       l.save!
 231     end
 232
 233     msg = current_project_writable ?
 234               "Created new collection in the project #{current_project.name}." :
 235               "Created new collection in your Home project."
 236
 237     redirect_to newc, flash: {'message' => msg}
 238   end
 239
 240   def report_issue_popup
 241     respond_to do |format|
 242       format.js
 243       format.html
 244     end
 245   end
 246
 247   def report_issue
 248     logger.warn "report_issue: #{params.inspect}"
 249
 250     respond_to do |format|
 251       IssueReporter.send_report(current_user, params).deliver
 252       format.js {render nothing: true}
 253     end
 254   end
 255
 256   def getting_started_popup
 257     respond_to do |format|
 258       format.js
 259     end
 260   end
 261
 262   protected
 263
 264   def derive_unique_filename filename, manifest_files
 265     filename_parts = filename.split('.')
 266     filename_part = filename_parts[0]
 267     counter = 1
 268     while true
 269       return filename if !manifest_files.include? filename
 270       filename_parts[0] = filename_part + "(" + counter.to_s + ")"
 271       filename = filename_parts.join('.')
 272       counter += 1
 273     end
 274   end
 275
 276 end