services/api/app/controllers/arvados/v1/containers_controller.rb

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: AGPL-3.0
   4
   5 class Arvados::V1::ContainersController < ApplicationController
   6   accept_attribute_as_json :environment, Hash
   7   accept_attribute_as_json :mounts, Hash
   8   accept_attribute_as_json :runtime_constraints, Hash
   9   accept_attribute_as_json :runtime_status, Hash
  10   accept_attribute_as_json :command, Array
  11   accept_attribute_as_json :scheduling_parameters, Hash
  12
  13   skip_before_action :find_object_by_uuid, only: [:current]
  14   skip_before_action :render_404_if_no_object, only: [:current]
  15
  16   def auth
  17     if @object.locked_by_uuid != Thread.current[:api_client_authorization].uuid
  18       raise ArvadosModel::PermissionDeniedError.new("Not locked by your token")
  19     end
  20     if @object.runtime_token.nil?
  21       @object = @object.auth
  22     else
  23       @object = ApiClientAuthorization.validate(token: @object.runtime_token)
  24       if @object.nil?
  25         raise ArvadosModel::PermissionDeniedError.new("Invalid runtime_token")
  26       end
  27     end
  28     show
  29   end
  30
  31   def update
  32     @object.with_lock do
  33       super
  34     end
  35   end
  36
  37   def find_objects_for_index
  38     super
  39     if action_name == 'lock' || action_name == 'unlock'
  40       # Avoid loading more fields than we need
  41       @objects = @objects.select(:id, :uuid, :state, :priority, :auth_uuid, :locked_by_uuid, :lock_count)
  42       @select = %w(uuid state priority auth_uuid locked_by_uuid)
  43     elsif action_name == 'update_priority'
  44       @objects = @objects.select(:id, :uuid)
  45       @select = %w(uuid)
  46     end
  47   end
  48
  49   def lock
  50     @object.lock
  51     show
  52   end
  53
  54   def unlock
  55     @object.unlock
  56     show
  57   end
  58
  59   def update_priority
  60     @object.update_priority!
  61     show
  62   end
  63
  64   def current
  65     if Thread.current[:api_client_authorization].nil?
  66       send_error("Not logged in", status: 401)
  67     else
  68       @object = Container.for_current_token
  69       if @object.nil?
  70         send_error("Token is not associated with a container.", status: 404)
  71       else
  72         show
  73       end
  74     end
  75   end
  76
  77   def secret_mounts
  78     c = Container.for_current_token
  79     if @object && c && @object.uuid == c.uuid
  80       send_json({"secret_mounts" => @object.secret_mounts})
  81     else
  82       send_error("Token is not associated with this container.", status: 403)
  83     end
  84   end
  85 end