Add proxy support

OpenConext · Aug 21, 2023 · 8337714 · 8337714
1 parent 6f0f2be
commit 8337714
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/manage-server/src/main/java/manage/control/DatabaseController.java b/manage-server/src/main/java/manage/control/DatabaseController.java
@@ -8,9 +8,14 @@
 import manage.web.PreemptiveAuthenticationHttpComponentsClientHttpRequestFactory;
 import org.apache.http.auth.AuthScope;
 import org.apache.http.auth.UsernamePasswordCredentials;
+import org.apache.http.HttpException;
+import org.apache.http.HttpHost;
+import org.apache.http.HttpRequest;
 import org.apache.http.impl.client.BasicCredentialsProvider;
 import org.apache.http.impl.client.CloseableHttpClient;
 import org.apache.http.impl.client.HttpClientBuilder;
+import org.apache.http.impl.conn.DefaultProxyRoutePlanner;
+import org.apache.http.protocol.HttpContext;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.core.env.Environment;
@@ -55,6 +60,9 @@ public class DatabaseController {
 
     private final Environment environment;
 
+    private String proxyHost;
+    private int proxyPort;
+
     @Autowired
     DatabaseController(MetaDataRepository metaDataRepository,
                        @Value("${push.eb.url}") String pushUri,
@@ -205,6 +213,25 @@ private ClientHttpRequestFactory getRequestFactory(String user, String password)
         BasicCredentialsProvider basicCredentialsProvider = new BasicCredentialsProvider();
         basicCredentialsProvider.setCredentials(AuthScope.ANY, new UsernamePasswordCredentials(user, password));
         httpClientBuilder.setDefaultCredentialsProvider(basicCredentialsProvider);
+
+        this.proxyHost = System.getProperty("http.proxyHost");
+        if (System.getProperty("http.proxyPort") != null) {
+            this.proxyPort = Integer.parseInt(System.getProperty("http.proxyPort"));
+        } else {
+            this.proxyPort = 8080;
+        }
+
+        if (this.proxyHost != null) {
+            HttpHost proxy = new HttpHost(this.proxyHost, this.proxyPort);
+
+            httpClientBuilder.setRoutePlanner(new DefaultProxyRoutePlanner(proxy) {
+                @Override
+                public HttpHost determineProxy(HttpHost target, HttpRequest request, HttpContext context) throws HttpException {
+                    return super.determineProxy(target, request, context);
+                }
+            });
+        }
+
         CloseableHttpClient httpClient = httpClientBuilder.build();
         return new PreemptiveAuthenticationHttpComponentsClientHttpRequestFactory(httpClient, pushUri);
     }