use std::fmt::Debug;

use serde::{Deserialize, Serialize};

use super::{MsgDataBound, MsgSource, Timestamp};

#[derive(Clone, Debug, Deserialize, PartialEq, Serialize)]
pub enum System {}

#[derive(Clone, Debug, Deserialize, PartialEq, Serialize)]
pub enum MsgType<TCustom> {
    System(System),
    Custom(TCustom),
}

#[derive(Clone, Debug, Deserialize, PartialEq, Serialize)]
pub struct Message<TData> {
    pub data: MsgType<TData>,
    pub key: String,
    pub ts: Timestamp,
    pub source: Option<MsgSource>,
    pub process: Option<MsgSource>,
}

impl<TCustom> Message<TCustom>
where
    TCustom: MsgDataBound,
{
    pub fn new(data: MsgType<TCustom>) -> Self {
        let key = define_key(&data);
        Self {
            data,
            key,
            ts: Default::default(),
            source: None,
            process: None,
        }
    }

    pub fn new_custom(custom_data: TCustom) -> Self {
        let data = MsgType::Custom(custom_data);
        let key = define_key(&data);
        Self {
            data,
            key,
            ts: Default::default(),
            source: None,
            process: None,
        }
    }

    pub fn cmp_set(&mut self, cmp: &MsgSource) {
        if self.source.is_none() {
            self.source = Some(cmp.clone());
        }
        self.process = Some(cmp.clone());
    }

    pub fn get_data(&self) -> Option<TCustom> {
        match &self.data {
            MsgType::System(_) => None,
            MsgType::Custom(data) => Some(data.clone()),
        }
    }
}

fn define_key<TCustom>(data: &MsgType<TCustom>) -> String
where
    TCustom: MsgDataBound,
{
    let full_str = format!("{:?}", data);
    let key = full_str.split('(').collect::<Vec<&str>>();
    // Убираем последний элемент. Если тип unit (), нужно убрать два последних элемента
    let skip = if key[key.len() - 2].is_empty() { 2 } else { 1 };
    key[0..key.len() - skip].join("-")
}