apps/workbench/app/controllers/users_controller.rb

   1 class UsersController < ApplicationController
   2   skip_before_filter :find_object_by_uuid, :only => [:welcome, :activity]
   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 show_pane_list
  67     if current_user.andand.is_admin
  68       super | %w(Admin)
  69     else
  70       super
  71     end
  72   end
  73
  74   def index_pane_list
  75     if current_user.andand.is_admin
  76       super | %w(Activity)
  77     else
  78       super
  79     end
  80   end
  81
  82   def sudo
  83     resp = $arvados_api_client.api(ApiClientAuthorization, '', {
  84                                      api_client_authorization: {
  85                                        owner_uuid: @object.uuid
  86                                      }
  87                                    })
  88     redirect_to root_url(api_token: resp[:api_token])
  89   end
  90
  91   def home
  92     @showallalerts = false
  93     @my_ssh_keys = AuthorizedKey.where(authorized_user_uuid: current_user.uuid)
  94     @my_tag_links = {}
  95
  96     @my_jobs = Job.
  97       limit(10).
  98       order('created_at desc').
  99       where(created_by: current_user.uuid)
 100
 101     @my_collections = Collection.
 102       limit(10).
 103       order('created_at desc').
 104       where(created_by: current_user.uuid)
 105     collection_uuids = @my_collections.collect &:uuid
 106
 107     @persist_state = {}
 108     collection_uuids.each do |uuid|
 109       @persist_state[uuid] = 'cache'
 110     end
 111
 112     Link.limit(1000).filter([['head_uuid', 'in', collection_uuids],
 113                              ['link_class', 'in', ['tag', 'resources']]]).
 114       each do |link|
 115       case link.link_class
 116       when 'tag'
 117         (@my_tag_links[link.head_uuid] ||= []) << link
 118       when 'resources'
 119         if link.name == 'wants'
 120           @persist_state[link.head_uuid] = 'persistent'
 121         end
 122       end
 123     end
 124
 125     @my_pipelines = PipelineInstance.
 126       limit(10).
 127       order('created_at desc').
 128       where(created_by: current_user.uuid)
 129
 130     respond_to do |f|
 131       f.js { render template: 'users/home.js' }
 132       f.html { render template: 'users/home' }
 133     end
 134   end
 135
 136   def unsetup
 137     if current_user.andand.is_admin
 138       @object.unsetup
 139     end
 140     show
 141   end
 142
 143   def setup
 144     respond_to do |format|
 145       if current_user.andand.is_admin
 146         setup_params = {}
 147         setup_params[:send_notification_email] = "#{Rails.configuration.send_user_setup_notification_email}"
 148         if params['user_uuid'] && params['user_uuid'].size>0
 149           setup_params[:uuid] = params['user_uuid']
 150         end
 151         if params['email'] && params['email'].size>0
 152           user = {email: params['email']}
 153           setup_params[:user] = user
 154         end
 155         if params['openid_prefix'] && params['openid_prefix'].size>0
 156           setup_params[:openid_prefix] = params['openid_prefix']
 157         end
 158         if params['repo_name'] && params['repo_name'].size>0
 159           setup_params[:repo_name] = params['repo_name']
 160         end
 161         if params['vm_uuid'] && params['vm_uuid'].size>0
 162           setup_params[:vm_uuid] = params['vm_uuid']
 163         end
 164
 165         if User.setup setup_params
 166           format.js
 167         else
 168           self.render_error status: 422
 169         end
 170       else
 171         self.render_error status: 422
 172       end
 173     end
 174   end
 175
 176   def setup_popup
 177     @vms = VirtualMachine.all.results
 178
 179     @current_selections = find_current_links @object
 180
 181     respond_to do |format|
 182       format.html
 183       format.js
 184     end
 185   end
 186
 187   protected
 188
 189   def find_current_links user
 190     current_selections = {}
 191
 192     if !user
 193       return current_selections
 194     end
 195
 196     # oid login perm
 197     oid_login_perms = Link.where(tail_uuid: user.email,
 198                                    head_kind: 'arvados#user',
 199                                    link_class: 'permission',
 200                                    name: 'can_login')
 201
 202     if oid_login_perms.any?
 203       prefix_properties = oid_login_perms.first.properties
 204       current_selections[:identity_url_prefix] = prefix_properties[:identity_url_prefix]
 205     end
 206
 207     # repo perm
 208     repo_perms = Link.where(tail_uuid: user.uuid,
 209                             head_kind: 'arvados#repository',
 210                             link_class: 'permission',
 211                             name: 'can_write')
 212     if repo_perms.any?
 213       repo_uuid = repo_perms.first.head_uuid
 214       repos = Repository.where(head_uuid: repo_uuid)
 215       if repos.any?
 216         repo_name = repos.first.name
 217         current_selections[:repo_name] = repo_name
 218       end
 219     end
 220
 221     # vm login perm
 222     vm_login_perms = Link.where(tail_uuid: user.uuid,
 223                               head_kind: 'arvados#virtualMachine',
 224                               link_class: 'permission',
 225                               name: 'can_login')
 226     if vm_login_perms.any?
 227       vm_uuid = vm_login_perms.first.head_uuid
 228       current_selections[:vm_uuid] = vm_uuid
 229     end
 230
 231     return current_selections
 232   end
 233
 234 end