const dotenv = require('dotenv'); dotenv.config(); const debugMode = process.env.DEBUG || true; const mysql = require('mysql'); /* 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 | | | | +-------------+--------------+------+-----+---------+----------------+ */ module.exports = { createGuildTables(guildId) { const 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}`; }); // 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); db.end(); return; } db.query(createLeaderboardTableQuery, (err) => { if (err) { reject("Error creating the leaderboard table: " + err.message); console.error("Offending query: " + createLeaderboardTableQuery); db.end(); return; } resolve({ "status": "Successfully checked both tables.", "data": null }); db.end(); }); }); }); }, getGuildInfo(guildId) { const 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}`; }); // 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); db.end(); 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"); db.end(); 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 }; db.end(); resolve({ "status": "Successfully fetched guild information", "data": guildInfo }); }); }); }, setGuildInfo(guildInfo) { const 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}`; }); // 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); db.end(); return; } db.end(); resolve({ "status": "Successfully set the guild information", "data": null }); }); }); }, setTreeInfo(guildInfo) { const 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}`; }); // 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); db.end(); reject("Error setting the guild info: " + err.message); return; } db.end(); resolve({ "status": "Successfully set the guild information", "data": null }); }); }); }, setLeaderboardInfo(guildInfo) { const 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}`; }); // 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); db.end(); reject("Error setting the guild info: " + err.message); return; } db.end(); resolve({ "status": "Successfully set the guild information", "data": null }); }); }); }, deleteGuildInfo(guildId) { const 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}`; }); // 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); db.end(); reject("Error deleting the guild info: " + err.message); return; } db.end(); resolve({ "status": "Successfully deleted the guild information", "data": null }); }); }); }, getLeaderboard(guildId) { const 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}`; }); // 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); db.end(); 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 }); }); db.end(); resolve({ "status": "Successfully fetched leaderboard.", "data": leaderboard }); }); }); }, uploadLeaderboard(leaderboard) { const 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}`; }); // 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); db.end(); console.error(err); return; } db.end(); resolve({ "status": "Successfully uploaded the leaderboard", "data": res }); }); }); } };