greed 0.6.1

A rust tool to automate trades 📈
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
118
119

use crate::asset::AssetSymbol;
use crate::error::GreedError;
use crate::platform::account::Account;
use crate::platform::bar::bar_request::BarRequest;
use crate::platform::bars::Bars;
use crate::platform::order::Order;
use crate::platform::position::Position;
use crate::platform::quote::Quote;
use crate::platform::request::OrderRequest;
use crate::platform::FinancialPlatform;
use async_trait::async_trait;
use std::sync::{Arc, Mutex};

pub struct MockPlatform {
    account: Account,
    positions: Vec<Position>,
    open_orders: Vec<Order>,
    recent_orders: Vec<Order>,
    quotes: Vec<Quote>,
    bars: Bars,
    place_order_result: Order,
    placed_orders: Mutex<Vec<OrderRequest>>,
}

impl Default for MockPlatform {
    fn default() -> Self {
        Self::new()
    }
}

impl MockPlatform {
    pub fn new() -> Self {
        Self {
            account: Account::default(),
            positions: Vec::new(),
            open_orders: Vec::new(),
            recent_orders: Vec::new(),
            quotes: Vec::new(),
            bars: Bars::default(),
            place_order_result: Order::default(),
            placed_orders: Mutex::new(Vec::new()),
        }
    }

    pub fn with_account(mut self, account: Account) -> Self {
        self.account = account;
        self
    }

    pub fn with_positions(mut self, positions: Vec<Position>) -> Self {
        self.positions = positions;
        self
    }

    pub fn with_open_orders(mut self, orders: Vec<Order>) -> Self {
        self.open_orders = orders;
        self
    }

    pub fn with_recent_orders(mut self, orders: Vec<Order>) -> Self {
        self.recent_orders = orders;
        self
    }

    pub fn with_quotes(mut self, quotes: Vec<Quote>) -> Self {
        self.quotes = quotes;
        self
    }

    pub fn with_bars(mut self, bars: Bars) -> Self {
        self.bars = bars;
        self
    }

    pub fn with_place_order_result(mut self, order: Order) -> Self {
        self.place_order_result = order;
        self
    }

    pub fn placed_orders(&self) -> Vec<OrderRequest> {
        self.placed_orders.lock().unwrap().clone()
    }

    pub fn arc(self) -> Arc<dyn FinancialPlatform> {
        Arc::new(self)
    }
}

#[async_trait]
impl FinancialPlatform for MockPlatform {
    async fn account(&self) -> Result<Account, GreedError> {
        Ok(self.account.clone())
    }

    async fn bars(&self, _bars_request: BarRequest) -> Result<Bars, GreedError> {
        Ok(self.bars.clone())
    }

    async fn latest_quotes(&self, _symbols: &[AssetSymbol]) -> Result<Vec<Quote>, GreedError> {
        Ok(self.quotes.clone())
    }

    async fn place_order(&self, order_request: OrderRequest) -> Result<Order, GreedError> {
        self.placed_orders.lock().unwrap().push(order_request);
        Ok(self.place_order_result.clone())
    }

    async fn positions(&self) -> Result<Vec<Position>, GreedError> {
        Ok(self.positions.clone())
    }

    async fn open_orders(&self) -> Result<Vec<Order>, GreedError> {
        Ok(self.open_orders.clone())
    }

    async fn recent_orders(&self) -> Result<Vec<Order>, GreedError> {
        Ok(self.recent_orders.clone())
    }
}