修正linux文件目录兼容
This commit is contained in:
@@ -1,4 +1,6 @@
|
||||
const fs = require("fs")
|
||||
const os = require('os');
|
||||
|
||||
function addMapping(router, mapping) {
|
||||
for (var url in mapping) {
|
||||
if (url.startsWith('GET ')) {
|
||||
@@ -16,7 +18,13 @@ function addMapping(router, mapping) {
|
||||
}
|
||||
|
||||
function addControllers(router,controllers_dir) {
|
||||
let dir = __dirname.replace("\\bin","")
|
||||
let ostype = os.type() == "Windows_NT"
|
||||
let dir;
|
||||
if(ostype){
|
||||
dir = __dirname.replace("\\bin","")
|
||||
}else{
|
||||
dir = __dirname.replace("/bin","")
|
||||
}
|
||||
|
||||
var files = fs.readdirSync(dir +"/"+ controllers_dir);
|
||||
var js_files = files.filter((f) => {
|
||||
@@ -29,9 +37,9 @@ function addControllers(router,controllers_dir) {
|
||||
addMapping(router, mapping);
|
||||
}
|
||||
}
|
||||
module.exports = function () {
|
||||
module.exports = function (dir) {
|
||||
let
|
||||
controllers_dir = 'controllers', // 如果不传参数,扫描目录默认为'controllers'
|
||||
controllers_dir = dir || 'controllers', // 如果不传参数,扫描目录默认为'controllers'
|
||||
router = require('koa-router')();
|
||||
addControllers(router, controllers_dir);
|
||||
return router.routes();
|
||||
|
||||
Reference in New Issue
Block a user