paste/controllers/pastes.js

var config = require('config');
var Paste = require('../models/paste');

module.exports = {
	*view() {
		try {
			let paste = yield Paste.findById(this.params.id).exec();
			let lang = Object.keys(this.query)[0];

			if (lang) {
				yield this.render('highlight', {
					pretty: config.prettyHtml,
					title: 'Paste ' + paste.id,
					paste: paste.paste,
					lang: lang
				});
			} else {
				this.type = 'text/plain';
				this.body = paste.paste;
			}
		} catch (ex) {
			this.throw('Paste Not Found', 404);
		}
	},

	*create() {
		if (this.request.body.fields) {
			if (this.request.body.fields.paste) {
				this.request.body.paste = this.request.body.fields.paste;
			}
			if (this.request.body.fields.highlight) {
				this.request.body.highlight = this.request.body.fields.highlight;
			}
			if (this.request.body.fields.expire) {
				this.request.body.expire = this.request.body.fields.expire;
			}
		}

		if (!this.request.body.expire) {
			this.request.body.expire = config.expiresDefault;
		}

		let paste = new Paste({
			paste: this.request.body.paste,
			expiresAt: new Date(Date.now() + this.request.body.expire * 1000)
		});

		yield paste.save();

		let link = paste.id;

		if (this.request.body.highlight) {
			link += '?' + this.request.body.highlight;
		}

		if (Object.keys(this.query).includes('redirect')) {
			this.redirect(link);
		} else {
			this.body = this.request.origin + '/' + link + '\n';
		}
	}
};