firebase-rust-sdk 0.1.0-beta

Unofficial Rust port of Firebase C++ SDK
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

//! Firestore CollectionReference type
//!
//! # C++ Reference
//! - `firestore/src/include/firebase/firestore/collection_reference.h:38`

use super::document_reference::DocumentReference;
use super::field_value::MapValue;
use super::query::{Query, QueryState};
use crate::error::FirebaseError;
use crate::firestore::firestore::FirestoreInner;

/// Reference to a Firestore collection
///
/// Implements the Query trait to support filtering, ordering, and pagination.
/// Following C++ SDK pattern where CollectionReference inherits from Query.
///
/// # C++ Reference
/// - `firebase-ios-sdk/Firestore/core/src/api/collection_reference.h:38`
/// - `query.h:61` - CollectionReference inherits from Query in C++
#[derive(Clone)]
pub struct CollectionReference {
    /// Internal query state
    pub(crate) state: QueryState,
}

impl CollectionReference {
    /// Create a new collection reference
    ///
    /// # C++ Reference
    /// - `firebase-ios-sdk/Firestore/core/src/api/collection_reference.cc:28`
    pub(crate) fn new(path: impl Into<String>, firestore: std::sync::Arc<FirestoreInner>) -> Self {
        Self { 
            state: QueryState::new(path.into(), firestore),
        }
    }

    /// Get collection ID (last segment of path)
    pub fn id(&self) -> &str {
        self.state.collection_path.rsplit('/').next().unwrap_or(&self.state.collection_path)
    }

    /// Get a document reference within this collection
    ///
    /// # C++ Reference
    /// - `firebase-ios-sdk/Firestore/core/src/api/collection_reference.cc:41` - Document()
    pub fn document(&self, document_id: impl AsRef<str>) -> DocumentReference {
        let path = format!("{}/{}", self.state.collection_path, document_id.as_ref());
        DocumentReference::new(path, std::sync::Arc::clone(&self.state.firestore))
    }

    /// Add a new document with auto-generated ID
    ///
    /// # C++ Reference
    /// - `firebase-ios-sdk/Firestore/core/src/api/collection_reference.cc:46` - AddDocument()
    pub async fn add(&self, data: MapValue) -> Result<DocumentReference, FirebaseError> {
        // Generate auto ID
        use rand::Rng;
        let auto_id: String = rand::thread_rng()
            .sample_iter(&rand::distributions::Alphanumeric)
            .take(20)
            .map(char::from)
            .collect();
        
        let doc_ref = self.document(&auto_id);
        doc_ref.set(data).await?;
        Ok(doc_ref)
    }
}

/// Implement Query trait for CollectionReference
///
/// Following C++ SDK pattern where CollectionReference inherits from Query.
/// All query methods return new CollectionReference instances with modified state.
///
/// # C++ Reference
/// - `query.h:61` - CollectionReference inherits from Query
/// - `collection_reference_main.h` - CollectionReferenceInternal inherits QueryInternal
#[allow(private_interfaces)]
impl Query for CollectionReference {
    fn query_state(&self) -> &QueryState {
        &self.state
    }

    fn with_state(&self, state: QueryState) -> Self {
        Self { state }
    }
}