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

use indexmap::map::IndexMap;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;

use crate::auth::KrakenAuth;
// Structs/Enums
use super::{EndpointInfo, KrakenInput, MethodType};
use crate::api::asset::KAsset;

// Traits
use super::{Input, MutateInput, Output, UpdateInput};

/// Request builder for the Get Account Balance endpoint
pub struct KIAccountBalance {
    params: IndexMap<String, String>,
}

impl KIAccountBalance {
    /// Constructor returning a [KrakenInput] builder for the get trade balance endpoint.
    /// There are no inputs to this endpoint so finish() is called for you
    pub fn build() -> KrakenInput {
        let account_balance = KIAccountBalance {
            params: IndexMap::new(),
        };
        account_balance.finish()
    }

    fn with_nonce(self) -> Self {
        self.update_input("nonce", KrakenAuth::nonce())
    }
}

impl Input for KIAccountBalance {
    fn finish(self) -> KrakenInput {
        KrakenInput {
            info: EndpointInfo {
                methodtype: MethodType::Private,
                endpoint: String::from("Balance"),
            },
            params: Some(self.with_nonce().params),
        }
    }

    fn finish_clone(self) -> (KrakenInput, Self) {
        let newself = self.with_nonce();
        (
            KrakenInput {
                info: EndpointInfo {
                    methodtype: MethodType::Private,
                    endpoint: String::from("Balance"),
                },
                params: Some(newself.params.clone()),
            },
            newself,
        )
    }
}

impl MutateInput for KIAccountBalance {
    fn list_mut(&mut self) -> &mut IndexMap<String, String> {
        &mut self.params
    }
}

impl UpdateInput for KIAccountBalance {}

/// Response from the Get Account Balance endpoint
#[derive(Deserialize, Serialize, Debug)]
pub struct KOAccountBalance {
    /// Map with the asset as the key and the asset's current balance as the value
    #[serde(flatten)]
    pub balances: HashMap<KAsset, String>,
}

impl Output for KOAccountBalance {}