services/api/lib/tasks/manage_long_lived_tokens.rake

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: AGPL-3.0
   4
   5 # Tasks that can be useful when changing token expiration policies by assigning
   6 # a non-zero value to Login.TokenLifetime config.
   7
   8 require 'set'
   9 require 'current_api_client'
  10
  11 namespace :db do
  12   desc "Apply expiration policy on long lived tokens"
  13   task fix_long_lived_tokens: :environment do
  14     lifetime = Rails.configuration.API.MaxTokenLifetime
  15     if lifetime.nil? or lifetime == 0
  16       lifetime = Rails.configuration.Login.TokenLifetime
  17     end
  18     if lifetime.nil? or lifetime == 0
  19       puts("No expiration policy set (API.MaxTokenLifetime nor Login.TokenLifetime is set), nothing to do.")
  20       # abort the rake task
  21       next
  22     end
  23     exp_date = Time.now + lifetime
  24     puts("Setting token expiration to: #{exp_date}")
  25     token_count = 0
  26     ll_tokens(lifetime).each do |auth|
  27       if auth.user.nil?
  28         printf("*** WARNING, found ApiClientAuthorization with invalid user: auth id: %d, user id: %d\n", auth.id, auth.user_id)
  29         # skip this token
  30         next
  31       end
  32       if (auth.user.uuid =~ /-tpzed-000000000000000/).nil?
  33         CurrentApiClientHelper.act_as_system_user do
  34           auth.update_attributes!(expires_at: exp_date)
  35         end
  36         token_count += 1
  37       end
  38     end
  39     puts("#{token_count} tokens updated.")
  40   end
  41
  42   desc "Show users with long lived tokens"
  43   task check_long_lived_tokens: :environment do
  44     lifetime = Rails.configuration.API.MaxTokenLifetime
  45     if lifetime.nil? or lifetime == 0
  46       lifetime = Rails.configuration.Login.TokenLifetime
  47     end
  48     if lifetime.nil? or lifetime == 0
  49       puts("No expiration policy set (API.MaxTokenLifetime nor Login.TokenLifetime is set), nothing to do.")
  50       # abort the rake task
  51       next
  52     end
  53     user_ids = Set.new()
  54     token_count = 0
  55     ll_tokens(lifetime).each do |auth|
  56       if auth.user.nil?
  57         printf("*** WARNING, found ApiClientAuthorization with invalid user: auth id: %d, user id: %d\n", auth.id, auth.user_id)
  58         # skip this token
  59         next
  60       end
  61       if not auth.user.nil? and (auth.user.uuid =~ /-tpzed-000000000000000/).nil?
  62         user_ids.add(auth.user_id)
  63         token_count += 1
  64       end
  65     end
  66
  67     if user_ids.size > 0
  68       puts("Found #{token_count} long-lived tokens from users:")
  69       user_ids.each do |uid|
  70         u = User.find(uid)
  71         puts("#{u.username},#{u.email},#{u.uuid}") if !u.nil?
  72       end
  73     else
  74       puts("No long-lived tokens found.")
  75     end
  76   end
  77
  78   def ll_tokens(lifetime)
  79     query = ApiClientAuthorization.where(expires_at: nil)
  80     query = query.or(ApiClientAuthorization.where("expires_at > ?", Time.now + lifetime))
  81     query
  82   end
  83 end