use crate::{
errors::MarketDataError,
models::IntradayCandlesResponse,
rest::client::RestClient,
};
pub struct CandlesRequestBuilder<'a> {
client: &'a RestClient,
symbol: Option<String>,
timeframe: Option<String>,
session: Option<String>,
}
impl<'a> CandlesRequestBuilder<'a> {
pub(crate) fn new(client: &'a RestClient) -> Self {
Self {
client,
symbol: None,
timeframe: None,
session: None,
}
}
pub fn symbol(mut self, symbol: &str) -> Self {
self.symbol = Some(symbol.to_string());
self
}
pub fn timeframe(mut self, timeframe: &str) -> Self {
self.timeframe = Some(timeframe.to_string());
self
}
pub fn after_hours(mut self) -> Self {
self.session = Some("afterhours".to_string());
self
}
pub fn send(self) -> Result<IntradayCandlesResponse, MarketDataError> {
let symbol = self.symbol.ok_or_else(|| MarketDataError::InvalidSymbol {
symbol: "(not provided)".to_string(),
})?;
let mut url = format!("{}/futopt/intraday/candles/{}", self.client.get_base_url(), symbol);
let mut query_params = Vec::new();
if let Some(timeframe) = &self.timeframe {
query_params.push(format!("timeframe={}", timeframe));
}
if let Some(session) = &self.session {
query_params.push(format!("session={}", session));
}
if !query_params.is_empty() {
url.push('?');
url.push_str(&query_params.join("&"));
}
let request = self.client.agent().get(&url);
let request = self.client.auth().apply_to_request(request);
let response = request.call()?;
let candles: IntradayCandlesResponse = response
.into_json()
.map_err(|e| MarketDataError::Other(e.into()))?;
Ok(candles)
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::rest::Auth;
#[test]
fn test_candles_builder_requires_symbol() {
let client = RestClient::new(Auth::SdkToken("test".to_string()));
let builder = CandlesRequestBuilder::new(&client);
let result = builder.send();
assert!(result.is_err());
assert!(matches!(result.unwrap_err(), MarketDataError::InvalidSymbol { .. }));
}
#[test]
fn test_candles_builder_with_params() {
let client = RestClient::new(Auth::SdkToken("test".to_string()));
let builder = CandlesRequestBuilder::new(&client)
.symbol("TXFC4")
.timeframe("5")
.after_hours();
assert_eq!(builder.symbol, Some("TXFC4".to_string()));
assert_eq!(builder.timeframe, Some("5".to_string()));
assert_eq!(builder.session, Some("afterhours".to_string()));
}
#[test]
fn test_candles_builder_timeframes() {
let client = RestClient::new(Auth::SdkToken("test".to_string()));
for tf in ["1", "5", "10", "15", "30", "60"] {
let builder = CandlesRequestBuilder::new(&client)
.symbol("TXFC4")
.timeframe(tf);
assert_eq!(builder.timeframe, Some(tf.to_string()));
}
}
}