From 00dfb4ce397d16350305c6254161d9eb30dfed7f Mon Sep 17 00:00:00 2001
From: hstyi <hstyi@foxmail.com>
Date: Wed, 25 Jun 2025 17:30:40 +0800
Subject: [PATCH] chore: support s3 proxy

---
 plugins/s3/build.gradle.kts                   |  2 +-
 .../termora/plugins/s3/S3ProtocolProvider.kt  | 35 +++++++++++++++++++
 2 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/plugins/s3/build.gradle.kts b/plugins/s3/build.gradle.kts
index 2d5db64..5257ba7 100644
--- a/plugins/s3/build.gradle.kts
+++ b/plugins/s3/build.gradle.kts
@@ -3,7 +3,7 @@ plugins {
 }
 
 
-project.version = "0.0.2"
+project.version = "0.0.3"
 
 
 dependencies {
diff --git a/plugins/s3/src/main/kotlin/app/termora/plugins/s3/S3ProtocolProvider.kt b/plugins/s3/src/main/kotlin/app/termora/plugins/s3/S3ProtocolProvider.kt
index 9d4404f..192e7b6 100644
--- a/plugins/s3/src/main/kotlin/app/termora/plugins/s3/S3ProtocolProvider.kt
+++ b/plugins/s3/src/main/kotlin/app/termora/plugins/s3/S3ProtocolProvider.kt
@@ -1,12 +1,18 @@
 package app.termora.plugins.s3
 
+import app.termora.AuthenticationType
 import app.termora.DynamicIcon
 import app.termora.Icons
+import app.termora.ProxyType
 import app.termora.protocol.PathHandler
 import app.termora.protocol.PathHandlerRequest
 import app.termora.protocol.TransferProtocolProvider
 import io.minio.MinioClient
+import okhttp3.*
 import org.apache.commons.lang3.StringUtils
+import java.net.InetSocketAddress
+import java.net.Proxy
+import java.time.Duration
 
 class S3ProtocolProvider private constructor() : TransferProtocolProvider {
 
@@ -32,6 +38,35 @@ class S3ProtocolProvider private constructor() : TransferProtocolProvider {
         if (StringUtils.isNotBlank(region)) {
             builder.region(region)
         }
+
+        if (host.proxy.type != ProxyType.No) {
+            val httpClient = OkHttpClient.Builder()
+                .connectTimeout(Duration.ofSeconds(10))
+                .callTimeout(Duration.ofSeconds(60))
+                .writeTimeout(Duration.ofSeconds(60))
+                .readTimeout(Duration.ofSeconds(60))
+
+            if (host.proxy.type == ProxyType.HTTP) {
+                httpClient.proxy(Proxy(Proxy.Type.HTTP, InetSocketAddress(host.proxy.host, host.proxy.port)))
+            } else if (host.proxy.type == ProxyType.SOCKS5) {
+                httpClient.proxy(Proxy(Proxy.Type.SOCKS, InetSocketAddress(host.proxy.host, host.proxy.port)))
+            }
+
+            if (host.proxy.authenticationType != AuthenticationType.No) {
+                httpClient.proxyAuthenticator(object : Authenticator {
+                    override fun authenticate(route: Route?, response: Response): Request? {
+                        val credential = Credentials.basic(host.proxy.username, host.proxy.password)
+                        return response.request.newBuilder()
+                            .header("Proxy-Authorization", credential)
+                            .build()
+                    }
+
+                })
+            }
+
+            builder.httpClient(httpClient.build(), true)
+        }
+
 //        val delimiter = host.options.extras["s3.delimiter"] ?: "/"
         val defaultPath = host.options.sftpDefaultDirectory
         val minioClient = builder.build()