ctrader-fix 0.5.4

use std::{
    collections::VecDeque,
    sync::{
        atomic::{AtomicBool, AtomicU32, Ordering},
        Arc,
    },
    time::Duration,
};

use async_std::{
    channel::{bounded, Receiver},
    io::{BufWriter, WriteExt},
    net::TcpStream,
    stream,
    stream::StreamExt,
    sync::RwLock,
    task,
};

use crate::types::{Config, Error, Field, InternalMDResult, SubID, DELIMITER};
use crate::{
    messages::{HeartbeatReq, LogonReq, LogoutReq, RequestMessage, ResponseMessage},
    types::ConnectionHandler,
};
use crate::{
    socket::Socket,
    types::{MarketCallback, TradeCallback},
};

pub struct FixApi {
    config: Config,
    stream: Option<Arc<TcpStream>>,
    seq: Arc<AtomicU32>,
    sub_id: SubID,

    is_connected: Arc<AtomicBool>,

    res_receiver: Option<Receiver<ResponseMessage>>,
    // pub container: Arc<RwLock<HashMap<String, Vec<ResponseMessage>>>>,

    // ReqMessage Container
    message_buffer: Arc<RwLock<VecDeque<(u32, String)>>>,

    //callback
    connection_handler: Option<Arc<dyn ConnectionHandler + Send + Sync>>,
    market_callback: Option<MarketCallback>,
    trade_callback: Option<TradeCallback>,
}

impl FixApi {
    pub fn new(
        sub_id: SubID,
        host: String,
        login: String,
        password: String,
        sender_comp_id: String,
        heartbeat_interval: Option<u32>,
    ) -> Self {
        Self {
            config: Config::new(
                host,
                login,
                password,
                sender_comp_id,
                heartbeat_interval.unwrap_or(30),
            ),
            stream: None,
            res_receiver: None,
            is_connected: Arc::new(AtomicBool::new(false)),
            seq: Arc::new(AtomicU32::new(1)),
            // container: Arc::new(RwLock::new(HashMap::new())),
            sub_id,

            message_buffer: Arc::new(RwLock::new(VecDeque::new())),
            connection_handler: None,
            market_callback: None,
            trade_callback: None,
        }
    }

    pub fn register_market_callback<F>(&mut self, callback: F)
    where
        F: Fn(InternalMDResult) -> () + Send + Sync + 'static,
    {
        self.market_callback = Some(Arc::new(move |mdresult: InternalMDResult| -> () {
            callback(mdresult)
        }));
    }

    pub fn register_trade_callback<F>(&mut self, callback: F)
    where
        F: Fn(ResponseMessage) -> () + Send + Sync + 'static,
    {
        self.trade_callback = Some(Arc::new(move |res: ResponseMessage| -> () {
            callback(res)
        }));
    }

    pub fn register_connection_handler_arc<T: ConnectionHandler + Send + Sync + 'static>(
        &mut self,
        handler: Arc<T>,
    ) {
        self.connection_handler = Some(handler);
    }

    pub fn register_connection_handler<T: ConnectionHandler + Send + Sync + 'static>(
        &mut self,
        handler: T,
    ) {
        self.connection_handler = Some(Arc::new(handler));
    }

    pub async fn disconnect(&mut self) -> Result<(), Error> {
        if let Some(stream) = self.stream.clone() {
            stream.shutdown(std::net::Shutdown::Both)?;
        }
        self.stream = None;
        self.res_receiver = None;
        self.is_connected.store(false, Ordering::Relaxed);
        self.message_buffer.write().await.clear();
        Ok(())
    }

    pub async fn connect(&mut self) -> Result<(), Error> {
        self.message_buffer.write().await.clear();
        let (sender, receiver) = bounded(1);
        let mut socket = Socket::connect(
            self.config.host.as_str(),
            if self.sub_id == SubID::QUOTE {
                5201
            } else {
                5202
            },
            sender,
        )
        .await?;
        self.is_connected.store(true, Ordering::Relaxed);
        log::debug!("stream connected");

        // notify connection
        if let Some(handler) = self.connection_handler.clone() {
            task::spawn(async move {
                handler.on_connect().await;
            });
        }

        self.res_receiver = Some(receiver);
        self.stream = Some(socket.stream.clone());

        let is_connected = self.is_connected.clone();

        let handler = self.connection_handler.clone();
        let _ = task::spawn(async move {
            socket.recv_loop(is_connected, handler).await.ok();
        });

        Ok(())
    }

    pub async fn send_message<R: RequestMessage>(&self, req: R) -> Result<(), Error> {
        let no_seq = self.seq.fetch_add(1, Ordering::Relaxed);
        let req = req.build(self.sub_id, no_seq, DELIMITER, &self.config);
        if let Some(stream) = self.stream.clone() {
            // FIXME
            self.message_buffer
                .write()
                .await
                .push_back((no_seq, req.clone()));
            self.message_buffer
                .write()
                .await
                .push_back((no_seq, req.clone()));
            if self.message_buffer.read().await.len() > 10 {
                self.message_buffer.write().await.pop_front();
            }

            log::debug!("Send request : {}", req);
            let mut writer = BufWriter::new(stream.as_ref());
            writer.write_all(req.as_bytes()).await?;
            writer.flush().await?;
        }
        Ok(())
    }

    pub fn is_connected(&self) -> bool {
        self.is_connected.load(Ordering::Relaxed)
    }

    pub async fn logon(&self, heartbeat: bool) -> Result<(), Error> {
        // res3et the seq
        self.seq.store(1, Ordering::Relaxed);
        self.send_message(LogonReq::new(Some(true))).await?;

        // wait to receive the response
        if let Some(recv) = &self.res_receiver {
            while let Ok(response) = recv.recv().await {
                // logon response
                let msg_type = response.get_message_type();
                match msg_type {
                    "A" => {
                        //
                        if let Some(handler) = self.connection_handler.clone() {
                            task::spawn(async move {
                                handler.on_logon().await;
                            });
                        }

                        let stream = self.stream.clone().unwrap();
                        let stream_clone = self.stream.clone().unwrap();
                        let sub_id = self.sub_id;
                        let config = self.config.clone();
                        let seq = self.seq.clone();
                        let msg_buffer = self.message_buffer.clone();
                        let is_connected = self.is_connected.clone();
                        let handler = self.connection_handler.clone();

                        let send_request = move |req: Box<dyn RequestMessage>| {
                            let stream = stream.clone();
                            let sub_id = sub_id;
                            let config = config.clone();
                            let seq = seq.clone();
                            let msg_buffer = msg_buffer.clone();
                            let is_connected = is_connected.clone();
                            let handler = handler.clone();
                            async move {
                                let msg_type = req.get_message_type();
                                let no_seq = seq.fetch_add(1, Ordering::Relaxed);
                                let req = req.build(sub_id, no_seq, DELIMITER, &config);
                                let handler = handler.clone();

                                // FIXME later
                                msg_buffer.write().await.push_back((no_seq, req.clone()));
                                if msg_buffer.read().await.len() > 10 {
                                    msg_buffer.write().await.pop_front();
                                }

                                let mut writer = BufWriter::new(stream.as_ref());
                                log::debug!(
                                    "[Session:MsgType({msg_type})] Sending request: {}",
                                    req
                                );
                                let _ = writer.write_all(req.as_bytes()).await;

                                match writer.flush().await {
                                    Ok(_) => {}
                                    Err(err) => {
                                        log::error!("Failed to send the request - {:?}", err);
                                        is_connected.store(false, Ordering::Relaxed);
                                        if let Err(err) = stream.shutdown(std::net::Shutdown::Both)
                                        {
                                            log::error!(
                                                "Failed to shutdown the stream - {:?}",
                                                err
                                            );
                                        }
                                        if let Some(handler) = handler {
                                            task::spawn(async move {
                                                handler.on_disconnect().await;
                                            });
                                        }
                                    }
                                }
                            }
                        };
                        let send_request_clone = send_request.clone();

                        if heartbeat {
                            let hb_interval = self.config.heart_beat as u64;

                            let is_connected = self.is_connected.clone();

                            //send heartbeat per hb_interval
                            task::spawn(async move {
                                let mut heartbeat_stream =
                                    stream::interval(Duration::from_secs(hb_interval));

                                while let Some(_) = heartbeat_stream.next().await {
                                    if !is_connected.load(Ordering::Relaxed) {
                                        break;
                                    }
                                    let req = HeartbeatReq::new(None);
                                    send_request(Box::new(req)).await;
                                }
                            });
                        }

                        //
                        // handle the responses

                        let recv = self.res_receiver.clone().unwrap();
                        let market_callback = self.market_callback.clone();
                        let trade_callback = self.trade_callback.clone();

                        let is_connected = self.is_connected.clone();
                        // let seq = self.seq.clone();
                        let msg_buffer = self.message_buffer.clone();
                        task::spawn(async move {
                            while let Ok(res) = recv.recv().await {
                                if !is_connected.load(Ordering::Relaxed) {
                                    break;
                                }

                                let msg_type = res.get_message_type();

                                // notify? or send? via channel?
                                match msg_type {
                                    "0" => {
                                        log::debug!(
                                            "[Session:MsyType({msg_type})] Received Heartbeat"
                                        );
                                    }
                                    "2" => {
                                        log::debug!(
                                            "[Session:MsyType({msg_type})] Received ResendRequest"
                                        );
                                        let begin = res
                                            .get_field_value(Field::BeginSeqNo)
                                            .map(|v| v.parse::<u32>().unwrap_or(0))
                                            .unwrap();

                                        let end = res
                                            .get_field_value(Field::EndSeqNo)
                                            .map(|v| v.parse::<u32>().unwrap_or(0))
                                            .unwrap();

                                        {
                                            for msg in msg_buffer
                                                .read()
                                                .await
                                                .iter()
                                                .filter(|(no, _)| {
                                                    if end == 0 {
                                                        *no >= begin
                                                    } else {
                                                        *no >= begin && *no <= end
                                                    }
                                                })
                                                .map(|(_, msg)| msg.clone())
                                            {
                                                let mut writer =
                                                    BufWriter::new(stream_clone.as_ref());
                                                log::debug!(
                                                 "[Session:MsgType({msg_type})] Send ResendRequest: {}",
                                                msg
                                                );
                                                let _ = writer.write_all(msg.as_bytes()).await;
                                                break;
                                            }
                                        }
                                    }
                                    "5" => {
                                        log::debug!(
                                            "[Session:MsyType({msg_type})] Received Logged out"
                                        );
                                        // 5 : logout
                                        //disconnect
                                        stream_clone.shutdown(std::net::Shutdown::Both).ok();
                                    }
                                    "1" => {
                                        log::debug!(
                                            "[Session:MsyType({msg_type})] Received TestRequest"
                                        );
                                        // send back with test request id
                                        if let Some(test_req_id) =
                                            res.get_field_value(Field::TestReqID)
                                        {
                                            send_request_clone(Box::new(HeartbeatReq::new(Some(
                                                test_req_id,
                                            ))))
                                            .await;
                                            log::debug!("Sent the heartbeat from test_req_id");
                                        }
                                    }
                                    "W" | "X" | "Y" => {
                                        // For market data
                                        let symbol_id = res
                                            .get_field_value(Field::Symbol)
                                            .unwrap_or("0".into())
                                            .parse::<u32>()
                                            .unwrap();
                                        // notify to callback
                                        if let Some(market_callback) = market_callback.clone() {
                                            let mdresult = if msg_type == "Y" {
                                                let md_req_id = res
                                                    .get_field_value(Field::MDReqID)
                                                    .map(|v| v.clone())
                                                    .unwrap_or("".into());
                                                let err_msg = res
                                                    .get_field_value(Field::Text)
                                                    .map(|v| v.clone())
                                                    .unwrap_or("".into());
                                                InternalMDResult::MDReject {
                                                    symbol_id,
                                                    md_req_id,
                                                    err_msg,
                                                }
                                            } else {
                                                let data = res.get_repeating_groups(
                                                    Field::NoMDEntries,
                                                    if msg_type == "W" {
                                                        Field::MDEntryType
                                                    } else {
                                                        Field::MDUpdateAction
                                                    },
                                                    None,
                                                );
                                                InternalMDResult::MD {
                                                    msg_type: msg_type.chars().next().unwrap(),
                                                    symbol_id,
                                                    data,
                                                }
                                            };

                                            market_callback(mdresult);
                                        }
                                    }
                                    _ => {
                                        log::debug!("{}", res.get_message());
                                        if let Some(trade_callback) = trade_callback.clone() {
                                            trade_callback(res);
                                        }
                                    }
                                }
                            }
                        });

                        break;
                    }
                    "5" => {
                        return Err(Error::LoggedOut);
                    }
                    _ => {}
                }
            }
        }
        Ok(())
    }

    pub async fn logout(&self) -> Result<(), Error> {
        self.send_message(LogoutReq::default()).await?;
        Ok(())
    }
}