-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp
executable file
·65 lines (52 loc) · 1.36 KB
/
app
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
#!/usr/bin/node
var Promise = require('bluebird');
var config = require('./knexfile')[process.env.NODE_ENV || 'production'];
var knex = require('knex')(config);
var stdin = process.stdin;
function readStdin() {
var payload = [];
var length = 0;
return new Promise(function(resolve, reject) {
function resolveWithPayload() {
return resolve(Buffer.concat(payload, length));
}
function collectChunks() {
var chunk;
while (chunk = stdin.read()) {
payload.push(chunk);
length += chunk.length;
}
}
if (stdin.isTTY) {
return resolveWithPayload();
}
stdin.on('readable', collectChunks);
stdin.on('end', resolveWithPayload);
stdin.on('error', reject);
});
}
function store(payload) {
payload || (payload = new Buffer(''));
var now = new Date();
var data = {
to: process.env.USER,
context: process.env.LABEL,
content_type: process.env.CONTENT_TYPE,
payload: payload,
created_at: now,
updated_at: now,
};
data.context == null && delete data.context;
data.content_type == null && delete data.content_type;
if (data.to) {
return knex('messages').insert(data);
}
}
function cleanup() {
knex.destroy();
}
function errorHandler(err) {
process.exitCode = 1;
process.stderr.write(err.toString());
}
readStdin().then(store).catch(errorHandler).finally(cleanup);