/
index.js
89 lines (72 loc) 路 1.97 KB
/
index.js
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
const _ = require('lodash');
const fs = require('fs');
const DB_URL = 'https://raw.githubusercontent.com/chamerling/hubot-lingif/master/db.json';
const DEFAULT_GIF = 'http://i.giphy.com/QHKMjJN1sG2sw.gif';
module.exports = (robot) => {
robot.hear(/lingif (.*)$/i, getGif);
robot.respond(/lingif (.*)$/i, getGif);
function getGif(msg) {
get(msg.match[1])
.then(gif => {
robot.logger.debug(`gif result to ${msg.match[1]} is ${gif}`);
msg.send(gif);
})
.catch(err => {
robot.logger.debug(err)
msg.send(`:kader: is always here for you, even when bot fails ${DEFAULT_GIF}`);
});
}
function get(text) {
return load().then(data => {
if (!data[text] || !data[text].length) {
throw new Error(`Can not find matching result for ${text}`);
}
return _.sample(data[text]);
});
}
function load() {
return loadRemoteDatabase()
.catch(err => {
robot.logger.error(err);
return loadLocalDatabase();
});
}
function loadLocalDatabase() {
const promise = new Promise((resolve, reject) => {
fs.readFile('./db.json', 'utf8', (err, data) => {
if (err) {
return reject(err);
}
resolve(JSON.parse(data));
});
});
return promise;
}
function loadRemoteDatabase() {
const promise = new Promise((resolve, reject) => {
robot
.http(DB_URL)
.header('Accept', 'application/json')
.get()((err, res, body) => {
if (body) {
const db = parseDB(JSON.parse(body));
return resolve(db);
}
reject(err);
});
});
return promise;
}
function parseDB(json) {
const result = {};
_.forEach(json, (value, key) => {
_.map(key.split(','), _.trim).forEach(key => {
if (!result[key]) {
result[key] = [];
}
result[key] = _.concat(result[key], value);
});
});
return result;
}
};