mtncloud-cli 1.0.0

data/lib/mtncloud/cli/commands/user_groups_command.rb

@@ -0,0 +1,512 @@
+require 'mtncloud/cli/cli_command'
+
+class MtnCloud::Cli::UserGroupsCommand
+  include MtnCloud::Cli::CliCommand
+  include MtnCloud::Cli::AccountsHelper
+
+  set_command_name :'user-groups'
+  register_subcommands :list, :get, :add, :update, :remove
+  register_subcommands :'add-user' => :add_user
+  register_subcommands :'remove-user' => :remove_user
+  
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    @user_groups_interface = @api_client.user_groups
+    @account_users_interface = @api_client.account_users
+    @accounts_interface = @api_client.accounts
+  end
+
+  def handle(args)
+    handle_subcommand(args)
+  end
+
+  def list(args)
+    options = {}
+    params = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[search phrase]")
+      build_standard_list_options(opts, options)
+      opts.footer = "List user groups."
+    end
+    optparse.parse!(args)
+    # verify_args!(args:args, optparse:optparse, count:0)
+    options[:phrase] = args.join(" ") if args.count > 0
+    connect(options)
+    
+    params.merge!(parse_list_options(options))
+    @user_groups_interface.setopts(options)
+    if options[:dry_run]
+      print_dry_run @user_groups_interface.dry.list(nil, params)
+      return
+    end
+    json_response = @user_groups_interface.list(nil, params)
+
+    render_response(json_response, options, "userGroups") do
+      if options[:json]
+        puts as_json(json_response, options, "userGroups")
+        return 0
+      elsif options[:csv]
+        puts records_as_csv(json_response["userGroups"], options)
+        return 0
+      elsif options[:yaml]
+        puts as_yaml(json_response, options, "userGroups")
+        return 0
+      end
+      user_groups = json_response['userGroups']
+      title = "User Groups"
+      subtitles = []
+      subtitles += parse_list_subtitles(options)
+      print_h1 title, subtitles
+      if user_groups.empty?
+        print cyan,"No user groups found.",reset,"\n"
+      else
+        print cyan
+        print as_pretty_table(user_groups, list_user_group_column_definitions, options)
+        print_results_pagination(json_response)
+      end
+      print reset,"\n"
+    end
+    return 0, nil
+  end
+  
+  def get(args)
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[name]")
+      build_common_options(opts, options, [:json, :yaml, :csv, :fields, :dry_run, :remote])
+    end
+    optparse.parse!(args)
+    if args.count < 1
+      puts optparse
+      exit 1
+    end
+    connect(options)
+    id_list = parse_id_list(args)
+    return run_command_for_each_arg(id_list) do |arg|
+      _get(arg, options)
+    end
+  end
+
+  def _get(id, options)
+
+    begin
+      user_group = find_user_group_by_name_or_id(nil, id)
+      if user_group.nil?
+        return 1
+      end
+      @user_groups_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @user_groups_interface.dry.get(nil, user_group['id'])
+        return
+      end
+      json_response = @user_groups_interface.get(nil, user_group['id'])
+      user_group = json_response['userGroup']
+      users = user_group['users'] || []
+      if options[:json]
+        puts as_json(json_response, options, "userGroup")
+        return 0
+      elsif options[:yaml]
+        puts as_yaml(json_response, options, "userGroup")
+        return 0
+      elsif options[:csv]
+        puts records_as_csv([json_response['userGroup']], options)
+        return 0
+      end
+
+      print_h1 "User Group Details"
+      print cyan
+      print_description_list(user_group_column_definitions, user_group, options)
+
+      ## Users
+      print_h2 "Users (#{users.size})"
+      if users.size == 0
+        print cyan,"No users",reset,"\n"
+      else
+        user_columns = [
+          {"ID" => lambda {|user| user['id'] } },
+          {"USERNAME" => lambda {|user| user['username'] } },
+          {"NAME" => lambda {|user| user['displayName'] } },
+        ]
+        print as_pretty_table(users, user_columns)
+      end
+
+      print reset,"\n"
+
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def add(args)
+    options = {}
+    params = {}
+    user_ids = nil
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[name]")
+      opts.on('--name VALUE', String, "Name") do |val|
+        params['name'] = val
+      end
+      opts.on('--description VALUE', String, "Description") do |val|
+        params['description'] = val
+      end
+      opts.on('--sudoUser [on|off]', String, "Sudo Access") do |val|
+        params['sudoUser'] = val.nil? || val.to_s == 'on' || val.to_s == 'true'
+      end
+      opts.on('--serverGroup VALUE', String, "Server Group") do |val|
+        params['name'] = val
+      end
+      opts.on('--users LIST', Array, "Users to include in this group, comma separated list of IDs or usernames") do |list|
+        if list.size == 1 && list[0] == 'null' # hacky way to clear it
+          user_ids = []
+        else
+          user_ids = list.collect {|it| it.to_s.strip.empty? ? nil : it.to_s.strip }.compact.uniq
+        end
+      end
+      build_common_options(opts, options, [:options, :payload, :json, :dry_run, :remote, :quiet])
+      opts.footer = "Create a new user group." + "\n" +
+                    "[name] is required and can be passed as --name instead."
+    end
+    optparse.parse!(args)
+    if args.count > 1
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error  "wrong number of arguments, expected 0-1 and got #{args.count}\n#{optparse}"
+      return 1
+    end
+    # support [name] as first argument
+    if args[0]
+      params['name'] = args[0]
+    end
+    connect(options)
+    begin
+      # construct payload
+      payload = nil
+      if options[:payload]
+        payload = options[:payload]
+      else
+        # merge -O options into normally parsed options
+        params.deep_merge!(options[:options].reject {|k,v| k.is_a?(Symbol) }) if options[:options]
+        users = []
+        if user_ids
+          user_ids.each do |user_id|
+            user = find_user_by_username_or_id(nil, user_id)
+            return 1 if user.nil?
+            users << user
+          end
+          params['users'] = users.collect {|it| it['id'] }
+        end
+        # todo: prompt?
+        payload = {'userGroup' => params}
+      end
+      @user_groups_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @user_groups_interface.dry.create(nil, payload)
+        return
+      end
+      json_response = @user_groups_interface.create(nil, payload)
+      if options[:json]
+        puts as_json(json_response, options)
+      elsif !options[:quiet]
+        user_group = json_response['userGroup']
+        print_green_success "Added user group #{user_group['name']}"
+        _get(user_group['id'], {})
+      end
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+
+  def update(args)
+    options = {}
+    params = {}
+    user_ids = nil
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[name]")
+      opts.on('--name VALUE', String, "Name for this user group") do |val|
+        params['name'] = val
+      end
+      opts.on('--description VALUE', String, "Description") do |val|
+        params['description'] = val
+      end
+      opts.on('--sudoUser [on|off]', String, "Sudo Access. Default is off.") do |val|
+        params['sudoUser'] = val.nil? || val.to_s == 'on' || val.to_s == 'true'
+      end
+      opts.on('--users LIST', Array, "Users to include in this group, comma separated list of IDs or usernames") do |list|
+        if list.size == 1 && list[0] == 'null' # hacky way to clear it
+          user_ids = []
+        else
+          user_ids = list.collect {|it| it.to_s.strip.empty? ? nil : it.to_s.strip }.compact.uniq
+        end
+      end
+      build_common_options(opts, options, [:options, :payload, :json, :dry_run, :remote, :quiet])
+      opts.footer = "Update a user group." + "\n" +
+                    "[name] is required. This is the name or id of a user group."
+    end
+    optparse.parse!(args)
+    if args.count != 1
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error  "wrong number of arguments, expected 1 and got #{args.count}\n#{optparse}"
+      return 1
+    end
+    connect(options)
+    begin
+      user_group = find_user_group_by_name_or_id(nil, args[0])
+      if user_group.nil?
+        return 1
+      end
+      # construct payload
+      payload = nil
+      if options[:payload]
+        payload = options[:payload]
+      else
+        # merge -O options into normally parsed options
+        params.deep_merge!(options[:options].reject {|k,v| k.is_a?(Symbol) }) if options[:options]
+        if user_ids
+          users = []
+          user_ids.each do |user_id|
+            user = find_user_by_username_or_id(nil, user_id)
+            return 1 if user.nil?
+            users << user
+          end
+          params['users'] = users.collect {|it| it['id'] }
+        else
+          # prevent server from clearing this!
+          params['users'] = user_group['users'] #.collect {|it| it['id'] }
+        end
+        # todo: prompt?
+        payload = {'userGroup' => params}
+      end
+      @user_groups_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @user_groups_interface.dry.update(nil, user_group["id"], payload)
+        return
+      end
+      json_response = @user_groups_interface.update(nil, user_group["id"], payload)
+      if options[:json]
+        puts as_json(json_response, options)
+      elsif !options[:quiet]
+        print_green_success "Updated user group #{user_group['name']}"
+        _get(user_group['id'], {})
+      end
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def add_user(args)
+    options = {}
+    params = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[name] [user]")
+      build_common_options(opts, options, [:json, :dry_run, :remote, :quiet])
+      opts.footer = "Add a user to a user group.\n" +
+                    "[name] is required. This is the name or id of a user group.\n" +
+                    "[user] is required. This is the username or id of a user. More than one can be passed."
+    end
+    optparse.parse!(args)
+    if args.count < 2
+      puts optparse
+      return 1
+    end
+    connect(options)
+    begin
+      user_group = find_user_group_by_name_or_id(nil, args[0])
+      if user_group.nil?
+        return 1
+      end
+      
+
+      # construct payload
+      payload = nil
+      if options[:payload]
+        payload = options[:payload]
+      else
+        user_ids = args[1..-1]
+        users = []
+        user_ids.each do |user_id|
+          user = find_user_by_username_or_id(nil, user_id)
+          return 1 if user.nil?
+          users << user
+        end
+        add_user_ids = users.collect {|it| it['id'] }
+        current_users = (user_group['users'] || [])
+        current_user_ids = current_users.collect {|it| it['id'] }
+        new_user_ids = (current_user_ids + add_user_ids).uniq
+        user_group_payload = {} # user_group
+        user_group_payload['users'] = new_user_ids
+        payload = {'userGroup' => user_group_payload}
+      end
+      @user_groups_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @user_groups_interface.dry.update(nil, user_group["id"], payload)
+        return 0
+      end
+      json_response = @user_groups_interface.update(nil, user_group["id"], payload)
+      if options[:json]
+        puts as_json(json_response, options)
+      elsif !options[:quiet]
+        if users.size == 1
+            print_green_success  "Added #{users[0]['username']} to user group #{user_group['name']}"
+          else
+            print_green_success "Added #{users.size} users to user group #{user_group['name']}"
+          end
+        _get(user_group['id'], {})
+      end
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def remove_user(args)
+    options = {}
+    params = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[name] [user]")
+      build_common_options(opts, options, [:json, :dry_run, :remote, :quiet])
+      opts.footer = "Remove a user from a user group.\n" +
+                    "[name] is required. This is the name or id of a user group.\n" +
+                    "[user] is required. This is the username or id of a user. More than one can be passed."
+    end
+    optparse.parse!(args)
+    if args.count < 2
+      puts optparse
+      return 1
+    end
+    connect(options)
+    begin
+      user_group = find_user_group_by_name_or_id(nil, args[0])
+      if user_group.nil?
+        return 1
+      end
+
+      # construct payload
+      payload = nil
+      if options[:payload]
+        payload = options[:payload]
+      else
+        user_ids = args[1..-1]
+        users = []
+        user_ids.each do |user_id|
+          user = find_user_by_username_or_id(nil, user_id)
+          return 1 if user.nil?
+          users << user
+        end
+        remove_user_ids = users.collect {|it| it['id'] }
+        current_users = (user_group['users'] || [])
+        current_user_ids = current_users.collect {|it| it['id'] }
+        new_user_ids = (current_user_ids - remove_user_ids).uniq
+        user_group_payload = {} # user_group
+        user_group_payload['users'] = new_user_ids
+        payload = {'userGroup' => user_group_payload}
+      end
+      @user_groups_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @user_groups_interface.dry.update(nil, user_group["id"], payload)
+        return 0
+      end
+      json_response = @user_groups_interface.update(nil, user_group["id"], payload)
+      if options[:json]
+        puts as_json(json_response, options)
+      elsif !options[:quiet]
+        if users.size == 1
+            print_green_success  "Added #{users[0]['username']} to user group #{user_group['name']}"
+          else
+            print_green_success "Added #{users.size} users to user group #{user_group['name']}"
+          end
+        _get(user_group['id'], {})
+      end
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def remove(args)
+    options = {}
+    params = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[name]")
+      build_common_options(opts, options, [:json, :dry_run, :quiet, :auto_confirm])
+    end
+    optparse.parse!(args)
+    if args.count < 1
+      puts optparse
+      return 127
+    end
+    connect(options)
+
+    begin
+      user_group = find_user_group_by_name_or_id(nil, args[0])
+      if user_group.nil?
+        return 1
+      end
+
+      unless options[:yes] || ::MtnCloud::Cli::OptionTypes::confirm("Are you sure you would like to delete user group '#{user_group['name']}'?", options)
+        return false
+      end
+
+      # payload = {
+      #   'userGroup' => {id: user_group["id"]}
+      # }
+      # payload['userGroup'].merge!(user_group)
+      payload = params
+      @user_groups_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @user_groups_interface.dry.destroy(nil, user_group["id"])
+        return
+      end
+
+      json_response = @user_groups_interface.destroy(nil, user_group["id"])
+      if options[:json]
+        puts as_json(json_response, options)
+      elsif !options[:quiet]
+        print_green_success "Deleted user group #{user_group['name']}"
+      end
+      return 0, nil
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      return 1
+    end
+  end
+
+
+  private
+
+  def print_user_groups_table(user_groups, opts={})
+    columns = [
+      {"ID" => lambda {|user_group| user_group['id'] } },
+      {"NAME" => lambda {|user_group| user_group['name'] } },
+      {"DESCRIPTION" => lambda {|user_group| user_group['description'] } },
+      {"USERS" => lambda {|user_group| 
+        users = user_group['users']
+        if users
+          n_users = 3
+          if users.size > n_users
+            users.first(n_users).collect { |user| user['username'] }.join(", ") + ", (#{users.size - n_users} more)"
+          else
+            users.collect { |user| user['username'] }.join(", ")
+          end
+        else
+          ""
+        end
+      } },
+      # {"KEY PAIR" => lambda {|user_group| 
+      #   user_group['keyPair'] ? user_group['keyPair']['name'] : user_group['sharedKeyPairId']
+      # } },
+      # {"SUDO" => lambda {|user_group| format_boolean(user_group['sudoUser']) } },
+    ]
+    if opts[:include_fields]
+      columns = opts[:include_fields]
+    end
+    print as_pretty_table(user_groups, columns, opts)
+  end
+
+end