path: root/filamento/src/roster.rs

use std::collections::HashSet;

use jid::JID;
use rusqlite::{
    ToSql,
    types::{FromSql, ToSqlOutput, Value},
};

pub struct ContactUpdate {
    pub name: Option<String>,
    pub groups: HashSet<String>,
}

#[derive(Debug, Clone)]
pub struct Contact {
    // jid is the id used to reference everything, but not the primary key
    pub user_jid: JID,
    pub subscription: Subscription,
    /// client user defined name
    pub name: Option<String>,
    // TODO: avatar, nickname
    /// nickname picked by contact
    // nickname: Option<String>,
    pub groups: HashSet<String>,
}

#[derive(Debug, Clone)]
pub enum Subscription {
    None,
    PendingOut,
    PendingIn,
    PendingInPendingOut,
    OnlyOut,
    OnlyIn,
    OutPendingIn,
    InPendingOut,
    Buddy,
    // TODO: perhaps don't need, just emit event to remove contact
    // Remove,
}

impl ToSql for Subscription {
    fn to_sql(&self) -> rusqlite::Result<rusqlite::types::ToSqlOutput<'_>> {
        Ok(match self {
            Subscription::None => ToSqlOutput::Owned(Value::Text("none".to_string())),
            Subscription::PendingOut => ToSqlOutput::Owned(Value::Text("pending-out".to_string())),
            Subscription::PendingIn => ToSqlOutput::Owned(Value::Text("pending-in".to_string())),
            Subscription::PendingInPendingOut => {
                ToSqlOutput::Owned(Value::Text("pending-in-pending-out".to_string()))
            }
            Subscription::OnlyOut => ToSqlOutput::Owned(Value::Text("only-out".to_string())),
            Subscription::OnlyIn => ToSqlOutput::Owned(Value::Text("only-in".to_string())),
            Subscription::OutPendingIn => {
                ToSqlOutput::Owned(Value::Text("out-pending-in".to_string()))
            }
            Subscription::InPendingOut => {
                ToSqlOutput::Owned(Value::Text("in-pending-out".to_string()))
            }
            Subscription::Buddy => ToSqlOutput::Owned(Value::Text("buddy".to_string())),
        })
    }
}

impl FromSql for Subscription {
    fn column_result(value: rusqlite::types::ValueRef<'_>) -> rusqlite::types::FromSqlResult<Self> {
        Ok(match value.as_str()? {
            "none" => Self::None,
            "pending-out" => Self::PendingOut,
            "pending-in" => Self::PendingIn,
            "pending-in-pending-out" => Self::PendingInPendingOut,
            "only-out" => Self::OnlyOut,
            "only-in" => Self::OnlyIn,
            "out-pending-in" => Self::OutPendingIn,
            "in-pending-out" => Self::InPendingOut,
            "buddy" => Self::Buddy,
            // TODO: don't have these lol
            value => panic!("unexpected subscription `{value}`"),
        })
    }
}

// none
// >
// >>
// <
// <<
// ><
// >><
// ><<
// >><<

impl From<stanza::roster::Item> for Contact {
    fn from(value: stanza::roster::Item) -> Self {
        let subscription = match value.ask {
            true => match value.subscription {
                Some(s) => match s {
                    stanza::roster::Subscription::Both => Subscription::Buddy,
                    stanza::roster::Subscription::From => Subscription::InPendingOut,
                    stanza::roster::Subscription::None => Subscription::PendingOut,
                    stanza::roster::Subscription::Remove => Subscription::PendingOut,
                    stanza::roster::Subscription::To => Subscription::OnlyOut,
                },
                None => Subscription::PendingOut,
            },
            false => match value.subscription {
                Some(s) => match s {
                    stanza::roster::Subscription::Both => Subscription::Buddy,
                    stanza::roster::Subscription::From => Subscription::OnlyIn,
                    stanza::roster::Subscription::None => Subscription::None,
                    stanza::roster::Subscription::Remove => Subscription::None,
                    stanza::roster::Subscription::To => Subscription::OnlyOut,
                },
                None => Subscription::None,
            },
        };
        Contact {
            user_jid: value.jid,
            subscription,
            name: value.name,
            groups: HashSet::from_iter(value.groups.into_iter().filter_map(|group| group.0)),
        }
    }
}