@onekeyfe/react-native-sni-connect 3.0.81-alpha.2 → 3.0.81-alpha.4

package/ios/SniConnectClient.swift

@@ -90,15 +90,22 @@ final class SniConnectPinnedResolverLease {
   }
 }
 
-final class SniConnectSessionInvalidationDelegate: NSObject, URLSessionDelegate {
+final class SniConnectSessionInvalidationDelegate: NSObject, URLSessionDataDelegate {
   private let hostname: String
   private let ip: String
   private let resolverLease: SniConnectPinnedResolverLease
-
-  init(hostname: String, ip: String, resolverLease: SniConnectPinnedResolverLease) {
+  private weak var forwardingDataDelegate: URLSessionDataDelegate?
+
+  init(
+    hostname: String,
+    ip: String,
+    resolverLease: SniConnectPinnedResolverLease,
+    forwardingDataDelegate: URLSessionDataDelegate? = nil
+  ) {
     self.hostname = hostname
     self.ip = ip
     self.resolverLease = resolverLease
+    self.forwardingDataDelegate = forwardingDataDelegate
   }
 
   func urlSession(_ session: URLSession, didBecomeInvalidWithError error: Error?) {
@@ -110,6 +117,64 @@ final class SniConnectSessionInvalidationDelegate: NSObject, URLSessionDelegate
       ("success", error == nil),
     ]))
   }
+
+  func urlSession(
+    _ session: URLSession,
+    task: URLSessionTask,
+    willPerformHTTPRedirection response: HTTPURLResponse,
+    newRequest request: URLRequest,
+    completionHandler: @escaping (URLRequest?) -> Void
+  ) {
+    if let forwardingDataDelegate {
+      forwardingDataDelegate.urlSession?(
+        session,
+        task: task,
+        willPerformHTTPRedirection: response,
+        newRequest: request,
+        completionHandler: completionHandler
+      )
+    } else {
+      completionHandler(nil)
+    }
+  }
+
+  func urlSession(
+    _ session: URLSession,
+    dataTask: URLSessionDataTask,
+    didReceive response: URLResponse,
+    completionHandler: @escaping (URLSession.ResponseDisposition) -> Void
+  ) {
+    forwardingDataDelegate?.urlSession?(
+      session,
+      dataTask: dataTask,
+      didReceive: response,
+      completionHandler: completionHandler
+    ) ?? completionHandler(.cancel)
+  }
+
+  func urlSession(
+    _ session: URLSession,
+    dataTask: URLSessionDataTask,
+    didReceive data: Data
+  ) {
+    forwardingDataDelegate?.urlSession?(
+      session,
+      dataTask: dataTask,
+      didReceive: data
+    )
+  }
+
+  func urlSession(
+    _ session: URLSession,
+    task: URLSessionTask,
+    didCompleteWithError error: Error?
+  ) {
+    forwardingDataDelegate?.urlSession?(
+      session,
+      task: task,
+      didCompleteWithError: error
+    )
+  }
 }
 
 /// Core HTTPS client that enforces IP direct connection with SNI.

package/ios/SniConnectPinnedTransport.swift

@@ -30,7 +30,8 @@ public final class SniConnectPinnedSession {
 public enum SniConnectPinnedTransport {
   static func makeResources(
     hostname: String,
-    ip: String
+    ip: String,
+    dataDelegate: URLSessionDataDelegate? = nil
   ) throws -> SniConnectPinnedTransportResources {
     try SniConnectValidation.validateHostname(hostname)
     try SniConnectValidation.validatePublicIP(ip)
@@ -65,7 +66,8 @@ public enum SniConnectPinnedTransport {
     let delegate = SniConnectSessionInvalidationDelegate(
       hostname: normalizedHostname,
       ip: ip,
-      resolverLease: resolverLease
+      resolverLease: resolverLease,
+      forwardingDataDelegate: dataDelegate
     )
     return SniConnectPinnedTransportResources(
       configuration: configuration,
@@ -76,10 +78,15 @@ public enum SniConnectPinnedTransport {
 
   public static func makeSession(
     hostname: String,
-    ip: String
+    ip: String,
+    dataDelegate: URLSessionDataDelegate? = nil
   ) throws -> SniConnectPinnedSession {
     SniConnectPinnedSession(
-      resources: try makeResources(hostname: hostname, ip: ip)
+      resources: try makeResources(
+        hostname: hostname,
+        ip: ip,
+        dataDelegate: dataDelegate
+      )
     )
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@onekeyfe/react-native-sni-connect",
-  "version": "3.0.81-alpha.2",
+  "version": "3.0.81-alpha.4",
   "description": "A React Native library for SNI-based HTTP requests with DNS caching and request management",
   "main": "./lib/module/index.js",
   "types": "./lib/typescript/src/index.d.ts",