finish main logic

index.js

@@ -0,0 +1 @@
+module.exports = require('./lib');

lib/index.js

@@ -0,0 +1,40 @@
+const Boom = require('boom');
+
+module.exports = (
+  env = 'production',
+  errStatusCodePropertie = ['status', 'statusCode', 'code']
+) => async (ctx, next) => {
+  try {
+    await next();
+    if (ctx.status === 404) {
+      throw Boom.notFound('API Not Found');
+    }
+    const response = ctx.body;
+    ctx.body = {
+      success: true,
+      data: response
+    };
+  } catch (err) {
+    if (!Boom.isBoom(err)) {
+      let statusCode = 500;
+      for (let property of errStatusCodePropertie) {
+        if (Number.isInteger(err[property])) {
+          statusCode = ~~err[property];
+        }
+      }
+      err = Boom.boomify(err, {
+        statusCode: statusCode
+      });
+    }
+
+    const payload = err.output.payload;
+    ctx.status = payload.statusCode;
+    ctx.body = {
+      success: false,
+      error: payload.error,
+      message:
+        (env !== 'production' ? err.message : payload.message) ||
+        payload.message
+    };
+  }
+};

package.json

@@ -22,5 +22,11 @@
   "bugs": {
     "url": "https://github.com/Maples7/koa-final-response/issues"
   },
-  "homepage": "https://github.com/Maples7/koa-final-response#readme"
+  "homepage": "https://github.com/Maples7/koa-final-response#readme",
+  "engines": {
+    "node": ">=7.6.0"
+  },
+  "dependencies": {
+    "boom": "^7.2.0"
+  }
 }