support for cookies on 302 redirect

Author: marekzebrowski

src/main/java/com/ning/http/client/RequestBuilder.java

@@ -160,4 +160,9 @@ public RequestBuilder setVirtualHost(String virtualHost) {
     public RequestBuilder setFollowRedirects(boolean followRedirects) {
         return super.setFollowRedirects(followRedirects);
     }
+
+    @Override
+    public RequestBuilder addOrReplaceCookie(Cookie c) {
+	return super.addOrReplaceCookie(c);		
+    }
 }

src/main/java/com/ning/http/client/RequestBuilderBase.java

@@ -563,4 +563,23 @@ private boolean allowBody(String method) {
             return true;
         }
     }
+    
+      public T addOrReplaceCookie(Cookie cookie) {
+    	String cookieKey=cookie.getName();
+    	boolean replace=false;
+    	int index=0;
+    	for(Cookie c : request.cookies) {
+    		if(c.getName().equals(cookieKey)){
+    			replace=true;
+    			break;
+    		};
+    		index++;
+    	}
+    	if(replace) {
+    		((ArrayList<Cookie>)request.cookies).set(index, cookie);
+    	} else {
+            request.cookies.add(cookie);
+    	}
+        return derived.cast(this);
+    }
 }

src/main/java/com/ning/http/client/providers/netty/NettyAsyncHttpProvider.java

@@ -1161,6 +1161,15 @@ public Object call() throws Exception {
                             final String newUrl = uri.toString();
 
                             log.debug("Redirecting to {}", newUrl);
+                            for(String cookieStr : future.getHttpResponse().getHeaders(HttpHeaders.Names.SET_COOKIE)){
+                                Cookie c=AsyncHttpProviderUtils.parseCookie(cookieStr);
+                                nBuilder.addOrReplaceCookie(c);
+                            }
+                            for(String cookieStr : future.getHttpResponse().getHeaders(HttpHeaders.Names.SET_COOKIE2)){
+                                Cookie c=AsyncHttpProviderUtils.parseCookie(cookieStr);
+                                nBuilder.addOrReplaceCookie(c);
+                            }
+    
 
                             AsyncCallable ac = new AsyncCallable(future) {
                                 public Object call() throws Exception {