LambdaRunner - interface between ES6 generators and lambda functions

co-mocha-runner.js

'use strict';

const co = require('co');

module.exports = function(lambda) {
    return function(e, ctx, cb) {
        co(
            function* () {
                let result = yield lambda(e, ctx);

                if (cb) {
                    cb(null, result)
                }

                ctx.succeed(result);
            }
        )
        .catch(
            function(err){
                let re = /Not found:/, msg;
                if( err  && re.test(err)) {
                  msg = err.message || err || "Not found: could not find resource";
                } else {
                  msg = `Error: ${err.message || 'Internal error'}`
                }
                ctx.fail(msg);
                if(cb) {
                    cb(err)
                }
            }
        );
    }
};