Skip to content

Commit a8b0f8c

Browse files
committed
[Feat] Add Firestore casting helers in FirestoreReference
1 parent 58f3dd6 commit a8b0f8c

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

Sources/SHFirestoreService/FirestoreReference.swift

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,4 +11,15 @@ import FirebaseFirestore
1111
public protocol FirestoreReference { }
1212
extension DocumentReference: FirestoreReference { }
1313
extension CollectionReference: FirestoreReference { }
14+
15+
// MARK: - Helpers
16+
extension FirestoreReference {
17+
var asCollectionRef: CollectionReference? {
18+
return self as? CollectionReference
19+
}
20+
21+
var asDocumentRef: DocumentReference? {
22+
return self as? DocumentReference
23+
}
24+
}
1425
#endif

0 commit comments

Comments
 (0)