tools.js

'use strict';

const fs = require('fs');
const https = require('https');
const mysql = require('mysql');
const sqlite3 = require('sqlite3');

let regdateCache = {};
let sqliteDb;
try {
	regdateCache = JSON.parse(fs.readFileSync('config/regdate.json', 'utf8'));
} catch (e) {}

if (!Config.mysql) {
	sqliteDb = new sqlite3.Database('./config/users.db', function () {
		sqliteDb.run("CREATE TABLE IF NOT EXISTS users (id TEXT, userid TEXT PRIMARY KEY, name TEXT, lastOnline INTEGER, lastOnlineServer TEXT, lastOnlineAction TEXT, room TEXT)");
		sqliteDb.run("CREATE TABLE IF NOT EXISTS tells (userid TEXT, sender TEXT, message TEXT, date INTEGER)");
	});
} else {
	// lets make sure the database is setup
	let connection = mysql.createConnection({
		host: Config.mysql.host,
		port: Config.mysql.port,
		user: Config.mysql.user,
		password: Config.mysql.password,
	});

	connection.connect(function (err) {
		if (err) {
			Config.mysql = false; // something went wrong, revert back to using sqlite
			return console.log('Error connecting to mysql server: ' + err.stack);
		}
		connection.query('CREATE DATABASE IF NOT EXISTS ' + Config.mysql.dbName + ';', function (err, results, fields) {
			if (err) {
				Config.mysql = false;
				return console.log('Error creating mysql database.');
			}
			connection.changeUser({database: Config.mysql.dbName}, function (err) {
				if (err) {
					Config.mysql = false;
					return console.log('Error changing mysql databases.');
				}
				connection.query('CREATE TABLE IF NOT EXISTS users (id VARCHAR(32), firstSeen BIGINT, userid VARCHAR(19) PRIMARY KEY, name VARCHAR(32), lastOnline BIGINT, lastOnlineServer VARCHAR(32), lastOnlineAction VARCHAR(100), room VARCHAR(100))', function (error, results, fields) {
					if (err) {
						Config.mysql = false;
						return console.log('Error creating mysql users table.');
					}
					connection.query('CREATE TABLE IF NOT EXISTS logs (id INT PRIMARY KEY AUTO_INCREMENT, date BIGINT, server VARCHAR(32), room VARCHAR(100), user VARCHAR(32), userid VARCHAR(19), messageType VARCHAR(20), message TEXT)', function (error, results, fields) {
						if (err) {
							Config.mysql = false;
							return console.log('Error creating mysql logs table.');
						}
						connection.query('CREATE TABLE IF NOT EXISTS tells (userid VARCHAR(19), sender VARCHAR(32), message TEXT, date BIGINT)', function (error, results, fields) {
							if (err) {
								Config.mysql = false;
								return console.log('Error creating mysql tells table.');
							}
						});
					});
				});
			});
		});
	});
}

module.exports = {
	query: function (queryString, options, callback) {
		if (!callback) callback = options;
		if (Config.mysql) {
			let connection = mysql.createConnection({
				host: Config.mysql.host,
				port: Config.mysql.port,
				user: Config.mysql.user,
				password: Config.mysql.password,
				database: Config.mysql.dbName,
			});

			connection.connect(function (err) {
				if (err) {
					console.log('Error connecting to mysql server: ' + err);
					return callback(err);
				}

				connection.query(queryString, function (err, rows) {
					if (err) {
						console.log('Error querying mysql database: ' + err);
						return callback(err);
					}
					return callback(false, rows);
				});
				connection.end(function (err) {
					if (err) console.log('Error ending mysql connection: ' + err);
				});
			});

			connection.on('error', function (err) {
				if (err) console.log('mysql err: ' + err);
			});
		} else {
			sqliteDb.all(queryString, options, function (err, rows) {
				if (err) {
					console.log('Error querying sqlite database: ' + err);
					return callback(err);
				}
				return callback(false, rows);
			});
		}
	},
	sanitize: function (message) {
		if (message.charAt(0) === '/') message = '/' + message;
		if (message.charAt(0) === '!' || message.substr(0, 2) === '>>') message = ' ' + message;
		return message.replace(/\n/g, '');
	},
	log: function (text, serverid, error) {
		if (error) {
			fs.appendFileSync('logs/error.txt', '[' + serverid + '] ' + text);
		}
		fs.appendFileSync('logs/' + serverid + '.log', text + '\n');
		console.log(text);
	},
	randomString: function (length) {
		return Math.round((Math.pow(36, length + 1) - Math.random() * Math.pow(36, length))).toString(36).slice(1);
	},
	toTimeStamp: function (date, options) {
		// Return a timestamp in the form {yyyy}-{MM}-{dd} {hh}:{mm}:{ss}.
		// Optionally reports hours in mod-12 format.
		const isHour12 = options && options.hour12;
		let parts = [date.getFullYear(), date.getMonth() + 1, date.getDate(), date.getHours(), date.getMinutes(), date.getSeconds()];
		if (isHour12) {
			parts.push(parts[3] >= 12 ? 'pm' : 'am');
			parts[3] = parts[3] % 12 || 12;
		}
		parts = parts.map(val => val < 10 ? '0' + val : '' + val);
		return parts.slice(0, 3).join("-") + " " + parts.slice(3, 6).join(":") + (isHour12 ? " " + parts[6] : "");
	},
	regdate: function (target, callback) {
		target = toId(target);
		if (regdateCache[target]) return callback(regdateCache[target]);
		let options = {
			host: 'pokemonshowdown.com',
			port: 443,
			path: '/users/' + target + '.json',
			method: 'GET',
		};
		https.get(options, function (res) {
			let data = '';
			res.on('data', function (chunk) {
				data += chunk;
			}).on('end', function () {
				if (data.charAt(0) !== '{') data = JSON.stringify({registertime: 0});
				data = JSON.parse(data);
				let date = data['registertime'];
				if (date !== 0 && date.toString().length < 13) {
					while (date.toString().length < 13) {
						date = Number(date.toString() + '0');
					}
				}
				if (date !== 0) {
					regdateCache[target] = date;
					fs.writeFileSync('config/regdate.json', JSON.stringify(regdateCache));
				}
				callback((date === 0 ? false : date));
			});
		});
	},
	// toDurationString from: https://github.com/Zarel/Pokemon-Showdown/blob/master/chat.js
	toDurationString: function (number, options) {
		// TODO: replace by Intl.DurationFormat or equivalent when it becomes available (ECMA-402)
		// https://github.com/tc39/ecma402/issues/47
		const date = new Date(+number);
		const parts = [date.getUTCFullYear() - 1970, date.getUTCMonth(), date.getUTCDate() - 1, date.getUTCHours(), date.getUTCMinutes(), date.getUTCSeconds()];
		const unitNames = ["second", "minute", "hour", "day", "month", "year"];
		const positiveIndex = parts.findIndex(elem => elem > 0);
		if (options && options.hhmmss) {
			let string = parts.slice(positiveIndex).map(value => value < 10 ? "0" + value : "" + value).join(":");
			return string.length === 2 ? "00:" + string : string;
		}
		return parts.slice(positiveIndex).reverse().map((value, index) => value ? value + " " + unitNames[index] + (value > 1 ? "s" : "") : "").reverse().join(" ").trim();
	},
	// uploadToHastebin function by TalkTakesTime (https://github.com/TalkTakesTime/Pokemon-Showdown-Bot)
	uploadToHastebin: function (toUpload, callback) {
		if (typeof callback !== 'function') return false;
		let reqOpts = {
			hostname: 'hastebin.com',
			port: 443,
			method: 'POST',
			path: '/documents',
		};

		let req = https.request(reqOpts, function (res) {
			res.on('data', function (chunk) {
				// CloudFlare can go to hell for sending the body in a header request like this
				if (typeof chunk === 'string' && chunk.substr(0, 15) === '<!DOCTYPE html>') return callback('Error uploading to Hastebin.');
				let filename = JSON.parse(chunk.toString()).key;
				callback('http://hastebin.com/raw/' + filename);
			});
		});
		req.on('error', function (e) {
			callback('Error uploading to Hastebin: ' + e.message);
		});

		req.write(toUpload);
		req.end();
	},
	getServers: function (callback) {
		https.get('https://pokemonshowdown.com/servers/', res => {
			let servers = [];
			let data = '';

			res.on('data', chunk => {
				data += chunk;
			}).on('end', () => {
				data = data.replace(/\n/g, '').replace(/\s+/g, ' ');
				//let servers = {};
				let rows = data.split('<tr>');
				rows.splice(0, 1);
				for (let u in rows) {
					if (rows[u].includes('(offline)')) continue;
					let serverName = rows[u].split('<strong>')[1].split('</strong>')[0];
					let serverid = rows[u].match(/<a href="\/servers\/[a-z]+/)[0].replace(/<a href="\/servers\//, '');
					let owners = (rows[u].match(/<small>([a,-z]+)<\/small>/) ? rows[u].match(/<small>([a,-z]+)<\/small>/)[0].replace(/(<small>|<\/small>)/g, '') : '');
					let userCount = rows[u].match(/\([0-9]+\)/)[0].replace(/(\(|\))/, '');
					servers.push({
						name: serverName,
						id: serverid,
						userCount: userCount,
						owners: owners,
					});
				}
				return callback(servers);
			});
		}).on('error', e => {
			console.log('Error retrieving server list: ' + e);
		});
	},
	getServerIp: function (id, callback) {
		if (id === 'showdown') {
			return callback({host: 'sim.smogon.com', port: 8000, name: 'Smogon University'});
		}
		https.get('https://play.pokemonshowdown.com/crossdomain.php?host=' + id + '.psim.us', res => {
			let data = '';

			res.on('data', chunk => {
				data += chunk;
			}).on('end', () => {
				let lines = data.split('\n');
				let config;
				for (let u in lines) {
					if (lines[u].includes('config')) {
						config = lines[u].replace(/\\"/g, '"').substr(14);
						config = config.substr(0, config.length - 2);
						break;
					}
				}
				return callback(JSON.parse(config));
			});
		});
	},
};