refactored compress()

Author: tj

lib/middleware/compress.js

@@ -72,12 +72,30 @@ module.exports = function compress(options) {
       , stream
       , method;
 
+    // proxy
+
+    res.write = function(chunk){
+      if (!res.headerSent) this._implicitHeader();
+      return stream.write(chunk);
+    };
+
+    res.end = function(chunk){
+      if (chunk) stream.write(chunk);
+      return stream.end();
+    };
+
+    function revert() {
+      stream = res;
+      res.write = write;
+      res.end = end;
+    }
+
     res.on('header', function(){
       // default request filter
-      if (!filter(req, res)) return;
+      if (!filter(req, res)) return revert();
 
       // SHOULD use identity
-      if (!accept) return;
+      if (!accept) return revert();
 
       // default to gzip
       if ('*' == accept.trim()) method = 'gzip';
@@ -93,7 +111,7 @@ module.exports = function compress(options) {
       }
 
       // compression method
-      if (!method) return;
+      if (!method) return revert();
 
       // compression stream
       stream = exports.methods[method](options);
@@ -102,17 +120,6 @@ module.exports = function compress(options) {
       res.setHeader('Content-Encoding', method);
       res.setHeader('Vary', 'Accept-Encoding');
 
-      // proxy
-
-      res.write = function(chunk){
-        return stream.write(chunk);
-      };
-
-      res.end = function(chunk){
-        if (chunk) stream.write(chunk);
-        return stream.end();
-      };
-
       // compression
 
       stream.on('data', function(chunk){