Separate PipelineSource and RealtimePipelineSource

Author: cherylEnkidu

Firestore/Swift/Source/SwiftAPI/Firestore+Pipeline.swift

@@ -23,15 +23,15 @@ import Foundation
 
 @objc public extension Firestore {
   @available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
-  @nonobjc func pipeline() -> PipelineSource<Pipeline> {
-    return PipelineSource<Pipeline>(db: self) { stages, db in
+  @nonobjc func pipeline() -> PipelineSource {
+    return PipelineSource(db: self) { stages, db in
       Pipeline(stages: stages, db: db)
     }
   }
 
   @available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
-  @nonobjc internal func realtimePipeline() -> PipelineSource<RealtimePipeline> {
-    return PipelineSource<RealtimePipeline>(db: self) { stages, db in
+  @nonobjc internal func realtimePipeline() -> RealtimePipelineSource {
+    return RealtimePipelineSource(db: self) { stages, db in
       RealtimePipeline(stages: stages, db: db)
     }
   }

Firestore/Swift/Source/SwiftAPI/Pipeline/PipelineSource.swift

@@ -13,44 +13,44 @@
 // limitations under the License.
 
 @available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
-public struct PipelineSource<P>: @unchecked Sendable {
+public struct PipelineSource: @unchecked Sendable {
   let db: Firestore
-  let factory: ([Stage], Firestore) -> P
+  let factory: ([Stage], Firestore) -> Pipeline
 
-  init(db: Firestore, factory: @escaping ([Stage], Firestore) -> P) {
+  init(db: Firestore, factory: @escaping ([Stage], Firestore) -> Pipeline) {
     self.db = db
     self.factory = factory
   }
 
-  public func collection(_ path: String) -> P {
+  public func collection(_ path: String) -> Pipeline {
     return factory([CollectionSource(collection: db.collection(path), db: db)], db)
   }
 
-  public func collection(_ coll: CollectionReference) -> P {
+  public func collection(_ coll: CollectionReference) -> Pipeline {
     return factory([CollectionSource(collection: coll, db: db)], db)
   }
 
-  public func collectionGroup(_ collectionId: String) -> P {
+  public func collectionGroup(_ collectionId: String) -> Pipeline {
     return factory(
       [CollectionGroupSource(collectionId: collectionId)],
       db
     )
   }
 
-  public func database() -> P {
+  public func database() -> Pipeline {
     return factory([DatabaseSource()], db)
   }
 
-  public func documents(_ docs: [DocumentReference]) -> P {
+  public func documents(_ docs: [DocumentReference]) -> Pipeline {
     return factory([DocumentsSource(docs: docs, db: db)], db)
   }
 
-  public func documents(_ paths: [String]) -> P {
+  public func documents(_ paths: [String]) -> Pipeline {
     let docs = paths.map { db.document($0) }
     return factory([DocumentsSource(docs: docs, db: db)], db)
   }
 
-  public func create(from query: Query) -> P {
+  public func create(from query: Query) -> Pipeline {
     let stageBridges = PipelineBridge.createStageBridges(from: query)
     let stages: [Stage] = stageBridges.map { bridge in
       switch bridge.name {

Firestore/Swift/Source/SwiftAPI/Pipeline/RealtimePipelineSource.swift

@@ -0,0 +1,48 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+@available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
+internal struct RealtimePipelineSource: @unchecked Sendable {
+  let db: Firestore
+  let factory: ([Stage], Firestore) -> RealtimePipeline
+
+  init(db: Firestore, factory: @escaping ([Stage], Firestore) -> RealtimePipeline) {
+    self.db = db
+    self.factory = factory
+  }
+
+  internal func collection(_ path: String) -> RealtimePipeline {
+    return factory([CollectionSource(collection: db.collection(path), db: db)], db)
+  }
+
+  internal func collection(_ coll: CollectionReference) -> RealtimePipeline {
+    return factory([CollectionSource(collection: coll, db: db)], db)
+  }
+
+  internal func collectionGroup(_ collectionId: String) -> RealtimePipeline {
+    return factory(
+      [CollectionGroupSource(collectionId: collectionId)],
+      db
+    )
+  }
+
+  internal func documents(_ docs: [DocumentReference]) -> RealtimePipeline {
+    return factory([DocumentsSource(docs: docs, db: db)], db)
+  }
+
+  internal func documents(_ paths: [String]) -> RealtimePipeline {
+    let docs = paths.map { db.document($0) }
+    return factory([DocumentsSource(docs: docs, db: db)], db)
+  }
+}