This repository has been archived on 2022-01-28. You can view files and clone it, but cannot push or open issues or pull requests.
beebot/index.js
2017-12-15 14:18:21 +01:00

247 lines
No EOL
6.6 KiB
JavaScript

var _ = require('lodash');
var express = require("express");
var request = require("request");
var Discord = require('discord.js');
const Canvas = require("canvas");
const Image = Canvas.Image;
const twemoji = require("./twemoji");
var app = express()
var config = require("./config.default.json");
try {
_.extend(config, require("./config"));
}catch(err) {
console.log("No config.json found!");
}
var cache = {};
function all(x, c) {
_.isArray(x) ? _.each(x,c) : c(x);
}
templates = config.templates;
for(templateName in templates) {
const data = templates[templateName];
all(data, template => {
template.image = new Image();
template.image.src = template.src;
})
}
// drawing: we keep the image fixed in its default position and draw the template on top/below it
// calculates the x or y position of the template to be drawn
// size = width or height of the template/image
// anchor = the corresponding anchor config
function calculatePosition(scale, anchor, imageSize) {
return imageSize * anchor.position / 100 - anchor.offset * scale;
}
function render(template, img, size) {
var imgWidth = img.width;
var imgHeight = img.height;
if (size && size.height) {
imgHeight = size.height;
if (!size.width) imgWidth = imgWidth * size.height / img.height;
}
if (size && size.width) {
imgWidth = size.width;
if (!size.height) imgHeight = imgHeight * size.width / img.width;
}
const xScale = imgWidth / template.anchor.x.size;
const yScale = imgHeight / template.anchor.y.size;
const templateScale = Math.max(0, Math.min(10, Math.max(xScale || 0, yScale || 0)));
console.log("templateScale",templateScale)
let templateOffsetX = calculatePosition(templateScale, template.anchor.x, imgWidth);
let templateOffsetY = calculatePosition(templateScale, template.anchor.y, imgHeight);
let imageOffsetX = 0;
let imageOffsetY = 0;
let resultingWidth = imgWidth; // start with the image boundaries as defined by the image
let resultingHeight = imgHeight;
if(templateOffsetX < 0) {
resultingWidth -= templateOffsetX;
imageOffsetX = -templateOffsetX;
templateOffsetX = 0;
}
if(templateOffsetY < 0) {
resultingHeight -= templateOffsetY;
imageOffsetY = -templateOffsetY;
templateOffsetY = 0;
}
if(templateOffsetX + template.image.width * templateScale > resultingWidth) {
resultingWidth = templateOffsetX + template.image.width * templateScale;
}
if(templateOffsetY + template.image.height * templateScale > resultingHeight) {
resultingHeight = templateOffsetY + template.image.height * templateScale;
}
const toDraw = [{
z: 1,
image: img,
x: imageOffsetX,
y: imageOffsetY,
h: imgHeight,
w: imgWidth,
name: "image"
}, {
z: template.z || 0,
image: template.image,
x: templateOffsetX,
y: templateOffsetY,
h: template.image.height * templateScale,
w: template.image.width * templateScale,
name: "template "+template.src
}].sort((u,v) => u.z > v.z);
var canvas = new Canvas(resultingWidth, resultingHeight);
var ctx = canvas.getContext("2d");
for(let i=0;i<toDraw.length;++i) {
const subject = toDraw[i];
console.log("Drawing "+subject.name)
try {
ctx.drawImage(subject.image, subject.x, subject.y, subject.w, subject.h);
} catch (err) {
console.error(err);
throw new Error(JSON.stringify({ status: 400, error: "Invalid template" }))
}
}
// return the image and cache it
return(canvas);
}
app.get("/:templateName/", async function (req, res) {
if(!templates[req.params.templateName]) return res.status(404).end();
try {
const canvas = render(templates[req.params.templateName], await loadImage(req.query.url))
console.log(canvas)
res.setHeader('Content-Type', 'image/png');
return canvas.pngStream().pipe(res);
} catch(err) {
console.log(err);
return res.status(400).end(err.message);
}
});
app.listen(3002, function () {
console.log('Beebot app listening on port 3002!')
})
// Discord stuff
var client = new Discord.Client({
autoReconnect: true
});
// manage roles permission is required
const invitelink = 'https://discordapp.com/oauth2/authorize?client_id='
+ config.discord.client_id + '&scope=bot&permissions=0';
const authlink = 'https://discordapp.com/oauth2/authorize?client_id='
+ config.discord.client_id + '&scope=email';
console.log("Bot invite link: " + invitelink);
client.login(config.discord.token).catch(function (error) {
if (error) {
console.error("Couldn't login: ", error);
process.exit(15);
}
});
function findEmoji(str) {
const discordEmote = /<:(\w+):(\d+)>/g.exec(str)
if (discordEmote) {
return {
name: discordEmote[1],
id: discordEmote[2],
url: `https://cdn.discordapp.com/emojis/${discordEmote[2]}.png`
}
}
let unicodeEmoji;
twemoji.parse(str, (name, emoji) => {
if (unicodeEmoji) return false;
unicodeEmoji = {
name,
id: name,
url: emoji.base + emoji.size + "/" + name + emoji.ext
}
return false;
});
return unicodeEmoji;
}
function loadImage(url) {
return new Promise((resolve, reject) => {
console.log("Getting " + url);
if (url) {
request.get({ url: url, encoding: null }, function (e, r, data) {
if (e) {
return reject({ status: (r && r.statusCode || 500), error: e });
}
var img = new Image();
img.src = data;
resolve(img);
})
}
});
}
client.on('message', async function (message) {
console.log(`[${message.guild.name} - ${message.channel.name}] ${message.author.username}#${message.author.discriminator}: ${message.cleanContent}`);
if(message.cleanContent.startsWith('/invite')) {
message.channel.send(`Invite link: <${invitelink}>`);
return;
}
if(message.cleanContent.startsWith('/help') || message.cleanContent.startsWith('/beebot')) {
let commands = Object.keys(templates).map(x => '/'+x).join(', ');
message.channel.send(`Available commands: ${commands}`);
return;
}
const messageSplit = message.cleanContent.split(" ");
const emoji = findEmoji(message.cleanContent);
let result = null;
let count = 0;
try {
if (emoji) {
let name = emoji.name;
for (var i = 0; i < messageSplit.length && count < 4; ++i) {
const commandParsed = /^\/(\w+)\b/.exec(messageSplit[i]);
if (commandParsed && templates[commandParsed[1]]) {
count++;
name += commandParsed[1];
if (result === null) result = await loadImage(emoji.url);
const templateData = templates[commandParsed[1]];
all(templateData, template => {
result = render(template, result);
})
} else {
if(i===0) return;
}
}
if (result) {
var messageOptions = {
files: [
{ attachment: result.toBuffer(), name: name + ".png" }
]
}
message.channel.send("", messageOptions);
}
}
} catch(err) {
console.error(err);
}
});