const path = require('path');
const fs = require('mz/fs');
const { promisify } = require('util');
const Reply = function (name, ...args) {
this.stack = [];
this[name](...args);
return this;
};
Reply.prototype.cookie = function (...args) {
this.stack.push(ctx => {
ctx.res.cookie(...args);
});
return this;
};
Reply.prototype.download = function (...args) {
// Guard clauses
if (args.length < 1) {
throw new Error('download() expects a path as the first argument');
}
if (args.length < 2) {
throw new Error('download() expects a filename as the second argument');
}
if (args.length > 2) {
throw new Error('download() only expects two arguments, path and filename. The rest of them will be ignored');
}
let [file, opts] = args;
if (!path.isAbsolute(file)) {
file = path.resolve(process.cwd(), file);
}
this.stack.push(async ({ res }) => {
if (!await fs.exists(file)) {
throw new Error(`The file "${file}" does not exist. Make sure that you set an absolute path or a relative path to the root of your project`);
}
return promisify(res.download.bind(res))(file, opts);
});
return this;
};
Reply.prototype.end = function () {
this.stack.push(ctx => {
ctx.res.end();
});
return this;
};
Reply.prototype.file = function (...args) {
// Guard clauses
if (args.length < 1) {
throw new Error('file() expects a path as the first argument');
}
if (args.length > 2) {
throw new Error(`file() only expects two arguments, the path and options, but ${args.length} were provided.`);
}
let [file, opts = {}] = args;
if (!path.isAbsolute(file)) {
file = path.resolve(process.cwd(), file);
}
this.stack.push(async ({ res }) => {
if (!await fs.exists(file)) {
throw new Error(`The file "${file}" does not exist. Make sure that you set an absolute path or a relative path to the root of your project`);
}
return promisify(res.sendFile.bind(res))(file, opts);
});
return this;
};
Reply.prototype.header = function (...args) {
this.stack.push(ctx => {
ctx.res.header(...args);
});
return this;
};
Reply.prototype.json = function (...args) {
this.stack.push(ctx => {
ctx.res.json(...args);
});
return this;
};
Reply.prototype.jsonp = function (...args) {
this.stack.push(ctx => {
ctx.res.jsonp(...args);
});
return this;
};
Reply.prototype.redirect = function (...args) {
this.stack.push(ctx => {
ctx.res.redirect(...args);
});
return this;
};
Reply.prototype.render = function (...args) {
// Guard clauses
if (args.length < 1) {
throw new Error('render() expects a path');
}
if (args.length > 2) {
throw new Error('render() expects a path and options but nothing else');
}
let [file, opts = {}] = args;
this.stack.push(async ({ res }) => {
const html = await promisify(res.render.bind(res))(file, opts);
res.send(html);
});
return this;
};
Reply.prototype.send = function (...args) {
// If we are trying to send the context
Iif (args[0] && args[0].close && args[0].close instanceof Function) {
throw new Error('Never send the context, request or response as those are a security risk');
}
this.stack.push(ctx => {
ctx.res.send(...args);
});
return this;
};
Reply.prototype.status = function (...args) {
this.stack.push(ctx => {
// In case there is no response, it'll respond with the status
ctx.res.explicitStatus = true;
ctx.res.status(...args);
});
return this;
};
Reply.prototype.type = function (...args) {
this.stack.push(ctx => {
ctx.res.type(...args);
});
return this;
};
Reply.prototype.exec = async function (ctx) {
for (let cb of this.stack) {
await cb(ctx);
}
this.stack = [];
};
// This will make that the first time a function is called it starts a new stack
module.exports = Reply;
|