rustdx-complete 0.6.0

use crate::tcp::Tdx;
use crate::bytes_helper::{u16_from_le_bytes, u32_from_le_bytes};

/// 获取股票财务信息。对应于 pytdx 中的 hq.get_finance_info、GetFinanceInfoCmd。
///
/// ## 注意
/// - 返回股票的基本面财务数据
/// - 包含股本结构、资产负债、利润表等信息
/// - market: 0=深市, 1=沪市
///
/// ## 示例
/// ```ignore
/// use rustdx::tcp::{Tcp, Tdx};
/// use rustdx::tcp::stock::FinanceInfo;
///
/// let mut tcp = Tcp::new()?;
/// let mut finance = FinanceInfo::new(0, "000001");
/// finance.recv_parsed(&mut tcp)?;
/// let info = &finance.result()[0];
/// println!("总股本: {:.0} 股", info.zongguben);
/// println!("净资产: {:.0} 元", info.jingzichan);
/// ```
#[derive(Debug, Clone)]
pub struct FinanceInfo<'d> {
    pub send: Box<[u8]>,
    pub market: u8,
    pub code: &'d str,
    pub response: Vec<u8>,
    pub data: Vec<FinanceInfoData>,
}

impl<'d> FinanceInfo<'d> {
    /// 创建一个新的财务信息请求。
    ///
    /// ## 参数
    /// - `market`: 市场代码（0=深市, 1=沪市）
    /// - `code`: 6位股票代码
    pub fn new(market: u8, code: &'d str) -> Self {
        assert_eq!(code.len(), 6, "股票代码必须是6位");

        let mut send = [0u8; Self::LEN];
        // 复制整个包头（14字节）
        send[0..14].copy_from_slice(Self::SEND);

        // 设置market（字节14）
        send[14] = market;
        // 设置code（字节15-20）
        send[15..21].copy_from_slice(code.as_bytes());

        Self {
            send: send.into(),
            market,
            code,
            response: Vec::new(),
            data: Vec::with_capacity(1),
        }
    }
}

impl<'a> Tdx for FinanceInfo<'a> {
    type Item = [FinanceInfoData];

    /// 获取财务信息的请求字节。
    ///
    /// ## 协议格式（基于pytdx源码分析）
    /// - 前14字节：固定包头
    /// - 字节14：market（市场代码）
    /// - 字节15-20：code（股票代码，6字节）
    const SEND: &'static [u8] = &[
        0x0c, 0x1f, 0x18, 0x76, 0x00, 0x01, 0x0b, 0x00, 0x0b, 0x00, 0x10, 0x00,
        0x01, 0x00, // 固定包头（14字节）
    ];

    const TAG: &'static str = "财务信息";
    const LEN: usize = 14 + 1 + 6; // 固定长度：包头14字节 + market(1) + code(6)

    fn send(&mut self) -> &[u8] {
        &self.send
    }

    /// 解析响应的字节。
    ///
    /// ## 响应格式（基于pytdx源码分析）
    /// - 前2字节：跳过
    /// - 7字节：market(1) + code(6)
    /// - 之后：32个财务字段
    fn parse(&mut self, v: Vec<u8>) {
        // 检查最小长度：2(跳过) + 7(market+code) + 32*4(字段) = 137
        if v.len() < 137 {
            eprintln!("⚠️  财务信息数据长度不足: {} 字节（需要至少 137 字节）", v.len());
            self.response = v;
            self.data = Vec::new();
            return;
        }

        let mut pos = 0;

        // 跳过前2字节
        pos += 2;

        // 读取market和code
        let market = v[pos];
        pos += 1;
        let code_bytes = &v[pos..pos + 6];
        pos += 6;
        let code = unsafe { std::str::from_utf8_unchecked(code_bytes) };
        let code = String::from(code);

        // 解析32个财务字段（混合浮点数和整数）
        // 格式: fHHIIffffffffffffffffffffffffffffff
        // 1个float, 2个u16, 2个u32, 26个float

        let liutongguben = f32_from_le_bytes(&v, pos); pos += 4;
        let province = u16_from_le_bytes(&v, pos); pos += 2;
        let industry = u16_from_le_bytes(&v, pos); pos += 2;
        let updated_date = u32_from_le_bytes(&v, pos); pos += 4;
        let ipo_date = u32_from_le_bytes(&v, pos); pos += 4;
        let zongguben = f32_from_le_bytes(&v, pos); pos += 4;
        let guojiagu = f32_from_le_bytes(&v, pos); pos += 4;
        let faqirenfarengu = f32_from_le_bytes(&v, pos); pos += 4;
        let farengu = f32_from_le_bytes(&v, pos); pos += 4;
        let bgu = f32_from_le_bytes(&v, pos); pos += 4;
        let hgu = f32_from_le_bytes(&v, pos); pos += 4;
        let zhigonggu = f32_from_le_bytes(&v, pos); pos += 4;
        let zongzichan = f32_from_le_bytes(&v, pos); pos += 4;
        let liudongzichan = f32_from_le_bytes(&v, pos); pos += 4;
        let gudingzichan = f32_from_le_bytes(&v, pos); pos += 4;
        let wuxingzichan = f32_from_le_bytes(&v, pos); pos += 4;
        let gudongrenshu = f32_from_le_bytes(&v, pos); pos += 4;
        let liudongfuzhai = f32_from_le_bytes(&v, pos); pos += 4;
        let changqifuzhai = f32_from_le_bytes(&v, pos); pos += 4;
        let zibengongjijin = f32_from_le_bytes(&v, pos); pos += 4;
        let jingzichan = f32_from_le_bytes(&v, pos); pos += 4;
        let zhuyingshouru = f32_from_le_bytes(&v, pos); pos += 4;
        let zhuyinglirun = f32_from_le_bytes(&v, pos); pos += 4;
        let yingshouzhangkuan = f32_from_le_bytes(&v, pos); pos += 4;
        let yingyelirun = f32_from_le_bytes(&v, pos); pos += 4;
        let touzishouyu = f32_from_le_bytes(&v, pos); pos += 4;
        let jingyingxianjinliu = f32_from_le_bytes(&v, pos); pos += 4;
        let zongxianjinliu = f32_from_le_bytes(&v, pos); pos += 4;
        let cunhuo = f32_from_le_bytes(&v, pos); pos += 4;
        let lirunzonghe = f32_from_le_bytes(&v, pos); pos += 4;
        let shuihoulirun = f32_from_le_bytes(&v, pos); pos += 4;
        let jinglirun = f32_from_le_bytes(&v, pos); pos += 4;
        let weifenpeilirun = f32_from_le_bytes(&v, pos); pos += 4;
        let baoliu1 = f32_from_le_bytes(&v, pos); pos += 4;
        let baoliu2 = f32_from_le_bytes(&v, pos);

        let info = FinanceInfoData {
            market,
            code,
            liutongguben: (liutongguben * 10000.0) as f64,
            province,
            industry,
            updated_date,
            ipo_date,
            zongguben: (zongguben * 10000.0) as f64,
            guojiagu: (guojiagu * 10000.0) as f64,
            faqirenfarengu: (faqirenfarengu * 10000.0) as f64,
            farengu: (farengu * 10000.0) as f64,
            bgu: (bgu * 10000.0) as f64,
            hgu: (hgu * 10000.0) as f64,
            zhigonggu: (zhigonggu * 10000.0) as f64,
            zongzichan: (zongzichan * 10000.0) as f64,
            liudongzichan: (liudongzichan * 10000.0) as f64,
            gudingzichan: (gudingzichan * 10000.0) as f64,
            wuxingzichan: (wuxingzichan * 10000.0) as f64,
            gudongrenshu: gudongrenshu as f64,
            liudongfuzhai: (liudongfuzhai * 10000.0) as f64,
            changqifuzhai: (changqifuzhai * 10000.0) as f64,
            zibengongjijin: (zibengongjijin * 10000.0) as f64,
            jingzichan: (jingzichan * 10000.0) as f64,
            zhuyingshouru: (zhuyingshouru * 10000.0) as f64,
            zhuyinglirun: (zhuyinglirun * 10000.0) as f64,
            yingshouzhangkuan: (yingshouzhangkuan * 10000.0) as f64,
            yingyelirun: (yingyelirun * 10000.0) as f64,
            touzishouyu: (touzishouyu * 10000.0) as f64,
            jingyingxianjinliu: (jingyingxianjinliu * 10000.0) as f64,
            zongxianjinliu: (zongxianjinliu * 10000.0) as f64,
            cunhuo: (cunhuo * 10000.0) as f64,
            lirunzonghe: (lirunzonghe * 10000.0) as f64,
            shuihoulirun: (shuihoulirun * 10000.0) as f64,
            jinglirun: (jinglirun * 10000.0) as f64,
            weifenpeilirun: (weifenpeilirun * 10000.0) as f64,
            meigujingzichan: baoliu1 as f64,
            baoliu2: baoliu2 as f64,
        };

        self.data.push(info);
        self.response = v;
    }

    fn result(&self) -> &Self::Item {
        &self.data
    }
}

/// 从字节数组中读取f32（小端序）
fn f32_from_le_bytes(data: &[u8], pos: usize) -> f32 {
    let bytes = [data[pos], data[pos + 1], data[pos + 2], data[pos + 3]];
    f32::from_le_bytes(bytes)
}

/// 财务信息数据。
#[derive(Debug, Default, Clone, serde::Serialize)]
pub struct FinanceInfoData {
    /// 市场代码
    pub market: u8,
    /// 股票代码
    pub code: String,
    /// 流通股本（股）
    pub liutongguben: f64,
    /// 所属省份代码
    pub province: u16,
    /// 所属行业代码
    pub industry: u16,
    /// 财务更新日期（YYYYMMDD格式）
    pub updated_date: u32,
    /// 上市日期（YYYYMMDD格式）
    pub ipo_date: u32,
    /// 总股本（股）
    pub zongguben: f64,
    /// 国家股（股）
    pub guojiagu: f64,
    /// 发起人法人股（股）
    pub faqirenfarengu: f64,
    /// 法人股（股）
    pub farengu: f64,
    /// B股（股）
    pub bgu: f64,
    /// H股（股）
    pub hgu: f64,
    /// 职工股（股）
    pub zhigonggu: f64,
    /// 总资产（元）
    pub zongzichan: f64,
    /// 流动资产（元）
    pub liudongzichan: f64,
    /// 固定资产（元）
    pub gudingzichan: f64,
    /// 无形资产（元）
    pub wuxingzichan: f64,
    /// 股东人数
    pub gudongrenshu: f64,
    /// 流动负债（元）
    pub liudongfuzhai: f64,
    /// 长期负债（元）
    pub changqifuzhai: f64,
    /// 资本公积金（元）
    pub zibengongjijin: f64,
    /// 净资产（元）
    pub jingzichan: f64,
    /// 主营收入（元）
    pub zhuyingshouru: f64,
    /// 主营利润（元）
    pub zhuyinglirun: f64,
    /// 应收账款（元）
    pub yingshouzhangkuan: f64,
    /// 营业利润（元）
    pub yingyelirun: f64,
    /// 投资收益（元）
    pub touzishouyu: f64,
    /// 经营现金流（元）
    pub jingyingxianjinliu: f64,
    /// 总现金流（元）
    pub zongxianjinliu: f64,
    /// 存货（元）
    pub cunhuo: f64,
    /// 利润总额（元）
    pub lirunzonghe: f64,
    /// 税后利润（元）
    pub shuihoulirun: f64,
    /// 净利润（元）
    pub jinglirun: f64,
    /// 未分配利润（元）
    pub weifenpeilirun: f64,
    /// 每股净资产（元）
    pub meigujingzichan: f64,
    /// 保留字段
    pub baoliu2: f64,
}

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

    #[test]
    fn test_finance_info_new() {
        let finance = FinanceInfo::new(0, "000001");
        assert_eq!(finance.market, 0);
        assert_eq!(finance.code, "000001");
        assert_eq!(finance.send.len(), 21);
    }

    #[test]
    fn test_finance_info_new_shanghai() {
        let finance = FinanceInfo::new(1, "600000");
        assert_eq!(finance.market, 1);
        assert_eq!(finance.code, "600000");
    }

    #[test]
    fn test_finance_info_send_bytes() {
        let finance = FinanceInfo::new(0, "000001");
        // 验证包头
        assert_eq!(&finance.send[0..14], &[0x0c, 0x1f, 0x18, 0x76, 0x00, 0x01, 0x0b, 0x00, 0x0b, 0x00, 0x10, 0x00, 0x01, 0x00]);
        // 验证market
        assert_eq!(finance.send[14], 0);
        // 验证code
        assert_eq!(&finance.send[15..21], b"000001");
    }

    #[test]
    #[should_panic(expected = "股票代码必须是6位")]
    fn test_finance_info_invalid_code() {
        FinanceInfo::new(0, "00001");
    }

    #[test]
    fn test_connection() {
        // 跳过集成测试（需要实际网络连接）
        if std::env::var("RUSTDX_SKIP_INTEGRATION_TESTS").is_ok() {
            println!("⚠️  跳过集成测试 (RUSTDX_SKIP_INTEGRATION_TESTS 已设置)");
            return;
        }
        println!("⚠️  集成测试需要手动验证（需要实际TCP连接）");
    }
}