-
Notifications
You must be signed in to change notification settings - Fork 0
/
chatdb.js
executable file
·83 lines (77 loc) · 2.12 KB
/
chatdb.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
const Chat = require('./chat');
const sqlite = require('sqlite3');
const DB = require('./db');
class ChatDB extends Chat {
constructor(dbFileName) {
super();
this.db = new DB();
}
/**
* Add message by user name
* @param {string} author
* @param {string} text
*/
add(author, text, repost) {
if (!repost) {
this.db.forTable('Messages').insert({
author,
text,
time: Date.now(),
repostText: '',
repostAuthor: ''
}, (id) => {
if (!id) {
console.log('error insert message');
}
});
} else {
this.db.forTable('Messages').insert({
author,
text,
time: Date.now(),
repostText: repost.message.text,
repostAuthor: repost.message.author
}, (id) => {
if (!id) {
console.log('error insert message');
}
});
}
}
/**
* Get messages with limit
* @param {string} currentUserName
* @param {Function} callback
* @param {number} limit
*/
getAll(currentUserName, callback, limit = 40) {
this.db.forTable('Messages').select((messages) => {
if (!messages) {
callback([]);
return;
}
const data = messages.map(message => ({
...message,
isMine: message.author === currentUserName
}));
callback(data);
}, limit)
}
get(messageId, callback) {
this.db.forTable('Messages')
.filter({id: messageId})
.select((messages) => {
if (!messages) {
callback([]);
return;
}
callback(messages[0]);
})
}
put(messageId, textM, callback) {
this.db.forTable('Messages')
.filter({id: messageId})
.update([textM, messageId], callback);
}
}
module.exports = ChatDB;