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

use super::types::ListCandlesResponse;
use crate::errors::{ProtocolError, Result};
use crate::graphql::list_candles;
use crate::types::{Candle, CandleInterval};
use chrono::{DateTime, Utc};
use std::str::FromStr;
use crate::protocol::traits::TryFromState;
use crate::protocol::state::State;
use std::sync::Arc;
use futures::lock::Mutex;
use async_trait::async_trait;


#[async_trait]
impl TryFromState<list_candles::ResponseData> for ListCandlesResponse {
    async fn from(response: list_candles::ResponseData, state: Arc<Mutex<State>>) -> Result<ListCandlesResponse> {
        let state = state.lock().await;
        let mut candles = Vec::new();
        // This unwrap is safe. ME_FIXME
        let market = state.get_market(&response.list_candles.market.unwrap().name)?;
        for candle_data in response.list_candles.candles {
            let a_volume = market.asset_a.with_amount(&candle_data.a_volume.amount)?;
            let b_volume = market.asset_b.with_amount(&candle_data.b_volume.amount)?;
            let low_price = market.asset_b.with_amount(&candle_data.low_price.amount)?;
            let open_price = market.asset_b.with_amount(&candle_data.open_price.amount)?;
            let close_price = market
                .asset_b
                .with_amount(&candle_data.close_price.amount)?;
            let high_price = market.asset_b.with_amount(&candle_data.high_price.amount)?;
            candles.push(Candle {
                a_volume,
                b_volume,
                high_price,
                low_price,
                open_price,
                close_price,
                interval: candle_data.interval.into(),
                interval_start: DateTime::<Utc>::from_str(&candle_data.interval_starting_at)
                    .map_err(|_| ProtocolError("Could not convert value to DateTime"))?,
            })
        }
        Ok(ListCandlesResponse {
            candles,
            next_page: response.list_candles.next.clone(),
        })
    }
}

impl From<list_candles::CandleInterval> for CandleInterval {
    fn from(interval: list_candles::CandleInterval) -> Self {
        match interval {
            list_candles::CandleInterval::FIFTEEN_MINUTE => CandleInterval::FifteenMinute,
            list_candles::CandleInterval::FIVE_MINUTE => CandleInterval::FiveMinute,
            list_candles::CandleInterval::FOUR_HOUR => CandleInterval::FourHour,
            list_candles::CandleInterval::ONE_DAY => CandleInterval::OneDay,
            list_candles::CandleInterval::ONE_HOUR => CandleInterval::OneHour,
            list_candles::CandleInterval::SIX_HOUR => CandleInterval::SixHour,
            list_candles::CandleInterval::THIRTY_MINUTE => CandleInterval::ThirtyMinute,
            list_candles::CandleInterval::THREE_HOUR => CandleInterval::ThreeHour,
            list_candles::CandleInterval::TWELVE_HOUR => CandleInterval::TwelveHour,
            list_candles::CandleInterval::ONE_MONTH => CandleInterval::OneMonth,
            list_candles::CandleInterval::ONE_WEEK => CandleInterval::OneWeek,
            list_candles::CandleInterval::ONE_MINUTE => CandleInterval::OneMinute,
            _ => panic!("Unsupported interval"),
        }
    }
}