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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use std::collections::HashMap;
use cynic::http::{CynicReqwestError};

use serde_json::json;
use crate::api::classes::LevelSpellcasting;
use crate::Character;

#[derive(Debug, thiserror::Error)]
pub enum ApiError {
    #[error("Reqwest error: {0}")]
    Reqwest(#[from] CynicReqwestError),
    #[error("Schema error")]
    Schema,
}

//noinspection RsCompileErrorMacro
#[cynic::schema("dnd5eapi")]
pub(super) mod schema {}

#[derive(Debug)]
pub enum CheckError{
    InvalidRace,
    InvalidClass,
    InvalidBackground,
    InvalidAlignment,
    InvalidAbilities
}

mod race_query {
    use cynic::http::ReqwestExt;
    use reqwest::Client;
    use crate::api::shared::ApiError;
    use cynic::QueryBuilder;
    use crate::Character;
    use super::schema;

    #[derive(cynic::QueryVariables, Debug)]
    struct SpeedQueryVariables {
        pub index: String,
    }

    #[derive(cynic::QueryFragment, Debug)]
    #[cynic(graphql_type = "Query", variables = "SpeedQueryVariables")]
    struct SpeedQuery {
        #[arguments(index: $index)]
        pub race: Option<RaceSpeed>,
    }

    #[derive(cynic::QueryFragment, Debug)]
    #[cynic(graphql_type = "Race")]
    struct RaceSpeed {
        pub speed: i32,
    }

    impl Character {
        pub async fn get_base_speed(&self) -> Result<i32, ApiError> {
            let op = SpeedQuery::build(SpeedQueryVariables {
                index: self.race_index.clone()
            });

            let speed = Client::new()
                .post("https://www.dnd5eapi.co/graphql")
                .run_graphql(op).await?
                .data.ok_or(ApiError::Schema)?
                .race.ok_or(ApiError::Schema)?
                .speed;

            Ok(speed)
        }
    }
}


impl Character {
    pub async fn get_spellcasting_slots(&self) -> Result<HashMap<String, LevelSpellcasting>, ApiError> {
        let mut spellcasting_slots = HashMap::new();
        for class in self.classes.0.iter() {
            let spellcasting_slots_class = class.1.get_spellcasting_slots().await?;
            if let Some(spellcasting_slots_class) = spellcasting_slots_class {
                spellcasting_slots.insert(class.0.clone(), spellcasting_slots_class);
            }
        }
        Ok(spellcasting_slots)
    }

    pub async fn get_features(&self, passive: bool) -> Result<Vec<String>, ApiError> {
        let mut features = Vec::new();
        for class in self.classes.0.iter() {
            let features_class = class.1.get_levels_features(None, passive).await?;
            features.extend(features_class);
        }
        Ok(features)
    }

    #[cfg(feature = "serde")]
    pub async fn rich_print(&self) -> Result<String, ApiError> {
        let spellcasting_slots = self.get_spellcasting_slots().await?;
        let features = self.get_features(true).await?;

        let mut character = json!(self);

        if !spellcasting_slots.is_empty() {
            character["spellcasting_slots"] = json!(spellcasting_slots);
        }

        if !features.is_empty() {
            character["features"] = json!(features);
        }

        Ok(character.to_string())
    }

    /// Call this method every day to reset daily vars
    pub async fn new_day(&mut self) {
        self.classes.new_day().await;
    }
}