Soutbound PM Rest Client

[oam/nf-oam-adopter.git] / ves-nf-oam-adopter / ves-nf-oam-adopter-pm-sb-rest-client / src / main / java / org / o / ran / oam / nf / oam / adopter / pm / sb / rest / client / http / TokenHandler.java

diff --git a/ves-nf-oam-adopter/ves-nf-oam-adopter-pm-sb-rest-client/src/main/java/org/o/ran/oam/nf/oam/adopter/pm/sb/rest/client/http/TokenHandler.java b/ves-nf-oam-adopter/ves-nf-oam-adopter-pm-sb-rest-client/src/main/java/org/o/ran/oam/nf/oam/adopter/pm/sb/rest/client/http/TokenHandler.java
new file mode 100644 (file)
index 0000000..26c62b5
--- /dev/null
+++ b/ves-nf-oam-adopter/ves-nf-oam-adopter-pm-sb-rest-client/src/main/java/org/o/ran/oam/nf/oam/adopter/pm/sb/rest/client/http/TokenHandler.java
@@ -0,0 +1,53 @@
+package org.o.ran.oam.nf.oam.adopter.pm.sb.rest.client.http;
+
+import static org.springframework.util.MimeTypeUtils.APPLICATION_JSON_VALUE;
+
+import com.google.gson.Gson;
+import java.nio.charset.StandardCharsets;
+import java.util.Base64;
+import java.util.concurrent.ExecutionException;
+import lombok.AccessLevel;
+import lombok.NoArgsConstructor;
+import org.apache.hc.client5.http.async.methods.SimpleHttpRequest;
+import org.apache.hc.client5.http.async.methods.SimpleHttpRequests;
+import org.apache.hc.client5.http.async.methods.SimpleHttpResponse;
+import org.apache.hc.client5.http.impl.async.CloseableHttpAsyncClient;
+import org.apache.hc.core5.http.HttpHeaders;
+import org.apache.hc.core5.http.HttpStatus;
+import org.apache.hc.core5.http.message.StatusLine;
+import org.o.ran.oam.nf.oam.adopter.pm.rest.manager.exceptions.TokenGenerationException;
+import org.o.ran.oam.nf.oam.adopter.pm.rest.manager.pojos.Adapter;
+import org.o.ran.oam.nf.oam.adopter.pm.sb.rest.client.pojos.TokenResponse;
+
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
+public final class TokenHandler {
+
+    public static final Gson GSON = new Gson();
+    public static final String HTTPS = "https://";
+
+    /**
+     * Request Token under defined endpoint.
+     */
+    public static synchronized String returnToken(final CloseableHttpAsyncClient client, final String tokenEndpoint,
+            final Adapter adapter) throws ExecutionException, InterruptedException {
+        final String host = HTTPS + adapter.getHostIpAddress();
+        final SimpleHttpRequest request = SimpleHttpRequests.post(host + tokenEndpoint);
+        final String basicAuth = Base64.getEncoder().encodeToString(
+                (adapter.getUsername() + ":" + adapter.getPassword()).getBytes(StandardCharsets.UTF_8));
+        request.setHeader(HttpHeaders.AUTHORIZATION, "Basic " + basicAuth);
+        request.addHeader(HttpHeaders.CONTENT_TYPE, APPLICATION_JSON_VALUE);
+        request.addHeader(HttpHeaders.ACCEPT, APPLICATION_JSON_VALUE);
+
+        final SimpleHttpResponse response = client.execute(request, null).get();
+        final String statusLine = new StatusLine(response).toString();
+        if (response.getCode() != HttpStatus.SC_OK) {
+            throw new TokenGenerationException("Failed to obtain a token for host " + host + ": " + statusLine);
+        }
+        final String output = response.getBody().getBodyText();
+        if (output.isEmpty()) {
+            throw new TokenGenerationException(
+                    "Failed to obtain a token for host " + host + ", empty output: " + statusLine);
+        }
+        return GSON.fromJson(output, TokenResponse.class).getToken();
+    }
+}