akshare 0.1.12

//! Historical data from Eastmoney (stock_zh_a_hist, stock_hk_hist, stock_us_hist, etc.)

use super::types::HistData;
use crate::client::AkShareClient;
use crate::error::{Error, Result};
use crate::market::eastmoney_secid;
use crate::util::parse_csv_line;
use crate::util::parse_f64_safe;

/// Parse raw kline strings into `HistData` records.
fn parse_hist_klines(klines: &[String]) -> Vec<HistData> {
    klines
        .iter()
        .filter_map(|s| {
            let f = parse_csv_line(s);
            if f.len() < 11 {
                return None;
            }
            Some(HistData {
                trade_date: f[0].to_string(),
                open: parse_f64_safe(f[1]),
                close: parse_f64_safe(f[2]),
                high: parse_f64_safe(f[3]),
                low: parse_f64_safe(f[4]),
                volume: parse_f64_safe(f[5]),
                amount: parse_f64_safe(f[6]),
                amplitude_pct: parse_f64_safe(f[7]),
                change_pct: parse_f64_safe(f[8]),
                change_amount: parse_f64_safe(f[9]),
                turnover_rate: parse_f64_safe(f[10]),
            })
        })
        .collect()
}

/// Resolve a period string to a klt code for daily/weekly/monthly.
fn daily_klt(period: &str) -> Result<&'static str> {
    match period {
        "daily" => Ok("101"),
        "weekly" => Ok("102"),
        "monthly" => Ok("103"),
        _ => Err(Error::invalid_input(format!(
            "unsupported period: {period}"
        ))),
    }
}

/// Resolve a period string to a klt code for intraday intervals.
fn minute_klt(period: &str) -> Result<&'static str> {
    match period {
        "1" => Ok("1"),
        "5" => Ok("5"),
        "15" => Ok("15"),
        "30" => Ok("30"),
        "60" => Ok("60"),
        _ => Err(Error::invalid_input(format!(
            "unsupported period: {period}"
        ))),
    }
}

/// Resolve an adjust string to a fqt code.
fn adjust_fqt(adjust: &str) -> &'static str {
    match adjust {
        "qfq" => "1",
        "hfq" => "2",
        _ => "0",
    }
}

impl AkShareClient {
    /// Shared fetch-and-parse logic for Eastmoney kline history.
    async fn hist_inner(
        &self,
        secid: &str,
        klt: &str,
        fqt: &str,
        start_date: &str,
        end_date: &str,
    ) -> Result<Vec<HistData>> {
        let beg = if start_date.is_empty() {
            "0"
        } else {
            start_date
        };
        let end = if end_date.is_empty() {
            "20500000"
        } else {
            end_date
        };
        let klines = self
            .kline_fetch(secid, klt, fqt, 1_000_000, &[("beg", beg), ("end", end)])
            .await?;
        Ok(parse_hist_klines(&klines))
    }

    /// 东方财富-沪深京A股-历史行情
    pub async fn stock_zh_a_hist(
        &self,
        symbol: &str,
        period: &str,
        adjust: &str,
        start_date: &str,
        end_date: &str,
    ) -> Result<Vec<HistData>> {
        let secid = eastmoney_secid(symbol)?;
        self.hist_inner(
            &secid,
            daily_klt(period)?,
            adjust_fqt(adjust),
            start_date,
            end_date,
        )
        .await
    }

    /// 东方财富-沪深A股-分时行情
    pub async fn stock_zh_a_hist_min(
        &self,
        symbol: &str,
        period: &str,
        adjust: &str,
        start_date: &str,
        end_date: &str,
    ) -> Result<Vec<HistData>> {
        let secid = eastmoney_secid(symbol)?;
        self.hist_inner(
            &secid,
            minute_klt(period)?,
            adjust_fqt(adjust),
            start_date,
            end_date,
        )
        .await
    }

    /// 东方财富-港股-历史行情
    pub async fn stock_hk_hist(
        &self,
        symbol: &str,
        period: &str,
        adjust: &str,
        start_date: &str,
        end_date: &str,
    ) -> Result<Vec<HistData>> {
        let secid = format!("116.{}", symbol.trim());
        self.hist_inner(
            &secid,
            daily_klt(period)?,
            adjust_fqt(adjust),
            start_date,
            end_date,
        )
        .await
    }

    /// 东方财富-港股-分时行情
    pub async fn stock_hk_hist_min(
        &self,
        symbol: &str,
        period: &str,
        adjust: &str,
        start_date: &str,
        end_date: &str,
    ) -> Result<Vec<HistData>> {
        let secid = format!("116.{}", symbol.trim());
        self.hist_inner(
            &secid,
            minute_klt(period)?,
            adjust_fqt(adjust),
            start_date,
            end_date,
        )
        .await
    }

    /// 东方财富-美股-历史行情
    pub async fn stock_us_hist(
        &self,
        symbol: &str,
        period: &str,
        adjust: &str,
        start_date: &str,
        end_date: &str,
    ) -> Result<Vec<HistData>> {
        // US stocks use 105/106 market codes
        let secid = if symbol.contains('.') {
            let parts: Vec<&str> = symbol.split('.').collect();
            if parts.len() == 2 {
                let code = parts[0];
                let suffix = parts[1].to_uppercase();
                let market = match suffix.as_str() {
                    "N" => "106",
                    _ => "105",
                };
                format!("{market}.{code}")
            } else {
                format!("105.{symbol}")
            }
        } else {
            format!("105.{symbol}")
        };
        self.hist_inner(
            &secid,
            daily_klt(period)?,
            adjust_fqt(adjust),
            start_date,
            end_date,
        )
        .await
    }

    /// 东方财富-美股-分时行情
    pub async fn stock_us_hist_min(
        &self,
        symbol: &str,
        period: &str,
        adjust: &str,
        start_date: &str,
        end_date: &str,
    ) -> Result<Vec<HistData>> {
        let secid = format!("105.{symbol}");
        self.hist_inner(
            &secid,
            minute_klt(period)?,
            adjust_fqt(adjust),
            start_date,
            end_date,
        )
        .await
    }
}