11185: Addressed several requests:
[arvados.git] / apps / workbench / app / controllers / work_units_controller.rb
1 class WorkUnitsController < ApplicationController
2   skip_around_filter :require_thread_api_token, if: proc { |ctrl|
3     Rails.configuration.anonymous_user_token and
4     'show_child_component' == ctrl.action_name
5   }
6
7   def find_objects_for_index
8     # If it's not the index rows partial display, just return
9     # The /index request will again be invoked to display the
10     # partial at which time, we will be using the objects found.
11     return if !params[:partial]
12
13     @limit = 20
14     @filters = @filters || []
15
16     # get next page of pipeline_instances
17     if PipelineInstance.api_exists?(:index)
18       filters = @filters + [["uuid", "is_a", ["arvados#pipelineInstance"]]]
19       pipelines = PipelineInstance.limit(@limit).order(["created_at desc"]).filter(filters)
20     end
21
22     # get next page of jobs
23     if Job.api_exists?(:index)
24       filters = @filters + [["uuid", "is_a", ["arvados#job"]]]
25       jobs = Job.limit(@limit).order(["created_at desc"]).filter(filters)
26     end
27
28     # get next page of container_requests
29     filters = @filters + [["uuid", "is_a", ["arvados#containerRequest"]]]
30     crs = ContainerRequest.limit(@limit).order(["created_at desc"]).filter(filters)
31     @objects = (jobs.to_a + pipelines.to_a + crs.to_a).sort_by(&:created_at).reverse.first(@limit)
32
33     if @objects.any?
34       @next_page_filters = next_page_filters('<=')
35       @next_page_href = url_for(partial: :all_processes_rows,
36                                 filters: @next_page_filters.to_json)
37       preload_links_for_objects(@objects.to_a)
38     else
39       @next_page_href = nil
40     end
41   end
42
43   def next_page_href with_params={}
44     @next_page_href
45   end
46
47   def create
48     template_uuid = params['work_unit']['template_uuid']
49
50     attrs = {}
51     rc = resource_class_for_uuid(template_uuid)
52     if rc == PipelineTemplate
53       model_class = PipelineInstance
54       attrs['pipeline_template_uuid'] = template_uuid
55     elsif rc == Workflow
56       # workflow json
57       workflow = Workflow.find? template_uuid
58       if workflow.definition
59         begin
60           wf_json = ActiveSupport::HashWithIndifferentAccess.new YAML::load(workflow.definition)
61         rescue => e
62           logger.error "Error converting definition yaml to json: #{e.message}"
63           raise ArgumentError, "Error converting definition yaml to json: #{e.message}"
64         end
65       end
66
67       model_class = ContainerRequest
68
69       attrs['name'] = "#{workflow['name']} container" if workflow['name'].present?
70       attrs['properties'] = {'template_uuid' => template_uuid}
71       attrs['priority'] = 1
72       attrs['state'] = "Uncommitted"
73
74       # required
75       attrs['command'] = ["arvados-cwl-runner", "--local", "--api=containers", "/var/lib/cwl/workflow.json#main", "/var/lib/cwl/cwl.input.json"]
76       attrs['container_image'] = "arvados/jobs"
77       attrs['cwd'] = "/var/spool/cwl"
78       attrs['output_path'] = "/var/spool/cwl"
79
80       input_defaults = {}
81       if wf_json
82         inputs = get_cwl_inputs(wf_json)
83         inputs.each do |input|
84           if input[:default]
85             input_defaults[cwl_shortname(input[:id])] = input[:default]
86           end
87         end
88       end
89
90       # mounts
91       mounts = {
92         "/var/lib/cwl/cwl.input.json" => {
93           "kind" => "json",
94           "content" => input_defaults
95         },
96         "stdout" => {
97           "kind" => "file",
98           "path" => "/var/spool/cwl/cwl.output.json"
99         },
100         "/var/spool/cwl" => {
101           "kind" => "collection",
102           "writable" => true
103         }
104       }
105       if wf_json
106         mounts["/var/lib/cwl/workflow.json"] = {
107           "kind" => "json",
108           "content" => wf_json
109         }
110       end
111       attrs['mounts'] = mounts
112
113       # runtime constriants
114       runtime_constraints = {
115         "vcpus" => 1,
116         "ram" => 256000000,
117         "API" => true
118       }
119       attrs['runtime_constraints'] = runtime_constraints
120     else
121       raise ArgumentError, "Unsupported template uuid: #{template_uuid}"
122     end
123
124     attrs['owner_uuid'] = params['work_unit']['owner_uuid']
125     @object ||= model_class.new attrs
126
127     if @object.save
128       redirect_to @object
129     else
130       render_error status: 422
131     end
132   end
133
134   def find_object_by_uuid
135     if params['object_type']
136       @object = params['object_type'].constantize.find(params['uuid'])
137     else
138       super
139     end
140   end
141
142   def show_child_component
143     data = JSON.load(params[:action_data])
144
145     current_obj = {}
146     current_obj_uuid = data['current_obj_uuid']
147     current_obj_name = data['current_obj_name']
148     current_obj_type = data['current_obj_type']
149     current_obj_parent = data['current_obj_parent']
150     if current_obj_uuid
151       resource_class = resource_class_for_uuid current_obj_uuid
152       obj = object_for_dataclass(resource_class, current_obj_uuid)
153       current_obj = obj if obj
154     end
155
156     if current_obj.is_a?(Hash) and !current_obj.any?
157       if current_obj_parent
158         resource_class = resource_class_for_uuid current_obj_parent
159         parent = object_for_dataclass(resource_class, current_obj_parent)
160         parent_wu = parent.work_unit
161         children = parent_wu.children
162         if current_obj_uuid
163           wu = children.select {|c| c.uuid == current_obj_uuid}.first
164         else current_obj_name
165           wu = children.select {|c| c.label.to_s == current_obj_name}.first
166         end
167       end
168     else
169       if current_obj_type == JobWorkUnit.to_s
170         wu = JobWorkUnit.new(current_obj, current_obj_name, current_obj_parent)
171       elsif current_obj_type == PipelineInstanceWorkUnit.to_s
172         wu = PipelineInstanceWorkUnit.new(current_obj, current_obj_name, current_obj_parent)
173       elsif current_obj_type == ContainerWorkUnit.to_s
174         wu = ContainerWorkUnit.new(current_obj, current_obj_name, current_obj_parent)
175       end
176     end
177
178     respond_to do |f|
179       f.html { render(partial: "show_component", locals: {wu: wu}) }
180     end
181   end
182 end