source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2021 SonarSource SA
   4  * mailto:info AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.alm.client.bitbucketserver;
  21
  22 import com.google.gson.Gson;
  23 import com.google.gson.GsonBuilder;
  24 import com.google.gson.JsonParseException;
  25 import com.google.gson.JsonSyntaxException;
  26 import com.google.gson.annotations.SerializedName;
  27 import java.io.IOException;
  28 import java.net.HttpURLConnection;
  29 import java.util.Optional;
  30 import java.util.function.Function;
  31 import java.util.stream.Collectors;
  32 import java.util.stream.Stream;
  33 import javax.annotation.Nullable;
  34 import okhttp3.HttpUrl;
  35 import okhttp3.MediaType;
  36 import okhttp3.OkHttpClient;
  37 import okhttp3.Request;
  38 import okhttp3.RequestBody;
  39 import okhttp3.Response;
  40 import okhttp3.ResponseBody;
  41 import org.sonar.alm.client.TimeoutConfiguration;
  42 import org.sonar.api.server.ServerSide;
  43 import org.sonar.api.utils.log.Logger;
  44 import org.sonar.api.utils.log.Loggers;
  45 import org.sonarqube.ws.client.OkHttpClientBuilder;
  46
  47 import static java.lang.String.format;
  48 import static java.util.Locale.ENGLISH;
  49 import static org.sonar.api.internal.apachecommons.lang.StringUtils.removeEnd;
  50
  51 @ServerSide
  52 public class BitbucketServerRestClient {
  53
  54   private static final Logger LOG = Loggers.get(BitbucketServerRestClient.class);
  55   private static final String GET = "GET";
  56   protected static final String UNABLE_TO_CONTACT_BITBUCKET_SERVER = "Unable to contact Bitbucket server";
  57
  58   protected final OkHttpClient client;
  59
  60   public BitbucketServerRestClient(TimeoutConfiguration timeoutConfiguration) {
  61     OkHttpClientBuilder okHttpClientBuilder = new OkHttpClientBuilder();
  62     client = okHttpClientBuilder
  63       .setConnectTimeoutMs(timeoutConfiguration.getConnectTimeout())
  64       .setReadTimeoutMs(timeoutConfiguration.getReadTimeout())
  65       .build();
  66   }
  67
  68   public RepositoryList getRepos(String serverUrl, String token, @Nullable String project, @Nullable String repo) {
  69     String projectOrEmpty = Optional.ofNullable(project).orElse("");
  70     String repoOrEmpty = Optional.ofNullable(repo).orElse("");
  71     HttpUrl url = buildUrl(serverUrl, format("/rest/api/1.0/repos?projectname=%s&name=%s", projectOrEmpty, repoOrEmpty));
  72     return doGet(token, url, r -> buildGson().fromJson(r.body().charStream(), RepositoryList.class));
  73   }
  74
  75   public Repository getRepo(String serverUrl, String token, String project, String repoSlug) {
  76     HttpUrl url = buildUrl(serverUrl, format("/rest/api/1.0/projects/%s/repos/%s", project, repoSlug));
  77     return doGet(token, url, r -> buildGson().fromJson(r.body().charStream(), Repository.class));
  78   }
  79
  80   public RepositoryList getRecentRepo(String serverUrl, String token) {
  81     HttpUrl url = buildUrl(serverUrl, "/rest/api/1.0/profile/recent/repos");
  82     return doGet(token, url, r -> buildGson().fromJson(r.body().charStream(), RepositoryList.class));
  83   }
  84
  85   public ProjectList getProjects(String serverUrl, String token) {
  86     HttpUrl url = buildUrl(serverUrl, "/rest/api/1.0/projects");
  87     return doGet(token, url, r -> buildGson().fromJson(r.body().charStream(), ProjectList.class));
  88   }
  89
  90   protected static HttpUrl buildUrl(@Nullable String serverUrl, String relativeUrl) {
  91     if (serverUrl == null || !(serverUrl.toLowerCase(ENGLISH).startsWith("http://") || serverUrl.toLowerCase(ENGLISH).startsWith("https://"))) {
  92       throw new IllegalArgumentException("url must start with http:// or https://");
  93     }
  94     return HttpUrl.parse(removeEnd(serverUrl, "/") + relativeUrl);
  95   }
  96
  97   protected <G> G doGet(String token, HttpUrl url, Function<Response, G> handler) {
  98     Request request = prepareRequestWithBearerToken(token, GET, url, null);
  99     return doCall(request, handler);
 100   }
 101
 102   protected static Request prepareRequestWithBearerToken(String token, String method, HttpUrl url, @Nullable RequestBody body) {
 103     return new Request.Builder()
 104       .method(method, body)
 105       .url(url)
 106       .addHeader("Authorization", "Bearer " + token)
 107       .addHeader("x-atlassian-token", "no-check")
 108       .build();
 109   }
 110
 111   protected <G> G doCall(Request request, Function<Response, G> handler) {
 112     try (Response response = client.newCall(request).execute()) {
 113       handleError(response);
 114       return handler.apply(response);
 115     } catch (JsonSyntaxException e) {
 116       throw new IllegalArgumentException(UNABLE_TO_CONTACT_BITBUCKET_SERVER + ", got an unexpected response", e);
 117     } catch (IOException e) {
 118       throw new IllegalArgumentException(UNABLE_TO_CONTACT_BITBUCKET_SERVER, e);
 119     }
 120   }
 121
 122   protected static void handleError(Response response) throws IOException {
 123     if (!response.isSuccessful()) {
 124       String errorMessage = getErrorMessage(response.body());
 125       LOG.debug(UNABLE_TO_CONTACT_BITBUCKET_SERVER + ": {} {}", response.code(), errorMessage);
 126       if (response.code() == HttpURLConnection.HTTP_UNAUTHORIZED) {
 127         throw new IllegalArgumentException("Invalid personal access token");
 128       }
 129       throw new IllegalArgumentException(UNABLE_TO_CONTACT_BITBUCKET_SERVER);
 130     }
 131   }
 132
 133   protected static boolean equals(@Nullable MediaType first, @Nullable MediaType second) {
 134     String s1 = first == null ? null : first.toString().toLowerCase(ENGLISH).replace(" ", "");
 135     String s2 = second == null ? null : second.toString().toLowerCase(ENGLISH).replace(" ", "");
 136     return s1 != null && s2 != null && s1.equals(s2);
 137   }
 138
 139   protected static String getErrorMessage(ResponseBody body) throws IOException {
 140     if (equals(MediaType.parse("application/json;charset=utf-8"), body.contentType())) {
 141       try {
 142         return Stream.of(buildGson().fromJson(body.charStream(), Errors.class).errorData)
 143           .map(e -> e.exceptionName + " " + e.message)
 144           .collect(Collectors.joining("\n"));
 145       } catch (JsonParseException e) {
 146         return body.string();
 147       }
 148     }
 149     return body.string();
 150   }
 151
 152   protected static Gson buildGson() {
 153     return new GsonBuilder()
 154       .create();
 155   }
 156
 157   protected static class Errors {
 158
 159     @SerializedName("errors")
 160     public Error[] errorData;
 161
 162     public Errors() {
 163       // http://stackoverflow.com/a/18645370/229031
 164     }
 165
 166     public static class Error {
 167       @SerializedName("message")
 168       public String message;
 169
 170       @SerializedName("exceptionName")
 171       public String exceptionName;
 172
 173       public Error() {
 174         // http://stackoverflow.com/a/18645370/229031
 175       }
 176     }
 177
 178   }
 179
 180 }