Actually stop the track when the broadcast extension socket is closed

Sources/LiveKit/Broadcast/BroadcastScreenCapturer.swift

@@ -62,6 +62,11 @@ class BroadcastScreenCapturer: BufferCapturer {
         frameReader.didCapture = { pixelBuffer, rotation in
             self.capture(pixelBuffer, rotation: rotation.toLKType())
         }
+        frameReader.didEnd = { [weak self] in
+            Task {
+                try? await self?.stopCapture()
+            }
+        }
         frameReader.startCapture(with: socketConnection)
         self.frameReader = frameReader
 

Sources/LiveKit/Broadcast/SocketConnectionFrameReader.swift

@@ -137,6 +137,7 @@ class SocketConnectionFrameReader: NSObject {
 
     private var message: Message?
     var didCapture: ((CVPixelBuffer, RTCVideoRotation) -> Void)?
+    var didEnd: (() -> Void)?
 
     override init() {}
 
@@ -227,6 +228,7 @@ extension SocketConnectionFrameReader: StreamDelegate {
         case .endEncountered:
             logger.log(level: .debug, "server stream end encountered")
             stopCapture()
+            didEnd?()
         case .errorOccurred:
             logger.log(level: .debug, "server stream error encountered: \(aStream.streamError?.localizedDescription ?? "")")
         default:

Sources/LiveKit/TrackPublications/LocalTrackPublication.swift

@@ -105,6 +105,20 @@ extension LocalTrackPublication: VideoCapturerDelegate {
             }
         }
     }
+
+    public func capturer(_ capturer: VideoCapturer, didUpdate state: VideoCapturer.CapturerState) {
+        // Broadcasts can always be stopped from system UI that bypasses our normal disable & unpublish methods.
+        // This check ensures that when this happens the track gets unpublished as well.
+        if state == .stopped && capturer is BroadcastScreenCapturer {
+            Task {
+                guard let participant = try await self.requireParticipant() as? LocalParticipant else {
+                    return
+                }
+
+                try await participant.unpublish(publication: self)
+            }
+        }
+    }
 }
 
 extension LocalTrackPublication {