AppFlowy/frontend/rust-lib/flowy-notification/src/entities/subject.rs

use flowy_derive::ProtoBuf;
use std::{fmt, fmt::Formatter};

#[derive(Debug, Clone, ProtoBuf, serde::Serialize)]
pub struct SubscribeObject {
    #[pb(index = 1)]
    pub source: String,

    #[pb(index = 2)]
    pub ty: i32,

    #[pb(index = 3)]
    pub id: String,

    #[pb(index = 4, one_of)]
    pub payload: Option<Vec<u8>>,

    #[pb(index = 5, one_of)]
    pub error: Option<Vec<u8>>,
}

impl std::fmt::Display for SubscribeObject {
    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
        f.write_str(&format!("{} changed: ", &self.source))?;
        if let Some(payload) = &self.payload {
            f.write_str(&format!("send {} payload", payload.len()))?;
        }

        if let Some(payload) = &self.error {
            f.write_str(&format!("receive {} error", payload.len()))?;
        }

        Ok(())
    }
}

impl std::default::Default for SubscribeObject {
    fn default() -> Self {
        Self {
            source: "".to_string(),
            ty: 0,
            id: "".to_string(),
            payload: None,
            error: None,
        }
    }
}