232 lines
12 KiB
JavaScript
232 lines
12 KiB
JavaScript
const dotenv = require('dotenv');
|
|
dotenv.config();
|
|
const debugMode = process.env.DEBUG || true;
|
|
const mysql = require('mysql');
|
|
let db = mysql.createConnection({
|
|
host : process.env.DBHOST,
|
|
user : process.env.DBUSER,
|
|
password : process.env.DBPASS,
|
|
database : process.env.DBNAME,
|
|
port : process.env.DBPORT
|
|
});
|
|
|
|
/* Table Structures
|
|
guild_info
|
|
+------------------------+-------------+------+-----+---------+----------------+
|
|
| Field | Type | Null | Key | Default | Extra |
|
|
+------------------------+-------------+------+-----+---------+----------------+
|
|
| guild_id | varchar(50) | NO | PRI | NULL | auto_increment |
|
|
| tree_name | varchar(100)| NO | | | |
|
|
| tree_height | varchar(10) | NO | | 0 | |
|
|
| tree_message_id | varchar(50) | NO | | | |
|
|
| tree_channel_id | varchar(50) | NO | | | |
|
|
| leaderboard_message_id | varchar(50) | NO | | | |
|
|
| leaderboard_channel_id | varchar(50) | NO | | | |
|
|
+------------------------+-------------+------+-----+---------+----------------+
|
|
*/
|
|
/*
|
|
leaderboard
|
|
+-------------+--------------+------+-----+---------+----------------+
|
|
| Field | Type | Null | Key | Default | Extra |
|
|
+-------------+--------------+------+-----+---------+----------------+
|
|
| id | int(10) | NO | PRI | NULL | auto_increment |
|
|
| guild_id | varchar(50) | NO | | | |
|
|
| tree_name | varchar(100) | NO | | | |
|
|
| tree_rank | int(10) | NO | | | |
|
|
| tree_height | int(10) | NO | | 1 | |
|
|
| has_pin | tinyint(1) | NO | | 0 | |
|
|
| timestamp | varchar(50) | NO | | | |
|
|
+-------------+--------------+------+-----+---------+----------------+
|
|
*/
|
|
|
|
db.connect((err) => {
|
|
if (err) throw `Error connecting to the database: ${err.message}`;
|
|
});
|
|
|
|
db.on('error', function(err) {
|
|
db = mysql.createConnection({
|
|
host : process.env.DBHOST,
|
|
user : process.env.DBUSER,
|
|
password : process.env.DBPASS,
|
|
database : process.env.DBNAME,
|
|
port : process.env.DBPORT
|
|
});
|
|
db.connect((err) => {
|
|
if (err) throw `Error connecting to the database: ${err.message}`;
|
|
});
|
|
});
|
|
|
|
|
|
module.exports = {
|
|
createGuildTables(guildId) {
|
|
// Create the guild-information and rank-information tables to be used.
|
|
const createGuildInfoTableQuery = "CREATE TABLE IF NOT EXISTS guild_info(guild_id VARCHAR(50) NOT NULL, tree_name VARCHAR(100) NOT NULL DEFAULT 'Run /setup where your tree is.', tree_height INT(10) NOT NULL DEFAULT 0, tree_message_id VARCHAR(50) NOT NULL DEFAULT 'Run /setup where your tree is.', tree_channel_id VARCHAR(50) NOT NULL DEFAULT 'Run /setup where your tree is.', leaderboard_message_id VARCHAR(50) NOT NULL DEFAULT 'Run /setup where your leaderboard is.', leaderboard_channel_id VARCHAR(50) NOT NULL DEFAULT 'Run /setup where your leaderboard is.', CONSTRAINT guild_pk PRIMARY KEY (guild_id))";
|
|
const createLeaderboardTableQuery = "CREATE TABLE IF NOT EXISTS leaderboard(id INT(10) NOT NULL AUTO_INCREMENT,guild_id VARCHAR(50) NOT NULL,tree_name VARCHAR(100) NOT NULL,tree_rank INT(10) NOT NULL,tree_height INT(10) NOT NULL DEFAULT 1,has_pin TINYINT(1) NOT NULL DEFAULT 0,timestamp DATETIME DEFAULT CURRENT_TIMESTAMP, CONSTRAINT id_pk PRIMARY KEY(id))";
|
|
// TODO run the queries, then add a call to this function at the beginning of main.js or functions.js
|
|
return new Promise((resolve, reject) => {
|
|
db.query(createGuildInfoTableQuery, (err) => {
|
|
if (err) {
|
|
reject("Error creating the guild_info table: " + err.message);
|
|
console.error("Offending query: " + createGuildInfoTableQuery);
|
|
return;
|
|
}
|
|
db.query(createLeaderboardTableQuery, (err) => {
|
|
if (err) {
|
|
reject("Error creating the leaderboard table: " + err.message);
|
|
console.error("Offending query: " + createLeaderboardTableQuery);
|
|
return;
|
|
}
|
|
resolve({ "status": "Successfully checked both tables.", "data": null })
|
|
});
|
|
});
|
|
});
|
|
},
|
|
getGuildInfo(guildId) {
|
|
// Get a server's tree information from the database
|
|
const selectGuildInfoQuery = `SELECT tree_name, tree_height, tree_message_id, tree_channel_id, leaderboard_message_id, leaderboard_channel_id FROM guild_info WHERE guild_id = ${db.escape(guildId)}`;
|
|
// TODO run this query and return a promise then structure the output into a GuildInfo object. resolve with { "status": , "data": guildInfo }
|
|
return new Promise((resolve, reject) => {
|
|
db.query(selectGuildInfoQuery, (err, res) => {
|
|
if (err) {
|
|
console.error(err);
|
|
reject("Error fetching guild information: " + err.message);
|
|
return;
|
|
}
|
|
/*const guildInfo = { "guildId": "123",
|
|
"treeName": "name",
|
|
"treeHeight": 123,
|
|
"treeMessageId": "123",
|
|
"treeChannelId": "123",
|
|
"leaderboardMessageId": "123",
|
|
"leaderboardChannelId": "123"
|
|
};*/
|
|
if (res.length == 0) {
|
|
reject("There is no database entry for your guild yet. Try running /setup");
|
|
return;
|
|
}
|
|
row = res[0];
|
|
const guildInfo = { "guildId": row.guild_id,
|
|
"treeName": row.tree_name,
|
|
"treeHeight": row.tree_height,
|
|
"treeMessageId": row.tree_message_id,
|
|
"treeChannelId": row.tree_channel_id,
|
|
"leaderboardMessageId": row.leaderboard_message_id,
|
|
"leaderboardChannelId": row.leaderboard_channel_id
|
|
};
|
|
resolve({ "status": "Successfully fetched guild information", "data": guildInfo })
|
|
});
|
|
});
|
|
},
|
|
setGuildInfo(guildInfo) {
|
|
// Returns a Promise, resolve({ "status": "", "data": null })
|
|
// guildInfo = { "guildId": "123", "treeName": "name", "treeHeight": 123, "treeMessageId": "123", "treeChannelId": "123", "leaderboardMessageId": "123", "leaderboardChannelId": "123"}
|
|
// Set a server's tree information in the database
|
|
const insertGuildInfoQuery = `INSERT INTO guild_info (guild_id, tree_name, tree_height, tree_message_id, tree_channel_id, leaderboard_message_id, leaderboard_channel_id) VALUES (${db.escape(guildInfo.guildId)}, ${db.escape(guildInfo.treeName)}, ${db.escape(guildInfo.treeHeight)},${db.escape(guildInfo.treeMessageId)}, ${db.escape(guildInfo.treeChannelId)}, ${db.escape(guildInfo.leaderboardMessageId)}, ${db.escape(guildInfo.leaderboardChannelId)}) ON DUPLICATE KEY UPDATE tree_name = ${db.escape(guildInfo.treeName)},tree_height = ${db.escape(guildInfo.treeHeight)},tree_message_id = ${db.escape(guildInfo.treeMessageId)},tree_channel_id = ${db.escape(guildInfo.treeChannelId)},leaderboard_message_id = ${db.escape(guildInfo.leaderboardMessageId)},leaderboard_channel_id = ${db.escape(guildInfo.leaderboardChannelId)}`;
|
|
// TODO run this query and return a promise, then resolve with { "status": , "data": null }
|
|
return new Promise((resolve, reject) => {
|
|
db.query(insertGuildInfoQuery, (err, res) => {
|
|
if (err) {
|
|
console.error(err);
|
|
reject("Error setting the guild info: " + err.message);
|
|
return;
|
|
}
|
|
resolve({ "status": "Successfully set the guild information", "data": null });
|
|
});
|
|
});
|
|
},
|
|
setTreeInfo(guildInfo) {
|
|
// Returns a Promise, resolve({ "status": "", "data": null })
|
|
// guildInfo = { "guildId": "123", "treeName": "name", "treeHeight": 123, "treeMessageId": "123", "treeChannelId": "123", "leaderboardMessageId": "123", "leaderboardChannelId": "123"}
|
|
// Set a server's tree information in the database)
|
|
const insertGuildInfoQuery = `INSERT INTO guild_info (guild_id, tree_name, tree_height, tree_message_id, tree_channel_id) VALUES (${db.escape(guildInfo.guildId)}, ${db.escape(guildInfo.treeName)}, ${db.escape(guildInfo.treeHeight)},${db.escape(guildInfo.treeMessageId)}, ${db.escape(guildInfo.treeChannelId)}) ON DUPLICATE KEY UPDATE tree_name = ${db.escape(guildInfo.treeName)},tree_height = ${db.escape(guildInfo.treeHeight)},tree_message_id = ${db.escape(guildInfo.treeMessageId)},tree_channel_id = ${db.escape(guildInfo.treeChannelId)}`;
|
|
// TODO run this query and return a promise, then resolve with { "status": , "data": null }
|
|
return new Promise((resolve, reject) => {
|
|
db.query(insertGuildInfoQuery, (err, res) => {
|
|
if (err) {
|
|
console.error(err);
|
|
reject("Error setting the guild info: " + err.message);
|
|
return;
|
|
}
|
|
resolve({ "status": "Successfully set the guild information", "data": null });
|
|
});
|
|
});
|
|
},
|
|
setLeaderboardInfo(guildInfo) {
|
|
// Returns a Promise, resolve({ "status": "", "data": null })
|
|
// guildInfo = { "guildId": "123", "treeName": "name", "treeHeight": 123, "treeMessageId": "123", "treeChannelId": "123", "leaderboardMessageId": "123", "leaderboardChannelId": "123"}
|
|
// Set a server's tree information in the database
|
|
const insertGuildInfoQuery = `INSERT INTO guild_info (guild_id, leaderboard_message_id, leaderboard_channel_id) VALUES (${db.escape(guildInfo.guildId)}, ${db.escape(guildInfo.leaderboardMessageId)}, ${db.escape(guildInfo.leaderboardChannelId)}) ON DUPLICATE KEY UPDATE leaderboard_message_id = ${db.escape(guildInfo.leaderboardMessageId)},leaderboard_channel_id = ${db.escape(guildInfo.leaderboardChannelId)}`;
|
|
// TODO run this query and return a promise, then resolve with { "status": , "data": null }
|
|
return new Promise((resolve, reject) => {
|
|
db.query(insertGuildInfoQuery, (err, res) => {
|
|
if (err) {
|
|
console.error(err);
|
|
reject("Error setting the guild info: " + err.message);
|
|
return;
|
|
}
|
|
resolve({ "status": "Successfully set the guild information", "data": null });
|
|
});
|
|
});
|
|
},
|
|
deleteGuildInfo(guildId) {
|
|
// Returns a Promise, resolve({ "status": "", "data": null })
|
|
// guildInfo = { "guildId": "123", "treeName": "name", "treeHeight": 123, "treeMessageId": "123", "treeChannelId": "123", "leaderboardMessageId": "123", "leaderboardChannelId": "123"}
|
|
// Set a server's tree information in the database
|
|
const deleteGuildInfoQuery = `DELETE FROM guild_info WHERE guild_id = ${db.escape(guildId)}`;
|
|
// TODO run this query and return a promise, then resolve with { "status": , "data": null }
|
|
return new Promise((resolve, reject) => {
|
|
db.query(deleteGuildInfoQuery, (err, res) => {
|
|
if (err) {
|
|
console.error(err);
|
|
reject("Error deleting the guild info: " + err.message);
|
|
return;
|
|
}
|
|
resolve({ "status": "Successfully deleted the guild information", "data": null });
|
|
});
|
|
});
|
|
},
|
|
getLeaderboard(guildId) {
|
|
// Returns a Promise, resolve({ "status": "", "data": leaderboard })
|
|
const selectLeaderboardQuery = `SELECT id, tree_name, tree_rank, tree_height, has_pin FROM leaderboard WHERE guild_id = ${db.escape(guildId)} ORDER BY id DESC LIMIT 10`;
|
|
// TODO run the query and return a promise then process the results. resolve with { "status": , "data": leaderboard }
|
|
return new Promise((resolve, reject) => {
|
|
db.query(selectLeaderboardQuery, (err, res) => {
|
|
if (err) {
|
|
console.error(err);
|
|
reject("Error fetching the most recent leaderboard: " + err.message);
|
|
return;
|
|
}
|
|
let leaderboard = [];
|
|
res.forEach(row => {
|
|
leaderboard.push({
|
|
"treeName": row.tree_name,
|
|
"treeRank": row.tree_rank,
|
|
"treeHeight": row.tree_height,
|
|
"hasPin": row.has_pin
|
|
});
|
|
});
|
|
resolve({ "status": "Successfully fetched leaderboard.", "data": leaderboard });
|
|
});
|
|
});
|
|
},
|
|
uploadLeaderboard(leaderboard) {
|
|
// Returns a Promise, resolve({ "status": "", "data": res })
|
|
// leaderboard = { "guildId": 1234, "entries": [ { "treeHeight": 12, "treeRank": 34, "treeName": "name", "hasPin": false }, {...}, {...} ] }
|
|
const insertLeaderboardQuery = "INSERT INTO `leaderboard` (guild_id, tree_name, tree_rank, tree_height, has_pin) VALUES ?";
|
|
const leaderboardValues = [];
|
|
leaderboard.entries.forEach(ranking => {
|
|
leaderboardValues.push([leaderboard.guildId, ranking.treeName, ranking.treeRank, ranking.treeHeight, ranking.hasPin]);
|
|
});
|
|
return new Promise((resolve, reject) => {
|
|
db.query(insertLeaderboardQuery, [leaderboardValues], (err, res) => {
|
|
if (err) {
|
|
reject("Error uploading the leaderboard: " + err.message);
|
|
console.error(err);
|
|
return;
|
|
}
|
|
resolve({ "status": "Successfully uploaded the leaderboard", "data": res });
|
|
});
|
|
});
|
|
},
|
|
db
|
|
}; |