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

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

use super::{
    EndpointInfo, Input, KAssetPair, KrakenInput, MethodType, MutateInput, OHLCInterval, Output,
    UpdateInput,
};

/// Request builder for the Get OHLC Data endpoint
pub struct KIOHLC {
    params: IndexMap<String, String>,
}

impl KIOHLC {
    /// Constructor returning a [KrakenInput] builder for the get OHLC data endpoint.
    /// * `pair` is the asset pair to query OHLC data for
    pub fn build(pair: KAssetPair) -> Self {
        let ohlc = KIOHLC {
            params: IndexMap::new(),
        };
        ohlc.update_pair(pair)
    }

    /// Update the asset pair to query OHLC data for
    pub fn update_pair(self, pair: KAssetPair) -> Self {
        self.update_input("pair", pair.to_string())
    }

    /// Return data from the given timestamp to the current timestamp, sampling data every
    /// `interval` minutes
    pub fn with_interval(self, interval: OHLCInterval) -> Self {
        self.update_input("interval", interval.to_string())
    }

    /// Unix timestamp to query OHLC data from. The [KOOHLC] member `last` can be used as input to
    /// `since()` to query OHLC data since the last time data was requested
    pub fn since(self, id: String) -> Self {
        self.update_input("since", id)
    }
}

impl Input for KIOHLC {
    fn finish(self) -> KrakenInput {
        KrakenInput {
            info: EndpointInfo {
                methodtype: MethodType::Public,
                endpoint: String::from("OHLC"),
            },
            params: Some(self.params),
        }
    }

    fn finish_clone(self) -> (KrakenInput, Self) {
        (
            KrakenInput {
                info: EndpointInfo {
                    methodtype: MethodType::Public,
                    endpoint: String::from("OHLC"),
                },
                params: Some(self.params.clone()),
            },
            self,
        )
    }
}

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

impl UpdateInput for KIOHLC {}

/// OHLC info data
#[derive(Deserialize, Serialize, Debug)]
pub struct KOOHLCData {
    pub timestamp: i64,
    pub open: String,
    pub high: String,
    pub low: String,
    pub close: String,
    pub vwap: String,
    pub volume: String,
    pub count: i64,
}

/// Response from the Get OHLC Data endpoint
#[derive(Deserialize, Serialize, Debug)]
pub struct KOOHLC {
    /// Map with the asset pair as the key and the pair's OHLC data as the value
    #[serde(flatten)]
    pub pair: HashMap<KAssetPair, Vec<KOOHLCData>>,
    /// ID to be used as "since" input to subsequent OHLC requests
    pub last: i64,
}

impl Output for KOOHLC {}