Add reconnect strategy callback interface (#1545)

Author: mobile-bungalow

Package.swift

@@ -25,7 +25,7 @@ let package = Package(
     dependencies: [
         // Dependencies declare other packages that this package depends on.
         .package(url: "https://github.com/apple/swift-async-algorithms", from: "1.0.0"),
-        .package(url: "https://github.com/liveview-native/liveview-native-core", exact: "0.4.1-rc-1"),
+        .package(url: "https://github.com/liveview-native/liveview-native-core", exact: "0.4.1-rc-2"),
 
         .package(url: "https://github.com/apple/swift-argument-parser", from: "1.5.0"),
 

Sources/LiveViewNative/Coordinators/LiveSessionConfiguration.swift

@@ -6,6 +6,7 @@
 //
 
 import SwiftUI
+import LiveViewNativeCore
 
 /// An object that configures the behavior of a ``LiveSessionCoordinator``.
 public struct LiveSessionConfiguration {
@@ -99,3 +100,22 @@ public struct LiveSessionConfiguration {
         public static let never: Self = .init()
     }
 }
+
+
+final class ReconnectStrategyAdapter: SocketReconnectStrategy {
+    private let behavior: LiveSessionConfiguration.ReconnectBehavior
+    
+    init(_ behavior: LiveSessionConfiguration.ReconnectBehavior) {
+        self.behavior = behavior
+    }
+
+    func sleepDuration(_ attempt: UInt64) -> TimeInterval {
+        guard let delay = behavior.delay else {
+            return TimeInterval.greatestFiniteMagnitude
+        }
+
+        let safeAttempt = min(Int(attempt), Int.max)
+        return delay(safeAttempt)
+     }
+}
+

Sources/LiveViewNative/Coordinators/LiveSessionCoordinator.swift

@@ -216,6 +216,8 @@ public class LiveSessionCoordinator<R: RootRegistry>: ObservableObject {
                 try await socket.shutdown()
             }
 
+            let adapter = ReconnectStrategyAdapter(self.configuration.reconnectBehavior)
+            
             self.liveSocket = try await LiveSocket(
                 originalURL.absoluteString,
                 LiveSessionParameters.platform,
@@ -224,7 +226,8 @@ public class LiveSessionCoordinator<R: RootRegistry>: ObservableObject {
                     body: httpBody,
                     method: httpMethod.flatMap(Method.init(_:)),
                     timeoutMs: 10_000
-                )
+                ),
+                adapter
             )
 
             // save cookies to storage
@@ -289,6 +292,17 @@ public class LiveSessionCoordinator<R: RootRegistry>: ObservableObject {
         }
     }
 
+    func overrideLiveReloadChannel(channel: LiveChannel) async throws {
+        
+        if let liveReloadChannel {
+            try await liveReloadChannel.shutdownParentSocket()
+            self.liveReloadChannel = nil
+        }
+        
+        self.liveReloadChannel = channel
+        self.bindLiveReloadListener()
+    }
+    
     func bindLiveReloadListener() {
         let eventListener = self.liveReloadChannel!.channel().events()
         self.liveReloadListenerLoop = Task { @MainActor [weak self] in

Sources/LiveViewNative/Live/LiveErrorView.swift

@@ -29,7 +29,7 @@ public struct LiveErrorView<Fallback: View>: View {
            case let .ConnectionError(trace) = error
         {
             SwiftUI.VStack {
-                WebErrorView(html: trace)
+                WebErrorView(html: trace.errorText)
                 #if os(watchOS) || os(tvOS)
                 SwiftUI.Button {
                     Task {

Sources/LiveViewNative/Live/LiveView.swift

@@ -8,6 +8,7 @@
 import Foundation
 import SwiftUI
 import Combine
+import LiveViewNativeCore
 
 /// Create a ``LiveView`` with a list of addons.
 ///
@@ -206,6 +207,16 @@ public struct LiveView<
         case .connecting:
             return .connecting
         case let .connectionFailed(error):
+            
+            if let error = error as? LiveViewNativeCore.LiveSocketError,
+               case let .ConnectionError(connectionError) = error {
+                if let channel = connectionError.livereloadChannel {
+                    Task { @MainActor [weak session] in
+                        try await session?.overrideLiveReloadChannel(channel: channel)
+                    }
+                }
+            }
+            
             return .error(error)
         case .setup:
             return .connecting