start-os/core/startos/src/notifications.rs

use std::collections::BTreeMap;
use std::fmt;
use std::str::FromStr;

use chrono::{DateTime, Utc};
use clap::builder::ValueParserFactory;
use clap::Parser;
use color_eyre::eyre::eyre;
use helpers::const_true;
use imbl_value::InternedString;
use models::{FromStrParser, PackageId};
use rpc_toolkit::{from_fn_async, Context, HandlerExt, ParentHandler};
use serde::{Deserialize, Serialize};
use tracing::instrument;
use ts_rs::TS;

use crate::backup::BackupReport;
use crate::context::{CliContext, RpcContext};
use crate::db::model::DatabaseModel;
use crate::prelude::*;
use crate::util::serde::HandlerExtSerde;

// #[command(subcommands(list, delete, delete_before, create))]
pub fn notification<C: Context>() -> ParentHandler<C> {
    ParentHandler::new()
        .subcommand(
            "list",
            from_fn_async(list)
                .with_display_serializable()
                .with_about("List notifications")
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "remove",
            from_fn_async(remove)
                .no_display()
                .with_about("Remove notification for given ids")
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "remove-before",
            from_fn_async(remove_before)
                .no_display()
                .with_about("Remove notifications preceding a given id")
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "mark-seen",
            from_fn_async(mark_seen)
                .no_display()
                .with_about("Mark given notifications as seen")
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "mark-seen-before",
            from_fn_async(mark_seen_before)
                .no_display()
                .with_about("Mark notifications preceding a given id as seen")
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "mark-unseen",
            from_fn_async(mark_unseen)
                .no_display()
                .with_about("Mark given notifications as unseen")
                .with_call_remote::<CliContext>(),
        )
        .subcommand(
            "create",
            from_fn_async(create)
                .no_display()
                .with_about("Persist a newly created notification")
                .with_call_remote::<CliContext>(),
        )
}

#[derive(Deserialize, Serialize, Parser, TS)]
#[serde(rename_all = "camelCase")]
#[command(rename_all = "kebab-case")]
pub struct ListNotificationParams {
    #[ts(type = "number | null")]
    before: Option<u32>,
    #[ts(type = "number | null")]
    limit: Option<usize>,
}
// #[command(display(display_serializable))]
#[instrument(skip_all)]
pub async fn list(
    ctx: RpcContext,
    ListNotificationParams { before, limit }: ListNotificationParams,
) -> Result<Vec<NotificationWithId>, Error> {
    ctx.db
        .mutate(|db| {
            let limit = limit.unwrap_or(40);
            match before {
                None => {
                    let records = db
                        .as_private()
                        .as_notifications()
                        .as_entries()?
                        .into_iter()
                        .rev()
                        .take(limit);
                    let notifs = records
                        .into_iter()
                        .map(|(id, notification)| {
                            Ok(NotificationWithId {
                                id,
                                notification: notification.de()?,
                            })
                        })
                        .collect::<Result<Vec<NotificationWithId>, Error>>()?;
                    db.as_public_mut()
                        .as_server_info_mut()
                        .as_unread_notification_count_mut()
                        .ser(&0)?;
                    Ok(notifs)
                }
                Some(before) => {
                    let records = db
                        .as_private()
                        .as_notifications()
                        .as_entries()?
                        .into_iter()
                        .filter(|(id, _)| *id < before)
                        .rev()
                        .take(limit);
                    records
                        .into_iter()
                        .map(|(id, notification)| {
                            Ok(NotificationWithId {
                                id,
                                notification: notification.de()?,
                            })
                        })
                        .collect()
                }
            }
        })
        .await
        .result
}

#[derive(Deserialize, Serialize, Parser, TS)]
#[serde(rename_all = "camelCase")]
#[command(rename_all = "kebab-case")]
pub struct ModifyNotificationParams {
    #[ts(type = "number[]")]
    ids: Vec<u32>,
}

pub async fn remove(
    ctx: RpcContext,
    ModifyNotificationParams { ids }: ModifyNotificationParams,
) -> Result<(), Error> {
    ctx.db
        .mutate(|db| {
            let n = db.as_private_mut().as_notifications_mut();
            for id in ids {
                n.remove(&id)?;
            }
            Ok(())
        })
        .await
        .result
}

#[derive(Deserialize, Serialize, Parser, TS)]
#[serde(rename_all = "camelCase")]
#[command(rename_all = "kebab-case")]
pub struct ModifyNotificationBeforeParams {
    #[ts(type = "number")]
    before: u32,
}

pub async fn remove_before(
    ctx: RpcContext,
    ModifyNotificationBeforeParams { before }: ModifyNotificationBeforeParams,
) -> Result<(), Error> {
    ctx.db
        .mutate(|db| {
            let n = db.as_private_mut().as_notifications_mut();
            for id in n.keys()?.range(..before) {
                n.remove(&id)?;
            }
            Ok(())
        })
        .await
        .result
}

pub async fn mark_seen(
    ctx: RpcContext,
    ModifyNotificationParams { ids }: ModifyNotificationParams,
) -> Result<(), Error> {
    ctx.db
        .mutate(|db| {
            let mut diff = 0;
            let n = db.as_private_mut().as_notifications_mut();
            for id in ids {
                if !n
                    .as_idx_mut(&id)
                    .or_not_found(lazy_format!("Notification #{id}"))?
                    .as_seen_mut()
                    .replace(&true)?
                {
                    diff += 1;
                }
            }
            db.as_public_mut()
                .as_server_info_mut()
                .as_unread_notification_count_mut()
                .mutate(|n| Ok(*n -= diff))?;
            Ok(())
        })
        .await
        .result
}

pub async fn mark_seen_before(
    ctx: RpcContext,
    ModifyNotificationBeforeParams { before }: ModifyNotificationBeforeParams,
) -> Result<(), Error> {
    ctx.db
        .mutate(|db| {
            let mut diff = 0;
            let n = db.as_private_mut().as_notifications_mut();
            for id in n.keys()?.range(..before) {
                if !n
                    .as_idx_mut(&id)
                    .or_not_found(lazy_format!("Notification #{id}"))?
                    .as_seen_mut()
                    .replace(&true)?
                {
                    diff += 1;
                }
            }
            db.as_public_mut()
                .as_server_info_mut()
                .as_unread_notification_count_mut()
                .mutate(|n| Ok(*n -= diff))?;
            Ok(())
        })
        .await
        .result
}

pub async fn mark_unseen(
    ctx: RpcContext,
    ModifyNotificationParams { ids }: ModifyNotificationParams,
) -> Result<(), Error> {
    ctx.db
        .mutate(|db| {
            let mut diff = 0;
            let n = db.as_private_mut().as_notifications_mut();
            for id in ids {
                if n.as_idx_mut(&id)
                    .or_not_found(lazy_format!("Notification #{id}"))?
                    .as_seen_mut()
                    .replace(&false)?
                {
                    diff += 1;
                }
            }
            db.as_public_mut()
                .as_server_info_mut()
                .as_unread_notification_count_mut()
                .mutate(|n| Ok(*n += diff))?;
            Ok(())
        })
        .await
        .result
}

#[derive(Deserialize, Serialize, Parser, TS)]
#[serde(rename_all = "camelCase")]
#[command(rename_all = "kebab-case")]
pub struct CreateParams {
    package: Option<PackageId>,
    level: NotificationLevel,
    title: String,
    message: String,
}

pub async fn create(
    ctx: RpcContext,
    CreateParams {
        package,
        level,
        title,
        message,
    }: CreateParams,
) -> Result<(), Error> {
    ctx.db
        .mutate(|db| notify(db, package, level, title, message, ()))
        .await
        .result
}

#[derive(Debug, Clone, PartialEq, Eq, Hash, serde::Serialize, serde::Deserialize, TS)]
#[serde(rename_all = "camelCase")]
pub enum NotificationLevel {
    Success,
    Info,
    Warning,
    Error,
}
impl fmt::Display for NotificationLevel {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            NotificationLevel::Success => write!(f, "success"),
            NotificationLevel::Info => write!(f, "info"),
            NotificationLevel::Warning => write!(f, "warning"),
            NotificationLevel::Error => write!(f, "error"),
        }
    }
}
impl ValueParserFactory for NotificationLevel {
    type Parser = FromStrParser<Self>;
    fn value_parser() -> Self::Parser {
        FromStrParser::new()
    }
}

pub struct InvalidNotificationLevel(String);
impl From<InvalidNotificationLevel> for crate::Error {
    fn from(val: InvalidNotificationLevel) -> Self {
        Error::new(
            eyre!("Invalid Notification Level: {}", val.0),
            ErrorKind::ParseDbField,
        )
    }
}
impl FromStr for NotificationLevel {
    type Err = InvalidNotificationLevel;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            s if s == "success" => Ok(NotificationLevel::Success),
            s if s == "info" => Ok(NotificationLevel::Info),
            s if s == "warning" => Ok(NotificationLevel::Warning),
            s if s == "error" => Ok(NotificationLevel::Error),
            s => Err(InvalidNotificationLevel(s.to_string())),
        }
    }
}
impl fmt::Display for InvalidNotificationLevel {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "Invalid Notification Level: {}", self.0)
    }
}

#[derive(Debug, Default, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Notifications(pub BTreeMap<u32, Notification>);
impl Notifications {
    pub fn new() -> Self {
        Self(BTreeMap::new())
    }
}
impl Map for Notifications {
    type Key = u32;
    type Value = Notification;
    fn key_str(key: &Self::Key) -> Result<impl AsRef<str>, Error> {
        Self::key_string(key)
    }
    fn key_string(key: &Self::Key) -> Result<InternedString, Error> {
        Ok(InternedString::from_display(key))
    }
}

#[derive(Debug, Serialize, Deserialize, HasModel)]
#[serde(rename_all = "camelCase")]
#[model = "Model<Self>"]
pub struct Notification {
    pub package_id: Option<PackageId>,
    pub created_at: DateTime<Utc>,
    pub code: u32,
    pub level: NotificationLevel,
    pub title: String,
    pub message: String,
    pub data: Value,
    #[serde(default = "const_true")]
    pub seen: bool,
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct NotificationWithId {
    id: u32,
    #[serde(flatten)]
    notification: Notification,
}

pub trait NotificationType:
    serde::Serialize + for<'de> serde::Deserialize<'de> + std::fmt::Debug
{
    const CODE: u32;
}

impl NotificationType for () {
    const CODE: u32 = 0;
}
impl NotificationType for BackupReport {
    const CODE: u32 = 1;
}
impl NotificationType for String {
    const CODE: u32 = 2;
}

#[instrument(skip(subtype, db))]
pub fn notify<T: NotificationType>(
    db: &mut DatabaseModel,
    package_id: Option<PackageId>,
    level: NotificationLevel,
    title: String,
    message: String,
    subtype: T,
) -> Result<(), Error> {
    let data = to_value(&subtype)?;
    db.as_public_mut()
        .as_server_info_mut()
        .as_unread_notification_count_mut()
        .mutate(|c| {
            *c += 1;
            Ok(())
        })?;
    let id = db
        .as_private()
        .as_notifications()
        .keys()?
        .into_iter()
        .max()
        .map_or(0, |id| id + 1);
    db.as_private_mut().as_notifications_mut().insert(
        &id,
        &Notification {
            package_id,
            created_at: Utc::now(),
            code: T::CODE,
            level,
            title,
            message,
            data,
            seen: false,
        },
    )
}

#[test]
fn serialization() {
    println!(
        "{}",
        serde_json::json!({ "test": "abcdefg", "num": 32, "nested": { "inner": null, "xyz": [0,2,4]}})
    )
}