1 files changed, 75 insertions, 0 deletions
diff --git a/src/main/java/io/trygvis/esper/testing/nexus/NexusClient.java b/src/main/java/io/trygvis/esper/testing/nexus/NexusClient.java
new file mode 100755
index 0000000..6477a80
--- /dev/null
+++ b/src/main/java/io/trygvis/esper/testing/nexus/NexusClient.java
@@ -0,0 +1,75 @@
+package io.trygvis.esper.testing.nexus;
+
+import fj.data.*;
+import static fj.data.Option.*;
+import static io.trygvis.esper.testing.nexus.NexusParser.*;
+import org.apache.commons.io.*;
+import org.apache.commons.lang.*;
+import static org.codehaus.httpcache4j.HTTPMethod.*;
+import org.codehaus.httpcache4j.*;
+import org.codehaus.httpcache4j.cache.*;
+import org.codehaus.httpcache4j.util.*;
+
+import java.io.*;
+import java.net.*;
+import javax.xml.stream.*;
+
+public class NexusClient {
+    private final HTTPCache http;
+    private final String nexusUrl;
+
+    public NexusClient(HTTPCache http, String nexusUrl) {
+        this.http = http;
+        this.nexusUrl = nexusUrl;
+    }
+
+    public ArtifactSearchResult fetchIndex(String groupId, Option<String> repositoryId) throws IOException {
+        ArtifactSearchResult aggregate = fetchIndexPage(groupId, repositoryId, Option.<Integer>none());
+        ArtifactSearchResult result = aggregate;
+
+        while(result.artifacts.size() > 0) {
+            result = fetchIndexPage(groupId, repositoryId, some(aggregate.artifacts.size()));
+            aggregate = aggregate.append(result);
+        }
+
+        return aggregate;
+    }
+
+    public ArtifactSearchResult fetchIndexPage(String groupId, Option<String> repositoryId, Option<Integer> from) throws IOException {
+        URIBuilder uriBuilder = URIBuilder.fromURI(URI.create(nexusUrl)).
+                addRawPath("/service/local/lucene/search").
+                addParameter("g", groupId + ".*");
+
+        if (repositoryId.isSome()) {
+            uriBuilder = uriBuilder.addParameter("repositoryId", repositoryId.some());
+        }
+
+        if (from.isSome()) {
+            uriBuilder = uriBuilder.addParameter("from", from.some().toString());
+        }
+
+        HTTPResponse response = http.execute(new HTTPRequest(uriBuilder.toURI(), GET));
+
+        int statusCode = response.getStatus().getCode();
+        if (statusCode != 200) {
+            throw new IOException("Got " + statusCode + " from Nexus search, expected 200.");
+        }
+
+        MIMEType mimeType = MIMEType.valueOf(StringUtils.trimToEmpty(response.getHeaders().getFirstHeaderValue("Content-Type")));
+        if (!mimeType.getPrimaryType().equals("application") || !mimeType.getSubType().equals("xml")) {
+            throw new IOException("Unexpected mime type: " + mimeType);
+        }
+
+        byte[] bytes = IOUtils.toByteArray(response.getPayload().getInputStream());
+
+        try {
+            ArtifactSearchResult result = parseDocument(new ByteArrayInputStream(bytes));
+            System.out.println("Parsed out " + result.artifacts.size() + " artifacts.");
+            return result;
+        } catch (XMLStreamException e) {
+            System.out.println("Unable to parse XML.");
+            System.out.println(new String(bytes));
+            throw new RuntimeException("Unable to parse XML.", e);
+        }
+    }
+}