xiao/commands/search/periodic-table.js

const Command = require('../../framework/Command');
const { PermissionFlagsBits } = require('discord.js');
const request = require('node-superfetch');
const { createCanvas } = require('@napi-rs/canvas');
const colors = 	{
	Solid: 'black',
	Liquid: 'blue',
	Gas: 'green'
};
const jerktonium = {
	name: 'Jerktonium',
	atomic_mass: 1999,
	number: 0,
	period: 'Bikini Bottom',
	phase: 'Solid',
	symbol: 'Jt'
};

module.exports = class PeriodicTableCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'periodic-table',
			aliases: ['element', 'p-table'],
			group: 'search',
			description: 'Finds an element on the periodic table.',
			throttling: {
				usages: 2,
				duration: 10
			},
			clientPermissions: [PermissionFlagsBits.AttachFiles],
			credit: [
				{
					name: 'Bowserinator',
					url: 'https://github.com/Bowserinator/',
					reason: 'Periodic Table Data',
					reasonURL: 'https://github.com/Bowserinator/Periodic-Table-JSON'
				},
				{
					name: 'Google',
					url: 'https://www.google.com/',
					reason: 'Noto Font',
					reasonURL: 'https://fonts.google.com/noto'
				}
			],
			args: [
				{
					key: 'element',
					type: 'string',
					validate: async element => {
						if (!this.table) await this.fetchTable();
						const num = Number.parseInt(element, 10);
						if (!Number.isNaN(num) && num >= 0 && num <= this.table.length - 1) return true;
						const search = element.toString().toLowerCase();
						if (this.table.find(e => e.name.toLowerCase() === search || e.symbol.toLowerCase() === search)) return true;
						return 'Invalid element, please enter a valid element symbol, name, or atomic number.';
					},
					parse: async element => {
						if (!this.table) await this.fetchTable();
						const num = Number.parseInt(element, 10);
						if (!Number.isNaN(num)) return this.table[num];
						const search = element.toLowerCase();
						return this.table.find(e => e.name.toLowerCase() === search || e.symbol.toLowerCase() === search);
					}
				}
			]
		});

		this.table = null;
	}

	async run(msg, { element }) {
		if (!this.table) await this.fetchTable();
		const canvas = createCanvas(500, 500);
		const ctx = canvas.getContext('2d');
		ctx.fillStyle = 'black';
		ctx.fillRect(0, 0, canvas.width, canvas.height);
		ctx.fillStyle = 'white';
		ctx.fillRect(10, 10, canvas.width - 20, canvas.height - 20);
		ctx.textAlign = 'center';
		ctx.font = this.client.fonts.get('Noto-Regular.ttf').toCanvasString(210);
		ctx.fillStyle = colors[element.phase] || 'gray';
		ctx.fillText(element.symbol, 250, 320);
		ctx.fillStyle = 'black';
		ctx.font = this.client.fonts.get('Noto-Regular.ttf').toCanvasString(45);
		ctx.fillText(element.number.toString(), 250, 100);
		ctx.fillText(element.name, 250, 450);
		ctx.font = this.client.fonts.get('Noto-Regular.ttf').toCanvasString(30);
		ctx.fillText(element.atomic_mass ? element.atomic_mass.toString() : '?', 250, 400);
		const period = element.number === 0
			? element.period
			: `period ${element.period}`;
		const phase = element.undiscovered ? `hypothetical ${element.phase || 'element'}` : element.phase;
		return msg.say(
			`**${element.name} (${element.symbol})** is a ${phase} in ${period}.`,
			{ files: [{ attachment: canvas.toBuffer('image/png'), name: `${element.name}.png` }] }
		);
	}

	async fetchTable() {
		if (this.table) return this.table;
		const { text } = await request
			.get('https://raw.githubusercontent.com/Bowserinator/Periodic-Table-JSON/master/PeriodicTableJSON.json');
		this.table = JSON.parse(text).elements;
		this.table.unshift(jerktonium);
		return this.table;
	}
};