lucien/odin
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

commit

Browse source
This commit is contained in:
Lukian LEIZOUR 2023-03-04 11:19:15 +01:00
parent 51e6f97c5f
commit e2f51ddee9
1 changed files with 83 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

66
app.js
View file

@ -25,25 +25,55 @@ const connection = mysql.createConnection({
//mysql functions
function addUserToDb(id, user) {
return new Promise((resolve, reject) => {
connection.query('INSERT INTO users(userid, username) VALUES("' + id + '","' + user + '")' , (error, results, fields) => {
if (error) {
console.error('Erreur dans la requête : ', error);
reject(error);
} else {
console.log('Les résultats de la requête : ', results);
return true;
resolve();
}
});
});
}
function incrementQuota(id) {
return new Promise((resolve, reject) => {
connection.query('UPDATE users SET quota = quota + 1 WHERE userid = ' + id, (error, results, fields) => {
if (error) {
console.error('Erreur dans la requête : ', error);
reject(error);
} else {
console.log('Les résultats de la requête : ', results);
resolve();
}
});
});
}
function usersInDb() {
return new Promise((resolve, reject) => {
connection.query('SELECT userid FROM users', (error, results, fields) => {
if (error) {
reject(error);
} else {
users = [];
results.forEach(element => {
users.push(element.userid);
});
resolve(users);
}
});
});
}
function getQuota(id) {
return new Promise((resolve, reject) => {
connection.query('SELECT quota FROM users WHERE userid = ' + id, (error, results, fields) => {
if (error) {
reject(error);
} else {
resolve(results[0].quota);
}
});
});
}
//Telegram commands
@ -188,30 +218,17 @@ client.on('interactionCreate', async interaction => {
if (interaction.commandName === 'gptrequest') {
await interaction.deferReply();
connection.query('SELECT userid FROM users', (error, results, fields) => {
if (error) {
console.error('Erreur dans la requête : ', error);
} else {
var users = [];
results.forEach(element => {
users.push(element.userid);
});
users = await usersInDb();
if (!(users.includes(interaction.member.user.id))) {
addUserToDb(interaction.member.user.id, interaction.member.user.username);
addToLogs('[Discord] Added user to the database : ' + interaction.member.user.username);
console.log('[Discord] Added user to the database : ' + interaction.member.user.username);
}
}
});
setTimeout(() => {
connection.query('SELECT quota FROM users WHERE userid = '+ interaction.member.user.id, (error, results, fields) => {
if (error) {
console.error('Erreur dans la requête : ', error);
} else {
if (results[0].quota >= 999) {
quota = await getQuota(interaction.member.user.id);
if (quota >= 999) {
interaction.editReply('Quota exceeded, please wait');
}
else {
@ -245,9 +262,6 @@ client.on('interactionCreate', async interaction => {
addToLogs('[Discord] Generating answer to : ' + interaction.options.get('question').value);
}
}
});
}, 100);
}
else if (interaction.commandName === 'info') {
console.log(interaction)