demotivational poster generator

commands/images/poster.js

@@ -0,0 +1,115 @@
+const { Command } = require('discord-akairo');
+let gm = require('gm');
+const os = require('os');
+const fetch = require('node-fetch');
+const fs = require('fs');
+
+class posterCommand extends Command {
+	constructor() {
+		super('poster', {
+			aliases: ['poster'],
+			category: 'images',
+			clientPermissions: ['SEND_MESSAGES', 'ATTACH_FILES'],
+			args: [
+				{
+					id: 'link',
+					prompt: {
+						start: 'Please input a link to use, say `cancel` to stop the command'
+					},
+					type: 'string',
+				},
+				{
+					id: 'message',
+					prompt: {
+						start: 'Please input a caption, say `cancel` to stop the command'
+					},
+					type: 'string',
+					match: 'rest'
+				}
+			],
+			description: {
+				content: 'Create demotivational poster (use ``|`` to separate top text and bottom text) WIP',
+				usage: '[link to image] [topText|bottomText]',
+				examples: ['']
+			}
+		});
+	}
+
+	async exec(message, args) {
+		let output = `${os.tmpdir()}/poster${message.id}.jpg`;
+
+		let options = args.message.trim().split('|');
+
+		if (options[0] == undefined)
+			options[0] = '';
+		else if (options[1] == undefined)
+			options[1] = '';
+
+		let url = args.link;
+
+
+		if (!url) {
+			return message.channel.send('You need an image to use this command!');
+		}
+
+		let loadingmsg = await message.channel.send('Processing <a:loadingmin:527579785212329984>');
+
+		// Create new graphicsmagick instance
+		fetch(url)
+			.then(res => {
+				const dest = fs.createWriteStream(`${os.tmpdir()}/${message.id}.jpg`);
+				res.body.pipe(dest);
+				dest.on('finish', async () => {
+
+					let img = gm(`${os.tmpdir()}/${message.id}.jpg`);
+
+					// Set some defaults
+					const TOP_TEXT = options[0];
+					const BOTTOM_TEXT = options[1];
+					const FONT = './asset/times.ttf';
+					const FONT_FILL = '#FFF';
+					const TEXT_POS = 'center';
+					const STROKE_COLOR = '#000';
+					const PADDING = 40;
+			
+					// Get the image size to calculate top and bottom text positions
+					img.size(function(err, value) {
+						// Set text position for top and bottom
+						const TOP_POS = Math.abs((value.height / 2) - PADDING) + 110;
+						const BOTTOM_POS = (value.height / 2) - PADDING + 180;
+						//const FONT_SIZE1 = 50;
+						const FONT_SIZE1 = (value.width / 12);
+						const FONT_SIZE2 = (value.width / 12) - 20;
+						//const FONT_SIZE2 = 30;
+			
+						// Write text on image using graphicsmagick
+						img.shave(1,1)
+							.borderColor('white')
+							.border(2,2)
+							.borderColor('black')
+							.border(50,200)
+							.fill(FONT_FILL)
+							.stroke(STROKE_COLOR, -2)
+							.font(FONT, FONT_SIZE1)
+							.drawText(0, TOP_POS, TOP_TEXT, TEXT_POS)
+							.font(FONT, FONT_SIZE2)
+							.drawText(0, BOTTOM_POS, BOTTOM_TEXT, TEXT_POS)
+							.write(output, function(err) {
+								loadingmsg.delete();
+								if (err) {
+									console.error(err);
+									return message.channel.send('An error just occured! is it a static image?');
+								}
+								return message.channel.send({files: [output]});
+							});
+					});
+				});
+			})
+			.catch((err) => {
+				console.error(err);
+				return message.channel.send(`Please input a correct link \`${err}\``);
+			});
+	}
+}
+
+module.exports = posterCommand;