apps/workbench/app/controllers/users_controller.rb

   1 class UsersController < ApplicationController
   2   skip_before_filter :find_object_by_uuid, :only => [:welcome, :activity, :storage]
   3   skip_around_filter :thread_with_mandatory_api_token, :only => :welcome
   4   before_filter :ensure_current_user_is_admin, only: [:sudo, :unsetup, :setup]
   5
   6   def welcome
   7     if current_user
   8       params[:action] = 'home'
   9       home
  10     end
  11   end
  12
  13   def activity
  14     @breadcrumb_page_name = nil
  15     @users = User.limit(params[:limit] || 1000).all
  16     @user_activity = {}
  17     @activity = {
  18       logins: {},
  19       jobs: {},
  20       pipeline_instances: {}
  21     }
  22     @total_activity = {}
  23     @spans = [['This week', Time.now.beginning_of_week, Time.now],
  24               ['Last week',
  25                Time.now.beginning_of_week.advance(weeks:-1),
  26                Time.now.beginning_of_week],
  27               ['This month', Time.now.beginning_of_month, Time.now],
  28               ['Last month',
  29                1.month.ago.beginning_of_month,
  30                Time.now.beginning_of_month]]
  31     @spans.each do |span, threshold_start, threshold_end|
  32       @activity[:logins][span] = Log.
  33         filter([[:event_type, '=', 'login'],
  34                 [:object_kind, '=', 'arvados#user'],
  35                 [:created_at, '>=', threshold_start],
  36                 [:created_at, '<', threshold_end]])
  37       @activity[:jobs][span] = Job.
  38         filter([[:created_at, '>=', threshold_start],
  39                 [:created_at, '<', threshold_end]])
  40       @activity[:pipeline_instances][span] = PipelineInstance.
  41         filter([[:created_at, '>=', threshold_start],
  42                 [:created_at, '<', threshold_end]])
  43       @activity.each do |type, act|
  44         records = act[span]
  45         @users.each do |u|
  46           @user_activity[u.uuid] ||= {}
  47           @user_activity[u.uuid][span + ' ' + type.to_s] ||= 0
  48         end
  49         records.each do |record|
  50           @user_activity[record.modified_by_user_uuid] ||= {}
  51           @user_activity[record.modified_by_user_uuid][span + ' ' + type.to_s] ||= 0
  52           @user_activity[record.modified_by_user_uuid][span + ' ' + type.to_s] += 1
  53           @total_activity[span + ' ' + type.to_s] ||= 0
  54           @total_activity[span + ' ' + type.to_s] += 1
  55         end
  56       end
  57     end
  58     @users = @users.sort_by do |a|
  59       [-@user_activity[a.uuid].values.inject(:+), a.full_name]
  60     end
  61     # Prepend a "Total" pseudo-user to the sorted list
  62     @user_activity[nil] = @total_activity
  63     @users = [OpenStruct.new(uuid: nil)] + @users
  64   end
  65
  66   def storage
  67     @breadcrumb_page_name = nil
  68     @users = User.limit(params[:limit] || 1000).all
  69     @user_storage = {}
  70     total_storage = {}
  71     @log_date = {}
  72     @users.each do |u|
  73       @user_storage[u.uuid] ||= {}
  74       storage_log = Log.
  75         filter([[:object_uuid, '=', u.uuid],
  76                 [:event_type, '=', 'user-storage-report']]).
  77         order(:created_at => :desc).
  78         limit(1)
  79       storage_log.each do |log_entry|
  80         # We expect this block to only execute once since we specified limit(1)
  81         @user_storage[u.uuid] = log_entry['properties']
  82         @log_date[u.uuid] = log_entry['event_at']
  83       end
  84       total_storage.merge!(@user_storage[u.uuid]) { |k,v1,v2| v1 + v2 }
  85     end
  86     @users = @users.sort_by { |u|
  87       [-@user_storage[u.uuid].values.push(0).inject(:+), u.full_name]}
  88     # Prepend a "Total" pseudo-user to the sorted list
  89     @users = [OpenStruct.new(uuid: nil)] + @users
  90     @user_storage[nil] = total_storage
  91   end
  92
  93   def show_pane_list
  94     if current_user.andand.is_admin
  95       super | %w(Admin)
  96     else
  97       super
  98     end
  99   end
 100
 101   def index_pane_list
 102     if current_user.andand.is_admin
 103       super | %w(Activity)
 104     else
 105       super
 106     end
 107   end
 108
 109   def sudo
 110     resp = $arvados_api_client.api(ApiClientAuthorization, '', {
 111                                      api_client_authorization: {
 112                                        owner_uuid: @object.uuid
 113                                      }
 114                                    })
 115     redirect_to root_url(api_token: resp[:api_token])
 116   end
 117
 118   def home
 119     @showallalerts = false
 120     @my_ssh_keys = AuthorizedKey.where(authorized_user_uuid: current_user.uuid)
 121     @my_tag_links = {}
 122
 123     @my_jobs = Job.
 124       limit(10).
 125       order('created_at desc').
 126       where(created_by: current_user.uuid)
 127
 128     @my_collections = Collection.
 129       limit(10).
 130       order('created_at desc').
 131       where(created_by: current_user.uuid)
 132
 133     Link.limit(1000).where(head_uuid: @my_collections.collect(&:uuid),
 134                            link_class: 'tag').each do |link|
 135       (@my_tag_links[link.head_uuid] ||= []) << link
 136     end
 137
 138     @my_pipelines = PipelineInstance.
 139       limit(10).
 140       order('created_at desc').
 141       where(created_by: current_user.uuid)
 142
 143
 144     # A Tutorial is a Link which has link_class "resources" and name
 145     # "wants", and is owned by the Tutorials Group (i.e., named
 146     # "Arvados Tutorials" and owned by the system user).
 147     @tutorial_group = Group.where(owner_uuid: User.system.uuid,
 148                                   name: 'Arvados Tutorials').first
 149     if @tutorial_group
 150       @tutorial_links = Link.where(tail_uuid: @tutorial_group.uuid,
 151                                    link_class: 'resources',
 152                                    name: 'wants')
 153     else
 154       @tutorial_links = []
 155     end
 156     @tutorial_complete = {
 157       'Run a job' => @my_last_job
 158     }
 159     respond_to do |f|
 160       f.js { render template: 'users/home.js' }
 161       f.html { render template: 'users/home' }
 162     end
 163   end
 164
 165   def unsetup
 166     if current_user.andand.is_admin
 167       @object.unsetup
 168     end
 169     show
 170   end
 171
 172   def setup
 173     respond_to do |format|
 174       if current_user.andand.is_admin
 175         setup_params = {}
 176         setup_params[:send_notification_email] = "#{Rails.configuration.send_user_setup_notification_email}"
 177         if params['user_uuid'] && params['user_uuid'].size>0
 178           setup_params[:uuid] = params['user_uuid']
 179         end
 180         if params['email'] && params['email'].size>0
 181           user = {email: params['email']}
 182           setup_params[:user] = user
 183         end
 184         if params['openid_prefix'] && params['openid_prefix'].size>0
 185           setup_params[:openid_prefix] = params['openid_prefix']
 186         end
 187         if params['repo_name'] && params['repo_name'].size>0
 188           setup_params[:repo_name] = params['repo_name']
 189         end
 190         if params['vm_uuid'] && params['vm_uuid'].size>0
 191           setup_params[:vm_uuid] = params['vm_uuid']
 192         end
 193
 194         if User.setup setup_params
 195           format.js
 196         else
 197           self.render_error status: 422
 198         end
 199       else
 200         self.render_error status: 422
 201       end
 202     end
 203   end
 204
 205   def setup_popup
 206     @vms = VirtualMachine.all.results
 207
 208     @current_selections = find_current_links @object
 209
 210     respond_to do |format|
 211       format.html
 212       format.js
 213     end
 214   end
 215
 216   protected
 217
 218   def find_current_links user
 219     current_selections = {}
 220
 221     if !user
 222       return current_selections
 223     end
 224
 225     # oid login perm
 226     oid_login_perms = Link.where(tail_uuid: user.email,
 227                                    head_kind: 'arvados#user',
 228                                    link_class: 'permission',
 229                                    name: 'can_login')
 230
 231     if oid_login_perms.any?
 232       prefix_properties = oid_login_perms.first.properties
 233       current_selections[:identity_url_prefix] = prefix_properties[:identity_url_prefix]
 234     end
 235
 236     # repo perm
 237     repo_perms = Link.where(tail_uuid: user.uuid,
 238                             head_kind: 'arvados#repository',
 239                             link_class: 'permission',
 240                             name: 'can_write')
 241     if repo_perms.any?
 242       repo_uuid = repo_perms.first.head_uuid
 243       repos = Repository.where(head_uuid: repo_uuid)
 244       if repos.any?
 245         repo_name = repos.first.name
 246         current_selections[:repo_name] = repo_name
 247       end
 248     end
 249
 250     # vm login perm
 251     vm_login_perms = Link.where(tail_uuid: user.uuid,
 252                               head_kind: 'arvados#virtualMachine',
 253                               link_class: 'permission',
 254                               name: 'can_login')
 255     if vm_login_perms.any?
 256       vm_uuid = vm_login_perms.first.head_uuid
 257       current_selections[:vm_uuid] = vm_uuid
 258     end
 259
 260     return current_selections
 261   end
 262
 263 end