Node.js 開發Rest的Api示例項目（二）

app.js

'use strict';

console.log('Hello world');
const Koa = require('koa');
const bodyParser = require('koa-bodyparser');

const app = new Koa();
const controller = require('./controller');
// parse request body:
app.use(bodyParser());
// add controller:
app.use(controller());
app.listen(5000);
console.log('app started at port 5000...');

controller.js

const fs = require('fs');

function addMapping(router, mapping) {
    for (var url in mapping) {
        if (url.startsWith('GET ')) {
            var path = url.substring(4);
            router.get(path, mapping[url]);
            console.log(`register URL mapping: GET ${path}`);
        } else if (url.startsWith('POST ')) {
            var path = url.substring(5);
            router.post(path, mapping[url]);
            console.log(`register URL mapping: POST ${path}`);
        } else {
            console.log(`invalid URL: ${url}`);
        }
    }
}

function addControllers(router) {
    var fs = require('fs');
    var files = fs.readdirSync(__dirname + '\\controllers');
    var js_files = files.filter((f) => {
        return f.endsWith('.js');
    });

    for (var f of js_files) {
        console.log(`process controller: ${f}...`);
        let mapping = require(__dirname + '\\controllers\\' + f);
        addMapping(router, mapping);
    }
}


module.exports = function (dir) {
    let
        controllers_dir = dir || 'controllers', // 如果不傳參數，掃描目錄默認爲'controllers'
        router = require('koa-router')();
    addControllers(router, controllers_dir);
    return router.routes();
};

controllers文件夾下的接口接口文件：

./controllers/hello.js

'use strict'
module.exports = {
    'POST /hello/:name': async (ctx, next) => {
        var name = ctx.request.body.name || '';
        ctx.response.body = `<h1>Hello post, ${name}!</h1>`;
    }
};

./controllers/index.js

'use strict'

var fn_index = async (ctx, next) => {
    ctx.response.body = `<h1>Index</h1>
        <form action="/signin" method="post">
            <p>Name: <input name="name" value="koa"></p>
            <p>Password: <input name="password" type="password"></p>
            <p><input type="submit" value="Submit"></p>
        </form>`;
};

var fn_signin = async (ctx, next) => {
    var
        name = ctx.request.body.name || '',
        password = ctx.request.body.password || '';
    console.log(`signin with name: ${name}, password: ${password}`);
    if (name === 'koa' && password === '12345') {
        ctx.response.body = `<h1>Welcome, ${name}!</h1>`;
    } else {
        ctx.response.body = `<h1>Login failed!</h1>
        <p><a href="/">Try again</a></p>`;
    }
};



module.exports = {
    'GET /': fn_index,
    'POST /signin': fn_signin
};