futucli/cmd/trade_ext/

cash_flow.rs

use anyhow::{Result, bail};
use chrono::Datelike;
use prost::Message as _;

use crate::cmd::account::{parse_trd_env, parse_trd_market};
use crate::common::connect_gateway;
use crate::output::OutputFormat;

/// v1.4.30 P2: 账户资金流水（对齐 py-futu-api `get_acc_cash_flow`）
pub async fn run_acc_cash_flow(
    gateway: &str,
    acc_id: u64,
    clearing_date: &str,
    env: &str,
    market: &str,
    direction: Option<i32>,
    _format: OutputFormat,
) -> Result<()> {
    let entries =
        fetch_acc_cash_flow_single(gateway, acc_id, clearing_date, env, market, direction).await?;
    println!(
        "account {} clearing={}  entries: {}",
        acc_id,
        clearing_date,
        entries.len()
    );
    for f in &entries {
        println!(
            "  date={:?} type={:?} amount={:?} remark={:?}",
            f.clearing_date, f.cash_flow_type, f.cash_flow_amount, f.cash_flow_remark
        );
    }
    Ok(())
}

/// v1.4.32: `--date-range FROM..TO` 便利封装。proto `clearing_date` 一次只接
/// 一天，这里在 CLI 层循环每个工作日调一次单日版，跳周末。
///
/// **服务端限流考虑**：31 天硬上限，防止 bot 脚本无意中打出几百个连续日期
/// 请求被服务端 rate-limit 掉。
pub struct AccCashFlowRangeCommand<'a> {
    pub gateway: &'a str,
    pub acc_id: u64,
    pub date_from: chrono::NaiveDate,
    pub date_to: chrono::NaiveDate,
    pub env: &'a str,
    pub market: &'a str,
    pub direction: Option<i32>,
}

pub async fn run_acc_cash_flow_range(input: AccCashFlowRangeCommand<'_>) -> Result<()> {
    let date_from = input.date_from;
    let date_to = input.date_to;
    if date_to < date_from {
        bail!("--date-range: 结束日期 {date_to} 不能早于开始日期 {date_from}");
    }
    let span_days = (date_to - date_from).num_days();
    if span_days > 31 {
        bail!(
            "--date-range: 跨度 {} 天超过 31 天上限（防 bot 脚本误调用被服务端限流）",
            span_days
        );
    }
    let mut day = date_from;
    let mut total_entries = 0usize;
    let mut failed_days = Vec::new();
    let mut success_days = 0usize; // v1.4.102 codex 45 F2: success ≠ total_entries
    while day <= date_to {
        // 跳周末（省 proto 请求；真·非交易日服务端会返空，也安全）
        if matches!(day.weekday(), chrono::Weekday::Sat | chrono::Weekday::Sun) {
            if !acc_cash_flow_advance_day(&mut day, date_to)? {
                break;
            }
            continue;
        }
        let date_str = day.format("%Y-%m-%d").to_string();
        match fetch_acc_cash_flow_single(
            input.gateway,
            input.acc_id,
            &date_str,
            input.env,
            input.market,
            input.direction,
        )
        .await
        {
            Ok(entries) => {
                // v1.4.102 codex 45 F2: 成功但空 (假日 / 无流水) 也算 success.
                // 之前 total_entries==0 + failed_days 非空 = all-failed 误判:
                // 部分成功 + 那些成功日恰好无流水 → 错认 all-failed.
                success_days += 1;
                if !entries.is_empty() {
                    println!(
                        "== {} ({}):  entries: {}",
                        date_str,
                        day.weekday(),
                        entries.len()
                    );
                    for f in &entries {
                        println!(
                            "  type={:?} amount={:?} remark={:?}",
                            f.cash_flow_type, f.cash_flow_amount, f.cash_flow_remark
                        );
                    }
                    total_entries += entries.len();
                }
            }
            Err(e) => {
                eprintln!("⚠️  {date_str}: {e}");
                failed_days.push(date_str);
                // 继续下一天，不 abort（某天 proto 错不影响其他天）
            }
        }
        if !acc_cash_flow_advance_day(&mut day, date_to)? {
            break;
        }
    }
    println!(
        "---\n{date_from}..{date_to}  attempted: {}  success_days: {success_days}  failed days: {}  total entries: {total_entries}",
        success_days + failed_days.len(),
        failed_days.len()
    );
    if !failed_days.is_empty() {
        eprintln!("failed days: {failed_days:?}");
    }
    // v1.4.102 codex 38 F5 + 45 F2 (P2): all-failed 必须返 Err 让 shell exit non-zero.
    // 之前 v1.4.102 codex 38 F5 用 `total_entries==0 + failed_days 非空` 判
    // all-failed, 但 codex 45 F2 指出: 成功的 daily fetch 也可能合法返空 entries
    // (假日 / 无流水); 部分成功 + 成功日空流水 → 误判 all-failed.
    // 真正 all-failed 判: success_days == 0 (没有任何一天成功).
    if success_days == 0 && !failed_days.is_empty() {
        anyhow::bail!(
            "acc-cash-flow --date-range {date_from}..{date_to} 所有 {} 天全部失败 \
             (no successful daily fetch). 失败 days: {:?}. 检查 acc_id / env / market / \
             daemon 连接, 或单日重试 (`futucli acc-cash-flow --date <day>`).",
            failed_days.len(),
            failed_days
        );
    }
    Ok(())
}

/// 拆出的单日拉取 helper：单日版和 range 版共用。
async fn fetch_acc_cash_flow_single(
    gateway: &str,
    acc_id: u64,
    clearing_date: &str,
    env: &str,
    market: &str,
    direction: Option<i32>,
) -> Result<Vec<futu_proto::trd_flow_summary::FlowSummaryInfo>> {
    let env_p = parse_trd_env(env)?;
    let market_p = parse_trd_market(market)?;
    let proto_header = futu_proto::trd_common::TrdHeader {
        trd_env: env_p as i32,
        acc_id,
        trd_market: market_p as i32,
        jp_acc_type: None,
    };
    let req = futu_proto::trd_flow_summary::Request {
        c2s: futu_proto::trd_flow_summary::C2s {
            header: proto_header,
            clearing_date: clearing_date.to_string(),
            cash_flow_direction: direction,
            start_create_date: None,
            end_create_date: None,
        },
    };
    let (client, _rx) = connect_gateway(gateway, "futucli-cash-flow").await?;
    let body = req.encode_to_vec();
    let frame = client
        .request(futu_core::proto_id::TRD_FLOW_SUMMARY, body)
        .await?;
    let resp = futu_proto::trd_flow_summary::Response::decode(frame.body.as_ref())
        .map_err(|e| anyhow::anyhow!("decode: {e}"))?;
    if resp.ret_type != 0 {
        bail!(
            "flow_summary ret_type={} msg={:?}",
            resp.ret_type,
            resp.ret_msg
        );
    }
    let s = resp.s2c.ok_or_else(|| anyhow::anyhow!("missing s2c"))?;
    Ok(s.flow_summary_info_list)
}

pub(crate) fn acc_cash_flow_advance_day(
    day: &mut chrono::NaiveDate,
    date_to: chrono::NaiveDate,
) -> Result<bool> {
    if *day >= date_to {
        return Ok(false);
    }
    let next = day
        .succ_opt()
        .ok_or_else(|| anyhow::anyhow!("acc-cash-flow date range cannot advance beyond {day}"))?;
    *day = next;
    Ok(true)
}