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