修正linux文件目录兼容

This commit is contained in:
2019-11-12 17:06:53 +08:00
parent 1eff89ee5b
commit cea9f385e9
4 changed files with 143 additions and 6 deletions

View File

@@ -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();