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

#![allow(clippy::vec_init_then_push, clippy::branches_sharing_code)]

mod constants;
mod cursor;
mod error;
mod filter;
mod interface;
mod join;
mod orderby;
mod output_meta;
mod projection;
mod query_builder;
mod query_strings;
mod root_queries;
mod value;

use error::MongoError;
use mongodb::{
    bson::{Bson, Document},
    ClientSession, SessionCursor,
};

pub use interface::*;

type Result<T> = std::result::Result<T, MongoError>;

trait IntoBson {
    fn into_bson(self) -> Result<Bson>;
}

trait BsonTransform {
    fn into_document(self) -> Result<Document>;
}

impl BsonTransform for Bson {
    fn into_document(self) -> Result<Document> {
        if let Bson::Document(doc) = self {
            Ok(doc)
        } else {
            Err(MongoError::ConversionError {
                from: format!("{self:?}"),
                to: "Bson::Document".to_string(),
            })
        }
    }
}

// Todo: Move to approriate place
/// Consumes a cursor stream until exhausted.
async fn vacuum_cursor(
    mut cursor: SessionCursor<Document>,
    session: &mut ClientSession,
) -> crate::Result<Vec<Document>> {
    let mut docs = vec![];

    while let Some(result) = cursor.next(session).await {
        match result {
            Ok(document) => docs.push(document),
            Err(e) => return Err(e.into()),
        }
    }

    Ok(docs)
}