Fix SB Replies

CustomModules/NodBot.js

@@ -46,6 +46,48 @@ module.exports = {
             this.content = typeof content === 'string' ? content : this.content;
             this.iconUrl = typeof iconUrl === 'string' ? iconUrl : this.iconUrl;
 
+            return this; // For chaining
+        }
+    },
+// StrainData
+    StrainData: class {
+        constructor() {
+            this.id = 0;
+            this.name = "Unknown";
+            this.type = "Unknown";
+            this.effects = "Unknown";
+            this.flavor = "Unknown";
+            this.rating = "0.0";
+            this.description = "Unknown";
+        }
+
+        // Initial Strain configuration
+        // Can also be used to update the data piecemeal
+        setInfo(name, details, id) {
+            /*  details contains all the optional pieces of data
+                This is to prevent having to call setInfo(name, undefined, undefined, undefined, 0) or something
+                details: {
+                    type: String,
+                    effects: String,
+                    rating: String,
+                    description: String
+                }
+            */
+
+            // Check for existing or incoming name
+            if ((this.name === "") && (typeof name !== 'string')) throw `Error: This Strain doesn't have existing name, and no name is going to be set.`;
+
+            // Property is set if the variable is the right type,
+            // otherwise it keeps the existing property
+            this.id = typeof id === 'number' ? id : this.id;
+            this.name = typeof name === 'string' ? name : this.name;
+            this.type = typeof details.type === 'string' ? details.type : this.type;
+            this.effects = typeof details.effects === 'string' ? details.effects : this.effects;
+            this.flavor = typeof details.flavor === 'string' ? details.flavor : this.flavor;
+            this.rating = typeof details.rating === 'string' ? details.rating : this.rating;
+            this.description = typeof details.description === 'string' ? details.description : this.description;
+            
+
             return this; // For chaining
         }
     }

Dockerfile

@@ -5,4 +5,4 @@ WORKDIR /usr/src/app
 COPY package.json ./
 RUN npm install
 COPY . .
-CMD ["/bin/sh", "-c", "node main.js 2> /logs/nodbot.error 1> /logs/nodbot.log"]
+CMD ["/bin/sh", "-c", "node main.js 2> /logs/$(date +%Y-%m-%d_%H-%M-%S)-error.txt 1> /logs/$(date +%Y-%m-%d_%H-%M-%S)-status.txt"]

dot-commands/spongebob.js

@@ -10,7 +10,10 @@ module.exports = {
 		// message.reply(fn.spongebob(commandData)).then(() => {
 		// 	message.delete();
 		// });
+		// Check if the message is replying to another message
 		if (message.reference != undefined) {
+			// Check if the message is just ".sb" or ".spongebob" (empty arguments)
+			if (commandData.args === "") {
 				const repliedMessageId = message.reference.messageId;
 				message.channel.messages.fetch(repliedMessageId)
 					.then(repliedMessage => {
@@ -21,6 +24,18 @@ module.exports = {
 					.catch(err => {
 						console.error(err);
 					});
+			} else { // If there is content before the .sb, meaning the reply is just for reference...
+				const repliedMessageId = message.reference.messageId;
+				message.channel.messages.fetch(repliedMessageId)
+					.then(repliedMessage => {
+						repliedMessage.reply(fn.spongebob({ args: commandData.args })).then(() => {
+							if (message.deletable) message.delete();
+						});
+					})
+					.catch(err => {
+						console.error(err);
+					});
+			}
 		} else {
 			message.channel.send(fn.spongebob(commandData)).then(() => {
 				if (message.deletable) message.delete();

functions.js

@@ -42,10 +42,11 @@ const config = require('./config.json');
 const strings = require('./strings.json');
 const slashCommandFiles = fs.readdirSync('./slash-commands/').filter(file => file.endsWith('.js'));
 const dotCommandFiles = fs.readdirSync('./dot-commands/').filter(file => file.endsWith('.js'));
+const https = require('https');
 
 // MySQL database connection
 const mysql = require('mysql');
-const { GifData, PastaData } = require('./CustomModules/NodBot');
+const { GifData, PastaData, StrainData } = require('./CustomModules/NodBot');
 const db = new mysql.createPool({
 	connectionLimit: 10,
 	host: dbHost,
@@ -151,11 +152,18 @@ const functions = {
 			if (!client.strains) client.strains = new Discord.Collection();
 			client.strains.clear();
 			for (const row of rows) {
-				const strain = {
-					id: row.id,
-					name: row.strain,
-				};
-				client.strains.set(strain.name, strain);
+				const strainData = new StrainData().setInfo(
+					row.strain,
+					{
+						type: row.type,
+						effects: row.effects,
+						rating: row.rating,
+						description: row.description,
+						flavor: row.flavor
+					},
+					row.id
+				);
+				client.strains.set(strainData.name, strainData);
 				// if (isDev) console.log(strain)
 			}
 			if (isDev) console.log('Strains Collection Built');
@@ -365,7 +373,7 @@ const functions = {
 				},
 			]);
 
-			interaction.reply({ embeds: [ strainEmbed ]});
+			return { embeds: [ strainEmbed ]};
 		},
 		dalle(prompt, imageUrl, size) {
 			const dalleEmbed = new Discord.MessageEmbed()
@@ -470,14 +478,8 @@ const functions = {
 				functions.download.medicalAdvice(client);
 			});
 		},
-		strain(interaction) {
-			const strain = db.escape(interaction.options.getString('name'));
-			const type = db.escape(interaction.options.getString('type'));
-			const effects = db.escape(( interaction.options.getString('effects') || 'Unkown' ));
-			const description = db.escape(( interaction.options.getString('description') || 'Unknown' ));
-			const flavor = db.escape(( interaction.options.getString('flavor') || 'Unknown' ));
-			const rating = db.escape(( interaction.options.getString('rating') || '3' ));
-			const strainQuery = `INSERT INTO strains (strain, type, effects, description, flavor, rating) VALUES (${strain}, ${type}, ${effects}, ${description}, ${flavor}, ${rating}) ON DUPLICATE KEY UPDATE strain=${db.escape(strain)}, type=${db.escape(type)}, effects=${db.escape(effects)}, description=${db.escape(description)}, flavor=${db.escape(flavor)}, rating=${db.escape(rating)}`;
+		strain(interaction, strainData) {
+			const strainQuery = `INSERT INTO strains (strain, type, effects, description, flavor, rating) VALUES (${db.escape(strainData.name)}, ${db.escape(strainData.type)}, ${db.escape(strainData.effects)}, ${db.escape(strainData.description)}, ${db.escape(strainData.flavor)}, ${db.escape(strainData.rating)}) ON DUPLICATE KEY UPDATE type=${db.escape(strainData.type)}, effects=${db.escape(strainData.effects)}, description=${db.escape(strainData.description)}, flavor=${db.escape(strainData.flavor)}, rating=${db.escape(strainData.rating)}`;
 			console.log(strainQuery);
 			return new Promise((resolve, reject) => {
 				db.query(strainQuery, (err, rows, fields) => {
@@ -495,16 +497,16 @@ const functions = {
 		}
 	},
 	download: {
-		requests(client) {
+		async requests(client) {
 			const query = 'SELECT * FROM requests WHERE status = \'Active\' ORDER BY id DESC';
-			db.query(query, (err, rows, fields) => {
+			await db.query(query, (err, rows, fields) => {
 				if (err) throw err;
 				functions.collections.requests(rows, client);
 			});
 		},
-		pastas(client) {
+		async pastas(client) {
 			const query = 'SELECT * FROM pastas ORDER BY id ASC';
-			db.query(query, (err, rows, fields) => {
+			await db.query(query, (err, rows, fields) => {
 				if (err) throw err;
 				functions.collections.pastas(rows, client);
 			});
@@ -516,16 +518,17 @@ const functions = {
 				functions.collections.gifs(rows, client);
 			});
 		},
-		joints(client) {
+		async joints(client) {
 			const query = 'SELECT * FROM joints ORDER BY id ASC';
-			db.query(query, (err, rows, fields) => {
+			await db.query(query, (err, rows, fields) => {
 				if (err) throw err;
 				functions.collections.joints(rows, client);
 			});
 		},
-		strain(strainName, interaction) {
+		async strain(strainName, interaction) {
+			return new Promise(async (resolve, reject) => {
 				const query = `SELECT id, strain, type, effects, description, flavor, rating FROM strains WHERE strain = ${db.escape(strainName)}`;
-			db.query(query, (err, rows, fields) => {
+				await db.query(query, (err, rows, fields) => {
 					if (rows != undefined) {
 						const strainInfo = {
 							id: `${rows[0].id}`,
@@ -536,20 +539,23 @@ const functions = {
 							flavor: `${rows[0].flavor}`,
 							rating: `${rows[0].rating}`,
 						};
-					functions.embeds.strain(strainInfo, interaction);
+						const embed = functions.embeds.strain(strainInfo, interaction);
+						resolve(embed);
 					}
 				});
+			});
+			
 		},
-		strains(client) {
-			const query = 'SELECT id, strain FROM strains';
-			db.query(query, (err, rows, fields) => {
+		async strains(client) {
+			const query = 'SELECT * FROM strains';
+			await db.query(query, (err, rows, fields) => {
 				if (err) throw err;
 				functions.collections.strains(rows, client);
 			});
 		},
-		medicalAdvice(client) {
+		async medicalAdvice(client) {
 			const query = 'SELECT * FROM medical_advice ORDER BY id ASC';
-			db.query(query, (err, rows, fields) => {
+			await db.query(query, (err, rows, fields) => {
 				if (err) throw err;
 				functions.collections.medicalAdvice(rows, client);
 			});
@@ -665,6 +671,19 @@ const functions = {
 		}
 		const errorId = digits.join("");
 		return errorId;
+	},
+	async sendHeartbeat(url) {
+		if (isDev) console.log("Heartbeat Sent: " + url);
+		https.get(url, async (response) => {
+			let data = '';
+
+			response.on('data', (chunk) => data += chunk);
+
+			response.on('end', () => {
+				parsedData = JSON.parse(data);
+				if ( !(parsedData.ok) ) console.error("Heartbeat failed");
+			});
+		}).on("error", (error) => console.error(error));
 	}
 };
 

main.js

@@ -3,9 +3,16 @@
 // dotenv for handling environment variables
 const dotenv = require('dotenv');
 dotenv.config();
+const isDev = process.env.isDev === 'true';
 const token = process.env.TOKEN;
 const statusChannelId = process.env.statusChannelId;
-
+const heartbeatUrl = process.env.HEARTBEAT_URL;
+const sendHeartbeat = typeof heartbeatUrl === 'string';
+if (sendHeartbeat) {
+	if (isDev) console.log("Hearbeat Enabled: " + heartbeatUrl);
+} else {
+	if (isDev) console.log("Hearbeat Disabled: " + heartbeatUrl + " (" + typeof heartbeatUrl + ")");
+}
 // Discord.JS
 const { Client, Intents } = require('discord.js');
 const client = new Client({
@@ -28,22 +35,28 @@ const fn = require('./functions.js');
 const config = require('./config.json');
 const strings = require('./strings.json');
 const { GifData } = require('./CustomModules/NodBot.js');
-const isDev = process.env.isDev;
 
-client.once('ready', () => {
+client.once('ready', async () => {
 	fn.collections.slashCommands(client);
 	fn.collections.dotCommands(client);
 	fn.collections.setvalidCommands(client);
-	fn.download.gifs(client);
-	fn.download.pastas(client);
-	fn.download.joints(client);
-	fn.download.requests(client);
-	fn.download.strains(client);
-	fn.download.medicalAdvice(client);
+	await fn.download.gifs(client);
+	await fn.download.pastas(client);
+	await fn.download.joints(client);
+	await fn.download.requests(client);
+	await fn.download.strains(client);
+	await fn.download.medicalAdvice(client);
 	console.log('Ready!');
-	client.channels.fetch(statusChannelId).then(channel => {
+	await client.channels.fetch(statusChannelId).then(channel => {
 		channel.send(`${new Date().toISOString()} -- <@${process.env.ownerId}>\nStartup Sequence Complete`);
 	});
+	// Heartbeat Timer
+	if (sendHeartbeat) {
+		setInterval(() => {
+			fn.sendHeartbeat(heartbeatUrl);
+		}, 30000);
+		if (isDev) console.log("Heartbeat interval set.");
+	}
 });
 
 // slash-commands
@@ -203,6 +216,14 @@ client.on('interactionCreate', async interaction => {
 							pastaChoices.map(choice => ({ name: choice, value: choice }))
 						);
 						break;
+					case "strain":
+						//TODO 
+						const strainQuery = interaction.options.getFocused();
+						const strainChoices = fn.weed.strain.lookup(strainQuery, interaction.client);
+						await interaction.respond(
+							strainChoices.map(choice => ({ name: choice, value: choice }))
+						);
+						break;
 					
 					default:
 						break;

slash-commands/edit.js

@@ -8,7 +8,7 @@ const tenor = require('tenorjs').client({
 
 const { SlashCommandBuilder } = require('@discordjs/builders');
 const { MessageActionRow, MessageButton } = require('discord.js');
-const { GifData, PastaData } = require('../CustomModules/NodBot');
+const { GifData, PastaData, StrainData } = require('../CustomModules/NodBot');
 const fn = require('../functions.js');
 const strings = require('../strings.json');
 const { emoji } = strings;
@@ -54,6 +54,54 @@ module.exports = {
 				.setDescription('The new content of the copypasta')
 				.setRequired(true)
 			)
+		)
+// Strain
+		.addSubcommand(subcommand =>
+			subcommand
+			.setName('strain')
+			.setDescription('Edit a strain\'s data')
+			.addStringOption(option =>
+				option
+				.setName('name')
+				.setDescription('Name of the strain')
+				.setRequired(true)
+				.setAutocomplete(true)
+			)
+			.addStringOption(option =>
+				option
+				.setName('type')
+				.setDescription('Indica/Sativa/Hybrid')
+				.setRequired(false)
+				.addChoices(
+					{ name: "Indica", value: "Indica" },
+					{ name: "Hybrid", value: "Hybrid" },
+					{ name: "Sativa", value: "Sativa" }
+				)
+			)
+			.addStringOption(option =>
+				option
+				.setName('effects')
+				.setDescription('What effects does the strain produce?')
+				.setRequired(false)
+			)
+			.addStringOption(option =>
+				option
+				.setName('flavor')
+				.setDescription('Flavor Profile')
+				.setRequired(false)
+			)
+			.addStringOption(option =>
+				option
+				.setName('rating')
+				.setDescription('Star rating of the strain on Leaf.ly')
+				.setRequired(false)
+			)
+			.addStringOption(option =>
+				option
+				.setName('description')
+				.setDescription('Summary of the strain')
+				.setRequired(false)
+			)
 		),
 	async execute(interaction) {
 		await interaction.deferReply({ ephemeral: true });
@@ -82,6 +130,7 @@ module.exports = {
 // Strain
 				case "strain":
 					//TODO 
+					await this.editStrain(interaction);
 					break;
 					break;
 // Default
@@ -106,5 +155,19 @@ module.exports = {
 		await fn.upload.pasta(pastaData, interaction.client);
 		await fn.download.gifs(interaction.client);
 		await interaction.editReply(`I've updated ${name}.pasta`);
+	},
+	async editStrain(interaction) {
+		const details = {
+			type: interaction.options.getString('type'),
+			effects: interaction.options.getString('effects'),
+			flavor: interaction.options.getString('flavor'),
+			rating: interaction.options.getString('rating'),
+			description: interaction.options.getString('description')
+		};
+		const strainData = interaction.client.strains.get(interaction.options.getString('name'));
+		strainData.setInfo(interaction.options.getString('name'), details);
+		await fn.upload.strain(interaction, strainData);
+		const reply = await fn.download.strain(strainData.name, interaction);
+		await interaction.editReply(reply);
 	}
 };

slash-commands/save.js

@@ -10,7 +10,7 @@ const { SlashCommandBuilder } = require('@discordjs/builders');
 const { MessageActionRow, MessageButton } = require('discord.js');
 const fn = require('../functions.js');
 const strings = require('../strings.json');
-const { GifData } = require('../CustomModules/NodBot.js');
+const { GifData, StrainData } = require('../CustomModules/NodBot.js');
 const { emoji } = strings;
 
 module.exports = {
@@ -213,7 +213,17 @@ module.exports = {
 // Strain
 				case "strain":
 					//TODO 
-					fn.upload.strain(interaction).then(res => {
+					const strainData = new StrainData().setInfo(
+						interaction.options.getString('name'),
+						{
+							type: interaction.options.getString('type'),
+							effects: interaction.options.getString('effects'),
+							flavor: interaction.options.getString('flavor'),
+							rating: interaction.options.getString('rating'),
+							description: interaction.options.getString('description')
+						}
+					);
+					fn.upload.strain(interaction, strainData).then(res => {
 						interaction.editReply({
 							content: `The strain information has been saved. (${interaction.options.getString('name')})`,
 							ephemeral: true

slash-commands/strain.js

@@ -12,6 +12,8 @@ module.exports = {
 				.setRequired(true)
 				.setAutocomplete(true)),
 	async execute(interaction) {
-		fn.download.strain(interaction.options.getString('name'), interaction);
+		await interaction.deferReply();
+		const reply = await fn.download.strain(interaction.options.getString('name'), interaction);
+		await interaction.editReply(reply);
 	},
 };