Skip to content

Instantly share code, notes, and snippets.

@johnbeynon
Created November 12, 2011 22:19
Show Gist options
  • Save johnbeynon/1361213 to your computer and use it in GitHub Desktop.
Save johnbeynon/1361213 to your computer and use it in GitHub Desktop.
Heroku Hubot Script
# A limited way to interact with the Heroku API.
#
# INSTALLATION:
# 1. Create file in scripts folder in hubot folder
# 2. Update package.json for hubot and add dependency on "sprintf": "0.1.1"
# 3. Set heroku config variable HEROKU_USER to heroku user account to use
# 4. Set heroku config variable HEROKU_APIKEY to heroku user account apikey (from My Account page)
#
# heroku status - Retrieve the most recent tweet from the @herokustatus account
# heroku ps --app <appname> - Get process information for application
# heroku ps:scale <process>=<quantity> --app <appname> - Scale the process to quantity for application
# heroku ps:restart <process> --app <appname> - Restart the specified process for application#
# heroku ps:stop <process> --app <appname> - Stop the specified process for application
#
# This is a WORK IN PROGRESS
#
sprintf = require("sprintf").sprintf
user = process.env.HEROKU_USER
apiKey = process.env.HEROKU_APIKEY
module.exports = (robot) ->
# heroku ps --app <appname> - returns process information of heroku app
robot.respond /heroku ps --app (.*)/i, (msg) ->
application = msg.match[1]
uri = 'https://api.heroku.com/apps/' + application + '/ps'
# get(uri)
msg.http(uri)
.headers(Authorization: "Basic #{new Buffer("#{user}:#{apiKey}").toString("base64")}", Accept: "application/json")
.get() (err, res, body) ->
if res.statusCode is 404
msg.send "Application not found"
return
results = JSON.parse(body)
if res.statusCode is 403
msg.send results.error
return
output = ""
output += "Process State Command\n"
output += "------------ ------------------ ------------------------------\n"
for process in results
output += sprintf("%-12s %-18s %-s", process.process, process.pretty_state, process.command) + "\n"
msg.send output
# heroku ps:scale <process>=<count> --app <appname>
robot.respond /heroku ps:scale (.*)=(\d+) --app (.*)/i, (msg) ->
type = msg.match[1]
qty = msg.match[2]
application = msg.match[3]
uri = 'https://api.heroku.com/apps/' + application + '/ps/scale'
msg.http(uri)
.headers(Authorization: "Basic #{new Buffer("#{user}:#{apiKey}").toString("base64")}", Accept: "application/json", 'Content-Length': 0)
.query(type: type, qty: qty)
.post() (err, res, body) ->
if res.statusCode is 404
msg.send "Application not found"
return
if res.statusCode is 403
msg.send JSON.parse(body).error
return
msg.send "Now running: " + body
# heroku ps:restart <process> --app <appname>
robot.respond /heroku ps:restart (.*) --app (.*)/i, (msg) ->
type = msg.match[1]
application = msg.match[2]
uri = 'https://api.heroku.com/apps/' + application + '/ps/restart'
msg.http(uri)
.headers(Authorization: "Basic #{new Buffer("#{user}:#{apiKey}").toString("base64")}", Accept: "application/json", 'Content-Length': 0)
.query(type: type)
.post() (err, res, body) ->
if res.statusCode is 404
msg.send "Application not found"
return
if res.statusCode is 403
msg.send JSON.parse(body).error
return
msg.send "Restarted " + type + " processes"
# heroku ps:stop <process> --app <appname>
robot.respond /heroku ps:stop (.*) --app (.*)/i, (msg) ->
type = msg.match[1]
application = msg.match[2]
uri = 'https://api.heroku.com/apps/' + application + '/ps/stop'
msg.http(uri)
.headers(Authorization: "Basic #{new Buffer("#{user}:#{apiKey}").toString("base64")}", Accept: "application/json", 'Content-Length': 0)
.query(type: type)
.post() (err, res, body) ->
if res.statusCode is 404
msg.send "Application not found"
return
if res.statusCode is 403
msg.send JSON.parse(body).error
return
msg.send "Stopped " + type + " processes"
# heroku status
robot.respond /heroku status/, (msg) ->
msg.http("http://api.twitter.com/1/statuses/user_timeline/herokustatus.json?count=1")
.get() (err, res, body) ->
response = JSON.parse body
if response[0]
msg.send "#{response[0]['created_at']} > #{response[0]['text']}"
else
msg.send "Error"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment