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

use key_path::path;
use teo_teon::Value;
use crate::connection::transaction;
use crate::model;
use crate::model::Model;

#[derive(Clone)]
pub struct Ctx {
    pub transaction_ctx: transaction::Ctx,
    pub model: &'static Model,
}

impl Ctx {

    pub async fn find_unique<T: From<model::Object>>(&self, finder: &Value) -> crate::path::Result<Option<T>> {
        self.transaction_ctx.find_unique(self.model, finder, None, path![]).await
    }

    pub async fn find_first<T: From<model::Object>>(&self, finder: &Value) -> crate::path::Result<Option<T>> {
        self.transaction_ctx.find_first(self.model, finder, None, path![]).await
    }

    pub async fn find_many<T: From<model::Object>>(&self, finder: &Value) -> crate::path::Result<Vec<T>> {
        self.transaction_ctx.find_many(self.model, finder, None, path![]).await
    }

    pub async fn count(&self, finder: &Value) -> crate::path::Result<usize> {
        self.transaction_ctx.count(self.model, finder, path![]).await
    }

    pub async fn aggregate<T, E>(&self, finder: &Value) -> crate::path::Result<T> where T: TryFrom<Value, Error=E>, crate::path::Error: From<E> {
        Ok(self.transaction_ctx.aggregate(self.model, finder, path![]).await?.try_into()?)
    }

    pub async fn group_by<T, E>(&self, finder: &Value) -> crate::path::Result<Vec<T>> where T: TryFrom<Value, Error=E>, crate::path::Error: From<E> {
        Ok(self.transaction_ctx.group_by(self.model, finder, path![]).await?.into_iter().map(|t| T::try_from(t)).collect::<Result<Vec<T>, E>>()?)
    }

    pub async fn create_object<T>(&self, input: &Value) -> crate::path::Result<T> where T: From<model::Object> {
        Ok(self.transaction_ctx.create_object(self.model, input, None).await?.into())
    }
}