refactor: use async function and support koa@2

.travis.yml

@@ -1,9 +1,8 @@
 sudo: false
 language: node_js
 node_js:
- - '4'
- - '6'
  - '8'
+ - '9'
 install:
  - npm i npminstall && npminstall
 script:

README.md

@@ -32,7 +32,7 @@ $ npm install koa-override --save
 ## Usage
 
 ```js
-const bodyParser = require('koa-body-parser')
+const bodyParser = require('koa-bodyparser')
 const override = require('koa-override')
 
 app.use(bodyParser())

appveyor.yml

@@ -11,6 +11,6 @@ install:
 test_script:
  - node --version
  - npm --version
- - npm run ci
+ - npm run test
 
 build: off

index.js

@@ -8,18 +8,18 @@ module.exports = options => {
  options = options || {};
  options.allowedMethods = options.allowedMethods || [ 'POST' ];
 
- return function* overrideMethod(next) {
- const orginalMethod = this.request.method;
- if (options.allowedMethods.indexOf(orginalMethod) === -1) return yield next;
+ return function overrideMethod(ctx, next) {
+ const orginalMethod = ctx.request.method;
+ if (options.allowedMethods.indexOf(orginalMethod) === -1) return next();
 
  let method;
  // body support
- const body = this.request.body;
+ const body = ctx.request.body;
  if (body && body._method) {
  method = body._method.toUpperCase();
  } else {
  // header support
- const header = this.get('x-http-method-override');
+ const header = ctx.get('x-http-method-override');
  if (header) {
  method = header.toUpperCase();
  }
@@ -30,11 +30,11 @@ module.exports = options => {
  // if you want to support other methods,
  // just create your own utility!
  if (methods.indexOf(method) === -1) {
- this.throw(400, `invalid overriden method: "${method}"`);
+ ctx.throw(400, `invalid overriden method: "${method}"`);
  }
- this.request.method = method;
+ ctx.request.method = method;
  }
 
- yield next;
+ return next();
  };
 };

package.json

@@ -7,7 +7,7 @@
  "index.js"
  ],
  "scripts": {
- "test": "mocha -r co-mocha -R spec -t 5000 test/*.test.js",
+ "test": "mocha -R spec -t 5000 test/*.test.js",
  "cov": "nyc -r text-summary -r lcov npm test",
  "lint": "eslint test *.js",
  "ci": "npm run lint && npm run cov",
@@ -18,12 +18,11 @@
  },
  "devDependencies": {
  "autod": "*",
- "co-mocha": "^1.2.0",
  "egg-ci": "^1.7.0",
  "eslint": "^4.0.0",
  "eslint-config-egg": "^4.2.1",
- "koa": "1",
- "koa-body-parser": "1",
+ "koa": "2",
+ "koa-bodyparser": "4",
  "mocha": "*",
  "nyc": "^11.0.2",
  "supertest": "3"

test/index.test.js

@@ -3,17 +3,17 @@
 const assert = require('assert');
 const request = require('supertest');
 const koa = require('koa');
-const bodyParser = require('koa-body-parser');
+const bodyParser = require('koa-bodyparser');
 const override = require('../');
 
 describe('override method middleware', () => {
  it('should override with x-http-method-override header', () => {
- const app = koa();
+ const app = new koa();
  app.use(override());
- app.use(function* () {
- this.body = {
- method: this.method,
- url: this.url,
+ app.use(ctx => {
+ ctx.body = {
+ method: ctx.method,
+ url: ctx.url,
  };
  });
 
@@ -28,14 +28,14 @@ describe('override method middleware', () => {
  });
 
  it('should override with body._method', () => {
- const app = koa();
+ const app = new koa();
  app.use(bodyParser());
  app.use(override());
- app.use(function* () {
- this.body = {
- method: this.method,
- url: this.url,
- body: this.request.body,
+ app.use(ctx => {
+ ctx.body = {
+ method: ctx.method,
+ url: ctx.url,
+ body: ctx.request.body,
  };
  });
 
@@ -57,7 +57,7 @@ describe('override method middleware', () => {
  });
 
  it('should throw invalid overriden method error', () => {
- const app = koa();
+ const app = new koa();
  app.on('error', function(err) {
  assert.equal(err.message, 'invalid overriden method: "SAVE"');
  });
@@ -71,14 +71,14 @@ describe('override method middleware', () => {
  });
 
  it('should dont override method on body._method and header both not match', () => {
- const app = koa();
+ const app = new koa();
  app.use(bodyParser());
  app.use(override());
- app.use(function* () {
- this.body = {
- method: this.method,
- url: this.url,
- body: this.request.body,
+ app.use(ctx => {
+ ctx.body = {
+ method: ctx.method,
+ url: ctx.url,
+ body: ctx.request.body,
  };
  });
 
@@ -95,12 +95,12 @@ describe('override method middleware', () => {
  });
 
  it('should not allow override with x-http-method-override header on GET request', () => {
- const app = koa();
+ const app = new koa();
  app.use(override());
- app.use(function* () {
- this.body = {
- method: this.method,
- url: this.url,
+ app.use(ctx => {
+ ctx.body = {
+ method: ctx.method,
+ url: ctx.url,
  };
  });
 
@@ -115,12 +115,12 @@ describe('override method middleware', () => {
  });
 
  it('should not allow override with x-http-method-override header on PUT request', () => {
- const app = koa();
+ const app = new koa();
  app.use(override());
- app.use(function* () {
- this.body = {
- method: this.method,
- url: this.url,
+ app.use(ctx => {
+ ctx.body = {
+ method: ctx.method,
+ url: ctx.url,
  };
  });
 
@@ -135,14 +135,14 @@ describe('override method middleware', () => {
  });
 
  it('should custom options.allowedMethods work', () => {
- const app = koa();
+ const app = new koa();
  app.use(override({
  allowedMethods: [ 'POST', 'PUT' ],
  }));
- app.use(function* () {
- this.body = {
- method: this.method,
- url: this.url,
+ app.use(ctx => {
+ ctx.body = {
+ method: ctx.method,
+ url: ctx.url,
  };
  });