Adding automatic redirect following.

request · May 19, 2010 · e66e90d · e66e90d
1 parent e7e37ad
commit e66e90d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/lib/main.js b/lib/main.js
@@ -11,7 +11,7 @@ function request (options, callback) {
       options.uri = url.parse(options.uri);
     }
   }
-
+  options.followRedirect = options.followRedirect ? options.followRedirect : true;
   options.method = options.method ? options.method : 'GET';
   options.headers = options.headers ? options.headers :  {};
   if (!options.headers.host) {
@@ -46,6 +46,14 @@ function request (options, callback) {
       buffer += chunk;
     })
     response.addListener("end", function () {
+      if (response.statusCode > 299 && response.statusCode < 400 && options.followRedirect && response.headers.location) {
+        options.uri = response.headers.location;
+        delete options.client; 
+        if (options.headers) {
+          delete options.headers.host;
+        }
+        request(options, callback);
+      }
       options.client.removeListener("error", clientErrorHandler);
       callback(null, response, buffer);
     })

diff --git a/tests/couch.js b/tests/couch.js
@@ -26,4 +26,8 @@ request({uri:'http://mikeal.couchone.com/testjs', method:'POST', body:'{"_id":"'
           if (!response.statusCode == 201) {
             throw new Error("Wrong status code "+response.statusCode);
           }
-        });
+        });
+
+request({uri:'http://gmail.com'}, function (error, response, body) {
+  sys.puts(error)
+})