1 files changed, 123 insertions, 0 deletions
diff --git a/src/main/java/net/sf/antcontrib/net/httpclient/GetCookieTask.java b/src/main/java/net/sf/antcontrib/net/httpclient/GetCookieTask.java
new file mode 100644
index 0000000..9287145
--- /dev/null
+++ b/src/main/java/net/sf/antcontrib/net/httpclient/GetCookieTask.java
@@ -0,0 +1,123 @@
+/*
+ * Copyright (c) 2001-2006 Ant-Contrib project.  All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package net.sf.antcontrib.net.httpclient;
+
+import org.apache.commons.httpclient.Cookie;
+import org.apache.commons.httpclient.HttpState;
+import org.apache.commons.httpclient.cookie.CookiePolicy;
+import org.apache.commons.httpclient.cookie.CookieSpec;
+import org.apache.tools.ant.BuildException;
+import org.apache.tools.ant.taskdefs.Property;
+
+public class GetCookieTask
+	extends AbstractHttpStateTypeTask {
+
+	private String property;
+    private String prefix;
+	private String cookiePolicy = CookiePolicy.DEFAULT;
+	
+	private String realm = null;
+    private int port = 80;
+    private String path = null;
+    private boolean secure = false;
+    private String name = null;
+    
+	public void setName(String name) {
+		this.name = name;
+	}
+
+	public void setCookiePolicy(String cookiePolicy) {
+		this.cookiePolicy = cookiePolicy;
+	}
+
+	public void setPath(String path) {
+		this.path = path;
+	}
+
+	public void setPort(int port) {
+		this.port = port;
+	}
+
+	public void setRealm(String realm) {
+		this.realm = realm;
+	}
+
+	public void setSecure(boolean secure) {
+		this.secure = secure;
+	}
+
+	public void setProperty(String property) {
+		this.property = property;
+	}
+
+	private Cookie findCookie(Cookie cookies[], String name) {
+		for (int i=0;i<cookies.length;i++) {
+			if (cookies[i].getName().equals(name)) {
+				return cookies[i];
+			}
+		}
+		return null;
+	}
+	protected void execute(HttpStateType stateType) throws BuildException {
+		
+		if (realm == null || path == null) {
+			throw new BuildException("'realm' and 'path' attributes are required");
+		}
+		
+		HttpState state = stateType.getState();
+		CookieSpec spec = CookiePolicy.getCookieSpec(cookiePolicy);
+		Cookie cookies[] = state.getCookies();
+		Cookie matches[] = spec.match(realm, port, path, secure, cookies);
+
+		if (name != null) {
+			Cookie c = findCookie(matches, name);
+			if (c != null) {
+				matches = new Cookie[] { c };
+			}
+			else {
+				matches = new Cookie[0];
+			}
+		}
+		
+		
+		if (property != null) {
+			if (matches != null && matches.length > 0) {
+				Property p = (Property)getProject().createTask("property");
+				p.setName(property);
+				p.setValue(matches[0].getValue());
+				p.perform();					
+			}
+		}
+		else if (prefix != null) {
+			if (matches != null && matches.length > 0) {
+				for (int i=0;i<matches.length;i++) {
+					String propName =
+						prefix +
+						matches[i].getName();
+					Property p = (Property)getProject().createTask("property");
+					p.setName(propName);
+					p.setValue(matches[i].getValue());
+					p.perform();
+				}
+			}
+		}
+		else {
+			throw new BuildException("Nothing to set");
+		}
+	}
+	
+	
+}