Add extra helper

rubenlagus · Apr 13, 2024 · d42ad30 · d42ad30
1 parent e9add4f
commit d42ad30
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/...src/main/java/org/telegram/telegrambots/longpolling/util/TelegramOkHttpClientFactory.java b/...src/main/java/org/telegram/telegrambots/longpolling/util/TelegramOkHttpClientFactory.java
@@ -45,7 +45,7 @@ protected static OkHttpClient.Builder getBaseClient() {
     }
 
     @RequiredArgsConstructor
-    public static class ProxyOkHttpClientCreator extends DefaultOkHttpClientCreator {
+    public static class HttpProxyOkHttpClientCreator extends DefaultOkHttpClientCreator {
         private final Supplier<Proxy> proxySupplier;
         private final Supplier<Authenticator> authenticatorSupplier;
 
@@ -60,4 +60,29 @@ public OkHttpClient get() {
             return okHttpClientBuilder.build();
         }
     }
+
+    @RequiredArgsConstructor
+    public static class SocksProxyOkHttpClientCreator extends DefaultOkHttpClientCreator {
+        private final Supplier<Proxy> proxySupplier;
+
+        @Override
+        public OkHttpClient get() {
+            OkHttpClient.Builder okHttpClientBuilder = getBaseClient();
+
+            // Proxy
+            ofNullable(proxySupplier.get()).ifPresent(okHttpClientBuilder::proxy);
+
+            return okHttpClientBuilder.build();
+        }
+    }
+
+    /**
+     * @deprecated Use {@link HttpProxyOkHttpClientCreator} instead
+     */
+    @Deprecated
+    public static class ProxyOkHttpClientCreator extends HttpProxyOkHttpClientCreator {
+        public ProxyOkHttpClientCreator(Supplier<Proxy> proxySupplier, Supplier<Authenticator> authenticatorSupplier) {
+            super(proxySupplier, authenticatorSupplier);
+        }
+    }
 }