Merge branch '10484-count-clients' closes #10484
[arvados.git] / apps / workbench / app / models / container_work_unit.rb
1 class ContainerWorkUnit < ProxyWorkUnit
2   attr_accessor :container
3
4   def initialize proxied, label, parent
5     super
6     if @proxied.is_a?(ContainerRequest)
7       container_uuid = get(:container_uuid)
8       if container_uuid
9         @container = Container.where(uuid: container_uuid).first
10       end
11     end
12   end
13
14   def children
15     return @my_children if @my_children
16
17     container_uuid = nil
18     container_uuid = if @proxied.is_a?(Container) then uuid else get(:container_uuid) end
19
20     items = []
21     if container_uuid
22       reqs = ContainerRequest.where(requesting_container_uuid: container_uuid).results
23       reqs.each do |cr|
24         items << cr.work_unit(cr.name || 'this container')
25       end
26     end
27
28     @my_children = items
29   end
30
31   def title
32     "container"
33   end
34
35   def uri
36     uuid = get(:uuid)
37
38     return nil unless uuid
39
40     if @proxied.class.respond_to? :table_name
41       "/#{@proxied.class.table_name}/#{uuid}"
42     else
43       resource_class = ArvadosBase.resource_class_for_uuid(uuid)
44       "#{resource_class.table_name}/#{uuid}" if resource_class
45     end
46   end
47
48   def can_cancel?
49     @proxied.is_a?(ContainerRequest) && @proxied.state == "Committed" && @proxied.priority > 0 && @proxied.editable?
50   end
51
52   def container_uuid
53     get(:container_uuid)
54   end
55
56   def priority
57     @proxied.priority
58   end
59
60   # For the following properties, use value from the @container if exists
61   # This applies to a ContainerRequest with container_uuid
62
63   def started_at
64     t = get_combined(:started_at)
65     t = Time.parse(t) if (t.is_a? String)
66     t
67   end
68
69   def modified_at
70     t = get_combined(:modified_at)
71     t = Time.parse(t) if (t.is_a? String)
72     t
73   end
74
75   def finished_at
76     t = get_combined(:finished_at)
77     t = Time.parse(t) if (t.is_a? String)
78     t
79   end
80
81   def state_label
82     ec = exit_code
83     return "Failed" if (ec && ec != 0)
84     state = get_combined(:state)
85     return "Ready" if ((priority == 0) and (["Queued", "Locked"].include?(state)))
86     state
87   end
88
89   def exit_code
90     get_combined(:exit_code)
91   end
92
93   def docker_image
94     get_combined(:container_image)
95   end
96
97   def runtime_constraints
98     get_combined(:runtime_constraints)
99   end
100
101   def log_collection
102     get_combined(:log)
103   end
104
105   def outputs
106     items = []
107     items << get_combined(:output) if get_combined(:output)
108     items
109   end
110
111   def command
112     get_combined(:command)
113   end
114
115   def cwd
116     get_combined(:cwd)
117   end
118
119   def environment
120     env = get_combined(:environment)
121     env = nil if env.andand.empty?
122     env
123   end
124
125   def mounts
126     mnt = get_combined(:mounts)
127     mnt = nil if mnt.andand.empty?
128     mnt
129   end
130
131   def output_path
132     get_combined(:output_path)
133   end
134
135   def log_object_uuids
136     [get(:uuid, @container), get(:uuid, @proxied)].compact
137   end
138
139   def render_log
140     collection = Collection.find(log_collection) rescue nil
141     if collection
142       return {log: collection, partial: 'collections/show_files', locals: {object: collection, no_checkboxes: true}}
143     end
144   end
145
146   def template_uuid
147     properties = get(:properties)
148     if properties
149       properties[:template_uuid]
150     end
151   end
152
153   # End combined properties
154
155   protected
156   def get_combined key
157     get(key, @container) || get(key, @proxied)
158   end
159 end