RestTemplate -> OkHttpClient

1 files changed, 37 insertions, 38 deletions

diff --git a/src/main/java/com/juick/service/ActivityPubService.java b/src/main/java/com/juick/service/ActivityPubService.java
index 9b93cab2..7eff123f 100644
--- a/src/main/java/com/juick/service/ActivityPubService.java
+++ b/src/main/java/com/juick/service/ActivityPubService.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2008-2020, Juick
+ * Copyright (C) 2008-2023, Juick
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU Affero General Public License as
@@ -17,16 +17,18 @@
 
 package com.juick.service;
 
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.juick.KeystoreManager;
+import com.juick.model.AnonymousUser;
 import com.juick.model.User;
 import com.juick.service.activities.DeleteUserEvent;
 import com.juick.util.DateFormattersHolder;
 import com.juick.www.api.activity.model.Context;
 import com.juick.www.api.activity.model.objects.Actor;
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.juick.KeystoreManager;
-import com.juick.model.AnonymousUser;
-
-import org.apache.commons.codec.binary.Base64;
+import okhttp3.MediaType;
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.RequestBody;
 import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -35,20 +37,18 @@ import org.springframework.cache.annotation.Cacheable;
 import org.springframework.context.ApplicationEventPublisher;
 import org.springframework.core.convert.ConversionService;
 import org.springframework.dao.DuplicateKeyException;
-import org.springframework.http.*;
+import org.springframework.http.HttpHeaders;
 import org.springframework.stereotype.Repository;
 import org.springframework.transaction.annotation.Transactional;
-import org.springframework.web.client.RestClientException;
-import org.springframework.web.client.RestTemplate;
 import org.springframework.web.util.UriComponents;
 import org.springframework.web.util.UriComponentsBuilder;
+import org.tomitribe.auth.signatures.Base64;
 import org.tomitribe.auth.signatures.MissingRequiredHeaderException;
 import org.tomitribe.auth.signatures.Signature;
 import org.tomitribe.auth.signatures.Verifier;
 
 import javax.annotation.Nonnull;
 import javax.inject.Inject;
-
 import java.io.IOException;
 import java.net.URI;
 import java.security.Key;
@@ -56,7 +56,6 @@ import java.security.MessageDigest;
 import java.security.NoSuchAlgorithmException;
 import java.security.SignatureException;
 import java.time.Instant;
-import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
@@ -71,7 +70,7 @@ public class ActivityPubService extends BaseJdbcService implements SocialService
     @Inject
     private UserService userService;
     @Inject
-    private RestTemplate restClient;
+    private OkHttpClient httpClient;
     @Inject
     private ObjectMapper jsonMapper;
     @Inject
@@ -138,28 +137,26 @@ public class ActivityPubService extends BaseJdbcService implements SocialService
         try {
             String signatureString = signatureService.addSignature(from, host, "get", contextUri.getPath(), requestDate,
                     "");
-            HttpHeaders requestHeaders = new HttpHeaders();
-            requestHeaders.add(HttpHeaders.DATE, requestDate);
-            requestHeaders.add(HttpHeaders.HOST, host);
-            requestHeaders.add("Signature", signatureString);
-            requestHeaders.setAccept(Collections.singletonList(MediaType.valueOf(Context.ACTIVITY_MEDIA_TYPE)));
-            HttpEntity<Void> activityRequest = new HttpEntity<>(requestHeaders);
-            var response = restClient.exchange(contextUri, HttpMethod.GET, activityRequest, Context.class);
-            if (response.getStatusCode().is2xxSuccessful()) {
-                var context = response.getBody();
-                if (context == null) {
-                    logger.warn("Cannot identify {}", contextUri);
-                    return Optional.empty();
+            var request = new Request.Builder()
+                    .url(contextUri.toURL())
+                    .addHeader(HttpHeaders.DATE, requestDate)
+                    .addHeader(HttpHeaders.HOST, host)
+                    .addHeader("Signature", signatureString)
+                    .addHeader(HttpHeaders.ACCEPT, Context.ACTIVITYSTREAMS_PROFILE_MEDIA_TYPE)
+                    .build();
+            try (var response = httpClient.newCall(request).execute()) {
+                if (response.isSuccessful() && response.body() != null) {
+                    var context = jsonMapper.readValue(response.body().string(), Context.class);
+                    return Optional.of(context);
                 }
-                return Optional.of(context);
             }
-        } catch (IOException | RestClientException e) {
+        } catch (IOException e) {
             logger.warn("HTTP Signature exception: {}", e.getMessage());
         }
         return Optional.empty();
     }
 
-    public HttpStatusCode post(Actor from, Actor to, Context data) throws IOException, NoSuchAlgorithmException {
+    public int post(Actor from, Actor to, Context data) throws IOException, NoSuchAlgorithmException, InterruptedException {
         UriComponentsBuilder uriComponentsBuilder = UriComponentsBuilder.fromUriString(to.getInbox());
         URI inbox = uriComponentsBuilder.build().toUri();
         Instant now = Instant.now();
@@ -171,18 +168,20 @@ public class ActivityPubService extends BaseJdbcService implements SocialService
         final String digestHeader = "SHA-256=" + new String(Base64.encodeBase64(digest));
         String signatureString = signatureService.addSignature(from, host, "post", inbox.getPath(), requestDate,
                 digestHeader);
-
-        HttpHeaders requestHeaders = new HttpHeaders();
-        requestHeaders.add(HttpHeaders.CONTENT_TYPE, Context.ACTIVITYSTREAMS_PROFILE_MEDIA_TYPE);
-        requestHeaders.add(HttpHeaders.DATE, requestDate);
-        requestHeaders.add(HttpHeaders.HOST, host);
-        requestHeaders.add("Digest", digestHeader);
-        requestHeaders.add("Signature", signatureString);
-        HttpEntity<String> request = new HttpEntity<>(payload, requestHeaders);
+        var body = RequestBody.create(payload, MediaType.get(Context.ACTIVITYSTREAMS_PROFILE_MEDIA_TYPE));
+        var request = new Request.Builder()
+                .url(inbox.toASCIIString())
+                .post(body)
+                .addHeader(HttpHeaders.DATE, requestDate)
+                .addHeader(HttpHeaders.CONTENT_TYPE, Context.ACTIVITYSTREAMS_PROFILE_MEDIA_TYPE)
+                .addHeader("Digest", digestHeader)
+                .addHeader("Signature", signatureString)
+                .build();
         logger.debug("Sending context to {}: {}", to.getId(), payload);
-        ResponseEntity<String> response = restClient.postForEntity(inbox, request, String.class);
-        logger.debug("Remote response: {} {}", response.getStatusCode(), response.getBody());
-        return response.getStatusCode();
+        try (var response = httpClient.newCall(request).execute()) {
+            logger.debug("Remote response: {} {}", response.code(), response.body());
+            return response.code();
+        }
     }
 
     public User verifyActor(String method, String path, Map<String, String> headers) {