-
Notifications
You must be signed in to change notification settings - Fork 3
/
resourcer.js
95 lines (76 loc) · 2.37 KB
/
resourcer.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
var fs = require('fs');
var assert = require('assert');
var join = require('path').join;
var mount = require('koa-mount');
var basename = require('path').basename;
var debug = require('debug')('koa-resourcer');
/**
* Auto mount all child resources
*
* @param {koa app} app
* @param {String} dir A directory name
* @param {Function} [fn] callback to invoke for each mounted resource
*/
module.exports = function koaResourcer(app, dir, fn) {
if (fn) {
assert.equal('function', typeof fn, 'fn must be a Function');
} else {
fn = noop;
}
walk(dir);
return;
function walk(fulldirname, prefix) {
var contents = fs.readdirSync(fulldirname);
var config = getConfig(contents, fulldirname, undefined !== prefix);
if (config && config.source) {
return handleApp(config, fulldirname, prefix);
}
if ('string' == typeof prefix) {
if (config && config.path) {
prefix = prefix + '/' + config.path;
} else {
prefix = prefix + '/' + basename(fulldirname);
}
} else {
prefix = '';
}
contents.forEach(function(name) {
var dir = join(fulldirname, name);
var stat = fs.statSync(dir);
if (!stat.isDirectory()) return;
walk(dir, prefix);
});
}
function getConfig(contents, dir, nested) {
if (~contents.indexOf('config.json')) {
var path = join(dir, 'config.json');
var config = require(path);
// In nested directories, we allow overridding the directory
// name to provide custom paths. If a user specifies "/" in
// a nested directory, it means "use the directory name" but
// at the top level directory, "/" means a literal "/".
if (nested && config && config.path) {
config.path = config.path.replace(/^\/+/, '');
}
return config;
}
}
function handleApp(config, fulldirname, prefix) {
assert(config.source, 'missing config.source');
var source = join(fulldirname, config.source);
var resource = require(source);
var suffix = config.path || basename(fulldirname);
var route;
if (prefix) {
route = prefix + '/' + suffix;
} else if ('/' == suffix.charAt(0)) {
route = suffix;
} else {
route = '/' + suffix;
}
debug('mounting %s => %s', route, source)
app.use(mount(route, resource));
fn({ path: route, resource: resource });
}
}
function noop() {};