koa-compress 1.0.9 → 2.0.0

package/README.md

@@ -6,9 +6,9 @@ Compress middleware for Koa
 
 ```js
 var compress = require('koa-compress')
-var koa = require('koa')
+var Koa = require('koa')
 
-var app = koa()
+var app = new Koa()
 app.use(compress({
   filter: function (content_type) {
   	return /text/i.test(content_type)
@@ -39,10 +39,8 @@ You can always disable compression by setting `this.compress = false`.
 This bypasses the filter check.
 
 ```js
-app.use(function (next) {
-  return function *() {
-    this.compress = true
-    this.body = fs.createReadStream(file)
-  }
+app.use((ctx, next) => {
+  ctx.compress = true
+  ctx.body = fs.createReadStream(file)
 })
 ```

package/index.js

@@ -1,3 +1,5 @@
+'use strict';
+
 /**
  * Module dependencies.
  */
@@ -26,7 +28,7 @@ var encodingMethods = {
  * @api public
  */
 
-module.exports = function (options) {
+module.exports = (options) => {
   options = options || {}
 
   var filter = options.filter || compressible
@@ -36,42 +38,42 @@ module.exports = function (options) {
     : typeof options.threshold === 'string' ? bytes(options.threshold)
     : 1024
 
-  return function* compress(next) {
-    this.vary('Accept-Encoding')
-
-    yield* next
+  return function compress(ctx, next) {
+    ctx.vary('Accept-Encoding')
 
-    var body = this.body
-    if (!body) return
-    if (this.compress === false) return
-    if (this.request.method === 'HEAD') return
-    if (status.empty[this.response.status]) return
-    if (this.response.get('Content-Encoding')) return
+    return next().then(() => {
+      var body = ctx.body
+      if (!body) return
+      if (ctx.compress === false) return
+      if (ctx.request.method === 'HEAD') return
+      if (status.empty[ctx.response.status]) return
+      if (ctx.response.get('Content-Encoding')) return
 
-    // forced compression or implied
-    if (!(this.compress === true || filter(this.response.type))) return
+      // forced compression or implied
+      if (!(ctx.compress === true || filter(ctx.response.type))) return
 
-    // identity
-    var encoding = this.acceptsEncodings('gzip', 'deflate', 'identity')
-    if (!encoding) this.throw(406, 'supported encodings: gzip, deflate, identity')
-    if (encoding === 'identity') return
+      // identity
+      var encoding = ctx.acceptsEncodings('gzip', 'deflate', 'identity')
+      if (!encoding) ctx.throw(406, 'supported encodings: gzip, deflate, identity')
+      if (encoding === 'identity') return
 
-    // json
-    if (isJSON(body)) body = this.body = JSON.stringify(body)
+      // json
+      if (isJSON(body)) body = ctx.body = JSON.stringify(body)
 
-    // threshold
-    if (threshold && this.response.length < threshold) return
+      // threshold
+      if (threshold && ctx.response.length < threshold) return
 
-    this.set('Content-Encoding', encoding)
-    this.res.removeHeader('Content-Length')
+      ctx.set('Content-Encoding', encoding)
+      ctx.res.removeHeader('Content-Length')
 
-    var stream =
-    this.body = encodingMethods[encoding](options)
+      var stream =
+      ctx.body = encodingMethods[encoding](options)
 
-    if (body instanceof Stream) {
-      body.pipe(stream)
-    } else {
-      stream.end(body)
-    }
-  }
+      if (body instanceof Stream) {
+        body.pipe(stream)
+      } else {
+        stream.end(body)
+      }
+    });
+  };
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koa-compress",
   "description": "Compress middleware for koa",
-  "version": "1.0.9",
+  "version": "2.0.0",
   "author": {
     "name": "Jonathan Ong",
     "email": "me@jongleberry.com",
@@ -18,7 +18,7 @@
   },
   "devDependencies": {
     "istanbul": "^0.4.2",
-    "koa": "^1.0.0",
+    "koa": "^2.0.0-alpha.3",
     "mocha": "^2.4.1",
     "should": "^3.0.0",
     "supertest": "^1.0.0"