#![allow(unused_imports)]
use async_trait::async_trait;
use derive_builder::Builder;
use reqwest;
use rust_decimal::prelude::*;
use serde::{Deserialize, Serialize};
use serde_json::{Value, json};
use std::collections::BTreeMap;
use crate::common::{
config::ConfigurationRestApi,
models::{ParamBuildError, RestApiResponse},
utils::send_request,
};
use crate::derivatives_trading_options::rest_api::models;
const HAS_TIME_UNIT: bool = false;
#[async_trait]
pub trait AccountApi: Send + Sync {
async fn account_funding_flow(
&self,
params: AccountFundingFlowParams,
) -> anyhow::Result<RestApiResponse<Vec<models::AccountFundingFlowResponseInner>>>;
async fn option_margin_account_information(
&self,
params: OptionMarginAccountInformationParams,
) -> anyhow::Result<RestApiResponse<models::OptionMarginAccountInformationResponse>>;
}
#[derive(Debug, Clone)]
pub struct AccountApiClient {
configuration: ConfigurationRestApi,
}
impl AccountApiClient {
pub fn new(configuration: ConfigurationRestApi) -> Self {
Self { configuration }
}
}
#[derive(Clone, Debug, Builder)]
#[builder(pattern = "owned", build_fn(error = "ParamBuildError"))]
pub struct AccountFundingFlowParams {
#[builder(setter(into))]
pub currency: String,
#[builder(setter(into), default)]
pub record_id: Option<i64>,
#[builder(setter(into), default)]
pub start_time: Option<i64>,
#[builder(setter(into), default)]
pub end_time: Option<i64>,
#[builder(setter(into), default)]
pub limit: Option<i64>,
#[builder(setter(into), default)]
pub recv_window: Option<i64>,
}
impl AccountFundingFlowParams {
#[must_use]
pub fn builder(currency: String) -> AccountFundingFlowParamsBuilder {
AccountFundingFlowParamsBuilder::default().currency(currency)
}
}
#[derive(Clone, Debug, Builder, Default)]
#[builder(pattern = "owned", build_fn(error = "ParamBuildError"))]
pub struct OptionMarginAccountInformationParams {
#[builder(setter(into), default)]
pub recv_window: Option<i64>,
}
impl OptionMarginAccountInformationParams {
#[must_use]
pub fn builder() -> OptionMarginAccountInformationParamsBuilder {
OptionMarginAccountInformationParamsBuilder::default()
}
}
#[async_trait]
impl AccountApi for AccountApiClient {
async fn account_funding_flow(
&self,
params: AccountFundingFlowParams,
) -> anyhow::Result<RestApiResponse<Vec<models::AccountFundingFlowResponseInner>>> {
let AccountFundingFlowParams {
currency,
record_id,
start_time,
end_time,
limit,
recv_window,
} = params;
let mut query_params = BTreeMap::new();
let body_params = BTreeMap::new();
query_params.insert("currency".to_string(), json!(currency));
if let Some(rw) = record_id {
query_params.insert("recordId".to_string(), json!(rw));
}
if let Some(rw) = start_time {
query_params.insert("startTime".to_string(), json!(rw));
}
if let Some(rw) = end_time {
query_params.insert("endTime".to_string(), json!(rw));
}
if let Some(rw) = limit {
query_params.insert("limit".to_string(), json!(rw));
}
if let Some(rw) = recv_window {
query_params.insert("recvWindow".to_string(), json!(rw));
}
send_request::<Vec<models::AccountFundingFlowResponseInner>>(
&self.configuration,
"/eapi/v1/bill",
reqwest::Method::GET,
query_params,
body_params,
if HAS_TIME_UNIT {
self.configuration.time_unit
} else {
None
},
true,
)
.await
}
async fn option_margin_account_information(
&self,
params: OptionMarginAccountInformationParams,
) -> anyhow::Result<RestApiResponse<models::OptionMarginAccountInformationResponse>> {
let OptionMarginAccountInformationParams { recv_window } = params;
let mut query_params = BTreeMap::new();
let body_params = BTreeMap::new();
if let Some(rw) = recv_window {
query_params.insert("recvWindow".to_string(), json!(rw));
}
send_request::<models::OptionMarginAccountInformationResponse>(
&self.configuration,
"/eapi/v1/marginAccount",
reqwest::Method::GET,
query_params,
body_params,
if HAS_TIME_UNIT {
self.configuration.time_unit
} else {
None
},
true,
)
.await
}
}
#[cfg(all(test, feature = "derivatives_trading_options"))]
mod tests {
use super::*;
use crate::TOKIO_SHARED_RT;
use crate::{errors::ConnectorError, models::DataFuture, models::RestApiRateLimit};
use async_trait::async_trait;
use std::collections::HashMap;
struct DummyRestApiResponse<T> {
inner: Box<dyn FnOnce() -> DataFuture<Result<T, ConnectorError>> + Send + Sync>,
status: u16,
headers: HashMap<String, String>,
rate_limits: Option<Vec<RestApiRateLimit>>,
}
impl<T> From<DummyRestApiResponse<T>> for RestApiResponse<T> {
fn from(dummy: DummyRestApiResponse<T>) -> Self {
Self {
data_fn: dummy.inner,
status: dummy.status,
headers: dummy.headers,
rate_limits: dummy.rate_limits,
}
}
}
struct MockAccountApiClient {
force_error: bool,
}
#[async_trait]
impl AccountApi for MockAccountApiClient {
async fn account_funding_flow(
&self,
_params: AccountFundingFlowParams,
) -> anyhow::Result<RestApiResponse<Vec<models::AccountFundingFlowResponseInner>>> {
if self.force_error {
return Err(ConnectorError::ConnectorClientError {
msg: "ResponseError".to_string(),
code: None,
}
.into());
}
let resp_json: Value = serde_json::from_str(r#"[{"id":1125899906842624000,"asset":"USDT","amount":"-0.552","type":"FEE","createDate":1592449456000},{"id":1125899906842624000,"asset":"USDT","amount":"100","type":"CONTRACT","createDate":1592449456000},{"id":1125899906842624000,"asset":"USDT","amount":"10000","type":"TRANSFER","createDate":1592448410000}]"#).unwrap();
let dummy_response: Vec<models::AccountFundingFlowResponseInner> =
serde_json::from_value(resp_json.clone())
.expect("should parse into Vec<models::AccountFundingFlowResponseInner>");
let dummy = DummyRestApiResponse {
inner: Box::new(move || Box::pin(async move { Ok(dummy_response) })),
status: 200,
headers: HashMap::new(),
rate_limits: None,
};
Ok(dummy.into())
}
async fn option_margin_account_information(
&self,
_params: OptionMarginAccountInformationParams,
) -> anyhow::Result<RestApiResponse<models::OptionMarginAccountInformationResponse>>
{
if self.force_error {
return Err(ConnectorError::ConnectorClientError {
msg: "ResponseError".to_string(),
code: None,
}
.into());
}
let resp_json: Value = serde_json::from_str(r#"{"asset":[{"asset":"USDT","marginBalance":"99998.87365244","equity":"99998.87365244","available":"96883.72734374","initialMargin":"3115.14630870","maintMargin":"0.00000000","unrealizedPNL":"0.00000000","adjustedEquity":"99998.87365244"}],"greek":[{"underlying":"BTCUSDT","delta":"0","theta":"0","gamma":"0","vega":"0"}],"time":1762843368098,"canTrade":true,"canDeposit":true,"canWithdraw":true,"reduceOnly":false,"tradeGroupId":-1}"#).unwrap();
let dummy_response: models::OptionMarginAccountInformationResponse =
serde_json::from_value(resp_json.clone())
.expect("should parse into models::OptionMarginAccountInformationResponse");
let dummy = DummyRestApiResponse {
inner: Box::new(move || Box::pin(async move { Ok(dummy_response) })),
status: 200,
headers: HashMap::new(),
rate_limits: None,
};
Ok(dummy.into())
}
}
#[test]
fn account_funding_flow_required_params_success() {
TOKIO_SHARED_RT.block_on(async {
let client = MockAccountApiClient { force_error: false };
let params = AccountFundingFlowParams::builder("currency_example".to_string(),).build().unwrap();
let resp_json: Value = serde_json::from_str(r#"[{"id":1125899906842624000,"asset":"USDT","amount":"-0.552","type":"FEE","createDate":1592449456000},{"id":1125899906842624000,"asset":"USDT","amount":"100","type":"CONTRACT","createDate":1592449456000},{"id":1125899906842624000,"asset":"USDT","amount":"10000","type":"TRANSFER","createDate":1592448410000}]"#).unwrap();
let expected_response : Vec<models::AccountFundingFlowResponseInner> = serde_json::from_value(resp_json.clone()).expect("should parse into Vec<models::AccountFundingFlowResponseInner>");
let resp = client.account_funding_flow(params).await.expect("Expected a response");
let data_future = resp.data();
let actual_response = data_future.await.unwrap();
assert_eq!(actual_response, expected_response);
});
}
#[test]
fn account_funding_flow_optional_params_success() {
TOKIO_SHARED_RT.block_on(async {
let client = MockAccountApiClient { force_error: false };
let params = AccountFundingFlowParams::builder("currency_example".to_string(),).record_id(1).start_time(1623319461670).end_time(1641782889000).limit(100).recv_window(5000).build().unwrap();
let resp_json: Value = serde_json::from_str(r#"[{"id":1125899906842624000,"asset":"USDT","amount":"-0.552","type":"FEE","createDate":1592449456000},{"id":1125899906842624000,"asset":"USDT","amount":"100","type":"CONTRACT","createDate":1592449456000},{"id":1125899906842624000,"asset":"USDT","amount":"10000","type":"TRANSFER","createDate":1592448410000}]"#).unwrap();
let expected_response : Vec<models::AccountFundingFlowResponseInner> = serde_json::from_value(resp_json.clone()).expect("should parse into Vec<models::AccountFundingFlowResponseInner>");
let resp = client.account_funding_flow(params).await.expect("Expected a response");
let data_future = resp.data();
let actual_response = data_future.await.unwrap();
assert_eq!(actual_response, expected_response);
});
}
#[test]
fn account_funding_flow_response_error() {
TOKIO_SHARED_RT.block_on(async {
let client = MockAccountApiClient { force_error: true };
let params = AccountFundingFlowParams::builder("currency_example".to_string())
.build()
.unwrap();
match client.account_funding_flow(params).await {
Ok(_) => panic!("Expected an error"),
Err(err) => {
assert_eq!(err.to_string(), "Connector client error: ResponseError");
}
}
});
}
#[test]
fn option_margin_account_information_required_params_success() {
TOKIO_SHARED_RT.block_on(async {
let client = MockAccountApiClient { force_error: false };
let params = OptionMarginAccountInformationParams::builder().build().unwrap();
let resp_json: Value = serde_json::from_str(r#"{"asset":[{"asset":"USDT","marginBalance":"99998.87365244","equity":"99998.87365244","available":"96883.72734374","initialMargin":"3115.14630870","maintMargin":"0.00000000","unrealizedPNL":"0.00000000","adjustedEquity":"99998.87365244"}],"greek":[{"underlying":"BTCUSDT","delta":"0","theta":"0","gamma":"0","vega":"0"}],"time":1762843368098,"canTrade":true,"canDeposit":true,"canWithdraw":true,"reduceOnly":false,"tradeGroupId":-1}"#).unwrap();
let expected_response : models::OptionMarginAccountInformationResponse = serde_json::from_value(resp_json.clone()).expect("should parse into models::OptionMarginAccountInformationResponse");
let resp = client.option_margin_account_information(params).await.expect("Expected a response");
let data_future = resp.data();
let actual_response = data_future.await.unwrap();
assert_eq!(actual_response, expected_response);
});
}
#[test]
fn option_margin_account_information_optional_params_success() {
TOKIO_SHARED_RT.block_on(async {
let client = MockAccountApiClient { force_error: false };
let params = OptionMarginAccountInformationParams::builder().recv_window(5000).build().unwrap();
let resp_json: Value = serde_json::from_str(r#"{"asset":[{"asset":"USDT","marginBalance":"99998.87365244","equity":"99998.87365244","available":"96883.72734374","initialMargin":"3115.14630870","maintMargin":"0.00000000","unrealizedPNL":"0.00000000","adjustedEquity":"99998.87365244"}],"greek":[{"underlying":"BTCUSDT","delta":"0","theta":"0","gamma":"0","vega":"0"}],"time":1762843368098,"canTrade":true,"canDeposit":true,"canWithdraw":true,"reduceOnly":false,"tradeGroupId":-1}"#).unwrap();
let expected_response : models::OptionMarginAccountInformationResponse = serde_json::from_value(resp_json.clone()).expect("should parse into models::OptionMarginAccountInformationResponse");
let resp = client.option_margin_account_information(params).await.expect("Expected a response");
let data_future = resp.data();
let actual_response = data_future.await.unwrap();
assert_eq!(actual_response, expected_response);
});
}
#[test]
fn option_margin_account_information_response_error() {
TOKIO_SHARED_RT.block_on(async {
let client = MockAccountApiClient { force_error: true };
let params = OptionMarginAccountInformationParams::builder()
.build()
.unwrap();
match client.option_margin_account_information(params).await {
Ok(_) => panic!("Expected an error"),
Err(err) => {
assert_eq!(err.to_string(), "Connector client error: ResponseError");
}
}
});
}
}