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   opt :n, 'Do not change anything, just probe'
  28 end
  29
  30 log.level = (ENV['DEBUG'] || opts.debug) ? Logger::DEBUG : Logger::WARN
  31
  32 if ARGV.count != 3
  33   Trollop::die "required arguments are missing"
  34 end
  35 user_arg, user_repo_name, vm_uuid = ARGV
  36
  37 require 'arvados'
  38 arv = Arvados.new(api_version: 'v1')
  39
  40 begin
  41         new_user = arv.user.create(user_param: user_arg, repo_name: user_repo_name, vm_uuid: vm_uuid, openid_prefix: opts.openid_prefix, just_probe: opts.n, user: {})
  42         log.warn new_user
  43 rescue Exception => e #Arvados::TransactionFailedError
  44         log.warn e.message
  45 end
  46