finance-query 3.0.0

//! News endpoints: stock news, FMP articles, press releases, crypto news, forex news.

use serde::{Deserialize, Serialize};

use crate::error::Result;

use crate::adapters::fmp::build_client;

// ============================================================================
// Response types
// ============================================================================

/// Stock news article.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[non_exhaustive]
pub struct StockNewsDTO {
    /// Ticker symbol.
    pub symbol: Option<String>,
    /// Published date.
    #[serde(rename = "publishedDate")]
    pub published_date: Option<String>,
    /// Article title.
    pub title: Option<String>,
    /// Article image URL.
    pub image: Option<String>,
    /// News site name.
    pub site: Option<String>,
    /// Publisher of the article.
    pub publisher: Option<String>,
    /// Article text / summary.
    pub text: Option<String>,
    /// Article URL.
    pub url: Option<String>,
}

/// Press release.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[non_exhaustive]
pub struct PressReleaseDTO {
    /// Ticker symbol.
    pub symbol: Option<String>,
    /// Publication date.
    #[serde(rename = "publishedDate")]
    pub date: Option<String>,
    /// Title.
    pub title: Option<String>,
    /// Full text.
    pub text: Option<String>,
    /// Wire service that carried the release.
    pub publisher: Option<String>,
    /// Release URL.
    pub url: Option<String>,
}

// ============================================================================
// Canonical conversion functions
// ============================================================================

/// Convert stock news DTOs into canonical News items.
fn stock_news_to_canonical(
    articles: Vec<StockNewsDTO>,
) -> Vec<crate::models::corporate::news::News> {
    articles
        .into_iter()
        .map(|a| crate::models::corporate::news::News {
            title: a.title.unwrap_or_default(),
            link: a.url.unwrap_or_default(),
            source: a.site.or(a.publisher).unwrap_or_default(),
            img: a.image.unwrap_or_default(),
            time: a.published_date.unwrap_or_default(),
            provider_id: Some(crate::providers::Provider::Fmp),
            #[cfg(feature = "sentiment")]
            sentiment: None,
        })
        .collect()
}

/// Fetch canonical news for a symbol.
pub async fn fetch_canonical_news(
    symbol: &str,
    limit: u32,
) -> Result<Vec<crate::models::corporate::news::News>> {
    let articles = stock_news(symbol, limit).await?;
    Ok(stock_news_to_canonical(articles))
}

// ============================================================================
// Public API
// ============================================================================

/// Fetch stock news articles.
///
/// * `tickers` - Comma-separated ticker symbols (e.g., `"AAPL,MSFT"`)
/// * `limit` - Number of results
pub async fn stock_news(tickers: &str, limit: u32) -> Result<Vec<StockNewsDTO>> {
    let client = build_client()?;
    let limit_str = limit.to_string();
    client
        .get(
            "/stable/news/stock",
            &[("symbols", tickers), ("limit", &limit_str)],
        )
        .await
}

/// Fetch press releases for a symbol.
pub async fn press_releases(symbol: &str, limit: u32) -> Result<Vec<PressReleaseDTO>> {
    let client = build_client()?;
    let limit_str = limit.to_string();
    client
        .get(
            "/stable/news/press-releases",
            &[("symbols", symbol), ("limit", &limit_str)],
        )
        .await
}

/// Fetch crypto news.
pub async fn crypto_news(limit: u32) -> Result<Vec<StockNewsDTO>> {
    let client = build_client()?;
    let limit_str = limit.to_string();
    client
        .get(
            "/stable/news/crypto-latest",
            &[("page", "0"), ("limit", &limit_str)],
        )
        .await
}

/// Fetch forex news.
pub async fn forex_news(limit: u32) -> Result<Vec<StockNewsDTO>> {
    let client = build_client()?;
    let limit_str = limit.to_string();
    client
        .get(
            "/stable/news/forex-latest",
            &[("page", "0"), ("limit", &limit_str)],
        )
        .await
}

/// Fetch canonical crypto news.
pub async fn fetch_crypto_news_response(
    limit: u32,
) -> Result<Vec<crate::models::corporate::news::News>> {
    Ok(stock_news_to_canonical(crypto_news(limit).await?))
}

/// Fetch canonical forex news.
pub async fn fetch_forex_news_response(
    limit: u32,
) -> Result<Vec<crate::models::corporate::news::News>> {
    Ok(stock_news_to_canonical(forex_news(limit).await?))
}

/// Fetch canonical press releases for a symbol.
pub async fn fetch_press_releases_response(
    symbol: &str,
    limit: u32,
) -> Result<Vec<crate::models::corporate::press_release::PressRelease>> {
    let dtos = press_releases(symbol, limit).await?;
    Ok(dtos
        .into_iter()
        .map(|d| crate::models::corporate::press_release::PressRelease {
            symbol: d.symbol.or_else(|| Some(symbol.to_string())),
            date: d.date,
            title: d.title,
            text: d.text,
        })
        .collect())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    async fn test_stock_news_mock() {
        let mut server = mockito::Server::new_async().await;
        let _mock = server
            .mock("GET", "/stable/news/stock")
            .match_query(mockito::Matcher::AllOf(vec![
                mockito::Matcher::UrlEncoded("apikey".into(), "test-key".into()),
                mockito::Matcher::UrlEncoded("symbols".into(), "AAPL".into()),
                mockito::Matcher::UrlEncoded("limit".into(), "5".into()),
            ]))
            .with_status(200)
            .with_body(
                r#"[{
                    "symbol": "AAPL",
                    "publishedDate": "2024-01-15 12:00:00",
                    "publisher": "Reuters",
                    "title": "Apple Reports Record Quarter",
                    "image": "https://example.com/image.jpg",
                    "site": "Reuters",
                    "text": "Apple Inc. reported record quarterly earnings...",
                    "url": "https://example.com/article"
                }]"#,
            )
            .create_async()
            .await;

        let client = crate::adapters::fmp::build_test_client(&server.url()).unwrap();
        let resp: Vec<StockNewsDTO> = client
            .get("/stable/news/stock", &[("symbols", "AAPL"), ("limit", "5")])
            .await
            .unwrap();

        let article = &resp[0];
        assert_eq!(article.symbol.as_deref(), Some("AAPL"));
        assert_eq!(article.site.as_deref(), Some("Reuters"));

        let news = stock_news_to_canonical(resp);
        assert_eq!(news[0].title, "Apple Reports Record Quarter");
        assert_eq!(news[0].link, "https://example.com/article");
        assert_eq!(news[0].source, "Reuters");
        assert_eq!(news[0].img, "https://example.com/image.jpg");
        assert_eq!(news[0].time, "2024-01-15 12:00:00");
    }

    #[tokio::test]
    async fn test_press_releases_mock() {
        let mut server = mockito::Server::new_async().await;
        let _mock = server
            .mock("GET", "/stable/news/press-releases")
            .match_query(mockito::Matcher::AllOf(vec![
                mockito::Matcher::UrlEncoded("apikey".into(), "test-key".into()),
                mockito::Matcher::UrlEncoded("limit".into(), "10".into()),
            ]))
            .with_status(200)
            .with_body(
                r#"[{
                    "symbol": "AAPL",
                    "publishedDate": "2024-01-15 08:15:00",
                    "publisher": "Business Wire",
                    "title": "Apple Announces New Product",
                    "text": "Cupertino, CA -- Apple today announced...",
                    "url": "https://example.com/pr"
                }]"#,
            )
            .create_async()
            .await;

        let client = crate::adapters::fmp::build_test_client(&server.url()).unwrap();
        let resp: Vec<PressReleaseDTO> = client
            .get("/stable/news/press-releases", &[("limit", "10")])
            .await
            .unwrap();

        let row = &resp[0];
        assert_eq!(row.symbol.as_deref(), Some("AAPL"));
        assert_eq!(row.date.as_deref(), Some("2024-01-15 08:15:00"));
        assert_eq!(row.title.as_deref(), Some("Apple Announces New Product"));
        assert!(row.text.is_some());
        assert_eq!(row.publisher.as_deref(), Some("Business Wire"));
        assert_eq!(row.url.as_deref(), Some("https://example.com/pr"));
    }
}