X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/8a0051a6fdd00f83e566cc87b88d2ac23ad5487d..edebf116a9f743a7a276d514c4cbad72c3406922:/apps/workbench/app/controllers/users_controller.rb diff --git a/apps/workbench/app/controllers/users_controller.rb b/apps/workbench/app/controllers/users_controller.rb index 72d44f51c2..d817b32711 100644 --- a/apps/workbench/app/controllers/users_controller.rb +++ b/apps/workbench/app/controllers/users_controller.rb @@ -1,8 +1,15 @@ class UsersController < ApplicationController skip_before_filter :find_object_by_uuid, :only => [:welcome, :activity, :storage] - skip_around_filter :thread_with_mandatory_api_token, :only => :welcome before_filter :ensure_current_user_is_admin, only: [:sudo, :unsetup, :setup] + def show + if params[:uuid] == current_user.uuid + redirect_to project_path(params[:uuid]) + else + super + end + end + def welcome if current_user params[:action] = 'home' @@ -67,22 +74,27 @@ class UsersController < ApplicationController @breadcrumb_page_name = nil @users = User.limit(params[:limit] || 1000).all @user_storage = {} + total_storage = {} + @log_date = {} @users.each do |u| @user_storage[u.uuid] ||= {} storage_log = Log. - # filter([[:object_uuid, '=', u.uuid], - # [:event_type, '=', 'user-storage-report']]) filter([[:object_uuid, '=', u.uuid], [:event_type, '=', 'user-storage-report']]). order(:created_at => :desc). limit(1) storage_log.each do |log_entry| + # We expect this block to only execute once since we specified limit(1) @user_storage[u.uuid] = log_entry['properties'] + @log_date[u.uuid] = log_entry['event_at'] end + total_storage.merge!(@user_storage[u.uuid]) { |k,v1,v2| v1 + v2 } end - @users = @users.sort_by do |u| - [-@user_storage[u.uuid].values.push(0).inject(:+), u.full_name] - end + @users = @users.sort_by { |u| + [-@user_storage[u.uuid].values.push(0).inject(:+), u.full_name]} + # Prepend a "Total" pseudo-user to the sorted list + @users = [OpenStruct.new(uuid: nil)] + @users + @user_storage[nil] = total_storage end def show_pane_list @@ -102,11 +114,11 @@ class UsersController < ApplicationController end def sudo - resp = $arvados_api_client.api(ApiClientAuthorization, '', { - api_client_authorization: { - owner_uuid: @object.uuid - } - }) + resp = arvados_api_client.api(ApiClientAuthorization, '', { + api_client_authorization: { + owner_uuid: @object.uuid + } + }) redirect_to root_url(api_token: resp[:api_token]) end @@ -124,10 +136,24 @@ class UsersController < ApplicationController limit(10). order('created_at desc'). where(created_by: current_user.uuid) + collection_uuids = @my_collections.collect &:uuid - Link.limit(1000).where(head_uuid: @my_collections.collect(&:uuid), - link_class: 'tag').each do |link| - (@my_tag_links[link.head_uuid] ||= []) << link + @persist_state = {} + collection_uuids.each do |uuid| + @persist_state[uuid] = 'cache' + end + + Link.limit(1000).filter([['head_uuid', 'in', collection_uuids], + ['link_class', 'in', ['tag', 'resources']]]). + each do |link| + case link.link_class + when 'tag' + (@my_tag_links[link.head_uuid] ||= []) << link + when 'resources' + if link.name == 'wants' + @persist_state[link.head_uuid] = 'persistent' + end + end end @my_pipelines = PipelineInstance. @@ -135,22 +161,6 @@ class UsersController < ApplicationController order('created_at desc'). where(created_by: current_user.uuid) - - # A Tutorial is a Link which has link_class "resources" and name - # "wants", and is owned by the Tutorials Group (i.e., named - # "Arvados Tutorials" and owned by the system user). - @tutorial_group = Group.where(owner_uuid: User.system.uuid, - name: 'Arvados Tutorials').first - if @tutorial_group - @tutorial_links = Link.where(tail_uuid: @tutorial_group.uuid, - link_class: 'resources', - name: 'wants') - else - @tutorial_links = [] - end - @tutorial_complete = { - 'Run a job' => @my_last_job - } respond_to do |f| f.js { render template: 'users/home.js' } f.html { render template: 'users/home' } @@ -208,6 +218,73 @@ class UsersController < ApplicationController end end + def manage_account + # repositories current user can read / write + repo_links = [] + Link.filter([['head_uuid', 'is_a', 'arvados#repository'], + ['tail_uuid', '=', current_user.uuid], + ['link_class', '=', 'permission'], + ['name', 'in', ['can_write', 'can_read']], + ]). + each do |perm_link| + repo_links << perm_link[:head_uuid] + end + @my_repositories = Repository.where(uuid: repo_links) + + # virtual machines the current user can login into + @my_vm_logins = {} + Link.where(tail_uuid: current_user.uuid, + link_class: 'permission', + name: 'can_login'). + each do |perm_link| + if perm_link.properties.andand[:username] + @my_vm_logins[perm_link.head_uuid] ||= [] + @my_vm_logins[perm_link.head_uuid] << perm_link.properties[:username] + end + end + @my_virtual_machines = VirtualMachine.where(uuid: @my_vm_logins.keys) + + # current user's ssh keys + @my_ssh_keys = AuthorizedKey.where(key_type: 'SSH', owner_uuid: current_user.uuid) + + respond_to do |f| + f.html { render template: 'users/manage_account' } + end + end + + def add_ssh_key_popup + respond_to do |format| + format.html + format.js + end + end + + def add_ssh_key + respond_to do |format| + key_params = {'key_type' => 'SSH'} + key_params['authorized_user_uuid'] = current_user.uuid + + if params['name'] && params['name'].size>0 + key_params['name'] = params['name'].strip + end + if params['public_key'] && params['public_key'].size>0 + key_params['public_key'] = params['public_key'].strip + end + + if !key_params['name'] && params['public_key'].andand.size>0 + split_key = key_params['public_key'].split + key_params['name'] = split_key[-1] if (split_key.size == 3) + end + + new_key = AuthorizedKey.create! key_params + if new_key + format.js + else + self.render_error status: 422 + end + end + end + protected def find_current_links user