forked from psycho237-prog/Psychobot
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdatabase.js
More file actions
72 lines (64 loc) · 2.25 KB
/
database.js
File metadata and controls
72 lines (64 loc) · 2.25 KB
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
// database.js
const sqlite3 = require('sqlite3').verbose();
const db = new sqlite3.Database('./bot.db');
const log = require('./logger')(module); // Utilise le logger pour la cohérence
db.serialize(() => {
log("Connexion à SQLite réussie.");
db.run(`
CREATE TABLE IF NOT EXISTS users (
id TEXT PRIMARY KEY,
name TEXT,
firstSeen TEXT,
commandCount INTEGER DEFAULT 0
)
`);
});
function getOrRegisterUser(userId, name) {
return new Promise((resolve, reject) => {
db.get("SELECT * FROM users WHERE id = ?", [userId], (err, row) => {
if (err) return reject(err);
if (row) {
resolve(row);
} else {
const firstSeen = new Date().toISOString();
db.run("INSERT INTO users (id, name, firstSeen) VALUES (?, ?, ?)", [userId, name, firstSeen], (err) => {
if (err) return reject(err);
log(`Nouvel utilisateur enregistré : ${name} (${userId})`);
resolve({ id: userId, name, firstSeen, commandCount: 0 });
});
}
});
});
}
function incrementCommandCount(userId) {
return new Promise((resolve, reject) => {
db.run("UPDATE users SET commandCount = commandCount + 1 WHERE id = ?", [userId], (err) => {
if (err) return reject(err);
resolve();
});
});
}
// --- LES FONCTIONS MANQUANTES SONT ICI ---
function getTotalUsers() {
return new Promise((resolve, reject) => {
db.get("SELECT COUNT(*) as count FROM users", (err, row) => {
if (err) return reject(err);
resolve(row.count || 0);
});
});
}
function getTotalCommands() {
return new Promise((resolve, reject) => {
db.get("SELECT COALESCE(SUM(commandCount), 0) as total FROM users", (err, row) => {
if (err) return reject(err);
resolve(row.total || 0);
});
});
}
// --- ON S'ASSURE QU'ELLES SONT BIEN EXPORTÉES ---
module.exports = {
getOrRegisterUser,
incrementCommandCount,
getTotalUsers, // <--- Ligne cruciale
getTotalCommands, // <--- Ligne cruciale
};