nuget-server/src/jetbrains/buildServer/nuget/server/tool/impl/AvailableOnDistNugetOrg.java

package jetbrains.buildServer.nuget.server.tool.impl;

import com.google.gson.Gson;
import com.intellij.openapi.diagnostic.Logger;
import jetbrains.buildServer.http.HttpUtil;
import jetbrains.buildServer.nuget.common.FeedConstants;
import jetbrains.buildServer.tools.available.AvailableToolsFetcher;
import jetbrains.buildServer.tools.available.DownloadableToolVersion;
import jetbrains.buildServer.tools.available.FetchAvailableToolsResult;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpStatus;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.http.HttpHeaders;
import org.jetbrains.annotations.NotNull;
import org.springframework.http.MediaType;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;

import static jetbrains.buildServer.nuget.common.FeedConstants.EXE_EXTENSION;

/**
 * @author Evgeniy.Koshkin
 */
public class AvailableOnDistNugetOrg implements AvailableToolsFetcher {

  private static final Logger LOG = Logger.getInstance(AvailableOnDistNugetOrg.class.getName());

  private static final int CONNECTION_TIMEOUT_SECONDS = 60;
  private final static String DIST_NUGET_ORG_INDEX_JSON_URL = "http://dist.nuget.org/index.json";
  private final static String DIST_NUGET_ORG_URL = "http://dist.nuget.org";
  private static final String NUGET_COMMANDLINE_ARTIFACT_NAME = "win-x86-commandline";

  @NotNull
  public FetchAvailableToolsResult fetchAvailable() {
    HttpClient client = HttpUtil.createHttpClient(CONNECTION_TIMEOUT_SECONDS);
    final GetMethod post = new GetMethod(DIST_NUGET_ORG_INDEX_JSON_URL);
    post.addRequestHeader(HttpHeaders.ACCEPT, MediaType.APPLICATION_JSON.toString());
    try {
      int status = client.executeMethod(post);
      if (status != HttpStatus.SC_OK)
        return FetchAvailableToolsResult.createError(String.format("Recieved http status %d when fetching available nuget.exe versions from %s", status, DIST_NUGET_ORG_URL));

      final String respText = post.getResponseBodyAsString();
      final Gson gson = new Gson();
      final AvailableArtifacts artifacts = gson.fromJson(respText, AvailableArtifacts.class);
      final Collection<DownloadableToolVersion> nugets = new ArrayList<DownloadableToolVersion>();
      for (final Artifact artifact : artifacts.getArtifacts()) {
        if(!artifact.getName().equalsIgnoreCase(NUGET_COMMANDLINE_ARTIFACT_NAME)) continue;
        for (final Version commandlineVersion : artifact.getVersions()) {
          nugets.add(new DownloadableNuGetTool(
                  commandlineVersion.getVersion(),
                  commandlineVersion.getUrl(),
                  FeedConstants.NUGET_COMMANDLINE + "." + commandlineVersion.getVersion() + EXE_EXTENSION));
        }
      }
      return FetchAvailableToolsResult.createSuccessful(nugets);
    } catch (IOException e) {
      LOG.debug(e);
      return FetchAvailableToolsResult.createError("Failed to fetch available nuget.exe versions from " + DIST_NUGET_ORG_INDEX_JSON_URL, e);
    }
  }

  private class AvailableArtifacts{
    private Collection<Artifact> artifacts;

    public Collection<Artifact> getArtifacts() {
      return artifacts;
    }
  }

  private class Artifact {
    private String name;
    private String displayName;
    private Collection<Version> versions;

    public String getName() {
      return name;
    }

    public String getDisplayName() {
      return displayName;
    }

    public Collection<Version> getVersions() {
      return versions;
    }
  }

  private class Version {
    private String displayName;
    private String version;
    private String url;
    private String releasedate;

    public String getDisplayName() {
      return displayName;
    }

    public String getVersion() {
      return version;
    }

    public String getUrl() {
      return url;
    }

    public String getReleaseDate() {
      return releasedate;
    }
  }
}