-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.js
77 lines (67 loc) · 1.43 KB
/
db.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
const sqlite3 = require('sqlite3').verbose();
const db = new sqlite3.Database('./db.sqlite3');
try {
db.run(
`CREATE TABLE IF NOT EXISTS messages (
id INTEGER PRIMARY KEY,
content TEXT,
paid BOOLEAN,
createdAt DATETIME
)`
)
} catch(error) {
console.error(error);
}
db.getAllMessages = () => {
return new Promise((resolve, reject) => {
db.all('SELECT * FROM messages', [], (err, messages) => {
if (err) {
reject(err);
} else {
resolve(messages);
}
});
});
};
db.getAllPaidMessages = () => {
return new Promise((resolve, reject) => {
db.all('SELECT * FROM messages WHERE paid=1', [], (err, messages) => {
if (err) {
reject(err);
} else {
resolve(messages);
}
});
});
};
db.createMessage = (fields) => {
return new Promise((resolve, reject) => {
db.run(
`INSERT INTO messages
(content, paid, createdAt)
VALUES(?, ?, ?)`,
Object.values(fields),
function(err) {
if (err) {
reject(err);
} else {
resolve(this.lastID);
}
});
});
};
db.markMessageAsPaid = (id) => {
return new Promise((resolve, reject) => {
db.run(
`UPDATE messages SET paid=1 WHERE id=?`,
[id],
function(err) {
if (err) {
reject(err);
} else {
resolve(this.lastID);
}
});
});
}
module.exports = db;