mirror of https://github.com/JoeBiellik/paste
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
135 lines
3.3 KiB
JavaScript
135 lines
3.3 KiB
JavaScript
const config = require('config');
|
|
const fs = require('fs').promises;
|
|
const isBinaryFile = require('isbinaryfile').isBinaryFile;
|
|
const Paste = require('../models/paste');
|
|
|
|
module.exports = {
|
|
async index(ctx) {
|
|
await ctx.render('index', {
|
|
pretty: config.prettyHtml,
|
|
title: config.name,
|
|
url: ctx.request.origin,
|
|
expires: config.expires,
|
|
highlights: config.highlights
|
|
});
|
|
},
|
|
|
|
async view(ctx) {
|
|
try {
|
|
const paste = await Paste.findById(ctx.params.id).exec();
|
|
const lang = (Object.keys(ctx.query)[0] || '').toLowerCase();
|
|
|
|
ctx.set('Cache-Control', 'public');
|
|
ctx.set('Expires', paste.expiresAt.toUTCString());
|
|
|
|
if (lang) {
|
|
await ctx.render('highlight', {
|
|
pretty: config.prettyHtml,
|
|
title: config.name + ' ' + paste.id,
|
|
paste: paste.paste,
|
|
lang: Object.keys(config.highlights).includes(lang) ? lang : 'unknown'
|
|
});
|
|
} else {
|
|
ctx.body = paste.paste;
|
|
}
|
|
} catch (ex) {
|
|
ctx.throw(404, 'Paste Not Found');
|
|
}
|
|
},
|
|
|
|
async create(ctx) {
|
|
ctx.set('Cache-Control', 'no-cache');
|
|
|
|
// Request body
|
|
if (!ctx.request.body.paste && ctx.request.files) {
|
|
try {
|
|
// Request body, xxx
|
|
let path = Object.values(ctx.request.files)[0].path;
|
|
|
|
// Request body, paste=xxx
|
|
if (ctx.request.files.paste) {
|
|
path = ctx.request.files.paste.path;
|
|
}
|
|
|
|
const data = await fs.readFile(path);
|
|
const stat = await fs.lstat(path);
|
|
|
|
if (isBinaryFile(data, stat.size)) ctx.throw();
|
|
|
|
ctx.request.body.paste = data;
|
|
|
|
try {
|
|
await fs.unlink(path);
|
|
} catch (ex) {
|
|
// Ignore
|
|
}
|
|
} catch (ex) {
|
|
ctx.throw(400, 'Bad Paste Body');
|
|
}
|
|
}
|
|
|
|
// Expiry multiplier
|
|
try {
|
|
if (ctx.request.body.expire && ctx.request.body.multiplier) {
|
|
ctx.request.body.expire = ctx.request.body.expire * ctx.request.body.multiplier;
|
|
}
|
|
} catch (ex) {
|
|
ctx.throw(400, 'Bad Paste Expiry');
|
|
}
|
|
|
|
// Raw request body
|
|
if (typeof ctx.request.body === 'string') {
|
|
ctx.request.body = {
|
|
paste: ctx.request.body
|
|
};
|
|
}
|
|
|
|
if (!ctx.request.body.paste) {
|
|
ctx.throw(400, 'No Paste Provided');
|
|
}
|
|
|
|
// /?expire=xxx
|
|
if (!ctx.request.body.expire && ctx.query.expire) {
|
|
ctx.request.body.expire = ctx.query.expire;
|
|
}
|
|
|
|
// No expire provided
|
|
if (!ctx.request.body.expire) {
|
|
ctx.request.body.expire = config.expires.default.value * config.expires.default.multiplier;
|
|
}
|
|
|
|
const paste = new Paste({
|
|
paste: ctx.request.body.paste,
|
|
expiresAt: new Date(Date.now() + ctx.request.body.expire * 1000)
|
|
});
|
|
|
|
try {
|
|
await paste.save();
|
|
} catch (ex) {
|
|
ctx.throw(500, 'Internal Storage Error');
|
|
}
|
|
|
|
// /?highlight=xxx
|
|
if (!ctx.request.body.highlight && ctx.query.highlight) {
|
|
ctx.request.body.highlight = ctx.query.highlight;
|
|
}
|
|
|
|
// /?xxx
|
|
if (!ctx.request.body.highlight && !ctx.query.highlight && ctx.query) {
|
|
ctx.request.body.highlight = Object.keys(ctx.query).filter(k => k != 'redirect' && k != 'expire')[0];
|
|
}
|
|
|
|
let highlight = '';
|
|
|
|
if (ctx.request.body.highlight && Object.keys(config.highlights).includes(ctx.request.body.highlight)) {
|
|
highlight = '?' + ctx.request.body.highlight;
|
|
}
|
|
|
|
if (Object.keys(ctx.query).includes('redirect')) {
|
|
ctx.redirect(ctx.request.origin + '/' + paste.id + highlight);
|
|
} else {
|
|
ctx.body = ctx.request.origin + '/' + paste.id + highlight + '\n';
|
|
}
|
|
}
|
|
};
|