puppeteer-discord-bot/index.js

console.log("start");
require("dotenv").config();
var Discord = require("discord.js");
var puppeteer = require("puppeteer");
var minimist = require("minimist");
var dhms = require("dhms");
var fs = require("fs");
var PREFIX = process.env.PREFIX || "p!";
(async () => {
	var browser = await puppeteer.launch(process.env.NO_SANDBOX && {args:["--no-sandbox"]});
	console.log("chromium launched");

	var client = new Discord.Client();
	await client.login(process.env.TOKEN);
	console.log("discord client ready");
	client.user.setActivity(`${PREFIX}help`);

	client.on("message", processMessage);
	/*client.on("messageUpdate", async (oldMessage, newMessage) => {
		if (newMessage.responses) newMessage.react('🚫');
	});*/
	client.on("messageDelete", async message => {
		if (message.responses) {
			console.log(`[${new Date().toLocaleString()}] [${message?.guild.id}(${message?.guild.name})] [${message.channel.id}(#${message.channel.name})] Deleted command from ${message.author.id} (${message.author.tag}): ${message.content}`);
			message.responses.forEach(async message => {
				(await message).delete();
			});
		}
	});

	async function processMessage(message) {
		if (!message.content.startsWith(PREFIX)) return;
		console.log(`[${new Date().toLocaleString()}] [${message?.guild.id}(${message?.guild.name})] [${message.channel.id}(#${message.channel.name})] User ${message.author.id} (${message.author.tag}) invoked command: ${message.content}`);
		message.responses = [];

		function respond() {
			message.author.pendingResponse = false;
			if (message.deleted) return;
			var r = message.channel.send.apply(message.channel, arguments)
			message.responses.push(r);
			return r;
		}

		var inp = message.content.slice(PREFIX.length);
		var args = inp.split(" ");
		var cmd = args[0].toLowerCase();
		var pargs = minimist(args.slice(1), {
			alias: {
				"wait": "w",
				"dimensions": "d",
				"fullpage": "f"
			},
			boolean: "fullpage"
		});
		var query = pargs._.join(' ');
		var wait_ms = Math.min(dhms(pargs.wait), 30000);
		var viewport = pargs.dimensions?.split('x');
		viewport = {
			width: Math.min(Number(viewport?.[0] || 1440), 10000),
			height: Math.min(Number(viewport?.[1] || 900), 10000)
		};

		switch (cmd) {
			case "help":
			case "h":
				respond({embed:{
					title: "Commands",
					description:
						`\n\`${PREFIX}screenshot <url>\`` +
						`\n\`${PREFIX}google <query>\`` +
						`\n\`${PREFIX}google-images <query>\`` +
						`\n\`${PREFIX}bing <query>\`` +
						`\n\`${PREFIX}bing-images <query>\`` +
						`\n\`${PREFIX}youtube <query>\`` +
						`\n\`${PREFIX}ebay <query>\`` +
						`\n\`${PREFIX}amazon <query>\`` +
						`\n\`${PREFIX}duckduckgo <query>\`` +
						`\n\`${PREFIX}duckduckgo-images <query>\`` +
						`\n\`${PREFIX}yahoo <query>\`` +
						`\n\`${PREFIX}yahoo-images <query>\`` +
						`\n\`${PREFIX}wikipedia <query>\`` +
						`\n Each command has a logical abbreviated alias (i.e. \`${PREFIX}gi\`)` +
						`\n` +
						`\n**Options**` +
						`\n\`--dimensions=<width>x<height>\` or i.e. \`-d 640x480\` (max 10000x10000)` +
						`\n\`--wait=<time>\` or i.e. \`-w 5s\` (max 30s)` +
						`\n\`--fullpage\` or \`-f\`` +
						`\n` +
						`\n[» Add this bot to your server](https://discordapp.com/oauth2/authorize?scope=bot&client_id=${client.user.id})` +
						`\n[» Suggest sites or report a bug](https://gitea.moe/lamp/puppeteer-discord-bot/issues/new)` +
						`\n[» Source code](https://gitea.moe/lamp/puppeteer-discord-bot/src/branch/master/index.js)`
						
				}});
				break;
			case "screenshot":
			case "ss":
				pup((query.startsWith("http://") || query.startsWith("https://")) ? query : `http://${query}`);
				break;
			case "google":
			case "g":
				pup(`https://www.google.com/search?q=${encodeURIComponent(query)}`);
				break;
			case "google-i'm-feeling-lucky": //todo this gives manual redirection confirmation page
			case "gifl":
				pup(`https://www.google.com/search?btnI=I%27m+Feeling+Lucky&q=${encodeURIComponent(query)}`);
				break;
			case "google-images":
			case "gi":
				pup(`https://www.google.com/search?q=${encodeURIComponent(query)}&tbm=isch&safe=${(message.channel.nsfw) ? 'off' : 'on'}`);
				break;
			case "bing":
			case "b":
				pup(`https://www.bing.com/search?q=${encodeURIComponent(query)}`);
				break;
			case "bing-images":
			case "bi":
				pup(`https://www.bing.com/images/search?q=${encodeURIComponent(query)}&safeSearch=${(message.channel.nsfw) ? 'off' : 'moderate'}`);
				break;
			case "youtube":
			case "yt":
				pup(`https://www.youtube.com/results?search_query=${encodeURIComponent(query)}`);
				break;
			case "ebay":
			case "e":
				pup(`https://www.ebay.com/sch/i.html?_nkw=${encodeURIComponent(query)}`);
				break;
			case "amazon":
			case "a":
				pup(`https://www.amazon.com/s?k=${encodeURIComponent(query)}`);
				break;
			case "duckduckgo":
			case "ddg":
				pup(`https://duckduckgo.com/?q=${encodeURIComponent(query)}`);
				break;
			case "duckduckgo-images":
			case "ddgi":
				pup(`https://duckduckgo.com/?q=${encodeURIComponent(query)}&t=h_&iax=images&ia=images&kp=${message.channel.nsfw ? '-1' : '-2'}`);
				break;
			case "yahoo":
			case "y":
				pup(`https://search.yahoo.com/search?p=${encodeURIComponent(query)}`);
				break;
			case "yahoo-images":
			case "yi":
				pup(`https://images.search.yahoo.com/search/images?p=${encodeURIComponent(query)}`); //todo don't know of query param for safe search, but defaults to on
				break;
			case "wikipedia":
			case "w":
				pup(`https://en.wikipedia.org/w/index.php?title=Special:Search&search=${encodeURIComponent(query)}`);
				break;
/*
			case "":
			case "":
				pup(``);
				break;
*/

			case "eval":
			case ">":
				if (message.author.id == (process.env.OWNER || "707359017252683896")) {
					try {
						respond(String(eval(query)));
					} catch(error) {
						respond(String(error));
					}
				}
				break;
		}

		async function pup(url) {
			//if (message.author.pendingResponse) { message.react('🚫'); return; }
			message.author.pendingResponse = true;
			message.react('🆗');
			try {
				var page = await browser.newPage();
				page.on("error", async error => {
					respond(`⚠️ ${error.message}`);
				});
				await page.setCacheEnabled(false)
				await page.setViewport(viewport);
				await page.goto(url);
				await new Promise(r => setTimeout(r, wait_ms));
				var screenshot = await page.screenshot({type: 'png', fullPage: Boolean(pargs.fullpage)});
				await respond({files:[{ attachment: screenshot, name: "screenshot.png" }]});
			} catch(error) {
				console.error(error);
				respond(`⚠️ ${error.message}`);
			} finally {
				try {
					await page.close();
				} catch(error) {
					console.error(error);
					process.exit(1);
				}
			}
		}
	}
})().catch(error => { console.error(error); process.exit(1); });