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

use serde_json::json;

use crate::error::TritiumError;
use crate::graphql::{GenericResponse, GenericSubscription};
use crate::tritium::Tritium;

pub type Variables = serde_json::Value;

impl Tritium {
    pub async fn query(
        &mut self,
        document: &str,
        variables: Option<Variables>,
    ) -> Result<GenericResponse, TritiumError> {
        let variables = variables.unwrap_or_else(|| json!(null));

        let query = self
            .client
            .generic_graphql_query(document, variables)
            .await?;
        let response = query.result.await?;

        Ok(response)
    }

    pub async fn subscription(
        &mut self,
        document: &str,
        variables: Option<Variables>,
    ) -> Result<GenericSubscription, TritiumError> {
        let variables = variables.unwrap_or_else(|| json!(null));

        let subscription = self
            .client
            .generic_graphql_subscription(document, variables)
            .await?;
        let sub = subscription.result.await?;

        Ok(sub)
    }
}