spus/lib/context.js

var formidable = require("formidable");
var crypto = require("crypto");
var preprocess = require("./preprocess.js");

var sessions = {};

function templatify(str, args, ctx) {
	str = preprocess(str, {
		session: ctx.session,
		template: function(key) {
			return ctx.template(key);
		}
	});

	if (args == undefined)
		return str;

	for (var i in args) {
		str = str.split("{{"+i+"}}").join(args[i]);
	}

	return str;
}

module.exports = function(options) {
	this.req = options.req;
	this.res = options.res;
	this.templates = options.templates;
	this.views = options.views;
	this.db = options.db;
	this.conf = options.conf;

	//Handle cookies
	this.cookies = {};
	this.req.headers.cookie = this.req.headers.cookie || "";
	this.req.headers.cookie.split(/;\s*/).forEach(function(elem) {
		var pair = elem.split("=");
		this.cookies[pair[0]] = decodeURIComponent(pair[1]);
	}.bind(this));

	//Handle sessions
	if (sessions[this.cookies.session]) {
		this.session = sessions[this.cookies.session];
	} else {
		var key;
		do {
			key = crypto.randomBytes(64).toString("hex");
		} while (sessions[key]);

		sessions[key] = {};
		this.res.setHeader("Set-Cookie", "session="+key);

		//Delete session after a while
		setTimeout(function() {
			delete sessions[key];
		}, this.conf.session_timeout);

		this.session = sessions[key];
	}
}

module.exports.prototype = {
	end: function(str) {
		if (this.statusCode)
			this.res.writeHead(this.statusCode);

		this.res.end(str);
	},

	succeed: function(obj) {
		obj = obj || {};
		obj.success = true;

		this.res.setHeader("Content-Type", "application/json");

		this.end(JSON.stringify(obj));
	},

	fail: function(err) {
		console.log("Sending error to client:");
		console.trace(err);

		this.res.setHeader("Content-Type", "application/json");

		obj = {};
		obj.success = false;
		obj.error = err.toString();
		this.end(JSON.stringify(obj));
	},

	template: function(name, args) {
		var str = this.templates[name];
		if (!str)
			throw new Error("No such template: "+name);

		return templatify(str, args, this);
	},

	view: function(name, args) {
		var str = this.views[name];
		if (!str)
			throw new Error("No such view: "+name);

		return templatify(str, args, this);
	},

	getPostData: function(cb) {
		if (this.postData)
			return cb(null, this.postData.data, this.postData.files);

		if (this.req.method.toUpperCase() != "POST")
			return cb(new Error("Expected POST request, got "+this.req.method));

		var form = new formidable.IncomingForm();
		form.parse(this.req, function(err, data, files) {
			if (err) return cb(err);

			this.postData = {
				data: data,
				files: files
			}

			cb(null, data, files);
		}.bind(this));
	},

	setStatus: function(code) {
		this.statusCode = code;
	}
}