ReferenceExtensions.swift

import FirebaseFirestore
import FirebaseFirestoreSwift
import Foundation

// MARK: - CollectionReference

public extension CollectionReference {
    @discardableResult
    func add<T: Encodable>(
        _ value: T,
        encoder: Firestore.Encoder = Firestore.Encoder()
    ) async throws -> DocumentReference {
        try await addDocument(data: try encoder.encode(value))
    }
}

// MARK: - DocumentReference

public extension DocumentReference {
    func update<T: Encodable>(
        _ value: T,
        encoder: Firestore.Encoder = Firestore.Encoder()
    ) async throws {
        try await updateData(try encoder.encode(value))
    }

    func set<T: Encodable>(
        _ value: T,
        encoder: Firestore.Encoder = Firestore.Encoder()
    ) async throws {
        try await setData(try encoder.encode(value))
    }
}