main: fix return functions

hex7c0 · Sep 9, 2015 · c5f9ccd · c5f9ccd
1 parent d0878ce
commit c5f9ccd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/index.js b/index.js
@@ -55,12 +55,11 @@ function wrapper(my) {
 
   var setHeader = require('setheaders').setWritableHeader;
 
-  function headers(req, res, next) {
+  return function headers(req, res, next) {
 
-    setHeader(res, header, output);
+    setHeader(res, header, output); // writable header, check if empty
     return next();
-  }
-  return headers;
+  };
 }
 
 /**

diff --git a/index.min.js b/index.min.js
@@ -1,9 +1,6 @@
 "use strict";
 
 function wrapper(my) {
-    function headers(req, res, next) {
-        return setHeader(res, header, output), next();
-    }
     var output, header = my.header;
     if (my.custom) output = String(my.custom); else {
         output = my.signature;
@@ -18,7 +15,9 @@ function wrapper(my) {
         return output;
     };
     var setHeader = require("setheaders").setWritableHeader;
-    return headers;
+    return function(req, res, next) {
+        return setHeader(res, header, output), next();
+    };
 }
 
 function signature(opt) {