SunRed/beebot
Archived
1
0
Fork 0
Code Issues 7 Pull requests Projects Releases Wiki Activity

Cause I can

Browse source
This commit is contained in:
CBenni 2017-12-10 15:56:37 +01:00
parent 4bb88b8296
commit cec95be43f
1 changed files with 89 additions and 75 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

164
index.js
View file

@ -34,77 +34,61 @@ for(templateName in templates) {
} }
function render(template, url, size) { function render(template, img, size) {
return new Promise((resolve, reject) => { var width = img.width;
console.log("Getting " + url); var height = img.height;
if (url) { if (size && size.height) {
request.get({ url: url, encoding: null }, function (e, r, data) { height = size.height;
if (e) { if (!size.width) width = width * size.height / img.height;
reject({ status: (r && r.statusCode || 500), error: e }); }
return; if (size && size.width) {
} width = size.width;
var img = new Image(); if (!size.height) height = height * size.width / img.width;
img.src = data; }
console.log("image loaded");
var width = img.width;
var height = img.height;
if (size && size.height) {
height = size.height;
if (!size.width) width = width * size.height / img.height;
}
if (size && size.width) {
width = size.width;
if (!size.height) height = height * size.width / img.width;
}
const templateScale = width / template.widthTarget; // scale the template to fit the image const templateScale = width / template.widthTarget; // scale the template to fit the image
let resultingWidth = template.image.width * templateScale; let resultingWidth = template.image.width * templateScale;
let resultingHeight = template.image.height * templateScale; let resultingHeight = template.image.height * templateScale;
let imgTop = template.bottomTarget * templateScale - 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 = template.leftOffset * templateScale - width / 2.0; const resultingImgLeft = template.leftOffset * templateScale - width / 2.0;
if(resultingImgLeft + width > resultingWidth) { if(resultingImgLeft + width > resultingWidth) {
resultingWidth = resultingImgLeft + width; resultingWidth = resultingImgLeft + width;
} }
const resultingTemplateTop = resultingHeight - template.image.height * templateScale; const resultingTemplateTop = resultingHeight - template.image.height * templateScale;
const resultingTemplateLeft = 0.0; 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 template") console.log("Drawing template "+template.template)
try { try {
ctx.drawImage(template.image, resultingTemplateLeft, resultingTemplateTop, template.image.width * templateScale, template.image.height * templateScale); 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 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); return(canvas);
});
} else {
reject({ status: 400, error: "No url specified" });
}
});
} }
const circleScale = 0.4; const circleScale = 0.4;
@ -112,9 +96,9 @@ 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("/:templateName/", function (req, res) { app.get("/:templateName/", async function (req, res) {
if(!templates[req.params.templateName]) return res.status(404).end(); if(!templates[req.params.templateName]) return res.status(404).end();
render(templates[req.params.templateName], req.query.url).then((canvas) => { render(templates[req.params.templateName], await loadImage(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);
@ -173,19 +157,49 @@ function findEmoji(str) {
return unicodeEmoji; return unicodeEmoji;
} }
client.on('message', function (message) { 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.cleanContent); console.log(message.cleanContent);
const commandParsed = /^\/(\w+)\b/.exec(message.cleanContent);
if (commandParsed && templates[commandParsed[1]]) { const messageSplit = message.cleanContent.split(" ");
// get emoji from message const emoji = findEmoji(message.cleanContent);
const emoji = findEmoji(message.cleanContent); let result = null;
if (emoji) render(templates[commandParsed[1]], emoji.url).then(canvas => { try {
var messageOptions = { if (emoji) {
files: [ let name = emoji.name;
{ attachment: canvas.toBuffer(), name: emoji.name + commandParsed[1] + ".png" } for (var i = 0; i < messageSplit.length && i<3; ++i) {
] const commandParsed = /^\/(\w+)\b/.exec(messageSplit[i]);
if (commandParsed && templates[commandParsed[1]]) {
name += commandParsed[1];
if (result === null) result = await loadImage(emoji.url);
result = render(templates[commandParsed[1]], result);
}
} }
message.channel.send("", messageOptions); if (result) {
}); var messageOptions = {
files: [
{ attachment: result.toBuffer(), name: name + ".png" }
]
}
message.channel.send("", messageOptions);
}
}
} catch(err) {
console.error(err);
} }
}); });