org.marid.applib.repository.maven.MavenArtifactFinder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of marid-webapp Show documentation
The newest version!
/*-
 * #%L
 * marid-webapp
 * %%
 * Copyright (C) 2012 - 2018 MARID software development group
 * %%
 * This program and the accompanying materials are made available under the terms of the Eclipse Public License v1.0
 * and Eclipse Distribution License v. 1.0 which accompanies this distribution.
 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 * and the Eclipse Distribution License is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 * #L%
 */
package org.marid.applib.repository.maven;

import org.marid.applib.json.MaridJackson;
import org.marid.applib.repository.Artifact;
import org.marid.applib.repository.ArtifactFinder;
import org.springframework.util.StringUtils;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class MavenArtifactFinder implements ArtifactFinder {

  private final URI url;

  MavenArtifactFinder(URI searchUrl) {
    url = searchUrl;
  }

  @Override
  public List find(String groupPattern, String artifactPattern, String classPattern) {
    final LinkedList conditions = new LinkedList<>();

    if (groupPattern != null && !groupPattern.isEmpty()) {
      conditions.add("g:\"" + groupPattern + "\"");
    }

    if (artifactPattern != null && !artifactPattern.isEmpty()) {
      conditions.add("a:\"" + artifactPattern + "\"");
    }

    if (classPattern != null && !classPattern.isEmpty()) {
      conditions.add("c:\"" + classPattern + "\"");
    }

    final var query = "q=" + String.join(" AND ", conditions) + "&wt=json";

    try {
      final var uri = new URI(
          url.getScheme(), url.getRawUserInfo(), url.getHost(), url.getPort(), url.getRawPath(), query, null
      );
      final var connection = uri.toURL().openConnection();
      connection.setAllowUserInteraction(false);
      connection.setConnectTimeout(10_000);
      connection.setUseCaches(false);
      connection.setReadTimeout(1_000);

      final JsonResponse response;
      try (final var stream = connection.getInputStream()) {
        response = MaridJackson.MAPPER.readValue(stream, JsonResponse.class);
      } finally {
        if (connection instanceof HttpURLConnection) {
          ((HttpURLConnection) connection).disconnect();
        }
      }

      return Stream.of(response.response.docs).parallel()
          .filter(d -> d.g != null && d.a != null && d.latestVersion != null)
          .flatMap(d -> Stream.of(d.ec)
              .map(StringUtils::stripFilenameExtension)
              .map(v -> v.startsWith("-") ? v.substring(1) : v)
              .distinct()
              .map(c -> new Artifact(d.g, d.a, d.latestVersion, c, d.p))
          )
          .sorted()
          .collect(Collectors.toUnmodifiableList());
    } catch (IOException | URISyntaxException x) {
      throw new IllegalStateException(x);
    }
  }
}