services/api/script/setup-new-user.rb

   1 #!/usr/bin/env ruby
   2
   3 abort 'Error: Ruby >= 1.9.3 required.' if RUBY_VERSION < '1.9.3'
   4
   5 require 'logger'
   6 require 'trollop'
   7 log = Logger.new STDERR
   8 log.progname = $0.split('/').last
   9
  10 opts = Trollop::options do
  11   banner ''
  12   banner "Usage: #{log.progname} " +
  13     "{user_uuid_or_email} {user_and_repo_name} {vm_uuid}"
  14   banner ''
  15   opt :debug, <<-eos
  16 Show debug messages.
  17   eos
  18   opt :create, <<-eos
  19 Create a new user with the given email address if an existing user \
  20 is not found.
  21   eos
  22   opt :openid_prefix, <<-eos, default: 'https://www.google.com/accounts/o8/id'
  23 If creating a new user record, require authentication from an OpenID \
  24 with this OpenID prefix *and* a matching email address in order to \
  25 claim the account.
  26   eos
  27 end
  28
  29 log.level = (ENV['DEBUG'] || opts.debug) ? Logger::DEBUG : Logger::WARN
  30
  31 if ARGV.count != 3
  32   Trollop::die "required arguments are missing"
  33 end
  34 user_arg, user_repo_name, vm_uuid = ARGV
  35
  36 require 'arvados'
  37 arv = Arvados.new(api_version: 'v1')
  38
  39 begin
  40   new_user = arv.user.create(user_param: user_arg, repo_name: user_repo_name, vm_uuid: vm_uuid, openid_prefix: opts.openid_prefix, user: {})
  41   log.warn new_user
  42 rescue Exception => e #Arvados::TransactionFailedError
  43   log.warn e.message
  44 end