Added turtles

This commit is contained in:
CBenni 2017-12-10 15:03:09 +01:00
parent 76a9510a7f
commit 4bb88b8296
2 changed files with 86 additions and 66 deletions

BIN
TurtleTemplate.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 46 KiB

144
index.js
View file

@ -12,17 +12,35 @@ var config = require("./config");
var cache = {}; var cache = {};
const beeTemplate = new Image(); templates = {
beeTemplate.src = "./BeeTemplate.png"; bee: {
template: "./BeeTemplate.png",
leftOffset: 475,
widthTarget: 600,
bottomTarget: 530
},
turtle: {
template: "./TurtleTemplate.png",
leftOffset: 210,
widthTarget: 150,
bottomTarget: 150
}
}
for(templateName in templates) {
const data = templates[templateName];
data.image = new Image();
data.image.src = data.template;
}
function bee(url, size) { function render(template, url, size) {
return new Promise((resolve,reject)=>{ return new Promise((resolve, reject) => {
console.log("Getting "+url); console.log("Getting " + url);
if(url) { if (url) {
request.get({url:url, encoding: null}, function(e, r, data) { request.get({ url: url, encoding: null }, function (e, r, data) {
if(e) { if (e) {
reject({status: (r && r.statusCode || 500), error: e}); reject({ status: (r && r.statusCode || 500), error: e });
return; return;
} }
var img = new Image(); var img = new Image();
@ -30,82 +48,83 @@ function bee(url, size) {
console.log("image loaded"); console.log("image loaded");
var width = img.width; var width = img.width;
var height = img.height; var height = img.height;
if(size && size.height) { if (size && size.height) {
height = size.height; height = size.height;
if(!size.width) width = width * size.height / img.height; if (!size.width) width = width * size.height / img.height;
} }
if(size && size.width) { if (size && size.width) {
width = size.width; width = size.width;
if(!size.height) height = height * size.width / img.width; if (!size.height) height = height * size.width / img.width;
} }
const leftOffset = 475.0; // relative to bee template width const templateScale = width / template.widthTarget; // scale the template to fit the image
const widthTarget = 600.0; // relative to bee template width
const bottomTarget = 530; // relative to bee template height
const beeScale = width / widthTarget; // scale the bee to fit the image
const resultingWidth = beeTemplate.width * beeScale; let resultingWidth = template.image.width * templateScale;
let resultingHeight = beeTemplate.height * beeScale; let resultingHeight = template.image.height * templateScale;
let imgTop = bottomTarget * beeScale - height; // naive top center position let imgTop = template.bottomTarget * templateScale - height; // naive top center position
if(imgTop < 0) { if (imgTop < 0) {
resultingHeight -= imgTop; resultingHeight -= imgTop;
imgTop = 0; imgTop = 0;
} }
const resultingImgTop = imgTop; const resultingImgTop = imgTop;
const resultingImgLeft = leftOffset * beeScale - width / 2.0; const resultingImgLeft = template.leftOffset * templateScale - width / 2.0;
const resultingBeeTop = resultingHeight - beeTemplate.height * beeScale; if(resultingImgLeft + width > resultingWidth) {
const resultingBeeLeft = 0.0; resultingWidth = resultingImgLeft + width;
}
const resultingTemplateTop = resultingHeight - template.image.height * templateScale;
const resultingTemplateLeft = 0.0;
var canvas = new Canvas(resultingWidth, resultingHeight); var canvas = new Canvas(resultingWidth, resultingHeight);
var ctx = canvas.getContext("2d"); var ctx = canvas.getContext("2d");
console.log("Drawing first image") console.log("Drawing template")
try { try {
ctx.drawImage(beeTemplate, resultingBeeLeft, resultingBeeTop, resultingWidth, beeTemplate.height * beeScale); ctx.drawImage(template.image, resultingTemplateLeft, resultingTemplateTop, template.image.width * templateScale, template.image.height * templateScale);
console.log("Drawing done.") console.log("Drawing done.")
} catch(err) { } catch (err) {
reject({status: 400, error: "Invalid template"}) reject({ status: 400, error: "Invalid template" })
} }
console.log("Drawing second image") console.log("Drawing image")
try { try {
ctx.drawImage(img, resultingImgLeft, resultingImgTop, width, height); ctx.drawImage(img, resultingImgLeft, resultingImgTop, width, height);
console.log("Drawing done.") console.log("Drawing done.")
} catch(err) { } catch (err) {
console.error(err); console.error(err);
reject({status: 400, error: "Invalid image"}) reject({ status: 400, error: "Invalid image" })
} }
// return the image and cache it // return the image and cache it
resolve(canvas); resolve(canvas);
}); });
} else { } else {
reject({status: 400, error: "No url specified"}); reject({ status: 400, error: "No url specified" });
} }
}); });
} }
const circleScale = 0.4; const circleScale = 0.4;
const lineWidthScale = 0.2; const lineWidthScale = 0.2;
const lowerRightFactor = 0.5 + 0.7071*circleScale; // %-ual location of the lower right end of the strikethrough line const lowerRightFactor = 0.5 + 0.7071 * circleScale; // %-ual location of the lower right end of the strikethrough line
const upperLeftFactor = 0.5 - 0.7071*circleScale; // %-ual location of the upper left end of the strikethrough line const upperLeftFactor = 0.5 - 0.7071 * circleScale; // %-ual location of the upper left end of the strikethrough line
app.get("/bee/", function(req, res) { app.get("/:templateName/", function (req, res) {
bee(req.query.url).then((canvas)=>{ if(!templates[req.params.templateName]) return res.status(404).end();
render(templates[req.params.templateName], req.query.url).then((canvas) => {
console.log(canvas) console.log(canvas)
res.setHeader('Content-Type', 'image/png'); res.setHeader('Content-Type', 'image/png');
canvas.pngStream().pipe(res); canvas.pngStream().pipe(res);
}).catch((err)=>{ }).catch((err) => {
res.status(err.status).end(err.error); res.status(err.status).end(err.error);
}) })
}); });
app.listen(3002, function () { app.listen(3002, function () {
console.log('Beebot app listening on port 3002!') console.log('Beebot app listening on port 3002!')
}) })
@ -114,26 +133,26 @@ app.listen(3002, function () {
var client = new Discord.Client({ var client = new Discord.Client({
autoReconnect: true autoReconnect: true
}); });
// manage roles permission is required // manage roles permission is required
const invitelink = 'https://discordapp.com/oauth2/authorize?client_id=' const invitelink = 'https://discordapp.com/oauth2/authorize?client_id='
+ config.discord.client_id + '&scope=bot&permissions=0'; + config.discord.client_id + '&scope=bot&permissions=0';
const authlink = 'https://discordapp.com/oauth2/authorize?client_id=' const authlink = 'https://discordapp.com/oauth2/authorize?client_id='
+ config.discord.client_id + '&scope=email'; + config.discord.client_id + '&scope=email';
console.log("Bot invite link: "+invitelink); console.log("Bot invite link: " + invitelink);
client.login(config.discord.token).catch(function (error) { client.login(config.discord.token).catch(function (error) {
if (error) { if (error) {
console.error("Couldn't login: ", error); console.error("Couldn't login: ", error);
process.exit(15); process.exit(15);
} }
}); });
function findEmoji(str) { function findEmoji(str) {
const discordEmote = /<:(\w+):(\d+)>/g.exec(str) const discordEmote = /<:(\w+):(\d+)>/g.exec(str)
if(discordEmote) { if (discordEmote) {
return { return {
name: discordEmote[1], name: discordEmote[1],
id: discordEmote[2], id: discordEmote[2],
@ -142,8 +161,8 @@ function findEmoji(str) {
} }
let unicodeEmoji; let unicodeEmoji;
twemoji.parse(str, (name, emoji) =>{ twemoji.parse(str, (name, emoji) => {
if(unicodeEmoji) return false; if (unicodeEmoji) return false;
unicodeEmoji = { unicodeEmoji = {
name, name,
id: name, id: name,
@ -156,16 +175,17 @@ function findEmoji(str) {
client.on('message', function (message) { client.on('message', function (message) {
console.log(message.cleanContent); console.log(message.cleanContent);
if (message.cleanContent.startsWith("/bee")) { const commandParsed = /^\/(\w+)\b/.exec(message.cleanContent);
// get emoji from message if (commandParsed && templates[commandParsed[1]]) {
const emoji = findEmoji(message.cleanContent); // get emoji from message
if(emoji) bee(emoji.url).then(canvas => { const emoji = findEmoji(message.cleanContent);
var messageOptions = { if (emoji) render(templates[commandParsed[1]], emoji.url).then(canvas => {
files: [ var messageOptions = {
{attachment: canvas.toBuffer(), name: emoji.name.toUpperCase()+"DETECTED.png"} files: [
] { attachment: canvas.toBuffer(), name: emoji.name + commandParsed[1] + ".png" }
} ]
message.channel.send("", messageOptions); }
}); message.channel.send("", messageOptions);
} });
}
}); });