diff --git a/commands/edit-image/emoji-face.js b/commands/edit-image/emoji-face.js new file mode 100644 index 00000000..f778dca5 --- /dev/null +++ b/commands/edit-image/emoji-face.js @@ -0,0 +1,98 @@ +const Command = require('../../structures/Command'); +const { GuildEmoji } = require('discord.js'); +const request = require('node-superfetch'); +const { createCanvas, loadImage } = require('canvas'); +const twemoji = require('twemoji-parser'); +const { base64 } = require('../../util/Util'); +const { FACEPLUSPLUS_KEY, FACEPLUSPLUS_SECRET } = process.env; + +module.exports = class EmojiFaceCommand extends Command { + constructor(client) { + super(client, { + name: 'emoji-face', + aliases: ['emoji-f', 'e-face'], + group: 'edit-image', + memberName: 'emoji-face', + description: 'Draws an emoji onto the faces in an image.', + throttling: { + usages: 2, + duration: 60 + }, + credit: [ + { + name: 'Face++ Cognitive Services', + url: 'https://www.faceplusplus.com/', + reason: 'Face Detection API', + reasonURL: 'https://www.faceplusplus.com/face-detection/' + } + ], + args: [ + { + key: 'emoji', + prompt: 'What emoji do you want to draw?', + type: 'default-emoji|custom-emoji' + }, + { + key: 'image', + prompt: 'What face would you like to scan?', + type: 'image-or-avatar' + } + ] + }); + } + + async run(msg, { emoji, image }) { + let emojiURL; + if (emoji instanceof GuildEmoji) { + emojiURL = emoji.url; + } else { + const parsed = twemoji.parse(emoji); + if (!parsed.length || !parsed[0] || !parsed[0].url) return msg.reply('This emoji is not yet supported.'); + emojiURL = parsed[0].url; + } + const emojiData = await request.get(emojiURL); + const emojiImg = await loadImage(emojiData); + const imgData = await request.get(image); + try { + const faces = await this.detect(imgData); + if (!faces) return msg.reply('There are no faces in this image.'); + if (faces === 'size') return msg.reply('This image is too large.'); + const base = await loadImage(imgData.body); + const canvas = createCanvas(base.width, base.height); + const ctx = canvas.getContext('2d'); + ctx.drawImage(base, 0, 0); + for (const face of faces) { + const landmarks = face.landmark; + const width = landmarks.contour_right1.x - landmarks.contour_left1.x; + const ratio = width / danny.width; + const height = danny.height * ratio; + ctx.drawImage( + emojiImg, + landmarks.contour_left1.x - (width * 0.25), + landmarks.contour_left1.y - (height / 2) - (height * 0.25), + width * 1.5, + height * 1.5 + ); + } + return msg.say({ files: [{ attachment: canvas.toBuffer(), name: 'emoji-face.png' }] }); + } catch (err) { + if (err.status === 400) return msg.reply('There are no faces in this image.'); + if (err.status === 403) return msg.reply('Hold your horses! The command is overloaded! Try again soon.'); + return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`); + } + } + + async detect(imgData) { + if (Buffer.byteLength(imgData.body) >= 2e+6) return 'size'; + const { body } = await request + .post('https://api-us.faceplusplus.com/facepp/v3/detect') + .attach('image_base64', base64(imgData.body)) + .query({ + api_key: FACEPLUSPLUS_KEY, + api_secret: FACEPLUSPLUS_SECRET, + return_landmark: 1 + }); + if (!body.faces || !body.faces.length) return null; + return body.faces; + } +}; diff --git a/package.json b/package.json index 61899afd..50ca8418 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "xiao", - "version": "134.8.0", + "version": "134.9.0", "description": "Your personal server companion.", "main": "Xiao.js", "private": true, @@ -80,6 +80,7 @@ "tesseract.js": "^2.1.4", "text-diff": "^1.0.1", "tictactoe-minimax-ai": "^1.2.1", + "twemoji-parser": "^13.0.0", "user-agents": "^1.0.611", "valid-url": "^1.0.9", "wavefile": "^11.0.0",