AGDIGGER-60 - allow to stream artifacts

Author: wtrocki

src/main/java/com/redhat/digkins/DiggerClient.java

@@ -1,6 +1,7 @@
 package com.redhat.digkins;
 
 import com.offbytwo.jenkins.JenkinsServer;
+import com.redhat.digkins.services.ArtifactsService;
 import com.redhat.digkins.services.CreateJobService;
 import com.redhat.digkins.services.TriggerBuildService;
 import com.redhat.digkins.util.DiggerClientException;
@@ -9,6 +10,7 @@
 import org.slf4j.LoggerFactory;
 
 import java.io.IOException;
+import java.io.InputStream;
 import java.net.URI;
 import java.net.URISyntaxException;
 
@@ -108,4 +110,16 @@ public long build(String jobName) throws DiggerClientException {
     return this.build(jobName, DEFAULT_BUILD_TIMEOUT);
   }
 
+  /**
+   * Fetch artifacts urls for specific job and build number
+   *
+   * @param jobName name of the job
+   * @param buildNumber job build number
+   * @param artifactName - name of the artifact to fetch - can be regexp
+   * @return InputStream with file contents
+   */
+  public InputStream fetchArtifact(String jobName, long buildNumber, String artifactName) {
+    ArtifactsService artifactsService = new ArtifactsService(jenkins);
+    return artifactsService.fetchArtifact(jobName,buildNumber,artifactName);
+  }
 }

src/main/java/com/redhat/digkins/services/ArtifactsService.java

@@ -0,0 +1,63 @@
+package com.redhat.digkins.services;
+
+import com.offbytwo.jenkins.JenkinsServer;
+import com.offbytwo.jenkins.model.Artifact;
+import com.offbytwo.jenkins.model.Build;
+import com.offbytwo.jenkins.model.BuildWithDetails;
+import org.jtwig.JtwigModel;
+import org.jtwig.JtwigTemplate;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.util.List;
+
+/**
+ * Service used to retrieve artifacts
+ */
+public class ArtifactsService {
+
+  private JenkinsServer jenkins;
+
+  /**
+   * @param jenkins jenkins api instance
+   */
+  public ArtifactsService(JenkinsServer jenkins) {
+    this.jenkins = jenkins;
+  }
+
+  private static final Logger LOG = LoggerFactory.getLogger(ArtifactsService.class);
+
+  /**
+   * Fetch artifacts urls for specific job and build number
+   *
+   * @param jobName name of the job
+   * @param buildNumber job build number
+   * @param artifactName - name of the artifact to fetch - can be regexp
+   * @return InputStream with file contents
+   */
+  public InputStream fetchArtifact(String jobName, long buildNumber, String artifactName) {
+    try {
+      Build build = jenkins.getBuild(jobName, buildNumber);
+      if (build instanceof BuildWithDetails) {
+        BuildWithDetails buildWithDetails = ((BuildWithDetails) build);
+        List<Artifact> artifacts = buildWithDetails.getArtifacts();
+        for (Artifact artifact : artifacts) {
+          if (artifact.getFileName().matches(artifactName)) {
+            LOG.debug("Streaming artifact {0}", artifactName);
+            return buildWithDetails.downloadArtifact(artifact);
+          }
+        }
+
+      }
+    } catch (Exception e) {
+      LOG.error("Problem when fetching artifacts for {0} {1} {2}", jobName, buildNumber, artifactName, e);
+    }
+    LOG.debug("Cannot find build for ", jobName, buildNumber, artifactName);
+    return null;
+  }
+
+}