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

use std::fmt;

use crate::{client::Client, error::Error};
use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize, PartialEq, Eq)]
pub enum OrderRequestStatus {
    SUCCESS,
    ERROR,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct StockOrderUnknownResponse {
    pub order_request_status: OrderRequestStatus,
    pub message: String,
    pub order_id: Option<String>,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct StockOrderResponse {
    pub message: String,
    pub order_id: String,
}

impl Into<StockOrderResponse> for StockOrderUnknownResponse {
    fn into(self) -> StockOrderResponse {
        StockOrderResponse {
            message: self.message,
            order_id: self.order_id.unwrap(),
        }
    }
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct StockOrder {
    pub orderbook_id: String,
    pub account_id: String,
    pub price: f64,
    pub volume: u32,
    pub side: Side,
}

#[derive(Debug, Serialize, Deserialize)]
pub enum Side {
    BUY,
    SELL,
}

impl fmt::Display for Side {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match *self {
            Side::BUY => write!(f, "BUY"),
            Side::SELL => write!(f, "SELL"),
        }
    }
}

impl Client {
    pub async fn execute_stock_order(
        &self,
        order: &StockOrder,
    ) -> Result<StockOrderResponse, Error> {
        let res = self
            .http_client
            .post(&self.config.urls.stock_order)
            .body_json(&order)?
            .recv_json::<StockOrderUnknownResponse>()
            .await?;

        if res.order_request_status == OrderRequestStatus::ERROR {
            return Err(Error::StockOrderError(format!(
                "Stock order failed with message: '{}'",
                res.message
            )));
        }

        let response: StockOrderResponse = res.into();

        return Ok(response);
    }
}