mod client;
pub mod models;
mod commodities;
mod core_stocks;
mod crypto;
mod economic_indicators;
mod forex;
mod fundamentals;
mod intelligence;
mod options;
mod technical_indicators;
use crate::error::{FinanceError, Result};
use crate::rate_limiter::RateLimiter;
use client::AlphaVantageClientBuilder;
use std::sync::{Arc, OnceLock};
use std::time::Duration;
pub use commodities::*;
pub use core_stocks::*;
pub use crypto::*;
pub use economic_indicators::*;
pub use forex::*;
pub use fundamentals::*;
pub use intelligence::*;
pub use options::*;
pub use technical_indicators::*;
pub use models::*;
const AV_RATE_PER_SEC: f64 = 1.0;
struct AlphaVantageSingleton {
api_key: String,
timeout: Duration,
limiter: Arc<RateLimiter>,
}
static AV_SINGLETON: OnceLock<AlphaVantageSingleton> = OnceLock::new();
pub fn init(api_key: impl Into<String>) -> Result<()> {
init_with_timeout(api_key, Duration::from_secs(30))
}
pub fn init_with_timeout(api_key: impl Into<String>, timeout: Duration) -> Result<()> {
AV_SINGLETON
.set(AlphaVantageSingleton {
api_key: api_key.into(),
timeout,
limiter: Arc::new(RateLimiter::new(AV_RATE_PER_SEC)),
})
.map_err(|_| FinanceError::InvalidParameter {
param: "alphavantage".to_string(),
reason: "Alpha Vantage client already initialized".to_string(),
})
}
pub(crate) fn build_client() -> Result<client::AlphaVantageClient> {
let s = AV_SINGLETON
.get()
.ok_or_else(|| FinanceError::InvalidParameter {
param: "alphavantage".to_string(),
reason: "Alpha Vantage not initialized. Call alphavantage::init(api_key) first."
.to_string(),
})?;
AlphaVantageClientBuilder::new(&s.api_key)
.timeout(s.timeout)
.build_with_limiter(Arc::clone(&s.limiter))
}
#[cfg(test)]
pub(crate) fn build_test_client(base_url: &str) -> Result<client::AlphaVantageClient> {
AlphaVantageClientBuilder::new("test-key")
.timeout(Duration::from_secs(5))
.base_url(base_url)
.build_with_limiter(Arc::new(RateLimiter::new(100.0)))
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_init_errors_on_double_init() {
let _ = init("test-key-1");
let result = init("test-key-2");
assert!(matches!(result, Err(FinanceError::InvalidParameter { .. })));
}
}