This repository has been archived by the owner on Jun 8, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 1.9k
/
Copy pathredis-brain.coffee
62 lines (52 loc) · 1.63 KB
/
redis-brain.coffee
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# Description:
# None
#
# Dependencies:
# "redis": "0.8.4"
#
# Configuration:
# REDISTOGO_URL or REDISCLOUD_URL or BOXEN_REDIS_URL or REDIS_URL.
# URL format: redis://<host>:<port>[/<brain_prefix>]
# If not provided, '<brain_prefix>' will default to 'hubot'.
#
# Commands:
# None
#
# Authors:
# atmos
# jan0sch
# spajus
Url = require "url"
Redis = require "redis"
module.exports = (robot) ->
info = Url.parse process.env.REDISTOGO_URL or process.env.REDISCLOUD_URL or process.env.BOXEN_REDIS_URL or process.env.REDIS_URL or 'redis://localhost:6379', true
client = Redis.createClient(info.port, info.hostname)
prefix = info.path?.replace('/', '') or 'hubot'
robot.brain.setAutoSave false
getData = ->
client.get "#{prefix}:storage", (err, reply) ->
if err
throw err
else if reply
robot.logger.info "Data for #{prefix} brain retrieved from Redis"
robot.brain.mergeData JSON.parse(reply.toString())
else
robot.logger.info "Initializing new data for #{prefix} brain"
robot.brain.mergeData {}
robot.brain.setAutoSave true
if info.auth
client.auth info.auth.split(":")[1], (err) ->
if err
robot.logger.error "Failed to authenticate to Redis"
else
robot.logger.info "Successfully authenticated to Redis"
getData()
client.on "error", (err) ->
robot.logger.error err
client.on "connect", ->
robot.logger.debug "Successfully connected to Redis"
getData() if not info.auth
robot.brain.on 'save', (data = {}) ->
client.set "#{prefix}:storage", JSON.stringify data
robot.brain.on 'close', ->
client.quit()