X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/310f6c81aad96a063f798d5940853dfad3409bba..c592f178659ed5ebfd03720a02940daa62931d14:/apps/workbench/app/controllers/application_controller.rb

diff --git a/apps/workbench/app/controllers/application_controller.rb b/apps/workbench/app/controllers/application_controller.rb
index 696e821607..1fc15807c9 100644
--- a/apps/workbench/app/controllers/application_controller.rb
+++ b/apps/workbench/app/controllers/application_controller.rb
@@ -94,7 +94,10 @@ class ApplicationController < ActionController::Base
         # Fall back to the default-setting code later.
       end
     end
+    @starred_projects ||= []
     @my_wanted_projects_tree ||= []
+    @my_project_tree ||= []
+    @shared_project_tree ||= []
     render_error(err_opts)
   end
 
@@ -443,6 +446,15 @@ class ApplicationController < ActionController::Base
     end
   end
 
+  helper_method :is_starred
+  def is_starred
+    links = Link.where(tail_uuid: current_user.uuid,
+               head_uuid: @object.uuid,
+               link_class: 'star')
+
+    return links.andand.any?
+  end
+
   protected
 
   helper_method :strip_token_from_path
@@ -832,51 +844,62 @@ class ApplicationController < ActionController::Base
     {collections: c, owners: own}
   end
 
+  helper_method :my_starred_projects
+  def my_starred_projects user
+    return if @starred_projects
+    links = Link.filter([['tail_uuid', '=', user.uuid],
+                         ['link_class', '=', 'star'],
+                         ['head_uuid', 'is_a', 'arvados#group']]).select(%w(head_uuid))
+    uuids =links.collect { |x| x.head_uuid }
+    starred_projects = Group.filter([['uuid', 'in', uuids]]).order('name')
+    @starred_projects = starred_projects.results
+  end
+
   # If there are more than 200 projects that are readable by the user,
-  # build the tree using only the top 200+ projects owned by the user.
+  # build the tree using only the top 200+ projects owned by the user,
+  # from the top three levels.
   # That is: get toplevel projects under home, get subprojects of
-  # these projects, and so on until we hit the limit
-  def my_wanted_projects page_size=100
+  # these projects, and so on until we hit the limit.
+  def my_wanted_projects user, page_size=100
     return @my_wanted_projects if @my_wanted_projects
 
-    all = Group.filter([['group_class','=','project']]).order('name').limit(page_size)
-    if all.items_available > page_size*3
-      @total_projects = all.items_available
-      from_top = []
-      uuids = [current_user.uuid]
-      while from_top.size <= page_size*2
-        current_level = Group.filter([['group_class','=','project'],
-                                      ['owner_uuid', 'in', uuids]])
-                             .order('name').limit(page_size*2)
-        break if current_level.results.size == 0
-        from_top.concat current_level.results
-        uuids = current_level.results.collect { |x| x.uuid }
-      end
-      @my_wanted_projects = from_top
-    else
-      if all.results.size == all.items_available
-        @my_wanted_projects = all
-      else
-        @my_wanted_projects = Group.filter([['group_class','=','project']]).order('name')
+    from_top = []
+    uuids = [user.uuid]
+    depth = 0
+    @too_many_projects = false
+    @reached_level_limit = false
+    while from_top.size <= page_size*2
+      current_level = Group.filter([['group_class','=','project'],
+                                    ['owner_uuid', 'in', uuids]])
+                      .order('name').limit(page_size*2)
+      break if current_level.results.size == 0
+      @too_many_projects = true if current_level.items_available > current_level.results.size
+      from_top.concat current_level.results
+      uuids = current_level.results.collect { |x| x.uuid }
+      depth += 1
+      if depth >= 3
+        @reached_level_limit = true
+        break
       end
     end
+    @my_wanted_projects = from_top
   end
 
   helper_method :my_wanted_projects_tree
   def my_wanted_projects_tree user, page_size=100
     build_my_wanted_projects_tree user, page_size
-    [@my_wanted_projects_tree, @total_projects]
+    [@my_wanted_projects_tree, @too_many_projects, @reached_level_limit]
   end
 
   def build_my_wanted_projects_tree user, page_size=100
     return @my_wanted_projects_tree if @my_wanted_projects_tree
 
     parent_of = {user.uuid => 'me'}
-    my_wanted_projects(page_size).each do |ob|
+    my_wanted_projects(user, page_size).each do |ob|
       parent_of[ob.uuid] = ob.owner_uuid
     end
     children_of = {false => [], 'me' => [user]}
-    my_wanted_projects(page_size).each do |ob|
+    my_wanted_projects(user, page_size).each do |ob|
       if ob.owner_uuid != user.uuid and
           not parent_of.has_key? ob.owner_uuid
         parent_of[ob.uuid] = false