deltachat/
reaction.rs

1//! # Reactions.
2//!
3//! Reactions are short messages consisting of emojis sent in reply to
4//! messages. Unlike normal messages which are added to the end of the chat,
5//! reactions are supposed to be displayed near the original messages.
6//!
7//! RFC 9078 specifies how reactions are transmitted in MIME messages.
8//!
9//! Reaction update semantics is not well-defined in RFC 9078, so
10//! Delta Chat uses the same semantics as in
11//! [XEP-0444](https://xmpp.org/extensions/xep-0444.html) section
12//! "3.2 Updating reactions to a message". Received reactions override
13//! all previously received reactions from the same user and it is
14//! possible to remove all reactions by sending an empty string as a reaction,
15//! even though RFC 9078 requires at least one emoji to be sent.
16
17use std::cmp::Ordering;
18use std::collections::BTreeMap;
19use std::fmt;
20
21use anyhow::Result;
22use serde::{Deserialize, Serialize};
23
24use crate::chat::{Chat, ChatId, send_msg};
25use crate::chatlist_events;
26use crate::contact::ContactId;
27use crate::context::Context;
28use crate::events::EventType;
29use crate::message::{Message, MsgId, rfc724_mid_exists};
30use crate::param::Param;
31
32/// A single reaction consisting of multiple emoji sequences.
33///
34/// It is guaranteed to have all emojis sorted and deduplicated inside.
35#[derive(Debug, Default, Clone, Deserialize, Eq, PartialEq, Serialize)]
36pub struct Reaction {
37    /// Canonical representation of reaction as a string of space-separated emojis.
38    reaction: String,
39}
40
41// We implement From<&str> instead of std::str::FromStr, because
42// FromStr requires error type and reaction parsing never returns an
43// error.
44impl From<&str> for Reaction {
45    /// Parses a string containing a reaction.
46    ///
47    /// Reaction string is separated by spaces or tabs (`WSP` in ABNF),
48    /// but this function accepts any ASCII whitespace, so even a CRLF at
49    /// the end of string is acceptable.
50    ///
51    /// Any short enough string is accepted as a reaction to avoid the
52    /// complexity of validating emoji sequences as required by RFC
53    /// 9078. On the sender side UI is responsible to provide only
54    /// valid emoji sequences via reaction picker. On the receiver
55    /// side, abuse of the possibility to use arbitrary strings as
56    /// reactions is not different from other kinds of spam attacks
57    /// such as sending large numbers of large messages, and should be
58    /// dealt with the same way, e.g. by blocking the user.
59    fn from(reaction: &str) -> Self {
60        let mut emojis: Vec<&str> = reaction
61            .split_ascii_whitespace()
62            .filter(|&emoji| emoji.len() < 30)
63            .collect();
64        emojis.sort_unstable();
65        emojis.dedup();
66        let reaction = emojis.join(" ");
67        Self { reaction }
68    }
69}
70
71impl Reaction {
72    /// Returns true if reaction contains no emojis.
73    pub fn is_empty(&self) -> bool {
74        self.reaction.is_empty()
75    }
76
77    /// Returns a vector of emojis composing a reaction.
78    pub fn emojis(&self) -> Vec<&str> {
79        self.reaction.split(' ').collect()
80    }
81
82    /// Returns space-separated string of emojis
83    pub fn as_str(&self) -> &str {
84        &self.reaction
85    }
86
87    /// Appends emojis from another reaction to this reaction.
88    pub fn add(&self, other: Self) -> Self {
89        let mut emojis: Vec<&str> = self.emojis();
90        emojis.append(&mut other.emojis());
91        emojis.sort_unstable();
92        emojis.dedup();
93        let reaction = emojis.join(" ");
94        Self { reaction }
95    }
96}
97
98/// Structure representing all reactions to a particular message.
99#[derive(Debug)]
100pub struct Reactions {
101    /// Map from a contact to its reaction to message.
102    reactions: BTreeMap<ContactId, Reaction>,
103}
104
105impl Reactions {
106    /// Returns vector of contacts that reacted to the message.
107    pub fn contacts(&self) -> Vec<ContactId> {
108        self.reactions.keys().copied().collect()
109    }
110
111    /// Returns reaction of a given contact to message.
112    ///
113    /// If contact did not react to message or removed the reaction,
114    /// this method returns an empty reaction.
115    pub fn get(&self, contact_id: ContactId) -> Reaction {
116        self.reactions.get(&contact_id).cloned().unwrap_or_default()
117    }
118
119    /// Returns true if the message has no reactions.
120    pub fn is_empty(&self) -> bool {
121        self.reactions.is_empty()
122    }
123
124    /// Returns a map from emojis to their frequencies.
125    pub fn emoji_frequencies(&self) -> BTreeMap<String, usize> {
126        let mut emoji_frequencies: BTreeMap<String, usize> = BTreeMap::new();
127        for reaction in self.reactions.values() {
128            for emoji in reaction.emojis() {
129                emoji_frequencies
130                    .entry(emoji.to_string())
131                    .and_modify(|x| *x += 1)
132                    .or_insert(1);
133            }
134        }
135        emoji_frequencies
136    }
137
138    /// Returns a vector of emojis
139    /// sorted in descending order of frequencies.
140    ///
141    /// This function can be used to display the reactions in
142    /// the message bubble in the UIs.
143    pub fn emoji_sorted_by_frequency(&self) -> Vec<(String, usize)> {
144        let mut emoji_frequencies: Vec<(String, usize)> =
145            self.emoji_frequencies().into_iter().collect();
146        emoji_frequencies.sort_by(|(a, a_count), (b, b_count)| {
147            match a_count.cmp(b_count).reverse() {
148                Ordering::Equal => a.cmp(b),
149                other => other,
150            }
151        });
152        emoji_frequencies
153    }
154}
155
156impl fmt::Display for Reactions {
157    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
158        let emoji_frequencies = self.emoji_sorted_by_frequency();
159        let mut first = true;
160        for (emoji, frequency) in emoji_frequencies {
161            if !first {
162                write!(f, " ")?;
163            }
164            first = false;
165            write!(f, "{emoji}{frequency}")?;
166        }
167        Ok(())
168    }
169}
170
171async fn set_msg_id_reaction(
172    context: &Context,
173    msg_id: MsgId,
174    chat_id: ChatId,
175    contact_id: ContactId,
176    timestamp: i64,
177    reaction: &Reaction,
178) -> Result<()> {
179    if reaction.is_empty() {
180        // Simply remove the record instead of setting it to empty string.
181        context
182            .sql
183            .execute(
184                "DELETE FROM reactions
185                 WHERE msg_id = ?1
186                 AND contact_id = ?2",
187                (msg_id, contact_id),
188            )
189            .await?;
190    } else {
191        context
192            .sql
193            .execute(
194                "INSERT INTO reactions (msg_id, contact_id, reaction)
195                 VALUES (?1, ?2, ?3)
196                 ON CONFLICT(msg_id, contact_id)
197                 DO UPDATE SET reaction=excluded.reaction",
198                (msg_id, contact_id, reaction.as_str()),
199            )
200            .await?;
201        let mut chat = Chat::load_from_db(context, chat_id).await?;
202        if chat
203            .param
204            .update_timestamp(Param::LastReactionTimestamp, timestamp)?
205        {
206            chat.param
207                .set_i64(Param::LastReactionMsgId, i64::from(msg_id.to_u32()));
208            chat.param
209                .set_i64(Param::LastReactionContactId, i64::from(contact_id.to_u32()));
210            chat.update_param(context).await?;
211        }
212    }
213
214    context.emit_event(EventType::ReactionsChanged {
215        chat_id,
216        msg_id,
217        contact_id,
218    });
219    chatlist_events::emit_chatlist_item_changed(context, chat_id);
220    Ok(())
221}
222
223/// Sends a reaction to message `msg_id`, overriding previously sent reactions.
224///
225/// `reaction` is a string consisting of space-separated emoji. Use
226/// empty string to retract a reaction.
227pub async fn send_reaction(context: &Context, msg_id: MsgId, reaction: &str) -> Result<MsgId> {
228    let msg = Message::load_from_db(context, msg_id).await?;
229    let chat_id = msg.chat_id;
230
231    let reaction: Reaction = reaction.into();
232    let mut reaction_msg = Message::new_text(reaction.as_str().to_string());
233    reaction_msg.set_reaction();
234    reaction_msg.in_reply_to = Some(msg.rfc724_mid);
235    reaction_msg.hidden = true;
236
237    // Send message first.
238    let reaction_msg_id = send_msg(context, chat_id, &mut reaction_msg).await?;
239
240    // Only set reaction if we successfully sent the message.
241    set_msg_id_reaction(
242        context,
243        msg_id,
244        msg.chat_id,
245        ContactId::SELF,
246        reaction_msg.timestamp_sort,
247        &reaction,
248    )
249    .await?;
250    Ok(reaction_msg_id)
251}
252
253/// Adds given reaction to message `msg_id` and sends an update.
254///
255/// This can be used to implement advanced clients that allow reacting
256/// with multiple emojis. For a simple messenger UI, you probably want
257/// to use [`send_reaction()`] instead so reacting with a new emoji
258/// removes previous emoji at the same time.
259pub async fn add_reaction(context: &Context, msg_id: MsgId, reaction: &str) -> Result<MsgId> {
260    let self_reaction = get_self_reaction(context, msg_id).await?;
261    let reaction = self_reaction.add(Reaction::from(reaction));
262    send_reaction(context, msg_id, reaction.as_str()).await
263}
264
265/// Updates reaction of `contact_id` on the message with `in_reply_to`
266/// Message-ID. If no such message is found in the database, reaction
267/// is ignored.
268///
269/// `reaction` is a space-separated string of emojis. It can be empty
270/// if contact wants to remove all reactions.
271pub(crate) async fn set_msg_reaction(
272    context: &Context,
273    in_reply_to: &str,
274    chat_id: ChatId,
275    contact_id: ContactId,
276    timestamp: i64,
277    reaction: Reaction,
278    is_incoming_fresh: bool,
279) -> Result<()> {
280    if let Some(msg_id) = rfc724_mid_exists(context, in_reply_to).await? {
281        set_msg_id_reaction(context, msg_id, chat_id, contact_id, timestamp, &reaction).await?;
282
283        if is_incoming_fresh
284            && !reaction.is_empty()
285            && msg_id.get_state(context).await?.is_outgoing()
286        {
287            context.emit_event(EventType::IncomingReaction {
288                chat_id,
289                contact_id,
290                msg_id,
291                reaction,
292            });
293        }
294    } else {
295        info!(
296            context,
297            "Can't assign reaction to unknown message with Message-ID {}", in_reply_to
298        );
299    }
300    Ok(())
301}
302
303/// Get our own reaction for a given message.
304async fn get_self_reaction(context: &Context, msg_id: MsgId) -> Result<Reaction> {
305    let reaction_str: Option<String> = context
306        .sql
307        .query_get_value(
308            "SELECT reaction
309             FROM reactions
310             WHERE msg_id=? AND contact_id=?",
311            (msg_id, ContactId::SELF),
312        )
313        .await?;
314    Ok(reaction_str
315        .as_deref()
316        .map(Reaction::from)
317        .unwrap_or_default())
318}
319
320/// Returns a structure containing all reactions to the message.
321pub async fn get_msg_reactions(context: &Context, msg_id: MsgId) -> Result<Reactions> {
322    let reactions: BTreeMap<ContactId, Reaction> = context
323        .sql
324        .query_map_collect(
325            "SELECT contact_id, reaction FROM reactions WHERE msg_id=?",
326            (msg_id,),
327            |row| {
328                let contact_id: ContactId = row.get(0)?;
329                let reaction: String = row.get(1)?;
330                Ok((contact_id, Reaction::from(reaction.as_str())))
331            },
332        )
333        .await?;
334    Ok(Reactions { reactions })
335}
336
337impl Chat {
338    /// Check if there is a reaction newer than the given timestamp.
339    ///
340    /// If so, reaction details are returned and can be used to create a summary string.
341    pub async fn get_last_reaction_if_newer_than(
342        &self,
343        context: &Context,
344        timestamp: i64,
345    ) -> Result<Option<(Message, ContactId, String)>> {
346        if self
347            .param
348            .get_i64(Param::LastReactionTimestamp)
349            .filter(|&reaction_timestamp| reaction_timestamp > timestamp)
350            .is_none()
351        {
352            return Ok(None);
353        };
354        let reaction_msg_id = MsgId::new(
355            self.param
356                .get_int(Param::LastReactionMsgId)
357                .unwrap_or_default() as u32,
358        );
359        let Some(reaction_msg) = Message::load_from_db_optional(context, reaction_msg_id).await?
360        else {
361            // The message reacted to may be deleted.
362            // These are no errors as `Param::LastReaction*` are just weak pointers.
363            // Instead, just return `Ok(None)` and let the caller create another summary.
364            return Ok(None);
365        };
366        let reaction_contact_id = ContactId::new(
367            self.param
368                .get_int(Param::LastReactionContactId)
369                .unwrap_or_default() as u32,
370        );
371        if let Some(reaction) = context
372            .sql
373            .query_get_value(
374                "SELECT reaction FROM reactions WHERE msg_id=? AND contact_id=?",
375                (reaction_msg.id, reaction_contact_id),
376            )
377            .await?
378        {
379            Ok(Some((reaction_msg, reaction_contact_id, reaction)))
380        } else {
381            Ok(None)
382        }
383    }
384}
385
386#[cfg(test)]
387mod tests {
388    use deltachat_contact_tools::ContactAddress;
389
390    use super::*;
391    use crate::chat::{forward_msgs, get_chat_msgs, marknoticed_chat, send_text_msg};
392    use crate::chatlist::Chatlist;
393    use crate::config::Config;
394    use crate::contact::{Contact, Origin};
395    use crate::download::DownloadState;
396    use crate::message::{MessageState, Viewtype, delete_msgs};
397    use crate::receive_imf::{receive_imf, receive_imf_from_inbox};
398    use crate::sql::housekeeping;
399    use crate::test_utils::E2EE_INFO_MSGS;
400    use crate::test_utils::TestContext;
401    use crate::test_utils::TestContextManager;
402    use crate::tools::SystemTime;
403    use std::time::Duration;
404
405    #[test]
406    fn test_parse_reaction() {
407        // Check that basic set of emojis from RFC 9078 is supported.
408        assert_eq!(Reaction::from("๐Ÿ‘").emojis(), vec!["๐Ÿ‘"]);
409        assert_eq!(Reaction::from("๐Ÿ‘Ž").emojis(), vec!["๐Ÿ‘Ž"]);
410        assert_eq!(Reaction::from("๐Ÿ˜€").emojis(), vec!["๐Ÿ˜€"]);
411        assert_eq!(Reaction::from("โ˜น").emojis(), vec!["โ˜น"]);
412        assert_eq!(Reaction::from("๐Ÿ˜ข").emojis(), vec!["๐Ÿ˜ข"]);
413
414        // Empty string can be used to remove all reactions.
415        assert!(Reaction::from("").is_empty());
416
417        // Short strings can be used as emojis, could be used to add
418        // support for custom emojis via emoji shortcodes.
419        assert_eq!(Reaction::from(":deltacat:").emojis(), vec![":deltacat:"]);
420
421        // Check that long strings are not valid emojis.
422        assert!(
423            Reaction::from(":foobarbazquuxaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa:").is_empty()
424        );
425
426        // Multiple reactions separated by spaces or tabs are supported.
427        assert_eq!(Reaction::from("๐Ÿ‘ โค").emojis(), vec!["โค", "๐Ÿ‘"]);
428        assert_eq!(Reaction::from("๐Ÿ‘\tโค").emojis(), vec!["โค", "๐Ÿ‘"]);
429
430        // Invalid emojis are removed, but valid emojis are retained.
431        assert_eq!(
432            Reaction::from("๐Ÿ‘\t:foo: โค").emojis(),
433            vec![":foo:", "โค", "๐Ÿ‘"]
434        );
435        assert_eq!(Reaction::from("๐Ÿ‘\t:foo: โค").as_str(), ":foo: โค ๐Ÿ‘");
436
437        // Duplicates are removed.
438        assert_eq!(Reaction::from("๐Ÿ‘ ๐Ÿ‘").emojis(), vec!["๐Ÿ‘"]);
439    }
440
441    #[test]
442    fn test_add_reaction() {
443        let reaction1 = Reaction::from("๐Ÿ‘ ๐Ÿ˜€");
444        let reaction2 = Reaction::from("โค");
445        let reaction_sum = reaction1.add(reaction2);
446
447        assert_eq!(reaction_sum.emojis(), vec!["โค", "๐Ÿ‘", "๐Ÿ˜€"]);
448    }
449
450    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
451    async fn test_receive_reaction() -> Result<()> {
452        let alice = TestContext::new_alice().await;
453
454        // Alice receives BCC-self copy of a message sent to Bob.
455        receive_imf(
456            &alice,
457            "To: bob@example.net\n\
458From: alice@example.org\n\
459Date: Today, 29 February 2021 00:00:00 -800\n\
460Message-ID: 12345@example.org\n\
461Subject: Meeting\n\
462\n\
463Can we chat at 1pm pacific, today?"
464                .as_bytes(),
465            false,
466        )
467        .await?;
468        let msg = alice.get_last_msg().await;
469        assert_eq!(msg.state, MessageState::OutDelivered);
470        let reactions = get_msg_reactions(&alice, msg.id).await?;
471        let contacts = reactions.contacts();
472        assert_eq!(contacts.len(), 0);
473
474        let bob_id = Contact::add_or_lookup(
475            &alice,
476            "",
477            &ContactAddress::new("bob@example.net")?,
478            Origin::ManuallyCreated,
479        )
480        .await?
481        .0;
482        let bob_reaction = reactions.get(bob_id);
483        assert!(bob_reaction.is_empty()); // Bob has not reacted to message yet.
484
485        // Alice receives reaction to her message from Bob.
486        receive_imf(
487            &alice,
488            "To: alice@example.org\n\
489From: bob@example.net\n\
490Date: Today, 29 February 2021 00:00:10 -800\n\
491Message-ID: 56789@example.net\n\
492In-Reply-To: 12345@example.org\n\
493Subject: Meeting\n\
494Mime-Version: 1.0 (1.0)\n\
495Content-Type: text/plain; charset=utf-8\n\
496Content-Disposition: reaction\n\
497\n\
498\u{1F44D}"
499                .as_bytes(),
500            false,
501        )
502        .await?;
503
504        let reactions = get_msg_reactions(&alice, msg.id).await?;
505        assert_eq!(reactions.to_string(), "๐Ÿ‘1");
506
507        let contacts = reactions.contacts();
508        assert_eq!(contacts.len(), 1);
509
510        assert_eq!(contacts.first(), Some(&bob_id));
511        let bob_reaction = reactions.get(bob_id);
512        assert_eq!(bob_reaction.is_empty(), false);
513        assert_eq!(bob_reaction.emojis(), vec!["๐Ÿ‘"]);
514        assert_eq!(bob_reaction.as_str(), "๐Ÿ‘");
515
516        // Alice receives reaction to her message from Bob with a footer.
517        receive_imf(
518            &alice,
519            "To: alice@example.org\n\
520From: bob@example.net\n\
521Date: Today, 29 February 2021 00:00:10 -800\n\
522Message-ID: 56790@example.net\n\
523In-Reply-To: 12345@example.org\n\
524Subject: Meeting\n\
525Mime-Version: 1.0 (1.0)\n\
526Content-Type: text/plain; charset=utf-8\n\
527Content-Disposition: reaction\n\
528\n\
529๐Ÿ˜€\n\
530\n\
531--\n\
532_______________________________________________\n\
533Here's my footer -- bob@example.net"
534                .as_bytes(),
535            false,
536        )
537        .await?;
538
539        let reactions = get_msg_reactions(&alice, msg.id).await?;
540        assert_eq!(reactions.to_string(), "๐Ÿ˜€1");
541
542        // Alice receives a message with reaction to her message from Bob.
543        let msg_bob = receive_imf(
544            &alice,
545            "To: alice@example.org\n\
546From: bob@example.net\n\
547Date: Today, 29 February 2021 00:00:10 -800\n\
548Message-ID: 56791@example.net\n\
549In-Reply-To: 12345@example.org\n\
550Mime-Version: 1.0\n\
551Content-Type: multipart/mixed; boundary=\"YiEDa0DAkWCtVeE4\"\n\
552Content-Disposition: inline\n\
553\n\
554--YiEDa0DAkWCtVeE4\n\
555Content-Type: text/plain; charset=utf-8\n\
556Content-Disposition: inline\n\
557\n\
558Reply + reaction\n\
559\n\
560--YiEDa0DAkWCtVeE4\n\
561Content-Type: text/plain; charset=utf-8\n\
562Content-Disposition: reaction\n\
563\n\
564\u{1F44D}\n\
565\n\
566--YiEDa0DAkWCtVeE4--"
567                .as_bytes(),
568            false,
569        )
570        .await?
571        .unwrap();
572        let msg_bob = Message::load_from_db(&alice, msg_bob.msg_ids[0]).await?;
573        assert_eq!(msg_bob.from_id, bob_id);
574        assert_eq!(msg_bob.chat_id, msg.chat_id);
575        assert_eq!(msg_bob.viewtype, Viewtype::Text);
576        assert_eq!(msg_bob.state, MessageState::InFresh);
577        assert_eq!(msg_bob.hidden, false);
578        assert_eq!(msg_bob.text, "Reply + reaction");
579        let reactions = get_msg_reactions(&alice, msg.id).await?;
580        assert_eq!(reactions.to_string(), "๐Ÿ‘1");
581
582        Ok(())
583    }
584
585    async fn expect_reactions_changed_event(
586        t: &TestContext,
587        expected_chat_id: ChatId,
588        expected_msg_id: MsgId,
589        expected_contact_id: ContactId,
590    ) -> Result<()> {
591        let event = t
592            .evtracker
593            .get_matching(|evt| {
594                matches!(
595                    evt,
596                    EventType::ReactionsChanged { .. } | EventType::IncomingMsg { .. }
597                )
598            })
599            .await;
600        match event {
601            EventType::ReactionsChanged {
602                chat_id,
603                msg_id,
604                contact_id,
605            } => {
606                assert_eq!(chat_id, expected_chat_id);
607                assert_eq!(msg_id, expected_msg_id);
608                assert_eq!(contact_id, expected_contact_id);
609            }
610            _ => panic!("Unexpected event {event:?}."),
611        }
612        Ok(())
613    }
614
615    async fn expect_incoming_reactions_event(
616        t: &TestContext,
617        expected_chat_id: ChatId,
618        expected_msg_id: MsgId,
619        expected_contact_id: ContactId,
620        expected_reaction: &str,
621    ) -> Result<()> {
622        let event = t
623            .evtracker
624            // Check for absence of `IncomingMsg` events -- it appeared that it's quite easy to make
625            // bugs when `IncomingMsg` is issued for reactions.
626            .get_matching(|evt| {
627                matches!(
628                    evt,
629                    EventType::IncomingReaction { .. } | EventType::IncomingMsg { .. }
630                )
631            })
632            .await;
633        match event {
634            EventType::IncomingReaction {
635                chat_id,
636                msg_id,
637                contact_id,
638                reaction,
639            } => {
640                assert_eq!(chat_id, expected_chat_id);
641                assert_eq!(msg_id, expected_msg_id);
642                assert_eq!(contact_id, expected_contact_id);
643                assert_eq!(reaction, Reaction::from(expected_reaction));
644            }
645            _ => panic!("Unexpected event {event:?}."),
646        }
647        Ok(())
648    }
649
650    /// Checks that no unwanted events remain after expecting "wanted" reaction events.
651    async fn expect_no_unwanted_events(t: &TestContext) {
652        let ev = t
653            .evtracker
654            .get_matching_opt(t, |evt| {
655                matches!(
656                    evt,
657                    EventType::IncomingReaction { .. }
658                        | EventType::IncomingMsg { .. }
659                        | EventType::MsgsChanged { .. }
660                )
661            })
662            .await;
663        if let Some(ev) = ev {
664            panic!("Unwanted event {ev:?}.")
665        }
666    }
667
668    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
669    async fn test_send_reaction() -> Result<()> {
670        let alice = TestContext::new_alice().await;
671        let bob = TestContext::new_bob().await;
672
673        // Test that the status does not get mixed up into reactions.
674        alice
675            .set_config(
676                Config::Selfstatus,
677                Some("Buy Delta Chat today and make this banner go away!"),
678            )
679            .await?;
680        bob.set_config(Config::Selfstatus, Some("Sent from my Delta Chat Pro. ๐Ÿ‘"))
681            .await?;
682
683        let chat_alice = alice.create_chat(&bob).await;
684        let alice_msg = alice.send_text(chat_alice.id, "Hi!").await;
685        let bob_msg = bob.recv_msg(&alice_msg).await;
686        assert_eq!(
687            get_chat_msgs(&alice, chat_alice.id).await?.len(),
688            E2EE_INFO_MSGS + 1
689        );
690        assert_eq!(
691            get_chat_msgs(&bob, bob_msg.chat_id).await?.len(),
692            E2EE_INFO_MSGS + 1
693        );
694
695        let alice_msg2 = alice.send_text(chat_alice.id, "Hi again!").await;
696        bob.recv_msg(&alice_msg2).await;
697        assert_eq!(
698            get_chat_msgs(&alice, chat_alice.id).await?.len(),
699            E2EE_INFO_MSGS + 2
700        );
701        assert_eq!(
702            get_chat_msgs(&bob, bob_msg.chat_id).await?.len(),
703            E2EE_INFO_MSGS + 2
704        );
705
706        bob_msg.chat_id.accept(&bob).await?;
707
708        bob.evtracker.clear_events();
709        send_reaction(&bob, bob_msg.id, "๐Ÿ‘").await.unwrap();
710        expect_reactions_changed_event(&bob, bob_msg.chat_id, bob_msg.id, ContactId::SELF).await?;
711        expect_no_unwanted_events(&bob).await;
712        assert_eq!(
713            get_chat_msgs(&bob, bob_msg.chat_id).await?.len(),
714            E2EE_INFO_MSGS + 2
715        );
716
717        let bob_reaction_msg = bob.pop_sent_msg().await;
718        let alice_reaction_msg = alice.recv_msg_hidden(&bob_reaction_msg).await;
719        assert_eq!(alice_reaction_msg.state, MessageState::InFresh);
720        assert_eq!(
721            get_chat_msgs(&alice, chat_alice.id).await?.len(),
722            E2EE_INFO_MSGS + 2
723        );
724
725        let reactions = get_msg_reactions(&alice, alice_msg.sender_msg_id).await?;
726        assert_eq!(reactions.to_string(), "๐Ÿ‘1");
727        let contacts = reactions.contacts();
728        assert_eq!(contacts.len(), 1);
729        let bob_id = contacts.first().unwrap();
730        let bob_reaction = reactions.get(*bob_id);
731        assert_eq!(bob_reaction.is_empty(), false);
732        assert_eq!(bob_reaction.emojis(), vec!["๐Ÿ‘"]);
733        assert_eq!(bob_reaction.as_str(), "๐Ÿ‘");
734        expect_reactions_changed_event(&alice, chat_alice.id, alice_msg.sender_msg_id, *bob_id)
735            .await?;
736        expect_incoming_reactions_event(
737            &alice,
738            chat_alice.id,
739            alice_msg.sender_msg_id,
740            *bob_id,
741            "๐Ÿ‘",
742        )
743        .await?;
744        expect_no_unwanted_events(&alice).await;
745
746        marknoticed_chat(&alice, chat_alice.id).await?;
747        assert_eq!(
748            alice_reaction_msg.id.get_state(&alice).await?,
749            MessageState::InSeen
750        );
751        // Reactions don't request MDNs.
752        assert_eq!(
753            alice
754                .sql
755                .count("SELECT COUNT(*) FROM smtp_mdns", ())
756                .await?,
757            0
758        );
759
760        // Alice reacts to own message.
761        send_reaction(&alice, alice_msg.sender_msg_id, "๐Ÿ‘ ๐Ÿ˜€")
762            .await
763            .unwrap();
764        let reactions = get_msg_reactions(&alice, alice_msg.sender_msg_id).await?;
765        assert_eq!(reactions.to_string(), "๐Ÿ‘2 ๐Ÿ˜€1");
766
767        assert_eq!(
768            reactions.emoji_sorted_by_frequency(),
769            vec![("๐Ÿ‘".to_string(), 2), ("๐Ÿ˜€".to_string(), 1)]
770        );
771
772        Ok(())
773    }
774
775    async fn assert_summary(t: &TestContext, expected: &str) {
776        let chatlist = Chatlist::try_load(t, 0, None, None).await.unwrap();
777        let summary = chatlist.get_summary(t, 0, None).await.unwrap();
778        assert_eq!(summary.text, expected);
779    }
780
781    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
782    async fn test_reaction_summary() -> Result<()> {
783        let mut tcm = TestContextManager::new();
784        let alice = tcm.alice().await;
785        let bob = tcm.bob().await;
786        alice.set_config(Config::Displayname, Some("ALICE")).await?;
787        bob.set_config(Config::Displayname, Some("BOB")).await?;
788        let alice_bob_id = alice.add_or_lookup_contact_id(&bob).await;
789
790        // Alice sends message to Bob
791        let alice_chat = alice.create_chat(&bob).await;
792        let alice_msg1 = alice.send_text(alice_chat.id, "Party?").await;
793        let bob_msg1 = bob.recv_msg(&alice_msg1).await;
794
795        // Bob reacts to Alice's message, this is shown in the summaries
796        SystemTime::shift(Duration::from_secs(10));
797        bob_msg1.chat_id.accept(&bob).await?;
798        send_reaction(&bob, bob_msg1.id, "๐Ÿ‘").await?;
799        let bob_send_reaction = bob.pop_sent_msg().await;
800        alice.recv_msg_hidden(&bob_send_reaction).await;
801        expect_incoming_reactions_event(
802            &alice,
803            alice_chat.id,
804            alice_msg1.sender_msg_id,
805            alice_bob_id,
806            "๐Ÿ‘",
807        )
808        .await?;
809        expect_no_unwanted_events(&alice).await;
810
811        let chatlist = Chatlist::try_load(&bob, 0, None, None).await?;
812        let summary = chatlist.get_summary(&bob, 0, None).await?;
813        assert_eq!(summary.text, "You reacted ๐Ÿ‘ to \"Party?\"");
814        assert_eq!(summary.timestamp, bob_msg1.get_timestamp()); // time refers to message, not to reaction
815        assert_eq!(summary.state, MessageState::InFresh); // state refers to message, not to reaction
816        assert!(summary.prefix.is_none());
817        assert!(summary.thumbnail_path.is_none());
818        assert_summary(&alice, "BOB reacted ๐Ÿ‘ to \"Party?\"").await;
819
820        // Alice reacts to own message as well
821        SystemTime::shift(Duration::from_secs(10));
822        send_reaction(&alice, alice_msg1.sender_msg_id, "๐Ÿฟ").await?;
823        let alice_send_reaction = alice.pop_sent_msg().await;
824        bob.evtracker.clear_events();
825        bob.recv_msg_opt(&alice_send_reaction).await;
826        expect_no_unwanted_events(&bob).await;
827
828        assert_summary(&alice, "You reacted ๐Ÿฟ to \"Party?\"").await;
829        assert_summary(&bob, "ALICE reacted ๐Ÿฟ to \"Party?\"").await;
830
831        // Alice sends a newer message, this overwrites reaction summaries
832        SystemTime::shift(Duration::from_secs(10));
833        let alice_msg2 = alice.send_text(alice_chat.id, "kewl").await;
834        bob.recv_msg(&alice_msg2).await;
835
836        assert_summary(&alice, "kewl").await;
837        assert_summary(&bob, "kewl").await;
838
839        // Reactions to older messages still overwrite newer messages
840        SystemTime::shift(Duration::from_secs(10));
841        send_reaction(&alice, alice_msg1.sender_msg_id, "๐Ÿค˜").await?;
842        let alice_send_reaction = alice.pop_sent_msg().await;
843        bob.recv_msg_opt(&alice_send_reaction).await;
844
845        assert_summary(&alice, "You reacted ๐Ÿค˜ to \"Party?\"").await;
846        assert_summary(&bob, "ALICE reacted ๐Ÿค˜ to \"Party?\"").await;
847
848        // Retracted reactions remove all summary reactions
849        SystemTime::shift(Duration::from_secs(10));
850        send_reaction(&alice, alice_msg1.sender_msg_id, "").await?;
851        let alice_remove_reaction = alice.pop_sent_msg().await;
852        bob.recv_msg_opt(&alice_remove_reaction).await;
853
854        assert_summary(&alice, "kewl").await;
855        assert_summary(&bob, "kewl").await;
856
857        // Alice adds another reaction and then deletes the message reacted to; this will also delete reaction summary
858        SystemTime::shift(Duration::from_secs(10));
859        send_reaction(&alice, alice_msg1.sender_msg_id, "๐Ÿงน").await?;
860        assert_summary(&alice, "You reacted ๐Ÿงน to \"Party?\"").await;
861
862        delete_msgs(&alice, &[alice_msg1.sender_msg_id]).await?; // this will leave a tombstone
863        assert_summary(&alice, "kewl").await;
864        housekeeping(&alice).await?; // this will delete the tombstone
865        assert_summary(&alice, "kewl").await;
866
867        Ok(())
868    }
869
870    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
871    async fn test_reaction_forwarded_summary() -> Result<()> {
872        let alice = TestContext::new_alice().await;
873
874        // Alice adds a message to "Saved Messages"
875        let self_chat = alice.get_self_chat().await;
876        let msg_id = send_text_msg(&alice, self_chat.id, "foo".to_string()).await?;
877        assert_summary(&alice, "foo").await;
878
879        // Alice reacts to that message
880        SystemTime::shift(Duration::from_secs(10));
881        send_reaction(&alice, msg_id, "๐Ÿซ").await?;
882        assert_summary(&alice, "You reacted ๐Ÿซ to \"foo\"").await;
883        let reactions = get_msg_reactions(&alice, msg_id).await?;
884        assert_eq!(reactions.reactions.len(), 1);
885
886        // Alice forwards that message to Bob: Reactions are not forwarded, the message is prefixed by "Forwarded".
887        let bob_id = Contact::create(&alice, "", "bob@example.net").await?;
888        let bob_chat_id = ChatId::create_for_contact(&alice, bob_id).await?;
889        forward_msgs(&alice, &[msg_id], bob_chat_id).await?;
890        assert_summary(&alice, "Forwarded: foo").await; // forwarded messages are prefixed
891        let chatlist = Chatlist::try_load(&alice, 0, None, None).await.unwrap();
892        let forwarded_msg_id = chatlist.get_msg_id(0)?.unwrap();
893        let reactions = get_msg_reactions(&alice, forwarded_msg_id).await?;
894        assert!(reactions.reactions.is_empty()); // reactions are not forwarded
895
896        // Alice reacts to forwarded message:
897        // For reaction summary neither original message author nor "Forwarded" prefix is shown
898        SystemTime::shift(Duration::from_secs(10));
899        send_reaction(&alice, forwarded_msg_id, "๐Ÿณ").await?;
900        assert_summary(&alice, "You reacted ๐Ÿณ to \"foo\"").await;
901        let reactions = get_msg_reactions(&alice, msg_id).await?;
902        assert_eq!(reactions.reactions.len(), 1);
903
904        Ok(())
905    }
906
907    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
908    async fn test_reaction_self_chat_multidevice_summary() -> Result<()> {
909        let alice0 = TestContext::new_alice().await;
910        let alice1 = TestContext::new_alice().await;
911        let chat = alice0.get_self_chat().await;
912
913        let msg_id = send_text_msg(&alice0, chat.id, "mom's birthday!".to_string()).await?;
914        alice1.recv_msg(&alice0.pop_sent_msg().await).await;
915
916        SystemTime::shift(Duration::from_secs(10));
917        send_reaction(&alice0, msg_id, "๐Ÿ‘†").await?;
918        let sync = alice0.pop_sent_msg().await;
919        receive_imf(&alice1, sync.payload().as_bytes(), false).await?;
920
921        assert_summary(&alice0, "You reacted ๐Ÿ‘† to \"mom's birthday!\"").await;
922        assert_summary(&alice1, "You reacted ๐Ÿ‘† to \"mom's birthday!\"").await;
923
924        Ok(())
925    }
926
927    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
928    async fn test_partial_download_and_reaction() -> Result<()> {
929        let alice = TestContext::new_alice().await;
930        let bob = TestContext::new_bob().await;
931
932        alice
933            .create_chat_with_contact("Bob", "bob@example.net")
934            .await;
935
936        let msg_header = "From: Bob <bob@example.net>\n\
937                    To: Alice <alice@example.org>\n\
938                    Chat-Version: 1.0\n\
939                    Subject: subject\n\
940                    Message-ID: <first@example.org>\n\
941                    Date: Sun, 14 Nov 2021 00:10:00 +0000\
942                    Content-Type: text/plain";
943        let msg_full = format!("{msg_header}\n\n100k text...");
944
945        // Alice downloads message from Bob partially.
946        let alice_received_message = receive_imf_from_inbox(
947            &alice,
948            "first@example.org",
949            msg_header.as_bytes(),
950            false,
951            Some(100000),
952        )
953        .await?
954        .unwrap();
955        let alice_msg_id = *alice_received_message.msg_ids.first().unwrap();
956
957        // Bob downloads own message on the other device.
958        let bob_received_message = receive_imf(&bob, msg_full.as_bytes(), false)
959            .await?
960            .unwrap();
961        let bob_msg_id = *bob_received_message.msg_ids.first().unwrap();
962
963        // Bob reacts to own message.
964        send_reaction(&bob, bob_msg_id, "๐Ÿ‘").await.unwrap();
965        let bob_reaction_msg = bob.pop_sent_msg().await;
966
967        // Alice receives a reaction.
968        alice.recv_msg_hidden(&bob_reaction_msg).await;
969
970        let reactions = get_msg_reactions(&alice, alice_msg_id).await?;
971        assert_eq!(reactions.to_string(), "๐Ÿ‘1");
972        let msg = Message::load_from_db(&alice, alice_msg_id).await?;
973        assert_eq!(msg.download_state(), DownloadState::Available);
974
975        // Alice downloads full message.
976        receive_imf_from_inbox(
977            &alice,
978            "first@example.org",
979            msg_full.as_bytes(),
980            false,
981            None,
982        )
983        .await?;
984
985        // Check that reaction is still on the message after full download.
986        let msg = Message::load_from_db(&alice, alice_msg_id).await?;
987        assert_eq!(msg.download_state(), DownloadState::Done);
988        let reactions = get_msg_reactions(&alice, alice_msg_id).await?;
989        assert_eq!(reactions.to_string(), "๐Ÿ‘1");
990
991        Ok(())
992    }
993
994    /// Regression test for reaction resetting self-status.
995    ///
996    /// Reactions do not contain the status,
997    /// but should not result in self-status being reset on other devices.
998    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
999    async fn test_reaction_status_multidevice() -> Result<()> {
1000        let mut tcm = TestContextManager::new();
1001        let alice1 = tcm.alice().await;
1002        let alice2 = tcm.alice().await;
1003
1004        alice1
1005            .set_config(Config::Selfstatus, Some("New status"))
1006            .await?;
1007
1008        let alice2_msg = tcm.send_recv(&alice1, &alice2, "Hi!").await;
1009        assert_eq!(
1010            alice2.get_config(Config::Selfstatus).await?.as_deref(),
1011            Some("New status")
1012        );
1013
1014        // Alice reacts to own message from second device,
1015        // first device receives rection.
1016        {
1017            send_reaction(&alice2, alice2_msg.id, "๐Ÿ‘").await?;
1018            let msg = alice2.pop_sent_msg().await;
1019            alice1.recv_msg_hidden(&msg).await;
1020        }
1021
1022        // Check that the status is still the same.
1023        assert_eq!(
1024            alice1.get_config(Config::Selfstatus).await?.as_deref(),
1025            Some("New status")
1026        );
1027        Ok(())
1028    }
1029
1030    #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
1031    async fn test_send_reaction_multidevice() -> Result<()> {
1032        let mut tcm = TestContextManager::new();
1033        let alice0 = tcm.alice().await;
1034        let alice1 = tcm.alice().await;
1035        let bob = tcm.bob().await;
1036        let chat_id = alice0.create_chat(&bob).await.id;
1037
1038        let alice0_msg_id = send_text_msg(&alice0, chat_id, "foo".to_string()).await?;
1039        let alice1_msg = alice1.recv_msg(&alice0.pop_sent_msg().await).await;
1040
1041        send_reaction(&alice0, alice0_msg_id, "๐Ÿ‘€").await?;
1042        alice1.recv_msg_hidden(&alice0.pop_sent_msg().await).await;
1043
1044        expect_reactions_changed_event(&alice0, chat_id, alice0_msg_id, ContactId::SELF).await?;
1045        expect_reactions_changed_event(&alice1, alice1_msg.chat_id, alice1_msg.id, ContactId::SELF)
1046            .await?;
1047        for a in [&alice0, &alice1] {
1048            expect_no_unwanted_events(a).await;
1049        }
1050        Ok(())
1051    }
1052}