Add Streaming Rest Call - returning the response input stream without…

… the wrappers

api/src/main/java/org/jfrog/artifactory/client/Artifactory.java

@@ -1,6 +1,8 @@
 package org.jfrog.artifactory.client;
 
 import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import org.apache.http.HttpResponse;
+import org.apache.http.client.methods.HttpUriRequest;
 
 import java.io.IOException;
 import java.io.InputStream;
@@ -42,10 +44,14 @@ public interface Artifactory extends ApiInterface, AutoCloseable {
 
     ArtifactoryResponse restCall(ArtifactoryRequest artifactoryRequest) throws IOException;
 
+    ArtifactoryStreamingResponse streamingRestCall(ArtifactoryRequest artifactoryRequest) throws IOException;
+
     InputStream getInputStream(String path) throws IOException;
 
     InputStream getInputStreamWithHeaders(String path, Map<String, String> headers) throws IOException;
 
+    HttpResponse execute(HttpUriRequest request) throws IOException;
+
     default public <T> T get(String path, Class<? extends T> object, Class<T> interfaceObject) throws IOException {
         return null;
     }

api/src/main/java/org/jfrog/artifactory/client/ArtifactoryResponse.java

@@ -1,23 +1,15 @@
 package org.jfrog.artifactory.client;
 
-import org.apache.http.Header;
-import org.apache.http.StatusLine;
 
 import java.io.IOException;
 
 /**
  * ArtifactoryResponse object returned from {@link Artifactory#restCall(ArtifactoryRequest)}.
  * acts as a wrapper for {@link org.apache.http.HttpResponse} but removes the need to handle response stream.
  */
-public interface ArtifactoryResponse {
-
-    Header[] getAllHeaders();
-
-    StatusLine getStatusLine();
-
+public interface ArtifactoryResponse extends BaseArtifactoryResponse {
     String getRawBody();
 
     <T> T parseBody(Class<T> toType) throws IOException;
 
-    boolean isSuccessResponse();
-}
+}

api/src/main/java/org/jfrog/artifactory/client/ArtifactoryStreamingResponse.java

@@ -0,0 +1,15 @@
+package org.jfrog.artifactory.client;
+
+import java.io.IOException;
+import java.io.InputStream;
+
+
+/**
+ * ArtifactoryStreamingResponse object returned from {@link Artifactory#streamingRestCall(ArtifactoryRequest)}.
+ * acts as a wrapper for {@link org.apache.http.HttpResponse}.
+ */
+public interface ArtifactoryStreamingResponse extends BaseArtifactoryResponse {
+
+    InputStream getInputStream() throws IOException;
+
+}

api/src/main/java/org/jfrog/artifactory/client/BaseArtifactoryResponse.java

@@ -0,0 +1,13 @@
+package org.jfrog.artifactory.client;
+import org.apache.http.Header;
+import org.apache.http.StatusLine;
+
+public interface BaseArtifactoryResponse {
+
+    Header[] getAllHeaders();
+
+    StatusLine getStatusLine();
+
+    boolean isSuccessResponse();
+
+}

services/src/main/groovy/org/jfrog/artifactory/client/impl/AbstractArtifactoryResponseImpl.java

@@ -0,0 +1,27 @@
+package org.jfrog.artifactory.client.impl;
+
+import org.apache.http.Header;
+import org.apache.http.HttpResponse;
+import org.apache.http.StatusLine;
+
+public abstract class AbstractArtifactoryResponseImpl {
+
+    private final HttpResponse httpResponse;
+
+    public AbstractArtifactoryResponseImpl(HttpResponse httpResponse) {
+        this.httpResponse = httpResponse;
+    }
+
+    public HttpResponse getHttpResponse() {
+        return httpResponse;
+    }
+
+    public Header[] getAllHeaders() {
+        return this.httpResponse.getAllHeaders();
+    }
+
+    public StatusLine getStatusLine() {
+        return this.httpResponse.getStatusLine();
+    }
+
+}

services/src/main/groovy/org/jfrog/artifactory/client/impl/ArtifactoryImpl.java

@@ -132,6 +132,24 @@ public ArtifactorySystem system() {
      */
     @Override
     public ArtifactoryResponse restCall(ArtifactoryRequest artifactoryRequest) throws IOException {
+        HttpResponse httpResponse = handleArtifactoryRequest(artifactoryRequest);
+        return new ArtifactoryResponseImpl(httpResponse);
+    }
+
+    /**
+     * Create a REST call to artifactory with a generic request
+     *
+     * @param artifactoryRequest that should be sent to artifactory
+     * @return {@link ArtifactoryStreamingResponse} Artifactory response in accordance with the request,
+     * which includes a reference to the inputStream.
+     */
+    @Override
+    public ArtifactoryStreamingResponse streamingRestCall(ArtifactoryRequest artifactoryRequest) throws IOException {
+        HttpResponse httpResponse = handleArtifactoryRequest(artifactoryRequest);
+        return new ArtifactoryStreamingResponseImpl(httpResponse);
+    }
+
+    private HttpResponse handleArtifactoryRequest(ArtifactoryRequest artifactoryRequest) throws IOException {
         HttpRequestBase httpRequest;
 
         String requestPath = "/" + artifactoryRequest.getApiUrl();
@@ -194,7 +212,7 @@ public ArtifactoryResponse restCall(ArtifactoryRequest artifactoryRequest) throw
         }
 
         HttpResponse httpResponse = execute(httpRequest);
-        return new ArtifactoryResponseImpl(httpResponse);
+        return httpResponse;
     }
 
     private void setEntity(HttpEntityEnclosingRequestBase httpRequest, Object body, ContentType contentType) throws JsonProcessingException {
@@ -369,6 +387,7 @@ public String delete(String path) throws IOException {
         return Util.responseToString(httpResponse);
     }
 
+    @Override
     public HttpResponse execute(HttpUriRequest request) throws IOException {
         HttpClientContext clientContext = HttpClientContext.create();
         if (clientContext.getAttribute(PreemptiveAuthInterceptor.ORIGINAL_HOST_CONTEXT_PARAM) == null) {

services/src/main/groovy/org/jfrog/artifactory/client/impl/ArtifactoryResponseImpl.java

@@ -1,25 +1,23 @@
 package org.jfrog.artifactory.client.impl;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
-import org.apache.http.Header;
 import org.apache.http.HttpEntity;
 import org.apache.http.HttpResponse;
-import org.apache.http.StatusLine;
 import org.apache.http.util.EntityUtils;
 import org.jfrog.artifactory.client.ArtifactoryResponse;
 import org.jfrog.artifactory.client.impl.util.Util;
 
 import java.io.IOException;
 
-public class ArtifactoryResponseImpl implements ArtifactoryResponse {
+public class ArtifactoryResponseImpl extends AbstractArtifactoryResponseImpl implements ArtifactoryResponse {
 
     private static final ObjectMapper objectMapper = new ObjectMapper();
 
-    private HttpResponse httpResponse;
+
     private String rawBody;
 
     ArtifactoryResponseImpl(HttpResponse httpResponse) throws IOException {
-        this.httpResponse = httpResponse;
+        super(httpResponse);
 
         HttpEntity entity = httpResponse.getEntity();
 
@@ -34,15 +32,7 @@ public class ArtifactoryResponseImpl implements ArtifactoryResponse {
         }
     }
 
-    @Override
-    public Header[] getAllHeaders() {
-        return this.httpResponse.getAllHeaders();
-    }
 
-    @Override
-    public StatusLine getStatusLine() {
-        return this.httpResponse.getStatusLine();
-    }
 
     @Override
     public String getRawBody() {
@@ -62,7 +52,7 @@ public <T> T parseBody(Class<T> toType) throws IOException {
     @Override
     public boolean isSuccessResponse() {
         int status = getStatusLine().getStatusCode();
-
         return status >= 200 && status < 300;
     }
+
 }

services/src/main/groovy/org/jfrog/artifactory/client/impl/ArtifactoryStreamingResponseImpl.java

@@ -0,0 +1,26 @@
+package org.jfrog.artifactory.client.impl;
+
+import org.apache.http.HttpResponse;
+import org.apache.http.HttpStatus;
+import org.jfrog.artifactory.client.ArtifactoryStreamingResponse;
+
+import java.io.IOException;
+import java.io.InputStream;
+
+public class ArtifactoryStreamingResponseImpl extends AbstractArtifactoryResponseImpl implements ArtifactoryStreamingResponse {
+
+    public ArtifactoryStreamingResponseImpl(HttpResponse httpResponse) {
+        super(httpResponse);
+    }
+
+    @Override
+    public InputStream getInputStream() throws IOException {
+        return getHttpResponse().getEntity().getContent();
+    }
+
+    public boolean isSuccessResponse() {
+        int status = getStatusLine().getStatusCode();
+        return (status == HttpStatus.SC_OK ||
+                status == HttpStatus.SC_PARTIAL_CONTENT);
+    }
+}

services/src/test/java/org/jfrog/artifactory/client/StreamingRestCallTest.java

@@ -0,0 +1,58 @@
+package org.jfrog.artifactory.client;
+
+import org.apache.commons.io.IOUtils;
+import org.jfrog.artifactory.client.impl.ArtifactoryRequestImpl;
+import org.testng.annotations.Test;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.charset.StandardCharsets;
+import java.util.HashMap;
+import java.util.Map;
+
+import static org.testng.Assert.*;
+
+public class StreamingRestCallTest extends ArtifactoryTestsBase {
+
+    @Test
+    public void testDownloadWithHeadersByStreamingRestCall() throws IOException {
+        InputStream inputStream = this.getClass().getResourceAsStream("/sample.txt");
+        assertNotNull(inputStream);
+        artifactory.repository(localRepository.getKey()).upload(PATH, inputStream).withProperty("color", "blue")
+                .withProperty("color", "red").doUpload();
+
+        Map<String, String> headers = new HashMap<>();
+        headers.put("Range", "bytes=0-10");
+        ArtifactoryRequest request = new ArtifactoryRequestImpl()
+                .apiUrl(localRepository.getKey() + "/" + PATH)
+                .method(ArtifactoryRequest.Method.GET)
+                .setHeaders(headers)
+                .requestType(ArtifactoryRequest.ContentType.JSON);
+
+        ArtifactoryStreamingResponse response = artifactory.streamingRestCall(request);
+        assertTrue(response.isSuccessResponse());
+
+        inputStream = response.getInputStream();
+        String actual = textFrom(inputStream);
+        assertEquals(actual, textFrom(this.getClass().getResourceAsStream("/sample.txt")).substring(0, 11));
+    }
+
+    @Test
+    public void testErrorStreamingRestCall() throws IOException {
+        ArtifactoryRequest request = new ArtifactoryRequestImpl()
+                .apiUrl(localRepository.getKey() + "/" + PATH + "shouldNotExist")
+                .method(ArtifactoryRequest.Method.GET)
+                .requestType(ArtifactoryRequest.ContentType.JSON);
+        ArtifactoryStreamingResponse response = artifactory.streamingRestCall(request);
+        assertFalse(response.isSuccessResponse());
+        assertEquals(response.getStatusLine().getStatusCode(), 404);
+        assertEquals(response.getStatusLine().getReasonPhrase(), "Not Found");
+        String raw = IOUtils.toString(response.getInputStream(), StandardCharsets.UTF_8);
+        assertEquals(raw, "{\n" +
+                "  \"errors\" : [ {\n" +
+                "    \"status\" : 404,\n" +
+                "    \"message\" : \"File not found.\"\n" +
+                "  } ]\n" +
+                "}");
+    }
+}