use std::future::Future;
use std::io::Cursor;
use std::pin::Pin;
use std::sync::Arc;
use log::error;
use crate::prudp::packet::PRUDPPacket;
use crate::prudp::socket::{ConnectionData, SocketData};
use crate::rmc::message::RMCMessage;
use crate::rmc::response::{RMCResponse, RMCResponseResult, send_response};
use crate::rmc::response::ErrorCode::Core_NotImplemented;

type ContainedProtocolList = Box<[Box<dyn for<'a> Fn(&'a RMCMessage, &'a mut ConnectionData) -> Pin<Box<dyn Future<Output = Option<RMCResponse>> + Send + 'a>> + Send + Sync>]>;

pub struct RMCProtocolServer(ContainedProtocolList);

impl RMCProtocolServer{
    pub fn new(protocols: ContainedProtocolList) -> Arc<Self>{
        Arc::new(Self(protocols))
    }

    pub async fn process_message(&self, packet: PRUDPPacket, socket: &SocketData, connection: &mut ConnectionData){
        let Ok(rmc) = RMCMessage::new(&mut Cursor::new(&packet.payload)) else {
            error!("error reading rmc message");
            return;
        };

        for proto in &self.0 {
            if let Some(response) = proto(&rmc, connection).await {
                send_response(&packet, &socket, connection, response).await;
                return;
            }
        }

        error!("tried to send message to unimplemented protocol {} with method id {}", rmc.protocol_id, rmc.method_id);

        send_response(&packet, &socket, connection, RMCResponse{
            protocol_id: rmc.protocol_id as u8,
            response_result: RMCResponseResult::Error {
                call_id: rmc.call_id,
                error_code: Core_NotImplemented
            }
        }).await;
    }
}