hubot-aws/scripts/autoscaling/put_schedule.coffee

Hubot masters aws commands

# Description:
#   Put autoscaling scheduled action
#
# Configurations:
#   HUBOT_AWS_AS_SCHEDULE : [optional] Path to csonfile to be performs to add a scheduled action based on. Required a config_path argument or this.
#
# Commands:
#   hubot autoscaling schedule put - Put an AutoScaling Scheduled Action
#
# Notes:
#   --group_name=***        : [optional] The name of the group. If omit it, the AutoScalingGroupName of config is used.
#   --config_path=***       : [optional] Config file path. If omit it, HUBOT_AWS_AS_SCHEDULE is referred to.
#   --dry-run               : [optional] Checks whether the api request is right. Recommend to set before applying to real asset.

fs   = require 'fs'
cson = require 'cson'
util = require 'util'

getArgParams = (arg) ->
  dry_run = if arg.match(/--dry-run/) then true else false

  group_name_capture = /--group_name=(.*?)( |$)/.exec(arg)
  group_name = if group_name_capture then group_name_capture[1] else null

  config_path_capture = /--config_path=(.*?)( |$)/.exec(arg)
  config_path = if config_path_capture then config_path_capture[1] else null

  return {dry_run: dry_run, group_name: group_name, config_path: config_path}

module.exports = (robot) ->
  robot.respond /autoscaling schedule put(.*)$/i, (msg) ->
    unless require('../../auth.coffee').canAccess(robot, msg.envelope.user)
      msg.send "You cannot access this feature. Please contact an admin."
      return

    arg_params = getArgParams(msg.match[1])

    dry_run     = arg_params.dry_run
    group_name  = arg_params.group_name
    config_path = arg_params.config_path

    config_path ||= process.env.HUBOT_AWS_AS_SCHEDULE
    unless fs.existsSync config_path
      msg.send "NOT FOUND HUBOT_AWS_AS_SCHEDULE"
      return
    params = cson.parseCSONFile config_path

    params.AutoScalingGroupName = group_name

    msg.send "Requesting scheduled action, group_name=#{group_name}, config_path=#{config_path}, dry-run=#{dry_run}..."

    if dry_run
      msg.send util.inspect(params, false, null)
      return

    aws = require('../../aws.coffee').aws()
    autoscaling = new aws.AutoScaling({apiVersion: '2011-01-01'})

    autoscaling.putScheduledUpdateGroupAction params, (err, res) ->
      if err
        msg.send "PutScheduledUpdateGroupActionError: #{err}"
        return
      msg.send util.inspect(res, false, null)