1
0
Fork 0
mirror of https://github.com/openstf/stf synced 2025-10-04 18:29:17 +02:00

Add Slack notifier.

This commit is contained in:
Gunther Brunner 2016-01-26 20:10:49 +09:00
parent 25f81aa80f
commit d61e31706e
3 changed files with 121 additions and 0 deletions

View file

@ -617,6 +617,45 @@ program
}) })
}) })
program
.command('notify-slack')
.description('start Slack notifier')
.option('-t, --token <token>'
, 'Slack API token (or $SLACK_TOKEN)'
, String
, process.env.SLACK_TOKEN)
.option('-c, --channel #<channel>'
, 'Slack channel (or $SLACK_CHANNEL)'
, String
, process.env.SLACK_CHANNEL)
.option('-p, --priority <level>'
, 'minimum log level'
, Number
, logger.Level.IMPORTANT)
.option('-s, --connect-sub <endpoint>'
, 'sub endpoint'
, cliutil.list)
.action(function(options) {
if (!options.token) {
this.missingArgument('--token')
}
if (!options.channel) {
this.missingArgument('--channel')
}
if (!options.connectSub) {
this.missingArgument('--connect-sub')
}
require('./units/notify/slack')({
token: options.token
, channel: options.channel
, priority: options.priority
, endpoints: {
sub: options.connectSub
}
})
})
program program
.command('log-rethinkdb') .command('log-rethinkdb')
.description('start a rethinkdb log recorder') .description('start a rethinkdb log recorder')

81
lib/units/notify/slack.js Normal file
View file

@ -0,0 +1,81 @@
var util = require('util')
var WebClient = require('slack-client').WebClient
var Promise = require('bluebird')
var logger = require('../../util/logger')
var wire = require('../../wire')
var wirerouter = require('../../wire/router')
var wireutil = require('../../wire/util')
var lifecycle = require('../../util/lifecycle')
var srv = require('../../util/srv')
var zmqutil = require('../../util/zmqutil')
module.exports = function(options) {
var log = logger.createLogger('notify-slack')
var client = new WebClient(options.token)
var buffer = []
var timer
// Input
var sub = zmqutil.socket('sub')
Promise.map(options.endpoints.sub, function(endpoint) {
return srv.resolve(endpoint).then(function(records) {
return srv.attempt(records, function(record) {
log.info('Receiving input from "%s"', record.url)
sub.connect(record.url)
return Promise.resolve(true)
})
})
})
// Establish always-on channels
;[wireutil.global].forEach(function(channel) {
log.info('Subscribing to permanent channel "%s"', channel)
sub.subscribe(channel)
})
function push() {
buffer.splice(0).forEach(function(entry) {
var format = entry.message.indexOf('\n') === -1 ? '`%s`' : '```%s```'
var message = util.format(format, entry.message)
client.chat.postMessage(options.channel, util.format(
'>>> *%s/%s* %d [*%s*] %s'
, logger.LevelLabel[entry.priority]
, entry.tag
, entry.pid
, entry.identifier
, message
)
, {
username: 'STF'
, icon_url: 'https://openstf.io/favicon.png'
}
)
})
}
sub.on('message', wirerouter()
.on(wire.DeviceLogMessage, function(channel, message) {
if (message.priority >= options.priority) {
buffer.push(message)
clearTimeout(timer)
timer = setTimeout(push, 1000)
}
})
.handler())
log.info('Listening for %s (or higher) level log messages',
logger.LevelLabel[options.priority])
lifecycle.observe(function() {
try {
sub.close()
}
catch (err) {
// No-op
}
})
}

View file

@ -71,6 +71,7 @@
"semver": "^5.0.1", "semver": "^5.0.1",
"serve-favicon": "^2.2.0", "serve-favicon": "^2.2.0",
"serve-static": "^1.9.2", "serve-static": "^1.9.2",
"slack-client": "^2.0.0-beta.3",
"socket.io": "1.4.4", "socket.io": "1.4.4",
"split": "^1.0.0", "split": "^1.0.0",
"stf-appstore-db": "^1.0.0", "stf-appstore-db": "^1.0.0",