/
store.js
79 lines (64 loc) · 1.77 KB
/
store.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
var fs = require('fs');
var db = 'scraps.json';
var userTokens = {};
var user= [];
var groupinfo = {};
exports.getUserToken = function (userId) {
return userTokens[userId];
}
exports.getid = function () {
return user;
}
// exports.removeuser = function (userid) {
// }
exports.saveGroup = function(res) {
groupinfo = res[2];
}
exports.returnGroup = function(){
return groupinfo;
}
exports.saveUserToken = function (userId, token) {
userTokens[userId] = token;
user.push(userId);
user.forEach(function(i){
console.log('int token call' + i)
})
}
var chats = {};
var conversationId = function (userId, chat) {
if (chat.startsWith('u:') || chat.startsWith('b:')) {
return [userId, chat].sort().join('|');
} else {
return chat;
}
}
exports.saveScrap = function (userId, chat, text) {
var cid = conversationId(userId, chat);
if (!chats[cid]) {
chats[cid] = [text];
} else {
chats[cid].push(text);
}
};
exports.listScraps = function (userId, chat) {
return chats[conversationId(userId, chat)];
};
var readDatabase = function () {
try {
var stringContent = fs.readFileSync(db);
var content = JSON.parse(stringContent);
userTokens = content.userTokens;
chats = content.chats;
} catch (e) {
console.log('Couldn\'t read db, starting empty.\nError:', e);
}
};
var saveDatabase = function () {
console.log('Saving db');
var stringContent = JSON.stringify({ userTokens: userTokens, chats: chats });
fs.writeFileSync(db, stringContent);
};
readDatabase();
process.on('SIGINT', function () { console.log('SIGINT'); process.exit(); });
process.on('SIGTERM', function () { console.log('SIGTERM'); process.exit(); });
process.on('exit', saveDatabase);