megalodon 1.3.0

Fediverse API client library for Rust.
Documentation
use super::{Account, Reaction, Status};
use crate::error::{Error, Kind};

use crate::entities as MegalodonEntities;
use chrono::{DateTime, Utc};
use core::str::FromStr;
use serde::{Deserialize, de, ser};
use std::fmt;

#[derive(Debug, Deserialize, Clone)]
pub struct Notification {
    account: Account,
    created_at: DateTime<Utc>,
    id: String,
    status: Option<Status>,
    r#type: NotificationType,
    emoji: Option<String>,
    emoji_url: Option<String>,
    target: Option<Account>,
}

impl Notification {
    fn map_reaction(&self) -> Option<Reaction> {
        let shortcode = self.emoji.clone()?;
        let name = shortcode.replace(":", "");
        Some(Reaction {
            count: 1,
            me: false,
            name,
            url: self.emoji_url.clone(),
            accounts: None,
            account_ids: None,
        })
    }
}

// https://git.pleroma.social/pleroma/pleroma/src/commit/683ab39160a2ff95d151887a89217bd1d4a6dcf5/lib/pleroma/notification.ex#L65-L78
#[derive(Debug, Clone)]
pub enum NotificationType {
    Follow,
    FollowRequest,
    Mention,
    Reblog,
    Favourite,
    Poll,
    PleromaEmojiReaction,
    Update,
    Move,
    Status,
}

impl fmt::Display for NotificationType {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            NotificationType::Follow => write!(f, "follow"),
            NotificationType::Mention => write!(f, "mention"),
            NotificationType::Reblog => write!(f, "reblog"),
            NotificationType::Favourite => write!(f, "favourite"),
            NotificationType::PleromaEmojiReaction => write!(f, "pleroma:emoji_reaction"),
            NotificationType::Poll => write!(f, "poll"),
            NotificationType::FollowRequest => write!(f, "follow_request"),
            NotificationType::Update => write!(f, "update"),
            NotificationType::Move => write!(f, "move"),
            NotificationType::Status => write!(f, "status"),
        }
    }
}

impl FromStr for NotificationType {
    type Err = Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "follow" => Ok(NotificationType::Follow),
            "mention" => Ok(NotificationType::Mention),
            "reblog" => Ok(NotificationType::Reblog),
            "favourite" => Ok(NotificationType::Favourite),
            "pleroma:emoji_reaction" => Ok(NotificationType::PleromaEmojiReaction),
            "poll" => Ok(NotificationType::Poll),
            "follow_request" => Ok(NotificationType::FollowRequest),
            "update" => Ok(NotificationType::Update),
            "move" => Ok(NotificationType::Move),
            "status" => Ok(NotificationType::Status),
            _ => Err(Error::new_own(
                s.to_owned(),
                Kind::ParseError,
                None,
                None,
                None,
            )),
        }
    }
}

impl ser::Serialize for NotificationType {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serializer.serialize_str(self.to_string().as_ref())
    }
}

impl<'de> de::Deserialize<'de> for NotificationType {
    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
    where
        D: de::Deserializer<'de>,
    {
        let s = String::deserialize(deserializer)?;
        match NotificationType::from_str(s.as_str()) {
            Ok(r) => Ok(r),
            Err(e) => Err(de::Error::custom(e)),
        }
    }
}

impl From<NotificationType> for MegalodonEntities::notification::NotificationType {
    fn from(val: NotificationType) -> Self {
        match val {
            NotificationType::Follow => MegalodonEntities::notification::NotificationType::Follow,
            NotificationType::FollowRequest => {
                MegalodonEntities::notification::NotificationType::FollowRequest
            }
            NotificationType::Mention => MegalodonEntities::notification::NotificationType::Mention,
            NotificationType::Reblog => MegalodonEntities::notification::NotificationType::Reblog,
            NotificationType::Favourite => {
                MegalodonEntities::notification::NotificationType::Favourite
            }
            NotificationType::Poll => {
                MegalodonEntities::notification::NotificationType::PollExpired
            }
            NotificationType::PleromaEmojiReaction => {
                MegalodonEntities::notification::NotificationType::Reaction
            }
            NotificationType::Update => MegalodonEntities::notification::NotificationType::Update,
            NotificationType::Move => MegalodonEntities::notification::NotificationType::Move,
            NotificationType::Status => MegalodonEntities::notification::NotificationType::Status,
        }
    }
}

impl From<Notification> for MegalodonEntities::Notification {
    fn from(val: Notification) -> Self {
        let reaction = val.clone().map_reaction();
        MegalodonEntities::Notification {
            account: Some(val.account.into()),
            created_at: val.created_at,
            id: val.id,
            status: val.status.map(|i| i.into()),
            reaction: reaction.map(|i| i.into()),
            target: val.target.map(|i| i.into()),
            r#type: val.r#type.into(),
        }
    }
}