4031: Find for collection dependencies even in the middle of script parameter

[arvados.git] / apps / workbench / app / controllers / jobs_controller.rb

diff --git a/apps/workbench/app/controllers/jobs_controller.rb b/apps/workbench/app/controllers/jobs_controller.rb
index 1cd3c3e28e894f4d188cbd52399834c5d44117ca..00ce0a53825fa0228c267cc9365080e62099a630 100644 (file)
--- a/apps/workbench/app/controllers/jobs_controller.rb
+++ b/apps/workbench/app/controllers/jobs_controller.rb
@@ -1,5 +1,74 @@
 class JobsController < ApplicationController
+
+  def generate_provenance(jobs)
+    return if params['tab_pane'] != "Provenance"
+
+    nodes = {}
+    collections = []
+    hashes = []
+    jobs.each do |j|
+      nodes[j[:uuid]] = j
+      hashes << j[:output]
+      ProvenanceHelper::find_collections(j[:script_parameters]) do |hash, uuid|
+        collections << uuid if uuid
+        hashes << hash if hash
+      end
+      nodes[j[:script_version]] = {:uuid => j[:script_version]}
+    end
+
+    Collection.where(uuid: collections).each do |c|
+      nodes[c[:portable_data_hash]] = c
+    end
+
+    Collection.where(portable_data_hash: hashes).each do |c|
+      nodes[c[:portable_data_hash]] = c
+    end
+
+    nodes.each do |n|
+      puts "\n#{n.inspect}"
+    end
+
+    @svg = ProvenanceHelper::create_provenance_graph nodes, "provenance_svg", {
+      :request => request,
+      :all_script_parameters => true,
+      :script_version_nodes => true}
+  end
+
   def index
-    @jobs = Job.all
+    @svg = ""
+    if params[:uuid]
+      @objects = Job.where(uuid: params[:uuid])
+      generate_provenance(@objects)
+      render_index
+    else
+      @limit = 20
+      super
+    end
+  end
+
+  def cancel
+    @object.cancel
+    if params[:return_to]
+      redirect_to params[:return_to]
+    else
+      redirect_to @object
+    end
+  end
+
+  def show
+    generate_provenance([@object])
+    super
+  end
+
+  def index_pane_list
+    if params[:uuid]
+      %w(Recent Provenance)
+    else
+      %w(Recent)
+    end
+  end
+
+  def show_pane_list
+    %w(Status Log Details Provenance Advanced)
   end
 end