94 lines
2.6 KiB
Rust
94 lines
2.6 KiB
Rust
|
use std::convert::{TryFrom, TryInto};
|
||
|
|
|
||
|
|
use async_trait::async_trait;
|
||
|
|
use bitfinex::api::Bitfinex;
|
||
|
|
use bitfinex::ticker::TradingPairTicker;
|
||
|
|
|
||
|
|
use crate::currency::{Symbol, SymbolPair};
|
||
|
|
use crate::orders::Order;
|
||
|
|
use crate::positions::{Position, PositionState};
|
||
|
|
use crate::BoxError;
|
||
|
|
|
||
|
|
#[async_trait]
|
||
|
|
pub trait Connector {
|
||
|
|
async fn active_positions(&self, pair: &SymbolPair) -> Result<Vec<Position>, BoxError>;
|
||
|
|
async fn current_prices(&self, pair: &SymbolPair) -> Result<TradingPairTicker, BoxError>;
|
||
|
|
async fn active_orders(&self, pair: &SymbolPair) -> Result<Vec<Order>, BoxError>;
|
||
|
|
async fn submit_order(&self) -> Result<u64, BoxError>;
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct BfxWrapper {
|
||
|
|
bfx: Bitfinex,
|
||
|
|
affiliate_code: Option<String>,
|
||
|
|
// account_info: String,
|
||
|
|
// ledger: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl BfxWrapper {
|
||
|
|
pub fn new(api_key: &str, api_secret: &str) -> Self {
|
||
|
|
BfxWrapper {
|
||
|
|
bfx: Bitfinex::new(Some(api_key.into()), Some(api_secret.into())),
|
||
|
|
affiliate_code: None,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn with_affiliate_code(mut self, affiliate_code: Option<String>) -> Self {
|
||
|
|
self.affiliate_code = affiliate_code;
|
||
|
|
self
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl TryInto<Position> for bitfinex::positions::Position {
|
||
|
|
type Error = BoxError;
|
||
|
|
|
||
|
|
fn try_into(self) -> Result<Position, Self::Error> {
|
||
|
|
let state = {
|
||
|
|
if self.status().to_lowercase().contains("active") {
|
||
|
|
PositionState::Open
|
||
|
|
} else {
|
||
|
|
PositionState::Closed
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
Ok(Position::new(
|
||
|
|
self.symbol().try_into()?,
|
||
|
|
state,
|
||
|
|
self.amount(),
|
||
|
|
self.base_price(),
|
||
|
|
self.pl(),
|
||
|
|
self.pl_perc(),
|
||
|
|
self.price_liq(),
|
||
|
|
self.position_id(),
|
||
|
|
)
|
||
|
|
.with_creation_date(self.mts_create())
|
||
|
|
.with_creation_update(self.mts_update()))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[async_trait]
|
||
|
|
impl Connector for BfxWrapper {
|
||
|
|
async fn active_positions(&self, pair: &SymbolPair) -> Result<Vec<Position>, BoxError> {
|
||
|
|
let active_positions = self.bfx.positions.active_positions().await?;
|
||
|
|
|
||
|
|
Ok(active_positions
|
||
|
|
.into_iter()
|
||
|
|
.filter_map(|x| x.try_into().ok())
|
||
|
|
.filter(|x: &Position| x.pair() == pair)
|
||
|
|
.collect())
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn current_prices(&self, pair: &SymbolPair) -> Result<TradingPairTicker, BoxError> {
|
||
|
|
let ticker: TradingPairTicker = self.bfx.ticker.trading_pair(pair.clone()).await?;
|
||
|
|
|
||
|
|
Ok(ticker)
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn active_orders(&self, pair: &SymbolPair) -> Result<Vec<Order>, BoxError> {
|
||
|
|
unimplemented!()
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn submit_order(&self) -> Result<u64, BoxError> {
|
||
|
|
unimplemented!()
|
||
|
|
}
|
||
|
|
}
|