1 # Copyright (C) The Arvados Authors. All rights reserved.
3 # SPDX-License-Identifier: AGPL-3.0
5 class ContainerRequestsController < ApplicationController
6 skip_around_action :require_thread_api_token, if: proc { |ctrl|
7 !Rails.configuration.Users.AnonymousUserToken.empty? and
8 'show' == ctrl.action_name
11 def generate_provenance(cr)
12 return if params['tab_pane'] != "Provenance"
18 col_uuids << cr[:output_uuid] if cr[:output_uuid]
19 col_pdhs += ProvenanceHelper::cr_input_pdhs(cr)
21 # Search for child CRs
22 if cr[:container_uuid]
23 child_crs = ContainerRequest.where(requesting_container_uuid: cr[:container_uuid]).with_count("none")
25 child_crs.each do |child|
26 nodes[child[:uuid]] = child
27 col_uuids << child[:output_uuid] if child[:output_uuid]
28 col_pdhs += ProvenanceHelper::cr_input_pdhs(child)
36 pdh_to_col = {} # Indexed by PDH
39 # Batch requests to get all related collections
40 # First fetch output collections by UUID.
41 Collection.filter([['uuid', 'in', col_uuids.uniq]]).with_count("none").each do |c|
42 output_pdhs << c[:portable_data_hash]
43 pdh_to_col[c[:portable_data_hash]] = c
46 # Next, get input collections by PDH.
48 [['portable_data_hash', 'in', col_pdhs - output_pdhs]]).with_count("none").each do |c|
49 nodes[c[:portable_data_hash]] = c
52 @svg = ProvenanceHelper::create_provenance_graph(
53 nodes, "provenance_svg",
56 :pdh_to_uuid => pdh_to_col,
62 panes = %w(Status Log Provenance Advanced)
63 if @object.andand.state == 'Uncommitted'
64 panes = %w(Inputs) + panes - %w(Log Provenance)
70 generate_provenance(@object)
75 if @object.container_uuid
76 c = Container.select(['state']).where(uuid: @object.container_uuid).with_count("none").first
77 if c && c.state != 'Running'
78 # If the container hasn't started yet, setting priority=0
79 # leaves our request in "Committed" state and doesn't cancel
80 # the container (even if no other requests are giving it
81 # priority). To avoid showing this container request as "on
82 # hold" after hitting the Cancel button, set state=Final too.
83 @object.state = 'Final'
86 @object.update_attributes! priority: 0
88 redirect_to params[:return_to]
95 @updates ||= params[@object.class.to_s.underscore.singularize.to_sym]
96 input_obj = @updates[:mounts].andand[:"/var/lib/cwl/cwl.input.json"].andand[:content]
98 workflow = @object.mounts[:"/var/lib/cwl/workflow.json"][:content]
99 get_cwl_inputs(workflow).each do |input_schema|
100 if not input_obj.include? cwl_shortname(input_schema[:id])
103 required, primary_type, param_id = cwl_input_info(input_schema)
104 if input_obj[param_id] == ""
105 input_obj[param_id] = nil
106 elsif primary_type == "boolean"
107 input_obj[param_id] = input_obj[param_id] == "true"
108 elsif ["int", "long"].include? primary_type
109 input_obj[param_id] = input_obj[param_id].to_i
110 elsif ["float", "double"].include? primary_type
111 input_obj[param_id] = input_obj[param_id].to_f
112 elsif ["File", "Directory"].include? primary_type
113 re = CollectionsHelper.match_uuid_with_optional_filepath(input_obj[param_id])
115 c = Collection.find(re[1])
116 input_obj[param_id] = {"class" => primary_type,
117 "location" => "keep:#{c.portable_data_hash}#{re[4]}",
118 "http://arvados.org/cwl#collectionUUID" => re[1]}
123 params[:merge] = true
125 if !@updates[:reuse_steps].nil?
126 if @updates[:reuse_steps] == "false"
127 @updates[:reuse_steps] = false
129 @updates[:command] ||= @object.command
130 if @updates[:reuse_steps]
131 @updates[:command] = @updates[:command] - ["--disable-reuse"]
132 @updates[:command].insert(1, '--enable-reuse')
134 @updates[:command] -= @updates[:command] - ["--enable-reuse"]
135 @updates[:command].insert(1, '--disable-reuse')
138 @updates.delete(:reuse_steps)
144 flash[:error] = e.to_s
152 @object = ContainerRequest.new
154 # set owner_uuid to that of source, provided it is a project and writable by current user
155 if params[:work_unit][:owner_uuid]
156 @object.owner_uuid = src.owner_uuid = params[:work_unit][:owner_uuid]
158 current_project = Group.find(src.owner_uuid) rescue nil
159 if (current_project && current_project.writable_by.andand.include?(current_user.uuid))
160 @object.owner_uuid = src.owner_uuid
164 if src.command[0] == 'arvados-cwl-runner'
165 command.each_with_index do |arg, i|
166 if arg.start_with? "--project-uuid="
167 command[i] = "--project-uuid=#{@object.owner_uuid}"
169 if arg == "--disable-reuse"
170 command[i] = "--enable-reuse"
175 # By default the copied CR won't be reusing containers, unless use_existing=true
177 command = src.command
178 if params[:use_existing]
179 @object.use_existing = true
180 # Pass the correct argument to arvados-cwl-runner command.
181 if src.command[0] == 'arvados-cwl-runner'
182 command = src.command - ['--disable-reuse']
183 command.insert(1, '--enable-reuse')
186 @object.use_existing = false
187 # Pass the correct argument to arvados-cwl-runner command.
188 if src.command[0] == 'arvados-cwl-runner'
189 command = src.command - ['--enable-reuse']
190 command.insert(1, '--disable-reuse')
194 @object.command = command
195 @object.container_image = src.container_image
196 @object.cwd = src.cwd
197 @object.description = src.description
198 @object.environment = src.environment
199 @object.mounts = src.mounts
200 @object.name = src.name
201 @object.output_path = src.output_path
203 @object.properties[:template_uuid] = src.properties[:template_uuid]
204 @object.runtime_constraints = src.runtime_constraints
205 @object.scheduling_parameters = src.scheduling_parameters
206 @object.state = 'Uncommitted'