com.vmware.pscoe.iac.artifact.rest.RestClientCs Maven / Gradle / Ivy
/*
* #%L
* artifact-manager
* %%
* Copyright (C) 2023 VMware
* %%
* Build Tools for VMware Aria
* Copyright 2023 VMware, Inc.
*
* This product is licensed to you under the BSD-2 license (the "License"). You may not use this product except in compliance with the BSD-2 License.
*
* This product may include a number of subcomponents with separate copyright notices and license terms. Your use of these subcomponents is subject to the terms and conditions of the subcomponent's license, as noted in the LICENSE file.
* #L%
*/
package com.vmware.pscoe.iac.artifact.rest;
import java.net.URI;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.jayway.jsonpath.JsonPath;
import com.vmware.pscoe.iac.artifact.configuration.Configuration;
import com.vmware.pscoe.iac.artifact.configuration.ConfigurationCs;
import com.vmware.pscoe.iac.artifact.rest.model.cs.CustomIntegrationVersion;
import com.vmware.pscoe.iac.artifact.rest.model.cs.Endpoint;
import com.vmware.pscoe.iac.artifact.rest.model.cs.Variable;
import org.apache.commons.lang3.StringUtils;
import org.apache.hc.core5.net.URIBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;
/**
* Rest client for Code Stream.
*/
public class RestClientCs extends RestClient {
private final Logger logger = LoggerFactory.getLogger(RestClientCs.class);
private static final String BASE = "/codestream/api";
private static final String API_VERSION = BASE + "/about";
private static final String VARIABLES_PATH = BASE + "/variables";
private static final String VARIABLE_PROJECT_NAME_PATH = BASE + "/variables/{0}/{1}";
private static final String CUSTOM_INTEGRATION_PATH = BASE + "/custom-integrations";
private static final String CUSTOM_INTEGRATION_VERSION_PATH = BASE + "/custom-integrations/{0}/versions";
private static final String ENDPOINTS_PATH = BASE + "/endpoints";
private static final String ENDPOINT_UPDATE_PATH = BASE + "/endpoints/{0}/{1}";
private static final String PIPELINE_PATH = BASE + "/pipelines";
private static final String PIPELINE_UPDATE_PATH = BASE + "/pipelines/{0}/{1}";
private static final String CLOUD_PROXY_PATH = BASE + "/cloud-proxy";
private static final String GET_PROJECT_INFO = "/iaas/api/projects";
private static final String GIT_PATH = BASE + "/git-webhooks";
private static final String GIT_PROJECT_NAME_PATH = GIT_PATH + "/{0}/{1}";
private static final String DOCKER_PATH = BASE + "/registry-webhooks";
private static final String DOCKER_PROJECT_NAME_PATH = DOCKER_PATH + "/{0}/{1}";
private static final String GERRIT_LISTENER_PATH = BASE + "/gerrit-listeners";
private static final String GERRIT_LISTENER_PROJECT_NAME_PATH = GERRIT_LISTENER_PATH + "/{0}/{1}";
private static final String GERRIT_TRIGGER_PATH = BASE + "/gerrit-triggers";
private static final String GERRIT_TRIGGER_PROJECT_NAME_PATH = GERRIT_TRIGGER_PATH + "/{0}/{1}";
private static final Integer PAGE_SIZE = 100;
private ConfigurationCs configuration;
private RestTemplate restTemplate;
private String apiVersion;
private String projectName;
private String projectId;
private String cloudProxyId;
protected RestClientCs(ConfigurationCs configuration, RestTemplate restTemplate) {
this.configuration = configuration;
this.restTemplate = restTemplate;
initProjectNameAndId();
initCloudProxyId();
}
private void initProjectNameAndId() {
String constraint = String.format("name eq '%s'", configuration.getProjectName());
URI url = getURI(getURIBuilder()
.setPath(GET_PROJECT_INFO)
.setParameter("$filter", constraint));
ResponseEntity response = restTemplate.exchange(url, HttpMethod.GET, getDefaultHttpEntity(),
String.class);
JsonElement jsonObj = JsonParser.parseString(response.getBody());
JsonArray projectJsonArray = jsonObj.getAsJsonObject().get("content").getAsJsonArray();
if (projectJsonArray.size() != 1) {
throw new RuntimeException("Project not found with given configuration!");
}
this.projectName = projectJsonArray.get(0).getAsJsonObject().get("name").getAsString();
this.projectId = projectJsonArray.get(0).getAsJsonObject().get("id").getAsString();
}
/**
* Get the project name.
*
* @return the project name
*/
public String getProjectName() {
return this.projectName;
}
/**
* Get the project id.
*
* @return the project id
*/
public String getProjectId() {
return projectId;
}
/**
* Get the cloud proxy id.
*
* @return the cloud proxy id
*/
public String getCloudProxyId() {
return cloudProxyId;
}
/**
* Get the rest template.
*
* @return the rest template
*/
public RestTemplate getRestTemplate() {
return restTemplate;
}
/**
* Get the configuration.
*
* @return the configuration
*/
@Override
protected Configuration getConfiguration() {
return configuration;
}
/**
* Get the API version.
*
* @return the API version
*/
@Override
public String getVersion() {
if (this.apiVersion != null && !this.apiVersion.isEmpty()) {
return this.apiVersion;
}
URI url = getURI(getURIBuilder().setPath(API_VERSION));
ResponseEntity response = restTemplate.exchange(url, HttpMethod.GET, getDefaultHttpEntity(),
String.class);
this.apiVersion = JsonPath.parse(response.getBody()).read("$.supportedApis[0].apiVersion");
logger.info("Detected API Version " + this.apiVersion);
return this.apiVersion;
}
/**
* Get the project variables.
*
* @param variableName the variable name
*
* @return the project variables
*/
public Variable getProjectVariableByName(String variableName) {
Gson gson = new Gson();
logger.info("Get variable " + variableName);
URI url = getURI(getURIBuilder()
.setPath(MessageFormat.format(VARIABLE_PROJECT_NAME_PATH, getProjectName(), variableName)));
ResponseEntity response = restTemplate.exchange(url, HttpMethod.GET, getDefaultHttpEntity(),
String.class);
logger.debug("Body: " + response.getBody());
return gson.fromJson(response.getBody(), Variable.class);
}
/**
* Get the project variables.
*
* @return the project variables
*/
public List getProjectVariables() {
logger.info("Get Variables");
Map params = new LinkedHashMap();
params.put("$filter", String.format("project eq '%s'", getProjectName()));
List result = getPagedResult(VARIABLES_PATH, params);
Gson gson = gson();
return result.stream()
.map(obj -> gson.fromJson(obj, Variable.class))
.collect(Collectors.toList());
}
/**
* Update the project variable.
*
* @param var the variable to update
*
* @return the response
*/
public String updateVariable(Variable var) {
logger.info("Importing pipeline " + var.getName());
URI url = getURI(getURIBuilder()
.setPath(MessageFormat.format(VARIABLE_PROJECT_NAME_PATH, getProjectName(), var.getName())));
HttpEntity entity = getJsonEntity(var);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.PUT, entity, String.class);
logger.debug("Body: " + response.getBody());
return response.getBody();
}
/**
* Create the project variable.
*
* @param var the variable to create
*
* @return the response
*/
public String createVariable(Variable var) {
logger.info("Importing variable " + var.getName());
URI url = getURI(getURIBuilder().setPath(VARIABLES_PATH));
HttpEntity entity = getJsonEntity(var);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.POST, entity, String.class);
logger.debug("Body: " + response.getBody());
return response.getBody();
}
/**
* Get the project endpoints.
*
* @return the project endpoints
*/
public List getProjectEndpoints() {
logger.info("Get Endpoints");
Map params = new LinkedHashMap();
params.put("$filter", String.format("project eq '%s'", getProjectName()));
List result = getPagedResult(ENDPOINTS_PATH, params);
Gson gson = gson();
return result.stream()
.map(obj -> gson.fromJson(obj, Endpoint.class))
.collect(Collectors.toList());
}
/**
* Update the project endpoint.
*
* @param var the endpoint to update
*
* @return the response
*/
public String updateEndpoint(Endpoint var) {
logger.info("Update endpoint: {} ", var.getName());
URI url = getURI(
getURIBuilder().setPath(MessageFormat.format(ENDPOINT_UPDATE_PATH, getProjectName(), var.getName())));
HttpEntity entity = getJsonEntity(var);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.PUT, entity, String.class);
logger.debug("Body: " + response.getBody());
return response.getBody();
}
/**
* Create the project endpoint.
*
* @param var the endpoint to create
*
* @return the response
*/
public String createEndpoint(Endpoint var) {
logger.info("Create endpoint " + var.getName());
URI url = getURI(getURIBuilder().setPath(ENDPOINTS_PATH));
HttpEntity entity = getJsonEntity(var);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.POST, entity, String.class);
logger.debug("Body: " + response.getBody());
return response.getBody();
}
/**
* Get the project custom integrations.
*
* @param ciName the custom integration name
*
* @return the project custom integrations
*/
public Optional getCustomIntegrationByName(String ciName) {
logger.info("Get Custom Integration: " + ciName);
Map params = new LinkedHashMap();
params.put("$filter", String.format("name eq '%s'", ciName));
List result = getPagedResult(CUSTOM_INTEGRATION_PATH, params);
Gson gson = gson();
return result.stream()
.map(obj -> gson.fromJson(obj, CustomIntegrationVersion.class))
.findFirst();
}
/**
* Get the project custom integrations.
*
* @return the project custom integrations
*/
public List getCustomIntegrations() {
logger.info("Get Custom Integrations");
List result = getPagedResult(CUSTOM_INTEGRATION_PATH);
Gson gson = gson();
return result.stream()
.map(obj -> gson.fromJson(obj, CustomIntegrationVersion.class))
.collect(Collectors.toList());
}
/**
* Get the project custom integrations.
*
* @param id the custom integration id
*
* @return the project custom integrations
*/
public List getCustomIntegrationVersions(String id) {
logger.info("Get Custom Integration Versions for: " + id);
List result = getPagedResult(MessageFormat.format(CUSTOM_INTEGRATION_VERSION_PATH, id));
Gson gson = gson();
return result.stream()
.map(obj -> gson.fromJson(obj, CustomIntegrationVersion.class))
.collect(Collectors.toList());
}
/**
* Delete the project custom integration.
*
* @param id the custom integration id
*/
public void deleteCustomIntegration(String id) {
logger.info("Delete custom integration" + id);
URI url = getURI(getURIBuilder().setPath(CUSTOM_INTEGRATION_PATH + "/" + id));
ResponseEntity response = restTemplate.exchange(url, HttpMethod.DELETE, getDefaultHttpEntity(),
String.class);
logger.debug("Body: " + response.getBody());
}
/**
* Create the project custom integration.
*
* @param ci the custom integration to create
*
* @return the response
*/
public CustomIntegrationVersion createCustomIntegration(CustomIntegrationVersion ci) {
logger.info("Create Custom Integration: " + ci.getName());
URI url = getURI(getURIBuilder().setPath(CUSTOM_INTEGRATION_PATH));
HttpEntity entity = getJsonEntity(ci);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.POST, entity, String.class);
logger.debug("Body: " + response.getBody());
Gson gson = new GsonBuilder().setLenient().serializeNulls().create();
return gson.fromJson(response.getBody(), CustomIntegrationVersion.class);
}
/**
* Update the project custom integration.
*
* @param ci the custom integration to update
*
* @return the response
*/
public CustomIntegrationVersion updateCustomIntegration(CustomIntegrationVersion ci) {
logger.info("Update Custom Integration: " + ci.getName());
URI url = getURI(getURIBuilder().setPath(CUSTOM_INTEGRATION_PATH + "/" + ci.getId()));
HttpEntity entity = getJsonEntity(ci);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.PUT, entity, String.class);
logger.debug("Body: " + response.getBody());
Gson gson = new GsonBuilder().setLenient().serializeNulls().create();
return gson.fromJson(response.getBody(), CustomIntegrationVersion.class);
}
/**
* Create the project custom integration version.
*
* @param id the custom integration id
* @param version the custom integration version to create
*
* @return the response
*/
public Object createCustomIntegrationVersion(String id, CustomIntegrationVersion version) {
logger.info(
String.format("Create Custom Integration Version: %s : %s ", version.getName(), version.getVersion()));
URI url = getURI(getURIBuilder().setPath(MessageFormat.format(CUSTOM_INTEGRATION_VERSION_PATH, id)));
HttpEntity entity = getJsonEntity(version);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.POST, entity, String.class);
logger.debug("Body: " + response.getBody());
Gson gson = new GsonBuilder().setLenient().serializeNulls().create();
return gson.fromJson(response.getBody(), CustomIntegrationVersion.class);
}
/**
* Get the project pipelines.
*/
public List getProjectPipelines() {
logger.info("Get Pipelines");
return getPagedResult(PIPELINE_PATH, getProjectFilterParam());
}
/**
* Get the project pipelines.
*
* @return the project pipelines
*/
public JsonObject getProjectPipelineByName(String name) {
logger.info("Get pipeline: " + name);
return requestForPath(getProjectNameUrl(PIPELINE_UPDATE_PATH, name), HttpMethod.GET, null);
}
/**
* Update the project pipeline.
*
* @param name the pipeline name
* @param var the pipeline to update
*
* @return the response
*/
public JsonObject updatePipeline(String name, JsonObject var) {
logger.info("Update pipeline: {}", name);
return requestForPath(getProjectNameUrl(PIPELINE_UPDATE_PATH, name), HttpMethod.PUT, var);
}
/**
* Patch the project pipeline.
*
* @param name the pipeline name
* @param var the pipeline to patch
*
* @return the response
*/
public JsonObject patchPipeline(String name, JsonObject var) {
logger.info("PATCH pipeline: {}", name);
return requestForPath(getProjectNameUrl(PIPELINE_UPDATE_PATH, name), HttpMethod.PATCH, var);
}
/**
* Create the project pipeline.
*
* @param var the pipeline to create
*
* @return the response
*/
public JsonObject createPipeline(JsonObject var) {
String name = var.get("name").getAsString();
logger.info("Create pipeline: {}", name);
return requestForPath(getURI(getURIBuilder().setPath(PIPELINE_PATH)), HttpMethod.POST, var);
}
/**
* Get project git webhooks.
*
* @return the project git webhooks
*/
public List getProjectGitWebhooks() {
logger.info("Get git Webhooks for project");
return getPagedResult(GIT_PATH, getProjectFilterParam());
}
/**
* Get the project git webhook
*
* @param name the git webhook name
*
* @return the project git webhook
*/
public JsonObject getProjectGitWebhook(String name) {
logger.info("Get Git webhook: " + name);
return requestForPath(getProjectNameUrl(GIT_PROJECT_NAME_PATH, name), HttpMethod.GET, null);
}
/**
* Update the project git webhook.
*
* @param name the git webhook name
* @param var the git webhook to update
*
* @return the response
*/
public JsonObject updateGitWebhook(String name, JsonObject var) {
logger.info("Update git Webhook: {}", name);
return requestForPath(getProjectNameUrl(GIT_PROJECT_NAME_PATH, name), HttpMethod.PUT, var);
}
/**
* Create the project git webhook.
*
* @param var the git webhook to create
*
* @return the response
*/
public JsonObject createGitWebhook(JsonObject var) {
String name = var.get("name").getAsString();
logger.info("Create Git Webhook: {}", name);
return requestForPath(getURI(getURIBuilder().setPath(GIT_PATH)), HttpMethod.POST, var);
}
/**
* Get the project docker webhooks.
*
* @return the project docker webhooks
*/
public List getProjectDockerWebhooks() {
logger.info("Get git Webhooks for project");
return getPagedResult(DOCKER_PATH, getProjectFilterParam());
}
/**
* Get the project docker webhooks.
*
* @return the project docker webhooks
*/
public JsonObject getProjectDockerWebhook(String name) {
logger.info("Get Docker webhook: " + name);
return requestForPath(getProjectNameUrl(DOCKER_PROJECT_NAME_PATH, name), HttpMethod.GET, null);
}
/**
* Update the project docker webhook.
*
* @param name the docker webhook name
* @param var the docker webhook to update
*
* @return the response
*/
public JsonObject updateDockerWebhook(String name, JsonObject var) {
logger.info("Update Docker Webhook: {}", name);
return requestForPath(getProjectNameUrl(DOCKER_PROJECT_NAME_PATH, name), HttpMethod.PUT, var);
}
/**
* Create the project docker webhook.
*
* @param var the docker webhook to create
*
* @return the response
*/
public JsonObject createDockerWebhook(JsonObject var) {
String name = var.get("name").getAsString();
logger.info("Create Docker Webhook: {}", name);
return requestForPath(getURI(getURIBuilder().setPath(DOCKER_PATH)), HttpMethod.POST, var);
}
/**
* Get the project gerrit listeners.
*
* @return the project gerrit listeners
*/
public List getProjectGerritListeners() {
logger.info("Get Gerrit Listener for project");
return getPagedResult(GERRIT_LISTENER_PATH, getProjectFilterParam());
}
/**
* Get the project gerrit listeners.
*
* @return the project gerrit listeners
*/
public JsonObject getProjectGerritListener(String name) {
logger.info("Get Gerrit Listener: " + name);
return requestForPath(getProjectNameUrl(GERRIT_LISTENER_PROJECT_NAME_PATH, name), HttpMethod.GET, null);
}
/**
* Update the project gerrit listener.
*
* @param name the gerrit listener name
* @param var the gerrit listener to update
*
* @return the response
*/
public JsonObject updateGerritListener(String name, JsonObject var) {
logger.info("Update Gerrit Listener: {}", name);
return requestForPath(getProjectNameUrl(GERRIT_LISTENER_PROJECT_NAME_PATH, name), HttpMethod.PUT, var);
}
/**
* Create the project gerrit listener.
*
* @param var the gerrit listener to create
*
* @return the response
*/
public JsonObject createGerritListener(JsonObject var) {
String name = var.get("name").getAsString();
logger.info("Create Gerrit Trigger: {}", name);
return requestForPath(getURI(getURIBuilder().setPath(GERRIT_LISTENER_PATH)), HttpMethod.POST, var);
}
/**
* Get the project gerrit triggers.
*
* @return the project gerrit triggers
*/
public List getProjectGerritTriggers() {
logger.info("Get Gerrit Triggers for project");
return getPagedResult(GERRIT_TRIGGER_PATH, getProjectFilterParam());
}
/**
* Get the project gerrit triggers.
*
* @return the project gerrit triggers
*/
public JsonObject getProjectGerritTrigger(String name) {
logger.info("Get Gerrit Trigger: " + name);
return requestForPath(getProjectNameUrl(GERRIT_TRIGGER_PROJECT_NAME_PATH, name), HttpMethod.GET, null);
}
/**
* Update the project gerrit trigger.
*
* @param name the gerrit trigger name
* @param var the gerrit trigger to update
*
* @return the response
*/
public JsonObject updateGerritTrigger(String name, JsonObject var) {
logger.info("Update Gerrit Trigger: {}", name);
return requestForPath(getProjectNameUrl(GERRIT_TRIGGER_PROJECT_NAME_PATH, name), HttpMethod.PUT, var);
}
/**
* Create the project gerrit trigger.
*
* @param var the gerrit trigger to create
*
* @return the response
*/
public JsonObject createGerritTrigger(JsonObject var) {
String name = var.get("name").getAsString();
logger.info("Create Gerrit Trigger: {}", name);
return requestForPath(getURI(getURIBuilder().setPath(GERRIT_TRIGGER_PATH)), HttpMethod.POST, var);
}
private void initCloudProxyId() {
if (StringUtils.isEmpty(configuration.getCloudProxyName())) {
return;
}
List result = getPagedResult(CLOUD_PROXY_PATH);
Optional optional = result.stream()
.peek(el -> logger.info(el
.get("customProperties")
.getAsJsonObject()
.get("proxyName")
.getAsString()))
.filter(el -> el
.get("customProperties")
.getAsJsonObject()
.get("proxyName")
.getAsString()
.equals(configuration.getCloudProxyName()))
.findFirst();
if (!optional.isPresent()) {
throw new RuntimeException("No cloud proxy id found for name: " + configuration.getCloudProxyName());
}
this.cloudProxyId = optional.get().get("id").getAsString();
}
/**
* Get the project cloud proxy id.
*
* @return the project cloud proxy id
*/
public List getPagedResult(String variablePath) {
return getPagedResult(variablePath, new LinkedHashMap());
}
private List getPagedResult(String variablePath, Map parameters) {
List result = new ArrayList<>();
int page = 0;
boolean hasMorePages = true;
do {
URIBuilder builder = getURIBuilder().setPath(variablePath)
.setParameter("$top", PAGE_SIZE.toString())
.setParameter("$skip", (PAGE_SIZE * page + ""));
parameters.forEach((key, value) -> builder.setParameter(key, value));
URI url = getURI(builder);
ResponseEntity response = restTemplate.exchange(url, HttpMethod.GET, getDefaultHttpEntity(),
String.class);
logger.debug("Body: " + response.getBody());
JsonObject root = JsonParser.parseString(response.getBody()).getAsJsonObject();
root.get("documents").getAsJsonObject()
.entrySet()
.stream()
.map(entry -> entry.getValue().getAsJsonObject())
.forEachOrdered(result::add);
int requestCount = root.get("count").getAsInt();
page++;
hasMorePages = requestCount == PAGE_SIZE;
} while (hasMorePages);
return result;
}
private Gson gson() {
return new GsonBuilder().setLenient().serializeNulls().create();
}
private HttpEntity getJsonEntity(Object body) {
Gson gson = new GsonBuilder().setLenient().serializeNulls().create();
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
return new HttpEntity(gson.toJson(body), headers);
}
private URI getProjectNameUrl(String path, String name) {
return getURI(getURIBuilder().setPath(MessageFormat.format(path, getProjectName(), name)));
}
/**
* Get the project filter param.
*
* @return the project filter param
*/
Map getProjectFilterParam() {
Map params = new LinkedHashMap();
params.put("$filter", String.format("project eq '%s'", getProjectName()));
return params;
}
private JsonObject requestForPath(URI url, HttpMethod method, JsonObject var) {
HttpEntity entity = var == null ? getDefaultHttpEntity() : getJsonEntity(var);
ResponseEntity response = restTemplate.exchange(url, method, entity, String.class);
logger.debug("Body: " + response.getBody());
return JsonParser.parseString(response.getBody()).getAsJsonObject();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy