coop/crates/chat_state/src/room.rs

use chrono::{Datelike, Local, TimeZone};
use common::{
    profile::NostrProfile,
    utils::{compare, random_name, room_hash},
};
use gpui::SharedString;
use nostr_sdk::prelude::*;
use std::collections::HashSet;

pub struct LastSeen(pub Timestamp);

impl LastSeen {
    pub fn ago(&self) -> SharedString {
        let now = Local::now();
        let input_time = Local.timestamp_opt(self.0.as_u64() as i64, 0).unwrap();
        let diff = (now - input_time).num_hours();

        if diff < 24 {
            let duration = now.signed_duration_since(input_time);

            if duration.num_seconds() < 60 {
                "now".to_string().into()
            } else if duration.num_minutes() == 1 {
                "1m".to_string().into()
            } else if duration.num_minutes() < 60 {
                format!("{}m", duration.num_minutes()).into()
            } else if duration.num_hours() == 1 {
                "1h".to_string().into()
            } else if duration.num_hours() < 24 {
                format!("{}h", duration.num_hours()).into()
            } else if duration.num_days() == 1 {
                "1d".to_string().into()
            } else {
                format!("{}d", duration.num_days()).into()
            }
        } else {
            input_time.format("%b %d").to_string().into()
        }
    }

    pub fn human_readable(&self) -> SharedString {
        let now = Local::now();
        let input_time = Local.timestamp_opt(self.0.as_u64() as i64, 0).unwrap();

        if input_time.day() == now.day() {
            format!("Today at {}", input_time.format("%H:%M %p")).into()
        } else if input_time.day() == now.day() - 1 {
            format!("Yesterday at {}", input_time.format("%H:%M %p")).into()
        } else {
            format!(
                "{}, {}",
                input_time.format("%d/%m/%y"),
                input_time.format("%H:%M %p")
            )
            .into()
        }
    }

    pub fn set(&mut self, created_at: Timestamp) {
        self.0 = created_at
    }
}

pub struct Room {
    pub id: u64,
    pub title: Option<SharedString>,
    pub owner: NostrProfile,        // Owner always match current user
    pub members: Vec<NostrProfile>, // Extract from event's tags
    pub last_seen: LastSeen,
    pub is_group: bool,
    pub new_messages: Vec<Event>, // Hold all new messages
}

impl PartialEq for Room {
    fn eq(&self, other: &Self) -> bool {
        let mut pubkeys: Vec<PublicKey> = self.members.iter().map(|m| m.public_key()).collect();
        pubkeys.push(self.owner.public_key());

        let mut pubkeys2: Vec<PublicKey> = other.members.iter().map(|m| m.public_key()).collect();
        pubkeys2.push(other.owner.public_key());

        compare(&pubkeys, &pubkeys2)
    }
}

impl Room {
    pub fn new(
        id: u64,
        owner: NostrProfile,
        members: Vec<NostrProfile>,
        title: Option<SharedString>,
        last_seen: LastSeen,
    ) -> Self {
        let is_group = members.len() > 1;
        let title = if title.is_none() {
            Some(random_name(2).into())
        } else {
            title
        };

        Self {
            id,
            owner,
            members,
            title,
            last_seen,
            is_group,
            new_messages: vec![],
        }
    }

    /// Convert nostr event to room
    pub fn parse(event: &Event) -> Room {
        let id = room_hash(event);
        let last_seen = LastSeen(event.created_at);

        // Always equal to current user
        let owner = NostrProfile::new(event.pubkey, Metadata::default());

        // Get all pubkeys that invole in this group
        let members: Vec<NostrProfile> = event
            .tags
            .public_keys()
            .collect::<HashSet<_>>()
            .into_iter()
            .map(|public_key| NostrProfile::new(*public_key, Metadata::default()))
            .collect();

        // Get title from event's tags
        let title = if let Some(tag) = event.tags.find(TagKind::Subject) {
            tag.content().map(|s| s.to_owned().into())
        } else {
            None
        };

        Self::new(id, owner, members, title, last_seen)
    }

    /// Set contact's metadata by public key
    pub fn set_metadata(&mut self, public_key: PublicKey, metadata: Metadata) {
        if self.owner.public_key() == public_key {
            self.owner.set_metadata(&metadata);
        }

        for member in self.members.iter_mut() {
            if member.public_key() == public_key {
                member.set_metadata(&metadata);
            }
        }
    }

    /// Get room's member by public key
    pub fn member(&self, public_key: &PublicKey) -> Option<NostrProfile> {
        if &self.owner.public_key() == public_key {
            Some(self.owner.clone())
        } else {
            self.members
                .iter()
                .find(|m| &m.public_key() == public_key)
                .cloned()
        }
    }

    /// Get room's display name
    pub fn name(&self) -> String {
        if self.members.len() <= 2 {
            self.members
                .iter()
                .map(|profile| profile.name())
                .collect::<Vec<_>>()
                .join(", ")
        } else {
            let name = self
                .members
                .iter()
                .take(2)
                .map(|profile| profile.name())
                .collect::<Vec<_>>()
                .join(", ");

            format!("{}, +{}", name, self.members.len() - 2)
        }
    }

    /// Get all public keys from current room
    pub fn pubkeys(&self) -> Vec<PublicKey> {
        let mut pubkeys: Vec<_> = self.members.iter().map(|m| m.public_key()).collect();
        pubkeys.push(self.owner.public_key());

        pubkeys
    }
}