ek_ui/

plugin.rs

//! `UiPlugin` —— 总线面:`eLog` 入环形日志、`ePlugUp/Down` 记事件 + rev、
//! `eBar`./`eBarUpdate.` 记行情库(K线窗数据面)、`eTick.` 记 tick 库(v4 行情带
//! 唯一源,S7 上屏地面)、
//! `eTimer`(1s 宿主自带)拉一次 `svctl.list` + 逐插件服务目录组快照 +
//! ask oms 全量查询组交易监控帧(S5),
//! 并辞行窗口面的操作指令队列(svctl 起停 + dm.reload 回灌 + sim 下单/撤单/流控)。
//!
//! 纪律:UI 不直调任何插件;唯一管制面是 ask 契约(svctl / dm.reload / sim 四口 /
//! oms 查询,全部与 headless 同源;`PluginContext` 只在 worker 可用)。

use std::sync::OnceLock;
use std::time::Duration;

use ek_core::context::PluginContext;
use ek_core::plugin::Plugin;
use ek_core::service::{PlugNotice, ServiceDesc};
use ek_core::{EVENT_PLUG_DOWN, EVENT_PLUG_UP};
use ek_ctp::plugin::{
    CancelOrderReply, CancelOrderReq, ConfigureReply, ConfigureReq, SendOrderReply, SendOrderReq,
    SubscribeReply as CtpSubscribeReply, SubscribeReq as CtpSubscribeReq,
};
use ek_data::account::AccountData;
use ek_data::event::{EVENT_BAR, EVENT_CONTRACT, EVENT_LOG, EVENT_TICK};
use ek_data::log::LogData;
use ek_data::order::OrderData;
use ek_data::position::PositionData;
use ek_data::trade::TradeData;
use ek_dm::plugin::{EVENT_BAR_UPDATE, ReloadReq};
use ek_event::EVENT_TIMER;
use ek_oms::plugin::{GetAllAccountsReq, GetAllOrdersReq, GetAllPositionsReq, GetAllTradesReq};
use ek_station::wire::{
    ListReq, PlugRow, RemoveReq, RemoveRes, SVC_LIST, SVC_REMOVE, SVC_START, StartReq, StartRes,
};

use crate::state::{BusSnap, UiShared};

/// 注册名 = 工厂名 = `"ui"`(BD-U5:两者一致)。
pub const UI_NAME: &str = "ui";

/// 常驻观测插件(窗口面在 [`crate::runtime`] 持 [`UiShared`] 渲染)。
pub struct UiPlugin {
    shared: std::sync::Arc<UiShared>,
    ctx: OnceLock<PluginContext>,
}

impl UiPlugin {
    /// 与窗口面共用的共享状态句柄。
    pub fn new(shared: std::sync::Arc<UiShared>) -> Self {
        Self {
            shared,
            ctx: OnceLock::new(),
        }
    }

    /// eTimer 节流回调:先辞行操作指令(有则记录),再组一次快照。
    fn refresh(&self) {
        let Some(ctx) = self.ctx.get() else {
            return; // 未 init 就吃到 eTimer:忽略,等下轮
        };
        self.run_cmds(ctx);
        match ctx.ask_service::<ListReq, Vec<PlugRow>>(SVC_LIST, ListReq, Duration::from_secs(2)) {
            Ok(rows) => {
                let mut services = Vec::with_capacity(rows.len());
                for r in &rows {
                    let ids = ctx
                        .services_of(&r.name)
                        .map(|v| v.iter().map(|s| s.id.clone()).collect::<Vec<_>>())
                        .unwrap_or_default();
                    services.push((r.name.clone(), ids));
                }
                self.shared.set_snapshot(BusSnap { rows, services });
            }
            Err(e) => self.shared.log_line(format!("[ui] svctl.list 失败: {e:?}")),
        }
        // S5 交易监控:ask oms 全量查询组帧(oms 不在/超时 → 空帧,不抛错)。
        let orders = ctx
            .ask_service::<GetAllOrdersReq, Vec<OrderData>>(
                "oms.get_all_orders",
                GetAllOrdersReq,
                Duration::from_secs(2),
            )
            .unwrap_or_default();
        let trades = ctx
            .ask_service::<GetAllTradesReq, Vec<TradeData>>(
                "oms.get_all_trades",
                GetAllTradesReq,
                Duration::from_secs(2),
            )
            .unwrap_or_default();
        let positions = ctx
            .ask_service::<GetAllPositionsReq, Vec<PositionData>>(
                "oms.get_all_positions",
                GetAllPositionsReq,
                Duration::from_secs(2),
            )
            .unwrap_or_default();
        let accounts = ctx
            .ask_service::<GetAllAccountsReq, Vec<AccountData>>(
                "oms.get_all_accounts",
                GetAllAccountsReq,
                Duration::from_secs(2),
            )
            .unwrap_or_default();
        self.shared.set_trade_snap(crate::state::TradeSnap {
            orders,
            trades,
            positions,
            accounts,
        });
    }

    /// 辞行指令队列(S2 svctl 操作面;回执进日志,面板语义可断言)。
    fn run_cmds(&self, ctx: &PluginContext) {
        for cmd in self.shared.drain_cmds() {
            match cmd {
                crate::state::UiCmd::Start { factory } => {
                    match ctx.ask_service::<StartReq, StartRes>(
                        SVC_START,
                        StartReq {
                            factory: factory.clone(),
                        },
                        Duration::from_secs(3),
                    ) {
                        Ok(StartRes::Started { name }) => {
                            self.shared
                                .log_line(format!("[ops] svctl.start {factory} → {name} 上线"));
                        }
                        Ok(StartRes::Rejected { reason }) => {
                            self.shared
                                .log_line(format!("[ops] start {factory} 被拒: {reason}"));
                        }
                        Err(e) => {
                            self.shared
                                .log_line(format!("[ops] start {factory} 失败: {e:?}"));
                        }
                    }
                }
                crate::state::UiCmd::Remove { name } => {
                    match ctx.ask_service::<RemoveReq, RemoveRes>(
                        SVC_REMOVE,
                        RemoveReq { name: name.clone() },
                        Duration::from_secs(3),
                    ) {
                        Ok(RemoveRes::Removed { name }) => {
                            self.shared
                                .log_line(format!("[ops] svctl.remove {name} 完成"));
                        }
                        Ok(RemoveRes::NoSuchName { name }) => {
                            self.shared
                                .log_line(format!("[ops] remove {name} 不存在/不可摘"));
                        }
                        Err(e) => {
                            self.shared
                                .log_line(format!("[ops] remove {name} 失败: {e:?}"));
                        }
                    }
                }
                crate::state::UiCmd::Reload {
                    interval,
                    symbol,
                    bars,
                } => {
                    // 回灌走 dm.reload ask(push=true 逐条广播 → 行情板实时见回执面)。
                    match ctx.ask_service::<ReloadReq, usize>(
                        "dm.reload",
                        ReloadReq {
                            interval,
                            symbol: symbol.clone(),
                            bars: bars.clone(),
                            size: Some(bars.len()),
                            push: true,
                        },
                        Duration::from_secs(5),
                    ) {
                        Ok(n) => {
                            self.shared.market().note_reload();
                            self.shared.log_line(format!(
                                "[ops] dm.reload {symbol}@{interval}s 回灌 {n} 条"
                            ));
                        }
                        Err(e) => {
                            self.shared
                                .log_line(format!("[ops] dm.reload {symbol} 失败: {e:?}"));
                        }
                    }
                }
                crate::state::UiCmd::SendOrder { req, gateway } => {
                    let svc = format!("{gateway}.send_order");
                    match ctx.ask_service::<SendOrderReq, SendOrderReply>(
                        &svc,
                        SendOrderReq { req: req.clone() },
                        Duration::from_secs(5),
                    ) {
                        Ok(SendOrderReply {
                            ok: true,
                            en_orderid,
                            ..
                        }) => {
                            self.shared.log_line(format!(
                                "[ops] {svc} {} {} {}x{} → {en_orderid}",
                                req.symbol,
                                req.direction.as_str(),
                                req.offset.as_str(),
                                req.volume,
                            ));
                        }
                        Ok(SendOrderReply { ok: false, msg, .. }) => {
                            self.shared.log_line(format!("[ops] 下单被拒: {msg}"));
                        }
                        Err(e) => {
                            self.shared.log_line(format!("[ops] {svc} 失败: {e:?}"));
                        }
                    }
                }
                crate::state::UiCmd::CancelOrder {
                    orderid,
                    symbol,
                    exchange,
                    gateway,
                } => {
                    let svc = format!("{gateway}.cancel_order");
                    match ctx.ask_service::<CancelOrderReq, CancelOrderReply>(
                        &svc,
                        CancelOrderReq {
                            req: ek_data::cancel_request::CancelRequest {
                                orderid: orderid.clone(),
                                symbol,
                                exchange,
                            },
                        },
                        Duration::from_secs(5),
                    ) {
                        Ok(CancelOrderReply { ok: true, msg }) => {
                            self.shared.log_line(format!("[ops] {msg}"));
                        }
                        Ok(CancelOrderReply { ok: false, msg }) => {
                            self.shared.log_line(format!("[ops] 撤单被拒: {msg}"));
                        }
                        Err(e) => {
                            self.shared.log_line(format!("[ops] {svc} 失败: {e:?}"));
                        }
                    }
                }
                crate::state::UiCmd::QueryRate => {
                    match ctx.ask_service::<crate::sim::QueryRateReq, crate::sim::QueryRateReply>(
                        "sim.query_rate",
                        crate::sim::QueryRateReq,
                        Duration::from_secs(2),
                    ) {
                        Ok(r) => {
                            self.shared.log_line(format!(
                                "[ops] sim.query_rate used={}/{} 余{}",
                                r.used, r.limit, r.remaining
                            ));
                        }
                        Err(e) => {
                            self.shared
                                .log_line(format!("[ops] sim.query_rate 失败: {e:?}"));
                        }
                    }
                }
                crate::state::UiCmd::CtpConfigure(form) => {
                    // S6 定盘口径:界面手工输入 → 插件自带配置口(插件不读文件)。
                    match ctx.ask_service::<ConfigureReq, ConfigureReply>(
                        "ctp.configure",
                        form.to_configure_req(),
                        Duration::from_secs(5),
                    ) {
                        Ok(ConfigureReply { ok: true, msg }) => {
                            self.shared
                                .log_line(format!("[ops] ctp.configure 受理:{msg}"));
                        }
                        Ok(ConfigureReply { ok: false, msg }) => {
                            self.shared
                                .log_line(format!("[ops] ctp.configure 被拒:{msg}"));
                        }
                        Err(e) => {
                            self.shared
                                .log_line(format!("[ops] ctp.configure 失败: {e:?}"));
                        }
                    }
                }
                crate::state::UiCmd::CtpSubscribe { symbols } => {
                    match ctx.ask_service::<CtpSubscribeReq, CtpSubscribeReply>(
                        "ctp.subscribe",
                        CtpSubscribeReq {
                            symbols: symbols.clone(),
                        },
                        Duration::from_secs(5),
                    ) {
                        Ok(CtpSubscribeReply { ok: true, msg }) => {
                            self.shared.log_line(format!(
                                "[ops] ctp.subscribe {} 受理:{}",
                                symbols.join(","),
                                msg
                            ));
                        }
                        Ok(CtpSubscribeReply { ok: false, msg }) => {
                            self.shared.log_line(format!(
                                "[ops] ctp.subscribe {} 被拒:{}",
                                symbols.join(","),
                                msg
                            ));
                        }
                        Err(e) => {
                            self.shared
                                .log_line(format!("[ops] ctp.subscribe 失败: {e:?}"));
                        }
                    }
                }
            }
        }
    }
}

impl Plugin for UiPlugin {
    fn name(&self) -> &str {
        UI_NAME
    }

    fn deps(&self) -> &[&str] {
        &[] // 纯观测;谁先上都能看到(bus 契约不依赖启动序)
    }

    fn routes(&self) -> Vec<std::borrow::Cow<'static, str>> {
        vec![
            std::borrow::Cow::Borrowed(EVENT_LOG),
            std::borrow::Cow::Borrowed(EVENT_PLUG_UP),
            std::borrow::Cow::Borrowed(EVENT_PLUG_DOWN),
            std::borrow::Cow::Borrowed(EVENT_BAR),
            std::borrow::Cow::Borrowed(EVENT_BAR_UPDATE),
            std::borrow::Cow::Borrowed(EVENT_TICK),
            std::borrow::Cow::Borrowed(EVENT_CONTRACT),
            std::borrow::Cow::Borrowed("eContractFinish"),
            std::borrow::Cow::Borrowed(EVENT_TIMER),
        ]
    }

    fn mailbox_cap(&self) -> usize {
        1024
    }

    fn version(&self) -> std::borrow::Cow<'static, str> {
        std::borrow::Cow::Borrowed(env!("CARGO_PKG_VERSION"))
    }

    fn services(&self) -> Vec<ServiceDesc> {
        Vec::new() // 纯视图零服务;管制面在 svctl
    }

    fn on_init(&self, ctx: &PluginContext) {
        let _ = self.ctx.set(ctx.clone());
    }

    fn on_start(&self) {
        self.shared.log_line("[ui] 界面插件上线");
    }

    fn on_stop(&self) {
        self.shared.log_line("[ui] 界面插件下线");
    }

    fn on_mail(&self, event: &ek_event::Event) {
        let ty = event.event_type.as_str();
        if ty == EVENT_LOG {
            if let Some(d) = event.data.downcast_ref::<LogData>() {
                let tag = if d.is_warning_or_above() {
                    "WARN"
                } else {
                    "INFO"
                };
                self.shared
                    .log_line(format!("[{tag}][{}] {}", d.gateway_name, d.msg));
            } else {
                self.shared.log_line("[eLog] 载荷类型异常(非 LogData)");
            }
            return;
        }
        if ty == EVENT_PLUG_UP || ty == EVENT_PLUG_DOWN {
            if let Some(n) = event.data.downcast_ref::<PlugNotice>() {
                let verb = if ty == EVENT_PLUG_UP {
                    "上线"
                } else {
                    "下线"
                };
                self.shared
                    .log_line(format!("[host] 插件{verb} {} v{}", n.name, n.version));
            }
            self.shared.bump();
            return;
        }
        if ty == EVENT_BAR {
            if let Some(f) = event.data.downcast_ref::<ek_dm::wire::BarsFrame>() {
                self.shared.market().on_frame(f);
            }
            self.shared.bump();
            return;
        }
        if ty == EVENT_BAR_UPDATE {
            if let Some(b) = event.data.downcast_ref::<ek_data::bar::BarData>() {
                self.shared.market().on_update(b);
            }
            self.shared.bump();
            return;
        }
        if ty == EVENT_TICK {
            if let Some(t) = event.data.downcast_ref::<ek_data::tick::TickData>() {
                self.shared.update_tick(t);
            }
            return;
        }
        if ty == EVENT_CONTRACT {
            if let Some(c) = event.data.downcast_ref::<ek_data::contract::ContractData>() {
                self.shared.update_contract(c);
            }
            return;
        }
        if ty == "eContractFinish" {
            let gw = event
                .data
                .downcast_ref::<String>()
                .cloned()
                .unwrap_or_else(|| "?".to_string());
            self.shared
                .log_line(format!("[ctp] 合约流完成 {gw}(查询合约窗可勾选订阅)"));
            return;
        }
        if ty == EVENT_TIMER {
            self.refresh();
        }
    }
}