1use std::cmp;
4use std::collections::{BTreeSet, HashMap};
5use std::fmt;
6use std::io::Cursor;
7use std::marker::Sync;
8use std::path::{Path, PathBuf};
9use std::str::FromStr;
10use std::time::Duration;
11
12use anyhow::{Context as _, Result, anyhow, bail, ensure};
13use chrono::TimeZone;
14use deltachat_contact_tools::{ContactAddress, sanitize_bidi_characters, sanitize_single_line};
15use humansize::{BINARY, format_size};
16use mail_builder::mime::MimePart;
17use serde::{Deserialize, Serialize};
18use strum_macros::EnumIter;
19
20use crate::blob::BlobObject;
21use crate::chatlist::Chatlist;
22use crate::chatlist_events;
23use crate::color::str_to_color;
24use crate::config::Config;
25use crate::constants::{
26 self, Blocked, Chattype, DC_CHAT_ID_ALLDONE_HINT, DC_CHAT_ID_ARCHIVED_LINK,
27 DC_CHAT_ID_LAST_SPECIAL, DC_CHAT_ID_TRASH, DC_RESEND_USER_AVATAR_DAYS, EDITED_PREFIX,
28 TIMESTAMP_SENT_TOLERANCE,
29};
30use crate::contact::{self, Contact, ContactId, Origin};
31use crate::context::Context;
32use crate::debug_logging::maybe_set_logging_xdc;
33use crate::download::{
34 DownloadState, PRE_MSG_ATTACHMENT_SIZE_THRESHOLD, PRE_MSG_SIZE_WARNING_THRESHOLD,
35};
36use crate::ephemeral::{Timer as EphemeralTimer, start_chat_ephemeral_timers};
37use crate::events::EventType;
38use crate::key::{Fingerprint, self_fingerprint};
39use crate::location;
40use crate::log::{LogExt, warn};
41use crate::logged_debug_assert;
42use crate::message::{self, Message, MessageState, MsgId, Viewtype};
43use crate::mimefactory::{MimeFactory, RenderedEmail};
44use crate::mimeparser::SystemMessage;
45use crate::param::{Param, Params};
46use crate::pgp::addresses_from_public_key;
47use crate::receive_imf::ReceivedMsg;
48use crate::smtp::{self, send_msg_to_smtp};
49use crate::stock_str;
50use crate::sync::{self, Sync::*, SyncData};
51use crate::tools::{
52 IsNoneOrEmpty, SystemTime, buf_compress, create_broadcast_secret, create_id,
53 create_outgoing_rfc724_mid, create_smeared_timestamp, create_smeared_timestamps, get_abs_path,
54 gm2local_offset, normalize_text, smeared_time, time, truncate_msg_text,
55};
56use crate::webxdc::StatusUpdateSerial;
57
58pub(crate) const PARAM_BROADCAST_SECRET: Param = Param::Arg3;
59
60#[derive(Debug, Copy, Clone, PartialEq, Eq)]
62pub enum ChatItem {
63 Message {
65 msg_id: MsgId,
67 },
68
69 DayMarker {
72 timestamp: i64,
74 },
75}
76
77#[derive(Debug, Clone, Copy, PartialEq, Eq)]
81pub(crate) enum CantSendReason {
82 SpecialChat,
84
85 DeviceChat,
87
88 ContactRequest,
90
91 ReadOnlyMailingList,
93
94 InBroadcast,
96
97 NotAMember,
99
100 MissingKey,
102}
103
104impl fmt::Display for CantSendReason {
105 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
106 match self {
107 Self::SpecialChat => write!(f, "the chat is a special chat"),
108 Self::DeviceChat => write!(f, "the chat is a device chat"),
109 Self::ContactRequest => write!(
110 f,
111 "contact request chat should be accepted before sending messages"
112 ),
113 Self::ReadOnlyMailingList => {
114 write!(f, "mailing list does not have a know post address")
115 }
116 Self::InBroadcast => {
117 write!(f, "Broadcast channel is read-only")
118 }
119 Self::NotAMember => write!(f, "not a member of the chat"),
120 Self::MissingKey => write!(f, "key is missing"),
121 }
122 }
123}
124
125#[derive(
130 Debug, Copy, Clone, Default, PartialEq, Eq, Serialize, Deserialize, Hash, PartialOrd, Ord,
131)]
132pub struct ChatId(u32);
133
134impl ChatId {
135 pub const fn new(id: u32) -> ChatId {
137 ChatId(id)
138 }
139
140 pub fn is_unset(self) -> bool {
144 self.0 == 0
145 }
146
147 pub fn is_special(self) -> bool {
151 (0..=DC_CHAT_ID_LAST_SPECIAL.0).contains(&self.0)
152 }
153
154 pub fn is_trash(self) -> bool {
161 self == DC_CHAT_ID_TRASH
162 }
163
164 pub fn is_archived_link(self) -> bool {
171 self == DC_CHAT_ID_ARCHIVED_LINK
172 }
173
174 pub fn is_alldone_hint(self) -> bool {
183 self == DC_CHAT_ID_ALLDONE_HINT
184 }
185
186 pub(crate) fn lookup_by_message(msg: &Message) -> Option<Self> {
188 if msg.chat_id == DC_CHAT_ID_TRASH {
189 return None;
190 }
191 if msg.download_state == DownloadState::Undecipherable {
192 return None;
193 }
194 Some(msg.chat_id)
195 }
196
197 pub async fn lookup_by_contact(
202 context: &Context,
203 contact_id: ContactId,
204 ) -> Result<Option<Self>> {
205 let Some(chat_id_blocked) = ChatIdBlocked::lookup_by_contact(context, contact_id).await?
206 else {
207 return Ok(None);
208 };
209
210 let chat_id = match chat_id_blocked.blocked {
211 Blocked::Not | Blocked::Request => Some(chat_id_blocked.id),
212 Blocked::Yes => None,
213 };
214 Ok(chat_id)
215 }
216
217 pub(crate) async fn get_for_contact(context: &Context, contact_id: ContactId) -> Result<Self> {
225 ChatIdBlocked::get_for_contact(context, contact_id, Blocked::Not)
226 .await
227 .map(|chat| chat.id)
228 }
229
230 pub async fn create_for_contact(context: &Context, contact_id: ContactId) -> Result<Self> {
235 ChatId::create_for_contact_with_blocked(context, contact_id, Blocked::Not).await
236 }
237
238 pub(crate) async fn create_for_contact_with_blocked(
242 context: &Context,
243 contact_id: ContactId,
244 create_blocked: Blocked,
245 ) -> Result<Self> {
246 let chat_id = match ChatIdBlocked::lookup_by_contact(context, contact_id).await? {
247 Some(chat) => {
248 if create_blocked != Blocked::Not || chat.blocked == Blocked::Not {
249 return Ok(chat.id);
250 }
251 chat.id.set_blocked(context, Blocked::Not).await?;
252 chat.id
253 }
254 None => {
255 if Contact::real_exists_by_id(context, contact_id).await?
256 || contact_id == ContactId::SELF
257 {
258 let chat_id =
259 ChatIdBlocked::get_for_contact(context, contact_id, create_blocked)
260 .await
261 .map(|chat| chat.id)?;
262 if create_blocked != Blocked::Yes {
263 info!(context, "Scale up origin of {contact_id} to CreateChat.");
264 ContactId::scaleup_origin(context, &[contact_id], Origin::CreateChat)
265 .await?;
266 }
267 chat_id
268 } else {
269 warn!(
270 context,
271 "Cannot create chat, contact {contact_id} does not exist."
272 );
273 bail!("Can not create chat for non-existing contact");
274 }
275 }
276 };
277 context.emit_msgs_changed_without_ids();
278 chatlist_events::emit_chatlist_changed(context);
279 chatlist_events::emit_chatlist_item_changed(context, chat_id);
280 Ok(chat_id)
281 }
282
283 pub(crate) async fn create_multiuser_record(
286 context: &Context,
287 chattype: Chattype,
288 grpid: &str,
289 grpname: &str,
290 create_blocked: Blocked,
291 param: Option<String>,
292 timestamp: i64,
293 ) -> Result<Self> {
294 let grpname = sanitize_single_line(grpname);
295 let timestamp = cmp::min(timestamp, smeared_time(context));
296 let row_id =
297 context.sql.insert(
298 "INSERT INTO chats (type, name, name_normalized, grpid, blocked, created_timestamp, protected, param) VALUES(?, ?, ?, ?, ?, ?, 0, ?)",
299 (
300 chattype,
301 &grpname,
302 normalize_text(&grpname),
303 grpid,
304 create_blocked,
305 timestamp,
306 param.unwrap_or_default(),
307 ),
308 ).await?;
309
310 let chat_id = ChatId::new(u32::try_from(row_id)?);
311 let chat = Chat::load_from_db(context, chat_id).await?;
312
313 if chat.is_encrypted(context).await? {
314 chat_id.add_e2ee_notice(context, timestamp).await?;
315 }
316
317 info!(
318 context,
319 "Created group/broadcast '{}' grpid={} as {}, blocked={}.",
320 &grpname,
321 grpid,
322 chat_id,
323 create_blocked,
324 );
325
326 Ok(chat_id)
327 }
328
329 async fn set_selfavatar_timestamp(self, context: &Context, timestamp: i64) -> Result<()> {
330 context
331 .sql
332 .execute(
333 "UPDATE contacts
334 SET selfavatar_sent=?
335 WHERE id IN(SELECT contact_id FROM chats_contacts WHERE chat_id=? AND add_timestamp >= remove_timestamp)",
336 (timestamp, self),
337 )
338 .await?;
339 Ok(())
340 }
341
342 pub(crate) async fn set_blocked(self, context: &Context, new_blocked: Blocked) -> Result<bool> {
346 if self.is_special() {
347 bail!("ignoring setting of Block-status for {self}");
348 }
349 let count = context
350 .sql
351 .execute(
352 "UPDATE chats SET blocked=?1 WHERE id=?2 AND blocked != ?1",
353 (new_blocked, self),
354 )
355 .await?;
356 Ok(count > 0)
357 }
358
359 pub async fn block(self, context: &Context) -> Result<()> {
361 self.block_ex(context, Sync).await
362 }
363
364 pub(crate) async fn block_ex(self, context: &Context, sync: sync::Sync) -> Result<()> {
365 let chat = Chat::load_from_db(context, self).await?;
366 let mut delete = false;
367
368 match chat.typ {
369 Chattype::OutBroadcast => {
370 bail!("Can't block chat of type {:?}", chat.typ)
371 }
372 Chattype::Single => {
373 for contact_id in get_chat_contacts(context, self).await? {
374 if contact_id != ContactId::SELF {
375 info!(
376 context,
377 "Blocking the contact {contact_id} to block 1:1 chat."
378 );
379 contact::set_blocked(context, Nosync, contact_id, true).await?;
380 }
381 }
382 }
383 Chattype::Group => {
384 info!(context, "Can't block groups yet, deleting the chat.");
385 delete = true;
386 }
387 Chattype::Mailinglist | Chattype::InBroadcast => {
388 if self.set_blocked(context, Blocked::Yes).await? {
389 context.emit_event(EventType::ChatModified(self));
390 }
391 }
392 }
393 chatlist_events::emit_chatlist_changed(context);
394
395 if sync.into() {
396 chat.sync(context, SyncAction::Block)
398 .await
399 .log_err(context)
400 .ok();
401 }
402 if delete {
403 self.delete_ex(context, Nosync).await?;
404 }
405 Ok(())
406 }
407
408 pub async fn unblock(self, context: &Context) -> Result<()> {
410 self.unblock_ex(context, Sync).await
411 }
412
413 pub(crate) async fn unblock_ex(self, context: &Context, sync: sync::Sync) -> Result<()> {
414 self.set_blocked(context, Blocked::Not).await?;
415
416 chatlist_events::emit_chatlist_changed(context);
417
418 if sync.into() {
419 let chat = Chat::load_from_db(context, self).await?;
420 chat.sync(context, SyncAction::Unblock)
424 .await
425 .log_err(context)
426 .ok();
427 }
428
429 Ok(())
430 }
431
432 pub async fn accept(self, context: &Context) -> Result<()> {
436 self.accept_ex(context, Sync).await
437 }
438
439 pub(crate) async fn accept_ex(self, context: &Context, sync: sync::Sync) -> Result<()> {
440 let chat = Chat::load_from_db(context, self).await?;
441
442 match chat.typ {
443 Chattype::Single | Chattype::Group | Chattype::OutBroadcast | Chattype::InBroadcast => {
444 let origin = match chat.typ {
450 Chattype::Group => Origin::IncomingTo,
451 _ => Origin::CreateChat,
452 };
453 for contact_id in get_chat_contacts(context, self).await? {
454 if contact_id != ContactId::SELF {
455 ContactId::scaleup_origin(context, &[contact_id], origin).await?;
456 }
457 }
458 }
459 Chattype::Mailinglist => {
460 }
462 }
463
464 if self.set_blocked(context, Blocked::Not).await? {
465 context.emit_event(EventType::ChatModified(self));
466 chatlist_events::emit_chatlist_item_changed(context, self);
467 }
468
469 if sync.into() {
470 chat.sync(context, SyncAction::Accept)
471 .await
472 .log_err(context)
473 .ok();
474 }
475 Ok(())
476 }
477
478 pub(crate) async fn add_e2ee_notice(self, context: &Context, timestamp: i64) -> Result<()> {
480 let text = stock_str::messages_e2ee_info_msg(context);
481
482 let sort_timestamp = 0;
486 add_info_msg_with_cmd(
487 context,
488 self,
489 &text,
490 SystemMessage::ChatE2ee,
491 Some(sort_timestamp),
492 timestamp,
493 None,
494 None,
495 None,
496 )
497 .await?;
498 Ok(())
499 }
500
501 pub(crate) async fn add_start_info_message(self, context: &Context, text: &str) -> Result<()> {
506 let sort_timestamp = 0;
507 add_info_msg_with_cmd(
508 context,
509 self,
510 text,
511 SystemMessage::Unknown,
512 Some(sort_timestamp),
513 time(),
514 None,
515 None,
516 None,
517 )
518 .await?;
519 Ok(())
520 }
521
522 pub async fn set_visibility(self, context: &Context, visibility: ChatVisibility) -> Result<()> {
524 self.set_visibility_ex(context, Sync, visibility).await
525 }
526
527 pub(crate) async fn set_visibility_ex(
528 self,
529 context: &Context,
530 sync: sync::Sync,
531 visibility: ChatVisibility,
532 ) -> Result<()> {
533 ensure!(
534 !self.is_special(),
535 "bad chat_id, can not be special chat: {self}"
536 );
537
538 context
539 .sql
540 .transaction(move |transaction| {
541 if visibility == ChatVisibility::Archived {
542 transaction.execute(
543 "UPDATE msgs SET state=? WHERE chat_id=? AND state=?;",
544 (MessageState::InNoticed, self, MessageState::InFresh),
545 )?;
546 }
547 transaction.execute(
548 "UPDATE chats SET archived=? WHERE id=?;",
549 (visibility, self),
550 )?;
551 Ok(())
552 })
553 .await?;
554
555 if visibility == ChatVisibility::Archived {
556 start_chat_ephemeral_timers(context, self).await?;
557 }
558
559 context.emit_msgs_changed_without_ids();
560 chatlist_events::emit_chatlist_changed(context);
561 chatlist_events::emit_chatlist_item_changed(context, self);
562
563 if sync.into() {
564 let chat = Chat::load_from_db(context, self).await?;
565 chat.sync(context, SyncAction::SetVisibility(visibility))
566 .await
567 .log_err(context)
568 .ok();
569 }
570 Ok(())
571 }
572
573 pub async fn unarchive_if_not_muted(
581 self,
582 context: &Context,
583 msg_state: MessageState,
584 ) -> Result<()> {
585 if msg_state != MessageState::InFresh {
586 context
587 .sql
588 .execute(
589 "UPDATE chats SET archived=0 WHERE id=? AND archived=1 \
590 AND NOT(muted_until=-1 OR muted_until>?)",
591 (self, time()),
592 )
593 .await?;
594 return Ok(());
595 }
596 let chat = Chat::load_from_db(context, self).await?;
597 if chat.visibility != ChatVisibility::Archived {
598 return Ok(());
599 }
600 if chat.is_muted() {
601 let unread_cnt = context
602 .sql
603 .count(
604 "SELECT COUNT(*)
605 FROM msgs
606 WHERE state=?
607 AND hidden=0
608 AND chat_id=?",
609 (MessageState::InFresh, self),
610 )
611 .await?;
612 if unread_cnt == 1 {
613 context.emit_msgs_changed_without_msg_id(DC_CHAT_ID_ARCHIVED_LINK);
615 }
616 return Ok(());
617 }
618 context
619 .sql
620 .execute("UPDATE chats SET archived=0 WHERE id=?", (self,))
621 .await?;
622 Ok(())
623 }
624
625 pub(crate) fn emit_msg_event(self, context: &Context, msg_id: MsgId, important: bool) {
628 if important {
629 debug_assert!(!msg_id.is_unset());
630
631 context.emit_incoming_msg(self, msg_id);
632 } else {
633 context.emit_msgs_changed(self, msg_id);
634 }
635 }
636
637 pub async fn delete(self, context: &Context) -> Result<()> {
643 self.delete_ex(context, Sync).await
644 }
645
646 pub(crate) async fn delete_ex(self, context: &Context, sync: sync::Sync) -> Result<()> {
647 ensure!(
648 !self.is_special(),
649 "bad chat_id, can not be a special chat: {self}"
650 );
651
652 let chat = Chat::load_from_db(context, self).await?;
653 let sync_id = match sync {
654 Nosync => None,
655 Sync => chat.get_sync_id(context).await?,
656 };
657
658 context
659 .sql
660 .transaction(|transaction| {
661 transaction.execute(
662 "UPDATE imap SET target='' WHERE rfc724_mid IN (SELECT rfc724_mid FROM msgs WHERE chat_id=? AND rfc724_mid!='')",
663 (self,),
664 )?;
665 transaction.execute(
666 "UPDATE imap SET target='' WHERE rfc724_mid IN (SELECT pre_rfc724_mid FROM msgs WHERE chat_id=? AND pre_rfc724_mid!='')",
667 (self,),
668 )?;
669 transaction.execute(
670 "DELETE FROM msgs_mdns WHERE msg_id IN (SELECT id FROM msgs WHERE chat_id=?)",
671 (self,),
672 )?;
673 transaction.execute(
676 "
677INSERT OR REPLACE INTO msgs (id, rfc724_mid, pre_rfc724_mid, timestamp, chat_id, deleted)
678SELECT id, rfc724_mid, pre_rfc724_mid, timestamp, ?, 1 FROM msgs WHERE chat_id=?
679 ",
680 (DC_CHAT_ID_TRASH, self),
681 )?;
682 transaction.execute("DELETE FROM chats_contacts WHERE chat_id=?", (self,))?;
683 transaction.execute("DELETE FROM chats WHERE id=?", (self,))?;
684 Ok(())
685 })
686 .await?;
687
688 context.emit_event(EventType::ChatDeleted { chat_id: self });
689 context.emit_msgs_changed_without_ids();
690
691 if let Some(id) = sync_id {
692 self::sync(context, id, SyncAction::Delete)
693 .await
694 .log_err(context)
695 .ok();
696 }
697
698 if chat.is_self_talk() {
699 let mut msg = Message::new_text(stock_str::self_deleted_msg_body(context));
700 add_device_msg(context, None, Some(&mut msg)).await?;
701 }
702 chatlist_events::emit_chatlist_changed(context);
703
704 context
705 .set_config_internal(Config::LastHousekeeping, None)
706 .await?;
707 context.scheduler.interrupt_smtp().await;
708
709 Ok(())
710 }
711
712 pub async fn set_draft(self, context: &Context, mut msg: Option<&mut Message>) -> Result<()> {
716 if self.is_special() {
717 return Ok(());
718 }
719
720 let changed = match &mut msg {
721 None => self.maybe_delete_draft(context).await?,
722 Some(msg) => self.do_set_draft(context, msg).await?,
723 };
724
725 if changed {
726 if msg.is_some() {
727 match self.get_draft_msg_id(context).await? {
728 Some(msg_id) => context.emit_msgs_changed(self, msg_id),
729 None => context.emit_msgs_changed_without_msg_id(self),
730 }
731 } else {
732 context.emit_msgs_changed_without_msg_id(self)
733 }
734 }
735
736 Ok(())
737 }
738
739 async fn get_draft_msg_id(self, context: &Context) -> Result<Option<MsgId>> {
741 let msg_id: Option<MsgId> = context
742 .sql
743 .query_get_value(
744 "SELECT id FROM msgs WHERE chat_id=? AND state=?;",
745 (self, MessageState::OutDraft),
746 )
747 .await?;
748 Ok(msg_id)
749 }
750
751 pub async fn get_draft(self, context: &Context) -> Result<Option<Message>> {
753 if self.is_special() {
754 return Ok(None);
755 }
756 match self.get_draft_msg_id(context).await? {
757 Some(draft_msg_id) => {
758 let msg = Message::load_from_db(context, draft_msg_id).await?;
759 Ok(Some(msg))
760 }
761 None => Ok(None),
762 }
763 }
764
765 async fn maybe_delete_draft(self, context: &Context) -> Result<bool> {
769 Ok(context
770 .sql
771 .execute(
772 "DELETE FROM msgs WHERE chat_id=? AND state=?",
773 (self, MessageState::OutDraft),
774 )
775 .await?
776 > 0)
777 }
778
779 async fn do_set_draft(self, context: &Context, msg: &mut Message) -> Result<bool> {
782 match msg.viewtype {
783 Viewtype::Unknown => bail!("Can not set draft of unknown type."),
784 Viewtype::Text => {
785 if msg.text.is_empty() && msg.in_reply_to.is_none_or_empty() {
786 bail!("No text and no quote in draft");
787 }
788 }
789 _ => {
790 if msg.viewtype == Viewtype::File
791 && let Some((better_type, _)) = message::guess_msgtype_from_suffix(msg)
792 .filter(|&(vt, _)| vt == Viewtype::Webxdc || vt == Viewtype::Vcard)
797 {
798 msg.viewtype = better_type;
799 }
800 if msg.viewtype == Viewtype::Vcard {
801 let blob = msg
802 .param
803 .get_file_blob(context)?
804 .context("no file stored in params")?;
805 msg.try_set_vcard(context, &blob.to_abs_path()).await?;
806 }
807 }
808 }
809
810 msg.state = MessageState::OutDraft;
813 msg.chat_id = self;
814
815 if !msg.id.is_special()
817 && let Some(old_draft) = self.get_draft(context).await?
818 && old_draft.id == msg.id
819 && old_draft.chat_id == self
820 && old_draft.state == MessageState::OutDraft
821 {
822 let affected_rows = context
823 .sql.execute(
824 "UPDATE msgs
825 SET timestamp=?1,type=?2,txt=?3,txt_normalized=?4,param=?5,mime_in_reply_to=?6
826 WHERE id=?7
827 AND (type <> ?2
828 OR txt <> ?3
829 OR txt_normalized <> ?4
830 OR param <> ?5
831 OR mime_in_reply_to <> ?6);",
832 (
833 time(),
834 msg.viewtype,
835 &msg.text,
836 normalize_text(&msg.text),
837 msg.param.to_string(),
838 msg.in_reply_to.as_deref().unwrap_or_default(),
839 msg.id,
840 ),
841 ).await?;
842 return Ok(affected_rows > 0);
843 }
844
845 let row_id = context
846 .sql
847 .transaction(|transaction| {
848 transaction.execute(
850 "DELETE FROM msgs WHERE chat_id=? AND state=?",
851 (self, MessageState::OutDraft),
852 )?;
853
854 transaction.execute(
856 "INSERT INTO msgs (
857 chat_id,
858 rfc724_mid,
859 from_id,
860 timestamp,
861 type,
862 state,
863 txt,
864 txt_normalized,
865 param,
866 hidden,
867 mime_in_reply_to)
868 VALUES (?,?,?,?,?,?,?,?,?,?,?);",
869 (
870 self,
871 &msg.rfc724_mid,
872 ContactId::SELF,
873 time(),
874 msg.viewtype,
875 MessageState::OutDraft,
876 &msg.text,
877 normalize_text(&msg.text),
878 msg.param.to_string(),
879 1,
880 msg.in_reply_to.as_deref().unwrap_or_default(),
881 ),
882 )?;
883
884 Ok(transaction.last_insert_rowid())
885 })
886 .await?;
887 msg.id = MsgId::new(row_id.try_into()?);
888 Ok(true)
889 }
890
891 pub async fn get_msg_cnt(self, context: &Context) -> Result<usize> {
893 let count = context
894 .sql
895 .count(
896 "SELECT COUNT(*) FROM msgs WHERE hidden=0 AND chat_id=?",
897 (self,),
898 )
899 .await?;
900 Ok(count)
901 }
902
903 pub async fn get_fresh_msg_cnt(self, context: &Context) -> Result<usize> {
905 let count = if self.is_archived_link() {
916 context
917 .sql
918 .count(
919 "SELECT COUNT(DISTINCT(m.chat_id))
920 FROM msgs m
921 LEFT JOIN chats c ON m.chat_id=c.id
922 WHERE m.state=10
923 and m.hidden=0
924 AND m.chat_id>9
925 AND c.blocked=0
926 AND c.archived=1
927 ",
928 (),
929 )
930 .await?
931 } else {
932 context
933 .sql
934 .count(
935 "SELECT COUNT(*)
936 FROM msgs
937 WHERE state=?
938 AND hidden=0
939 AND chat_id=?;",
940 (MessageState::InFresh, self),
941 )
942 .await?
943 };
944 Ok(count)
945 }
946
947 pub(crate) async fn created_timestamp(self, context: &Context) -> Result<i64> {
948 Ok(context
949 .sql
950 .query_get_value("SELECT created_timestamp FROM chats WHERE id=?", (self,))
951 .await?
952 .unwrap_or(0))
953 }
954
955 pub(crate) async fn join_timestamp(self, context: &Context) -> Result<Option<i64>> {
957 context
958 .sql
959 .query_get_value(
960 "SELECT add_timestamp FROM chats_contacts WHERE chat_id=? AND contact_id=?",
961 (self, ContactId::SELF),
962 )
963 .await
964 }
965
966 pub(crate) async fn get_timestamp(self, context: &Context) -> Result<Option<i64>> {
969 let timestamp = context
970 .sql
971 .query_get_value(
972 "SELECT MAX(timestamp)
973 FROM msgs
974 WHERE chat_id=?
975 HAVING COUNT(*) > 0",
976 (self,),
977 )
978 .await?;
979 Ok(timestamp)
980 }
981
982 #[expect(clippy::arithmetic_side_effects)]
988 pub async fn get_similar_chat_ids(self, context: &Context) -> Result<Vec<(ChatId, f64)>> {
989 let intersection = context
991 .sql
992 .query_map_vec(
993 "SELECT y.chat_id, SUM(x.contact_id = y.contact_id)
994 FROM chats_contacts as x
995 JOIN chats_contacts as y
996 WHERE x.contact_id > 9
997 AND y.contact_id > 9
998 AND x.add_timestamp >= x.remove_timestamp
999 AND y.add_timestamp >= y.remove_timestamp
1000 AND x.chat_id=?
1001 AND y.chat_id<>x.chat_id
1002 AND y.chat_id>?
1003 GROUP BY y.chat_id",
1004 (self, DC_CHAT_ID_LAST_SPECIAL),
1005 |row| {
1006 let chat_id: ChatId = row.get(0)?;
1007 let intersection: f64 = row.get(1)?;
1008 Ok((chat_id, intersection))
1009 },
1010 )
1011 .await
1012 .context("failed to calculate member set intersections")?;
1013
1014 let chat_size: HashMap<ChatId, f64> = context
1015 .sql
1016 .query_map_collect(
1017 "SELECT chat_id, count(*) AS n
1018 FROM chats_contacts
1019 WHERE contact_id > ? AND chat_id > ?
1020 AND add_timestamp >= remove_timestamp
1021 GROUP BY chat_id",
1022 (ContactId::LAST_SPECIAL, DC_CHAT_ID_LAST_SPECIAL),
1023 |row| {
1024 let chat_id: ChatId = row.get(0)?;
1025 let size: f64 = row.get(1)?;
1026 Ok((chat_id, size))
1027 },
1028 )
1029 .await
1030 .context("failed to count chat member sizes")?;
1031
1032 let our_chat_size = chat_size.get(&self).copied().unwrap_or_default();
1033 let mut chats_with_metrics = Vec::new();
1034 for (chat_id, intersection_size) in intersection {
1035 if intersection_size > 0.0 {
1036 let other_chat_size = chat_size.get(&chat_id).copied().unwrap_or_default();
1037 let union_size = our_chat_size + other_chat_size - intersection_size;
1038 let metric = intersection_size / union_size;
1039 chats_with_metrics.push((chat_id, metric))
1040 }
1041 }
1042 chats_with_metrics.sort_unstable_by(|(chat_id1, metric1), (chat_id2, metric2)| {
1043 metric2
1044 .partial_cmp(metric1)
1045 .unwrap_or(chat_id2.cmp(chat_id1))
1046 });
1047
1048 let mut res = Vec::new();
1050 let now = time();
1051 for (chat_id, metric) in chats_with_metrics {
1052 if let Some(chat_timestamp) = chat_id.get_timestamp(context).await?
1053 && now > chat_timestamp + 42 * 24 * 3600
1054 {
1055 continue;
1057 }
1058
1059 if metric < 0.1 {
1060 break;
1062 }
1063
1064 let chat = Chat::load_from_db(context, chat_id).await?;
1065 if chat.typ != Chattype::Group {
1066 continue;
1067 }
1068
1069 match chat.visibility {
1070 ChatVisibility::Normal | ChatVisibility::Pinned => {}
1071 ChatVisibility::Archived => continue,
1072 }
1073
1074 res.push((chat_id, metric));
1075 if res.len() >= 5 {
1076 break;
1077 }
1078 }
1079
1080 Ok(res)
1081 }
1082
1083 pub async fn get_similar_chatlist(self, context: &Context) -> Result<Chatlist> {
1087 let chat_ids: Vec<ChatId> = self
1088 .get_similar_chat_ids(context)
1089 .await
1090 .context("failed to get similar chat IDs")?
1091 .into_iter()
1092 .map(|(chat_id, _metric)| chat_id)
1093 .collect();
1094 let chatlist = Chatlist::from_chat_ids(context, &chat_ids).await?;
1095 Ok(chatlist)
1096 }
1097
1098 pub(crate) async fn get_param(self, context: &Context) -> Result<Params> {
1099 let res: Option<String> = context
1100 .sql
1101 .query_get_value("SELECT param FROM chats WHERE id=?", (self,))
1102 .await?;
1103 Ok(res
1104 .map(|s| s.parse().unwrap_or_default())
1105 .unwrap_or_default())
1106 }
1107
1108 pub(crate) async fn is_unpromoted(self, context: &Context) -> Result<bool> {
1110 let param = self.get_param(context).await?;
1111 let unpromoted = param.get_bool(Param::Unpromoted).unwrap_or_default();
1112 Ok(unpromoted)
1113 }
1114
1115 pub(crate) async fn is_promoted(self, context: &Context) -> Result<bool> {
1117 let promoted = !self.is_unpromoted(context).await?;
1118 Ok(promoted)
1119 }
1120
1121 pub async fn is_self_talk(self, context: &Context) -> Result<bool> {
1123 Ok(self.get_param(context).await?.exists(Param::Selftalk))
1124 }
1125
1126 pub async fn is_device_talk(self, context: &Context) -> Result<bool> {
1128 Ok(self.get_param(context).await?.exists(Param::Devicetalk))
1129 }
1130
1131 async fn parent_query<T, F>(
1132 self,
1133 context: &Context,
1134 fields: &str,
1135 state_out_min: MessageState,
1136 f: F,
1137 ) -> Result<Option<T>>
1138 where
1139 F: Send + FnOnce(&rusqlite::Row) -> rusqlite::Result<T>,
1140 T: Send + 'static,
1141 {
1142 let sql = &context.sql;
1143 let query = format!(
1144 "SELECT {fields} \
1145 FROM msgs \
1146 WHERE chat_id=? \
1147 AND ((state BETWEEN {} AND {}) OR (state >= {})) \
1148 AND NOT hidden \
1149 AND download_state={} \
1150 AND from_id != {} \
1151 ORDER BY timestamp DESC, id DESC \
1152 LIMIT 1;",
1153 MessageState::InFresh as u32,
1154 MessageState::InSeen as u32,
1155 state_out_min as u32,
1156 DownloadState::Done as u32,
1159 ContactId::INFO.to_u32(),
1162 );
1163 sql.query_row_optional(&query, (self,), f).await
1164 }
1165
1166 async fn get_parent_mime_headers(
1167 self,
1168 context: &Context,
1169 state_out_min: MessageState,
1170 ) -> Result<Option<(String, String, String)>> {
1171 self.parent_query(
1172 context,
1173 "rfc724_mid, mime_in_reply_to, IFNULL(mime_references, '')",
1174 state_out_min,
1175 |row: &rusqlite::Row| {
1176 let rfc724_mid: String = row.get(0)?;
1177 let mime_in_reply_to: String = row.get(1)?;
1178 let mime_references: String = row.get(2)?;
1179 Ok((rfc724_mid, mime_in_reply_to, mime_references))
1180 },
1181 )
1182 .await
1183 }
1184
1185 pub async fn get_encryption_info(self, context: &Context) -> Result<String> {
1193 let chat = Chat::load_from_db(context, self).await?;
1194 if !chat.is_encrypted(context).await? {
1195 return Ok(stock_str::encr_none(context));
1196 }
1197
1198 let mut ret = stock_str::messages_are_e2ee(context) + "\n";
1199
1200 for &contact_id in get_chat_contacts(context, self)
1201 .await?
1202 .iter()
1203 .filter(|&contact_id| !contact_id.is_special())
1204 {
1205 let contact = Contact::get_by_id(context, contact_id).await?;
1206 let addr = contact.get_addr();
1207 logged_debug_assert!(
1208 context,
1209 contact.is_key_contact(),
1210 "get_encryption_info: contact {contact_id} is not a key-contact."
1211 );
1212 let fingerprint = contact
1213 .fingerprint()
1214 .context("Contact does not have a fingerprint in encrypted chat")?;
1215 if let Some(public_key) = contact.public_key(context).await? {
1216 if let Some(relay_addrs) = addresses_from_public_key(&public_key) {
1217 let relays = relay_addrs.join(",");
1218 ret += &format!("\n{addr}({relays})\n{fingerprint}\n");
1219 } else {
1220 ret += &format!("\n{addr}\n{fingerprint}\n");
1221 }
1222 } else {
1223 ret += &format!("\n{addr}\n(key missing)\n{fingerprint}\n");
1224 }
1225 }
1226
1227 Ok(ret.trim().to_string())
1228 }
1229
1230 pub fn to_u32(self) -> u32 {
1235 self.0
1236 }
1237
1238 pub(crate) async fn reset_gossiped_timestamp(self, context: &Context) -> Result<()> {
1239 context
1240 .sql
1241 .execute("DELETE FROM gossip_timestamp WHERE chat_id=?", (self,))
1242 .await?;
1243 Ok(())
1244 }
1245
1246 pub(crate) async fn calc_sort_timestamp(
1253 self,
1254 context: &Context,
1255 message_timestamp: i64,
1256 always_sort_to_bottom: bool,
1257 ) -> Result<i64> {
1258 let mut sort_timestamp = cmp::min(message_timestamp, smeared_time(context));
1259
1260 let last_msg_time: Option<i64> = if always_sort_to_bottom {
1261 context
1267 .sql
1268 .query_get_value(
1269 "SELECT MAX(timestamp)
1270 FROM msgs
1271 WHERE chat_id=? AND state!=?
1272 HAVING COUNT(*) > 0",
1273 (self, MessageState::OutDraft),
1274 )
1275 .await?
1276 } else {
1277 None
1278 };
1279
1280 if let Some(last_msg_time) = last_msg_time
1281 && last_msg_time > sort_timestamp
1282 {
1283 sort_timestamp = last_msg_time;
1284 }
1285
1286 if let Some(join_timestamp) = self.join_timestamp(context).await? {
1287 Ok(std::cmp::max(sort_timestamp, join_timestamp))
1293 } else {
1294 Ok(sort_timestamp)
1295 }
1296 }
1297}
1298
1299impl std::fmt::Display for ChatId {
1300 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
1301 if self.is_trash() {
1302 write!(f, "Chat#Trash")
1303 } else if self.is_archived_link() {
1304 write!(f, "Chat#ArchivedLink")
1305 } else if self.is_alldone_hint() {
1306 write!(f, "Chat#AlldoneHint")
1307 } else if self.is_special() {
1308 write!(f, "Chat#Special{}", self.0)
1309 } else {
1310 write!(f, "Chat#{}", self.0)
1311 }
1312 }
1313}
1314
1315impl rusqlite::types::ToSql for ChatId {
1320 fn to_sql(&self) -> rusqlite::Result<rusqlite::types::ToSqlOutput<'_>> {
1321 let val = rusqlite::types::Value::Integer(i64::from(self.0));
1322 let out = rusqlite::types::ToSqlOutput::Owned(val);
1323 Ok(out)
1324 }
1325}
1326
1327impl rusqlite::types::FromSql for ChatId {
1329 fn column_result(value: rusqlite::types::ValueRef) -> rusqlite::types::FromSqlResult<Self> {
1330 i64::column_result(value).and_then(|val| {
1331 if 0 <= val && val <= i64::from(u32::MAX) {
1332 Ok(ChatId::new(val as u32))
1333 } else {
1334 Err(rusqlite::types::FromSqlError::OutOfRange(val))
1335 }
1336 })
1337 }
1338}
1339
1340#[derive(Debug, Clone, Deserialize, Serialize)]
1345pub struct Chat {
1346 pub id: ChatId,
1348
1349 pub typ: Chattype,
1351
1352 pub name: String,
1354
1355 pub visibility: ChatVisibility,
1357
1358 pub grpid: String,
1361
1362 pub blocked: Blocked,
1364
1365 pub param: Params,
1367
1368 is_sending_locations: bool,
1370
1371 pub mute_duration: MuteDuration,
1373}
1374
1375impl Chat {
1376 pub async fn load_from_db(context: &Context, chat_id: ChatId) -> Result<Self> {
1378 let mut chat = context
1379 .sql
1380 .query_row(
1381 "SELECT c.type, c.name, c.grpid, c.param, c.archived,
1382 c.blocked, c.locations_send_until, c.muted_until
1383 FROM chats c
1384 WHERE c.id=?;",
1385 (chat_id,),
1386 |row| {
1387 let c = Chat {
1388 id: chat_id,
1389 typ: row.get(0)?,
1390 name: row.get::<_, String>(1)?,
1391 grpid: row.get::<_, String>(2)?,
1392 param: row.get::<_, String>(3)?.parse().unwrap_or_default(),
1393 visibility: row.get(4)?,
1394 blocked: row.get::<_, Option<_>>(5)?.unwrap_or_default(),
1395 is_sending_locations: row.get(6)?,
1396 mute_duration: row.get(7)?,
1397 };
1398 Ok(c)
1399 },
1400 )
1401 .await
1402 .context(format!("Failed loading chat {chat_id} from database"))?;
1403
1404 if chat.id.is_archived_link() {
1405 chat.name = stock_str::archived_chats(context);
1406 } else {
1407 if chat.typ == Chattype::Single && chat.name.is_empty() {
1408 let mut chat_name = "Err [Name not found]".to_owned();
1411 match get_chat_contacts(context, chat.id).await {
1412 Ok(contacts) => {
1413 if let Some(contact_id) = contacts.first()
1414 && let Ok(contact) = Contact::get_by_id(context, *contact_id).await
1415 {
1416 contact.get_display_name().clone_into(&mut chat_name);
1417 }
1418 }
1419 Err(err) => {
1420 error!(
1421 context,
1422 "Failed to load contacts for {}: {:#}.", chat.id, err
1423 );
1424 }
1425 }
1426 chat.name = chat_name;
1427 }
1428 if chat.param.exists(Param::Selftalk) {
1429 chat.name = stock_str::saved_messages(context);
1430 } else if chat.param.exists(Param::Devicetalk) {
1431 chat.name = stock_str::device_messages(context);
1432 }
1433 }
1434
1435 Ok(chat)
1436 }
1437
1438 pub fn is_self_talk(&self) -> bool {
1440 self.param.exists(Param::Selftalk)
1441 }
1442
1443 pub fn is_device_talk(&self) -> bool {
1445 self.param.exists(Param::Devicetalk)
1446 }
1447
1448 pub fn is_mailing_list(&self) -> bool {
1450 self.typ == Chattype::Mailinglist
1451 }
1452
1453 pub(crate) async fn why_cant_send(&self, context: &Context) -> Result<Option<CantSendReason>> {
1457 self.why_cant_send_ex(context, &|_| false).await
1458 }
1459
1460 pub(crate) async fn why_cant_send_ex(
1461 &self,
1462 context: &Context,
1463 skip_fn: &(dyn Send + Sync + Fn(&CantSendReason) -> bool),
1464 ) -> Result<Option<CantSendReason>> {
1465 use CantSendReason::*;
1466 if self.id.is_special() {
1469 let reason = SpecialChat;
1470 if !skip_fn(&reason) {
1471 return Ok(Some(reason));
1472 }
1473 }
1474 if self.is_device_talk() {
1475 let reason = DeviceChat;
1476 if !skip_fn(&reason) {
1477 return Ok(Some(reason));
1478 }
1479 }
1480 if self.is_contact_request() {
1481 let reason = ContactRequest;
1482 if !skip_fn(&reason) {
1483 return Ok(Some(reason));
1484 }
1485 }
1486 if self.is_mailing_list() && self.get_mailinglist_addr().is_none_or_empty() {
1487 let reason = ReadOnlyMailingList;
1488 if !skip_fn(&reason) {
1489 return Ok(Some(reason));
1490 }
1491 }
1492 if self.typ == Chattype::InBroadcast {
1493 let reason = InBroadcast;
1494 if !skip_fn(&reason) {
1495 return Ok(Some(reason));
1496 }
1497 }
1498
1499 let reason = NotAMember;
1501 if !skip_fn(&reason) && !self.is_self_in_chat(context).await? {
1502 return Ok(Some(reason));
1503 }
1504
1505 let reason = MissingKey;
1506 if !skip_fn(&reason) && self.typ == Chattype::Single {
1507 let contact_ids = get_chat_contacts(context, self.id).await?;
1508 if let Some(contact_id) = contact_ids.first() {
1509 let contact = Contact::get_by_id(context, *contact_id).await?;
1510 if contact.is_key_contact() && contact.public_key(context).await?.is_none() {
1511 return Ok(Some(reason));
1512 }
1513 }
1514 }
1515
1516 Ok(None)
1517 }
1518
1519 pub async fn can_send(&self, context: &Context) -> Result<bool> {
1523 Ok(self.why_cant_send(context).await?.is_none())
1524 }
1525
1526 pub async fn is_self_in_chat(&self, context: &Context) -> Result<bool> {
1530 match self.typ {
1531 Chattype::Single | Chattype::OutBroadcast | Chattype::Mailinglist => Ok(true),
1532 Chattype::Group | Chattype::InBroadcast => {
1533 is_contact_in_chat(context, self.id, ContactId::SELF).await
1534 }
1535 }
1536 }
1537
1538 pub(crate) async fn update_param(&mut self, context: &Context) -> Result<()> {
1539 context
1540 .sql
1541 .execute(
1542 "UPDATE chats SET param=? WHERE id=?",
1543 (self.param.to_string(), self.id),
1544 )
1545 .await?;
1546 Ok(())
1547 }
1548
1549 pub fn get_id(&self) -> ChatId {
1551 self.id
1552 }
1553
1554 pub fn get_type(&self) -> Chattype {
1556 self.typ
1557 }
1558
1559 pub fn get_name(&self) -> &str {
1561 &self.name
1562 }
1563
1564 pub fn get_mailinglist_addr(&self) -> Option<&str> {
1566 self.param.get(Param::ListPost)
1567 }
1568
1569 pub async fn get_profile_image(&self, context: &Context) -> Result<Option<PathBuf>> {
1571 if self.id.is_archived_link() {
1572 return Ok(Some(get_archive_icon(context).await?));
1575 } else if self.is_device_talk() {
1576 return Ok(Some(get_device_icon(context).await?));
1577 } else if self.is_self_talk() {
1578 return Ok(Some(get_saved_messages_icon(context).await?));
1579 } else if !self.is_encrypted(context).await? {
1580 return Ok(Some(get_abs_path(
1582 context,
1583 Path::new(&get_unencrypted_icon(context).await?),
1584 )));
1585 } else if self.typ == Chattype::Single {
1586 let contacts = get_chat_contacts(context, self.id).await?;
1590 if let Some(contact_id) = contacts.first() {
1591 let contact = Contact::get_by_id(context, *contact_id).await?;
1592 return contact.get_profile_image(context).await;
1593 }
1594 } else if let Some(image_rel) = self.param.get(Param::ProfileImage) {
1595 if !image_rel.is_empty() {
1597 return Ok(Some(get_abs_path(context, Path::new(&image_rel))));
1598 }
1599 }
1600 Ok(None)
1601 }
1602
1603 pub async fn get_color(&self, context: &Context) -> Result<u32> {
1609 let mut color = 0;
1610
1611 if self.typ == Chattype::Single {
1612 let contacts = get_chat_contacts(context, self.id).await?;
1613 if let Some(contact_id) = contacts.first()
1614 && let Ok(contact) = Contact::get_by_id(context, *contact_id).await
1615 {
1616 color = contact.get_color();
1617 }
1618 } else if !self.grpid.is_empty() {
1619 color = str_to_color(&self.grpid);
1620 } else {
1621 color = str_to_color(&self.name);
1622 }
1623
1624 Ok(color)
1625 }
1626
1627 pub async fn get_info(&self, context: &Context) -> Result<ChatInfo> {
1632 let draft = match self.id.get_draft(context).await? {
1633 Some(message) => message.text,
1634 _ => String::new(),
1635 };
1636 Ok(ChatInfo {
1637 id: self.id,
1638 type_: self.typ as u32,
1639 name: self.name.clone(),
1640 archived: self.visibility == ChatVisibility::Archived,
1641 param: self.param.to_string(),
1642 is_sending_locations: self.is_sending_locations,
1643 color: self.get_color(context).await?,
1644 profile_image: self
1645 .get_profile_image(context)
1646 .await?
1647 .unwrap_or_else(std::path::PathBuf::new),
1648 draft,
1649 is_muted: self.is_muted(),
1650 ephemeral_timer: self.id.get_ephemeral_timer(context).await?,
1651 })
1652 }
1653
1654 pub fn get_visibility(&self) -> ChatVisibility {
1656 self.visibility
1657 }
1658
1659 pub fn is_contact_request(&self) -> bool {
1664 self.blocked == Blocked::Request
1665 }
1666
1667 pub fn is_unpromoted(&self) -> bool {
1669 self.param.get_bool(Param::Unpromoted).unwrap_or_default()
1670 }
1671
1672 pub fn is_promoted(&self) -> bool {
1675 !self.is_unpromoted()
1676 }
1677
1678 pub async fn is_encrypted(&self, context: &Context) -> Result<bool> {
1680 let is_encrypted = self.is_self_talk()
1681 || match self.typ {
1682 Chattype::Single => {
1683 match context
1684 .sql
1685 .query_row_optional(
1686 "SELECT cc.contact_id, c.fingerprint<>''
1687 FROM chats_contacts cc LEFT JOIN contacts c
1688 ON c.id=cc.contact_id
1689 WHERE cc.chat_id=?
1690 ",
1691 (self.id,),
1692 |row| {
1693 let id: ContactId = row.get(0)?;
1694 let is_key: bool = row.get(1)?;
1695 Ok((id, is_key))
1696 },
1697 )
1698 .await?
1699 {
1700 Some((id, is_key)) => is_key || id == ContactId::DEVICE,
1701 None => true,
1702 }
1703 }
1704 Chattype::Group => {
1705 !self.grpid.is_empty()
1707 }
1708 Chattype::Mailinglist => false,
1709 Chattype::OutBroadcast | Chattype::InBroadcast => true,
1710 };
1711 Ok(is_encrypted)
1712 }
1713
1714 pub fn is_sending_locations(&self) -> bool {
1716 self.is_sending_locations
1717 }
1718
1719 pub fn is_muted(&self) -> bool {
1721 match self.mute_duration {
1722 MuteDuration::NotMuted => false,
1723 MuteDuration::Forever => true,
1724 MuteDuration::Until(when) => when > SystemTime::now(),
1725 }
1726 }
1727
1728 pub(crate) async fn member_list_timestamp(&self, context: &Context) -> Result<i64> {
1730 if let Some(member_list_timestamp) = self.param.get_i64(Param::MemberListTimestamp) {
1731 Ok(member_list_timestamp)
1732 } else {
1733 Ok(self.id.created_timestamp(context).await?)
1734 }
1735 }
1736
1737 pub(crate) async fn member_list_is_stale(&self, context: &Context) -> Result<bool> {
1743 let now = time();
1744 let member_list_ts = self.member_list_timestamp(context).await?;
1745 let is_stale = now.saturating_add(TIMESTAMP_SENT_TOLERANCE)
1746 >= member_list_ts.saturating_add(60 * 24 * 3600);
1747 Ok(is_stale)
1748 }
1749
1750 async fn prepare_msg_raw(
1756 &mut self,
1757 context: &Context,
1758 msg: &mut Message,
1759 update_msg_id: Option<MsgId>,
1760 ) -> Result<()> {
1761 let mut to_id = 0;
1762 let mut location_id = 0;
1763
1764 if msg.rfc724_mid.is_empty() {
1765 msg.rfc724_mid = create_outgoing_rfc724_mid();
1766 }
1767
1768 if self.typ == Chattype::Single {
1769 if let Some(id) = context
1770 .sql
1771 .query_get_value(
1772 "SELECT contact_id FROM chats_contacts WHERE chat_id=?;",
1773 (self.id,),
1774 )
1775 .await?
1776 {
1777 to_id = id;
1778 } else {
1779 error!(
1780 context,
1781 "Cannot send message, contact for {} not found.", self.id,
1782 );
1783 bail!("Cannot set message, contact for {} not found.", self.id);
1784 }
1785 } else if matches!(self.typ, Chattype::Group | Chattype::OutBroadcast)
1786 && self.param.get_int(Param::Unpromoted).unwrap_or_default() == 1
1787 {
1788 msg.param.set_int(Param::AttachChatAvatarAndDescription, 1);
1789 self.param
1790 .remove(Param::Unpromoted)
1791 .set_i64(Param::GroupNameTimestamp, msg.timestamp_sort)
1792 .set_i64(Param::GroupDescriptionTimestamp, msg.timestamp_sort);
1793 self.update_param(context).await?;
1794 }
1795
1796 let is_bot = context.get_config_bool(Config::Bot).await?;
1797 msg.param
1798 .set_optional(Param::Bot, Some("1").filter(|_| is_bot));
1799
1800 let new_references;
1804 if self.is_self_talk() {
1805 new_references = String::new();
1808 } else if let Some((parent_rfc724_mid, parent_in_reply_to, parent_references)) =
1809 self
1815 .id
1816 .get_parent_mime_headers(context, MessageState::OutPending)
1817 .await?
1818 {
1819 if msg.in_reply_to.is_none() && !parent_rfc724_mid.is_empty() {
1823 msg.in_reply_to = Some(parent_rfc724_mid.clone());
1824 }
1825
1826 let parent_references = if parent_references.is_empty() {
1836 parent_in_reply_to
1837 } else {
1838 parent_references
1839 };
1840
1841 let mut references_vec: Vec<&str> = parent_references.rsplit(' ').take(2).collect();
1844 references_vec.reverse();
1845
1846 if !parent_rfc724_mid.is_empty()
1847 && !references_vec.contains(&parent_rfc724_mid.as_str())
1848 {
1849 references_vec.push(&parent_rfc724_mid)
1850 }
1851
1852 if references_vec.is_empty() {
1853 new_references = msg.rfc724_mid.clone();
1856 } else {
1857 new_references = references_vec.join(" ");
1858 }
1859 } else {
1860 new_references = msg.rfc724_mid.clone();
1866 }
1867
1868 if msg.param.exists(Param::SetLatitude)
1870 && let Ok(row_id) = context
1871 .sql
1872 .insert(
1873 "INSERT INTO locations \
1874 (timestamp,from_id,chat_id, latitude,longitude,independent)\
1875 VALUES (?,?,?, ?,?,1);",
1876 (
1877 msg.timestamp_sort,
1878 ContactId::SELF,
1879 self.id,
1880 msg.param.get_float(Param::SetLatitude).unwrap_or_default(),
1881 msg.param.get_float(Param::SetLongitude).unwrap_or_default(),
1882 ),
1883 )
1884 .await
1885 {
1886 location_id = row_id;
1887 }
1888
1889 let ephemeral_timer = if msg.param.get_cmd() == SystemMessage::EphemeralTimerChanged {
1890 EphemeralTimer::Disabled
1891 } else {
1892 self.id.get_ephemeral_timer(context).await?
1893 };
1894 let ephemeral_timestamp = match ephemeral_timer {
1895 EphemeralTimer::Disabled => 0,
1896 EphemeralTimer::Enabled { duration } => time().saturating_add(duration.into()),
1897 };
1898
1899 let (msg_text, was_truncated) = truncate_msg_text(context, msg.text.clone()).await?;
1900 let new_mime_headers = if msg.has_html() {
1901 msg.param.get(Param::SendHtml).map(|s| s.to_string())
1902 } else {
1903 None
1904 };
1905 let new_mime_headers: Option<String> = new_mime_headers.map(|s| {
1906 let html_part = MimePart::new("text/html", s);
1907 let mut buffer = Vec::new();
1908 let cursor = Cursor::new(&mut buffer);
1909 html_part.write_part(cursor).ok();
1910 String::from_utf8_lossy(&buffer).to_string()
1911 });
1912 let new_mime_headers = new_mime_headers.or_else(|| match was_truncated {
1913 true => Some("Content-Type: text/plain; charset=utf-8\r\n\r\n".to_string() + &msg.text),
1917 false => None,
1918 });
1919 let new_mime_headers = match new_mime_headers {
1920 Some(h) => Some(tokio::task::block_in_place(move || {
1921 buf_compress(h.as_bytes())
1922 })?),
1923 None => None,
1924 };
1925
1926 msg.chat_id = self.id;
1927 msg.from_id = ContactId::SELF;
1928
1929 if let Some(update_msg_id) = update_msg_id {
1931 context
1932 .sql
1933 .execute(
1934 "UPDATE msgs
1935 SET rfc724_mid=?, chat_id=?, from_id=?, to_id=?, timestamp=?, type=?,
1936 state=?, txt=?, txt_normalized=?, subject=?, param=?,
1937 hidden=?, mime_in_reply_to=?, mime_references=?, mime_modified=?,
1938 mime_headers=?, mime_compressed=1, location_id=?, ephemeral_timer=?,
1939 ephemeral_timestamp=?
1940 WHERE id=?;",
1941 params_slice![
1942 msg.rfc724_mid,
1943 msg.chat_id,
1944 msg.from_id,
1945 to_id,
1946 msg.timestamp_sort,
1947 msg.viewtype,
1948 msg.state,
1949 msg_text,
1950 normalize_text(&msg_text),
1951 &msg.subject,
1952 msg.param.to_string(),
1953 msg.hidden,
1954 msg.in_reply_to.as_deref().unwrap_or_default(),
1955 new_references,
1956 new_mime_headers.is_some(),
1957 new_mime_headers.unwrap_or_default(),
1958 location_id as i32,
1959 ephemeral_timer,
1960 ephemeral_timestamp,
1961 update_msg_id
1962 ],
1963 )
1964 .await?;
1965 msg.id = update_msg_id;
1966 } else {
1967 let raw_id = context
1968 .sql
1969 .insert(
1970 "INSERT INTO msgs (
1971 rfc724_mid,
1972 chat_id,
1973 from_id,
1974 to_id,
1975 timestamp,
1976 type,
1977 state,
1978 txt,
1979 txt_normalized,
1980 subject,
1981 param,
1982 hidden,
1983 mime_in_reply_to,
1984 mime_references,
1985 mime_modified,
1986 mime_headers,
1987 mime_compressed,
1988 location_id,
1989 ephemeral_timer,
1990 ephemeral_timestamp)
1991 VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,1,?,?,?);",
1992 params_slice![
1993 msg.rfc724_mid,
1994 msg.chat_id,
1995 msg.from_id,
1996 to_id,
1997 msg.timestamp_sort,
1998 msg.viewtype,
1999 msg.state,
2000 msg_text,
2001 normalize_text(&msg_text),
2002 &msg.subject,
2003 msg.param.to_string(),
2004 msg.hidden,
2005 msg.in_reply_to.as_deref().unwrap_or_default(),
2006 new_references,
2007 new_mime_headers.is_some(),
2008 new_mime_headers.unwrap_or_default(),
2009 location_id as i32,
2010 ephemeral_timer,
2011 ephemeral_timestamp
2012 ],
2013 )
2014 .await?;
2015 context.new_msgs_notify.notify_one();
2016 msg.id = MsgId::new(u32::try_from(raw_id)?);
2017
2018 maybe_set_logging_xdc(context, msg, self.id).await?;
2019 context
2020 .update_webxdc_integration_database(msg, context)
2021 .await?;
2022 }
2023 context.scheduler.interrupt_ephemeral_task().await;
2024 Ok(())
2025 }
2026
2027 pub(crate) async fn sync_contacts(&self, context: &Context) -> Result<()> {
2029 if self.is_encrypted(context).await? {
2030 let self_fp = self_fingerprint(context).await?;
2031 let fingerprint_addrs = context
2032 .sql
2033 .query_map_vec(
2034 "SELECT c.id, c.fingerprint, c.addr
2035 FROM contacts c INNER JOIN chats_contacts cc
2036 ON c.id=cc.contact_id
2037 WHERE cc.chat_id=? AND cc.add_timestamp >= cc.remove_timestamp",
2038 (self.id,),
2039 |row| {
2040 if row.get::<_, ContactId>(0)? == ContactId::SELF {
2041 return Ok((self_fp.to_string(), String::new()));
2042 }
2043 let fingerprint = row.get(1)?;
2044 let addr = row.get(2)?;
2045 Ok((fingerprint, addr))
2046 },
2047 )
2048 .await?;
2049 self.sync(context, SyncAction::SetPgpContacts(fingerprint_addrs))
2050 .await?;
2051 } else {
2052 let addrs = context
2053 .sql
2054 .query_map_vec(
2055 "SELECT c.addr \
2056 FROM contacts c INNER JOIN chats_contacts cc \
2057 ON c.id=cc.contact_id \
2058 WHERE cc.chat_id=? AND cc.add_timestamp >= cc.remove_timestamp",
2059 (self.id,),
2060 |row| {
2061 let addr: String = row.get(0)?;
2062 Ok(addr)
2063 },
2064 )
2065 .await?;
2066 self.sync(context, SyncAction::SetContacts(addrs)).await?;
2067 }
2068 Ok(())
2069 }
2070
2071 async fn get_sync_id(&self, context: &Context) -> Result<Option<SyncId>> {
2073 match self.typ {
2074 Chattype::Single => {
2075 if self.is_device_talk() {
2076 return Ok(Some(SyncId::Device));
2077 }
2078
2079 let mut r = None;
2080 for contact_id in get_chat_contacts(context, self.id).await? {
2081 if contact_id == ContactId::SELF && !self.is_self_talk() {
2082 continue;
2083 }
2084 if r.is_some() {
2085 return Ok(None);
2086 }
2087 let contact = Contact::get_by_id(context, contact_id).await?;
2088 if let Some(fingerprint) = contact.fingerprint() {
2089 r = Some(SyncId::ContactFingerprint(fingerprint.hex()));
2090 } else {
2091 r = Some(SyncId::ContactAddr(contact.get_addr().to_string()));
2092 }
2093 }
2094 Ok(r)
2095 }
2096 Chattype::OutBroadcast
2097 | Chattype::InBroadcast
2098 | Chattype::Group
2099 | Chattype::Mailinglist => {
2100 if !self.grpid.is_empty() {
2101 return Ok(Some(SyncId::Grpid(self.grpid.clone())));
2102 }
2103
2104 let Some((parent_rfc724_mid, parent_in_reply_to, _)) = self
2105 .id
2106 .get_parent_mime_headers(context, MessageState::OutDelivered)
2107 .await?
2108 else {
2109 warn!(
2110 context,
2111 "Chat::get_sync_id({}): No good message identifying the chat found.",
2112 self.id
2113 );
2114 return Ok(None);
2115 };
2116 Ok(Some(SyncId::Msgids(vec![
2117 parent_in_reply_to,
2118 parent_rfc724_mid,
2119 ])))
2120 }
2121 }
2122 }
2123
2124 pub(crate) async fn sync(&self, context: &Context, action: SyncAction) -> Result<()> {
2126 if let Some(id) = self.get_sync_id(context).await? {
2127 sync(context, id, action).await?;
2128 }
2129 Ok(())
2130 }
2131}
2132
2133pub(crate) async fn sync(context: &Context, id: SyncId, action: SyncAction) -> Result<()> {
2134 context
2135 .add_sync_item(SyncData::AlterChat { id, action })
2136 .await?;
2137 context.scheduler.interrupt_smtp().await;
2138 Ok(())
2139}
2140
2141#[derive(Debug, Copy, Eq, PartialEq, Clone, Serialize, Deserialize, EnumIter, Default)]
2143#[repr(i8)]
2144pub enum ChatVisibility {
2145 #[default]
2147 Normal = 0,
2148
2149 Archived = 1,
2151
2152 Pinned = 2,
2154}
2155
2156impl rusqlite::types::ToSql for ChatVisibility {
2157 fn to_sql(&self) -> rusqlite::Result<rusqlite::types::ToSqlOutput<'_>> {
2158 let val = rusqlite::types::Value::Integer(*self as i64);
2159 let out = rusqlite::types::ToSqlOutput::Owned(val);
2160 Ok(out)
2161 }
2162}
2163
2164impl rusqlite::types::FromSql for ChatVisibility {
2165 fn column_result(value: rusqlite::types::ValueRef) -> rusqlite::types::FromSqlResult<Self> {
2166 i64::column_result(value).map(|val| {
2167 match val {
2168 2 => ChatVisibility::Pinned,
2169 1 => ChatVisibility::Archived,
2170 0 => ChatVisibility::Normal,
2171 _ => ChatVisibility::Normal,
2173 }
2174 })
2175 }
2176}
2177
2178#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
2180#[non_exhaustive]
2181pub struct ChatInfo {
2182 pub id: ChatId,
2184
2185 #[serde(rename = "type")]
2192 pub type_: u32,
2193
2194 pub name: String,
2196
2197 pub archived: bool,
2199
2200 pub param: String,
2204
2205 pub is_sending_locations: bool,
2207
2208 pub color: u32,
2212
2213 pub profile_image: std::path::PathBuf,
2218
2219 pub draft: String,
2227
2228 pub is_muted: bool,
2232
2233 pub ephemeral_timer: EphemeralTimer,
2235 }
2241
2242async fn get_asset_icon(context: &Context, name: &str, bytes: &[u8]) -> Result<PathBuf> {
2243 ensure!(name.starts_with("icon-"));
2244 if let Some(icon) = context.sql.get_raw_config(name).await? {
2245 return Ok(get_abs_path(context, Path::new(&icon)));
2246 }
2247
2248 let blob =
2249 BlobObject::create_and_deduplicate_from_bytes(context, bytes, &format!("{name}.png"))?;
2250 let icon = blob.as_name().to_string();
2251 context.sql.set_raw_config(name, Some(&icon)).await?;
2252
2253 Ok(get_abs_path(context, Path::new(&icon)))
2254}
2255
2256pub(crate) async fn get_saved_messages_icon(context: &Context) -> Result<PathBuf> {
2257 get_asset_icon(
2258 context,
2259 "icon-saved-messages",
2260 include_bytes!("../assets/icon-saved-messages.png"),
2261 )
2262 .await
2263}
2264
2265pub(crate) async fn get_device_icon(context: &Context) -> Result<PathBuf> {
2266 get_asset_icon(
2267 context,
2268 "icon-device",
2269 include_bytes!("../assets/icon-device.png"),
2270 )
2271 .await
2272}
2273
2274pub(crate) async fn get_archive_icon(context: &Context) -> Result<PathBuf> {
2275 get_asset_icon(
2276 context,
2277 "icon-archive",
2278 include_bytes!("../assets/icon-archive.png"),
2279 )
2280 .await
2281}
2282
2283pub(crate) async fn get_unencrypted_icon(context: &Context) -> Result<PathBuf> {
2286 get_asset_icon(
2287 context,
2288 "icon-unencrypted",
2289 include_bytes!("../assets/icon-unencrypted.png"),
2290 )
2291 .await
2292}
2293
2294async fn update_special_chat_name(
2295 context: &Context,
2296 contact_id: ContactId,
2297 name: String,
2298) -> Result<()> {
2299 if let Some(ChatIdBlocked { id: chat_id, .. }) =
2300 ChatIdBlocked::lookup_by_contact(context, contact_id).await?
2301 {
2302 context
2304 .sql
2305 .execute(
2306 "UPDATE chats SET name=?, name_normalized=? WHERE id=? AND name!=?",
2307 (&name, normalize_text(&name), chat_id, &name),
2308 )
2309 .await?;
2310 }
2311 Ok(())
2312}
2313
2314pub(crate) async fn update_special_chat_names(context: &Context) -> Result<()> {
2315 update_special_chat_name(
2316 context,
2317 ContactId::DEVICE,
2318 stock_str::device_messages(context),
2319 )
2320 .await?;
2321 update_special_chat_name(context, ContactId::SELF, stock_str::saved_messages(context)).await?;
2322 Ok(())
2323}
2324
2325#[derive(Debug)]
2333pub(crate) struct ChatIdBlocked {
2334 pub id: ChatId,
2336
2337 pub blocked: Blocked,
2339}
2340
2341impl ChatIdBlocked {
2342 pub async fn lookup_by_contact(
2346 context: &Context,
2347 contact_id: ContactId,
2348 ) -> Result<Option<Self>> {
2349 ensure!(context.sql.is_open().await, "Database not available");
2350 ensure!(
2351 contact_id != ContactId::UNDEFINED,
2352 "Invalid contact id requested"
2353 );
2354
2355 context
2356 .sql
2357 .query_row_optional(
2358 "SELECT c.id, c.blocked
2359 FROM chats c
2360 INNER JOIN chats_contacts j
2361 ON c.id=j.chat_id
2362 WHERE c.type=100 -- 100 = Chattype::Single
2363 AND c.id>9 -- 9 = DC_CHAT_ID_LAST_SPECIAL
2364 AND j.contact_id=?;",
2365 (contact_id,),
2366 |row| {
2367 let id: ChatId = row.get(0)?;
2368 let blocked: Blocked = row.get(1)?;
2369 Ok(ChatIdBlocked { id, blocked })
2370 },
2371 )
2372 .await
2373 }
2374
2375 pub async fn get_for_contact(
2380 context: &Context,
2381 contact_id: ContactId,
2382 create_blocked: Blocked,
2383 ) -> Result<Self> {
2384 ensure!(context.sql.is_open().await, "Database not available");
2385 ensure!(
2386 contact_id != ContactId::UNDEFINED,
2387 "Invalid contact id requested"
2388 );
2389
2390 if let Some(res) = Self::lookup_by_contact(context, contact_id).await? {
2391 return Ok(res);
2393 }
2394
2395 let contact = Contact::get_by_id(context, contact_id).await?;
2396 let chat_name = contact.get_display_name().to_string();
2397 let mut params = Params::new();
2398 match contact_id {
2399 ContactId::SELF => {
2400 params.set_int(Param::Selftalk, 1);
2401 }
2402 ContactId::DEVICE => {
2403 params.set_int(Param::Devicetalk, 1);
2404 }
2405 _ => (),
2406 }
2407
2408 let smeared_time = create_smeared_timestamp(context);
2409
2410 let chat_id = context
2411 .sql
2412 .transaction(move |transaction| {
2413 transaction.execute(
2414 "INSERT INTO chats
2415 (type, name, name_normalized, param, blocked, created_timestamp)
2416 VALUES(?, ?, ?, ?, ?, ?)",
2417 (
2418 Chattype::Single,
2419 &chat_name,
2420 normalize_text(&chat_name),
2421 params.to_string(),
2422 create_blocked as u8,
2423 smeared_time,
2424 ),
2425 )?;
2426 let chat_id = ChatId::new(
2427 transaction
2428 .last_insert_rowid()
2429 .try_into()
2430 .context("chat table rowid overflows u32")?,
2431 );
2432
2433 transaction.execute(
2434 "INSERT INTO chats_contacts
2435 (chat_id, contact_id)
2436 VALUES((SELECT last_insert_rowid()), ?)",
2437 (contact_id,),
2438 )?;
2439
2440 Ok(chat_id)
2441 })
2442 .await?;
2443
2444 let chat = Chat::load_from_db(context, chat_id).await?;
2445 if chat.is_encrypted(context).await?
2446 && !chat.param.exists(Param::Devicetalk)
2447 && !chat.param.exists(Param::Selftalk)
2448 {
2449 chat_id.add_e2ee_notice(context, smeared_time).await?;
2450 }
2451
2452 Ok(Self {
2453 id: chat_id,
2454 blocked: create_blocked,
2455 })
2456 }
2457}
2458
2459async fn prepare_msg_blob(context: &Context, msg: &mut Message) -> Result<()> {
2460 if msg.viewtype == Viewtype::Text || msg.viewtype == Viewtype::Call {
2461 } else if msg.viewtype.has_file() {
2463 let viewtype_orig = msg.viewtype;
2464 let mut blob = msg
2465 .param
2466 .get_file_blob(context)?
2467 .with_context(|| format!("attachment missing for message of type #{}", msg.viewtype))?;
2468 let mut maybe_image = false;
2469
2470 if msg.viewtype == Viewtype::File
2471 || msg.viewtype == Viewtype::Image
2472 || msg.viewtype == Viewtype::Sticker && !msg.param.exists(Param::ForceSticker)
2473 {
2474 if let Some((better_type, _)) = message::guess_msgtype_from_suffix(msg) {
2481 if msg.viewtype == Viewtype::Sticker {
2482 if better_type != Viewtype::Image {
2483 msg.param.set_int(Param::ForceSticker, 1);
2485 }
2486 } else if better_type == Viewtype::Image {
2487 maybe_image = true;
2488 } else if better_type != Viewtype::Webxdc
2489 || context
2490 .ensure_sendable_webxdc_file(&blob.to_abs_path())
2491 .await
2492 .is_ok()
2493 {
2494 msg.viewtype = better_type;
2495 }
2496 }
2497 } else if msg.viewtype == Viewtype::Webxdc {
2498 context
2499 .ensure_sendable_webxdc_file(&blob.to_abs_path())
2500 .await?;
2501 }
2502
2503 if msg.viewtype == Viewtype::Vcard {
2504 msg.try_set_vcard(context, &blob.to_abs_path()).await?;
2505 }
2506 if msg.viewtype == Viewtype::File && maybe_image
2507 || msg.viewtype == Viewtype::Image
2508 || msg.viewtype == Viewtype::Sticker && !msg.param.exists(Param::ForceSticker)
2509 {
2510 let new_name = blob
2511 .check_or_recode_image(context, msg.get_filename(), &mut msg.viewtype)
2512 .await?;
2513 msg.param.set(Param::Filename, new_name);
2514 msg.param.set(Param::File, blob.as_name());
2515 }
2516
2517 if !msg.param.exists(Param::MimeType)
2518 && let Some((viewtype, mime)) = message::guess_msgtype_from_suffix(msg)
2519 {
2520 let mime = match viewtype != Viewtype::Image
2523 || matches!(msg.viewtype, Viewtype::Image | Viewtype::Sticker)
2524 {
2525 true => mime,
2526 false => "application/octet-stream",
2527 };
2528 msg.param.set(Param::MimeType, mime);
2529 }
2530
2531 msg.try_calc_and_set_dimensions(context).await?;
2532
2533 let filename = msg.get_filename().context("msg has no file")?;
2534 let suffix = Path::new(&filename)
2535 .extension()
2536 .and_then(|e| e.to_str())
2537 .unwrap_or("dat");
2538 let filename: String = match viewtype_orig {
2542 Viewtype::Voice => format!(
2543 "voice-messsage_{}.{}",
2544 chrono::Utc
2545 .timestamp_opt(msg.timestamp_sort, 0)
2546 .single()
2547 .map_or_else(
2548 || "YY-mm-dd_hh:mm:ss".to_string(),
2549 |ts| ts.format("%Y-%m-%d_%H-%M-%S").to_string()
2550 ),
2551 &suffix
2552 ),
2553 Viewtype::Image | Viewtype::Gif => format!(
2554 "image_{}.{}",
2555 chrono::Utc
2556 .timestamp_opt(msg.timestamp_sort, 0)
2557 .single()
2558 .map_or_else(
2559 || "YY-mm-dd_hh:mm:ss".to_string(),
2560 |ts| ts.format("%Y-%m-%d_%H-%M-%S").to_string(),
2561 ),
2562 &suffix,
2563 ),
2564 Viewtype::Video => format!(
2565 "video_{}.{}",
2566 chrono::Utc
2567 .timestamp_opt(msg.timestamp_sort, 0)
2568 .single()
2569 .map_or_else(
2570 || "YY-mm-dd_hh:mm:ss".to_string(),
2571 |ts| ts.format("%Y-%m-%d_%H-%M-%S").to_string()
2572 ),
2573 &suffix
2574 ),
2575 _ => filename,
2576 };
2577 msg.param.set(Param::Filename, filename);
2578
2579 info!(
2580 context,
2581 "Attaching \"{}\" for message type #{}.",
2582 blob.to_abs_path().display(),
2583 msg.viewtype
2584 );
2585 } else {
2586 bail!("Cannot send messages of type #{}.", msg.viewtype);
2587 }
2588 Ok(())
2589}
2590
2591pub async fn is_contact_in_chat(
2593 context: &Context,
2594 chat_id: ChatId,
2595 contact_id: ContactId,
2596) -> Result<bool> {
2597 let exists = context
2604 .sql
2605 .exists(
2606 "SELECT COUNT(*) FROM chats_contacts
2607 WHERE chat_id=? AND contact_id=?
2608 AND add_timestamp >= remove_timestamp",
2609 (chat_id, contact_id),
2610 )
2611 .await?;
2612 Ok(exists)
2613}
2614
2615pub async fn send_msg(context: &Context, chat_id: ChatId, msg: &mut Message) -> Result<MsgId> {
2622 ensure!(
2623 !chat_id.is_special(),
2624 "chat_id cannot be a special chat: {chat_id}"
2625 );
2626
2627 if msg.state != MessageState::Undefined && msg.state != MessageState::OutPreparing {
2628 msg.param.remove(Param::GuaranteeE2ee);
2629 msg.param.remove(Param::ForcePlaintext);
2630 }
2632
2633 if msg.is_system_message() {
2635 msg.text = sanitize_bidi_characters(&msg.text);
2636 }
2637
2638 if !prepare_send_msg(context, chat_id, msg).await?.is_empty() {
2639 if !msg.hidden {
2640 context.emit_msgs_changed(msg.chat_id, msg.id);
2641 }
2642
2643 if msg.param.exists(Param::SetLatitude) {
2644 context.emit_location_changed(Some(ContactId::SELF)).await?;
2645 }
2646
2647 context.scheduler.interrupt_smtp().await;
2648 }
2649
2650 Ok(msg.id)
2651}
2652
2653pub async fn send_msg_sync(context: &Context, chat_id: ChatId, msg: &mut Message) -> Result<MsgId> {
2658 let rowids = prepare_send_msg(context, chat_id, msg).await?;
2659 if rowids.is_empty() {
2660 return Ok(msg.id);
2661 }
2662 let mut smtp = crate::smtp::Smtp::new();
2663 for rowid in rowids {
2664 send_msg_to_smtp(context, &mut smtp, rowid)
2665 .await
2666 .context("failed to send message, queued for later sending")?;
2667 }
2668 context.emit_msgs_changed(msg.chat_id, msg.id);
2669 Ok(msg.id)
2670}
2671
2672async fn prepare_send_msg(
2676 context: &Context,
2677 chat_id: ChatId,
2678 msg: &mut Message,
2679) -> Result<Vec<i64>> {
2680 let mut chat = Chat::load_from_db(context, chat_id).await?;
2681
2682 let skip_fn = |reason: &CantSendReason| match reason {
2683 CantSendReason::ContactRequest => {
2684 msg.param.get_cmd() == SystemMessage::SecurejoinMessage
2687 }
2688 CantSendReason::NotAMember => msg.param.get_cmd() == SystemMessage::MemberRemovedFromGroup,
2692 CantSendReason::InBroadcast => {
2693 matches!(
2694 msg.param.get_cmd(),
2695 SystemMessage::MemberRemovedFromGroup | SystemMessage::SecurejoinMessage
2696 )
2697 }
2698 CantSendReason::MissingKey => msg
2699 .param
2700 .get_bool(Param::ForcePlaintext)
2701 .unwrap_or_default(),
2702 _ => false,
2703 };
2704 if let Some(reason) = chat.why_cant_send_ex(context, &skip_fn).await? {
2705 bail!("Cannot send to {chat_id}: {reason}");
2706 }
2707
2708 if chat.typ != Chattype::Single
2713 && !context.get_config_bool(Config::Bot).await?
2714 && let Some(quoted_message) = msg.quoted_message(context).await?
2715 && quoted_message.chat_id != chat_id
2716 {
2717 bail!(
2718 "Quote of message from {} cannot be sent to {chat_id}",
2719 quoted_message.chat_id
2720 );
2721 }
2722
2723 let update_msg_id = if msg.state == MessageState::OutDraft {
2725 msg.hidden = false;
2726 if !msg.id.is_special() && msg.chat_id == chat_id {
2727 Some(msg.id)
2728 } else {
2729 None
2730 }
2731 } else {
2732 None
2733 };
2734
2735 if matches!(
2736 msg.state,
2737 MessageState::Undefined | MessageState::OutPreparing
2738 )
2739 && msg.param.get_cmd() != SystemMessage::SecurejoinMessage
2741 && chat.is_encrypted(context).await?
2742 {
2743 msg.param.set_int(Param::GuaranteeE2ee, 1);
2744 if !msg.id.is_unset() {
2745 msg.update_param(context).await?;
2746 }
2747 }
2748 msg.state = MessageState::OutPending;
2749
2750 msg.timestamp_sort = create_smeared_timestamp(context);
2751 prepare_msg_blob(context, msg).await?;
2752 if !msg.hidden {
2753 chat_id.unarchive_if_not_muted(context, msg.state).await?;
2754 }
2755 chat.prepare_msg_raw(context, msg, update_msg_id).await?;
2756
2757 let row_ids = create_send_msg_jobs(context, msg)
2758 .await
2759 .context("Failed to create send jobs")?;
2760 if !row_ids.is_empty() {
2761 donation_request_maybe(context).await.log_err(context).ok();
2762 }
2763 Ok(row_ids)
2764}
2765
2766async fn render_mime_message_and_pre_message(
2773 context: &Context,
2774 msg: &mut Message,
2775 mimefactory: MimeFactory,
2776) -> Result<(Option<RenderedEmail>, RenderedEmail)> {
2777 let needs_pre_message = msg.viewtype.has_file()
2778 && mimefactory.will_be_encrypted() && msg
2780 .get_filebytes(context)
2781 .await?
2782 .context("filebytes not available, even though message has attachment")?
2783 > PRE_MSG_ATTACHMENT_SIZE_THRESHOLD;
2784
2785 if needs_pre_message {
2786 info!(
2787 context,
2788 "Message {} is large and will be split into pre- and post-messages.", msg.id,
2789 );
2790
2791 let mut mimefactory_post_msg = mimefactory.clone();
2792 mimefactory_post_msg.set_as_post_message();
2793 let rendered_msg = Box::pin(mimefactory_post_msg.render(context))
2794 .await
2795 .context("Failed to render post-message")?;
2796
2797 let mut mimefactory_pre_msg = mimefactory;
2798 mimefactory_pre_msg.set_as_pre_message_for(&rendered_msg);
2799 let rendered_pre_msg = Box::pin(mimefactory_pre_msg.render(context))
2800 .await
2801 .context("pre-message failed to render")?;
2802
2803 if rendered_pre_msg.message.len() > PRE_MSG_SIZE_WARNING_THRESHOLD {
2804 warn!(
2805 context,
2806 "Pre-message for message {} is larger than expected: {}.",
2807 msg.id,
2808 rendered_pre_msg.message.len()
2809 );
2810 }
2811
2812 Ok((Some(rendered_pre_msg), rendered_msg))
2813 } else {
2814 Ok((None, Box::pin(mimefactory.render(context)).await?))
2815 }
2816}
2817
2818pub(crate) async fn create_send_msg_jobs(context: &Context, msg: &mut Message) -> Result<Vec<i64>> {
2828 let cmd = msg.param.get_cmd();
2829 if cmd == SystemMessage::GroupNameChanged || cmd == SystemMessage::GroupDescriptionChanged {
2830 msg.chat_id
2831 .update_timestamp(
2832 context,
2833 if cmd == SystemMessage::GroupNameChanged {
2834 Param::GroupNameTimestamp
2835 } else {
2836 Param::GroupDescriptionTimestamp
2837 },
2838 msg.timestamp_sort,
2839 )
2840 .await?;
2841 }
2842
2843 let needs_encryption = msg.param.get_bool(Param::GuaranteeE2ee).unwrap_or_default();
2844 let mimefactory = match MimeFactory::from_msg(context, msg.clone()).await {
2845 Ok(mf) => mf,
2846 Err(err) => {
2847 message::set_msg_failed(context, msg, &err.to_string())
2849 .await
2850 .ok();
2851 return Err(err);
2852 }
2853 };
2854 let attach_selfavatar = mimefactory.attach_selfavatar;
2855 let mut recipients = mimefactory.recipients();
2856
2857 let from = context.get_primary_self_addr().await?;
2858 let lowercase_from = from.to_lowercase();
2859
2860 recipients.retain(|x| x.to_lowercase() != lowercase_from);
2861
2862 if (msg.param.get_int(Param::WebxdcIntegration).is_some() && msg.hidden)
2864 || (!context.get_config_bool(Config::BccSelf).await? && recipients.is_empty())
2866 {
2867 info!(
2868 context,
2869 "Message {} has no recipient, skipping smtp-send.", msg.id
2870 );
2871 msg.param.set_int(Param::GuaranteeE2ee, 1);
2872 msg.update_param(context).await?;
2873 msg.id.set_delivered(context).await?;
2874 msg.state = MessageState::OutDelivered;
2875 return Ok(Vec::new());
2876 }
2877
2878 let (rendered_pre_msg, rendered_msg) =
2879 match render_mime_message_and_pre_message(context, msg, mimefactory).await {
2880 Ok(res) => Ok(res),
2881 Err(err) => {
2882 message::set_msg_failed(context, msg, &err.to_string()).await?;
2883 Err(err)
2884 }
2885 }?;
2886
2887 if let (post_msg, Some(pre_msg)) = (&rendered_msg, &rendered_pre_msg) {
2888 info!(
2889 context,
2890 "Message {} sizes: pre-message: {}; post-message: {}.",
2891 msg.id,
2892 format_size(pre_msg.message.len(), BINARY),
2893 format_size(post_msg.message.len(), BINARY),
2894 );
2895 msg.pre_rfc724_mid = pre_msg.rfc724_mid.clone();
2896 } else {
2897 info!(
2898 context,
2899 "Message {} will be sent in one shot (no pre- and post-message). Size: {}.",
2900 msg.id,
2901 format_size(rendered_msg.message.len(), BINARY),
2902 );
2903 }
2904
2905 if context.get_config_bool(Config::BccSelf).await? {
2906 smtp::add_self_recipients(context, &mut recipients, rendered_msg.is_encrypted).await?;
2907 }
2908
2909 if needs_encryption && !rendered_msg.is_encrypted {
2910 message::set_msg_failed(
2912 context,
2913 msg,
2914 "End-to-end-encryption unavailable unexpectedly.",
2915 )
2916 .await?;
2917 bail!(
2918 "e2e encryption unavailable {} - {:?}",
2919 msg.id,
2920 needs_encryption
2921 );
2922 }
2923
2924 let now = smeared_time(context);
2925
2926 if rendered_msg.last_added_location_id.is_some()
2927 && let Err(err) = location::set_kml_sent_timestamp(context, msg.chat_id, now).await
2928 {
2929 error!(context, "Failed to set kml sent_timestamp: {err:#}.");
2930 }
2931
2932 if attach_selfavatar && let Err(err) = msg.chat_id.set_selfavatar_timestamp(context, now).await
2933 {
2934 error!(context, "Failed to set selfavatar timestamp: {err:#}.");
2935 }
2936
2937 if rendered_msg.is_encrypted {
2938 msg.param.set_int(Param::GuaranteeE2ee, 1);
2939 } else {
2940 msg.param.remove(Param::GuaranteeE2ee);
2941 }
2942 msg.subject.clone_from(&rendered_msg.subject);
2943 context
2945 .sql
2946 .execute(
2947 "
2948UPDATE msgs SET
2949 timestamp=(
2950 SELECT MAX(timestamp) FROM msgs WHERE
2951 -- From `InFresh` to `OutMdnRcvd` inclusive except `OutDraft`.
2952 state IN(10,13,16,18,20,24,26,28) AND
2953 hidden IN(0,1) AND
2954 chat_id=?
2955 ),
2956 pre_rfc724_mid=?, subject=?, param=?
2957WHERE id=?
2958 ",
2959 (
2960 msg.chat_id,
2961 &msg.pre_rfc724_mid,
2962 &msg.subject,
2963 msg.param.to_string(),
2964 msg.id,
2965 ),
2966 )
2967 .await?;
2968
2969 let chunk_size = context.get_max_smtp_rcpt_to().await?;
2970 let trans_fn = |t: &mut rusqlite::Transaction| {
2971 let mut row_ids = Vec::<i64>::new();
2972
2973 if let Some(sync_ids) = rendered_msg.sync_ids_to_delete {
2974 t.execute(
2975 &format!("DELETE FROM multi_device_sync WHERE id IN ({sync_ids})"),
2976 (),
2977 )?;
2978 }
2979 let mut stmt = t.prepare(
2980 "INSERT INTO smtp (rfc724_mid, recipients, mime, msg_id)
2981 VALUES (?1, ?2, ?3, ?4)",
2982 )?;
2983 for recipients_chunk in recipients.chunks(chunk_size) {
2984 let recipients_chunk = recipients_chunk.join(" ");
2985 if let Some(pre_msg) = &rendered_pre_msg {
2986 let row_id = stmt.execute((
2987 &pre_msg.rfc724_mid,
2988 &recipients_chunk,
2989 &pre_msg.message,
2990 msg.id,
2991 ))?;
2992 row_ids.push(row_id.try_into()?);
2993 }
2994 let row_id = stmt.execute((
2995 &rendered_msg.rfc724_mid,
2996 &recipients_chunk,
2997 &rendered_msg.message,
2998 msg.id,
2999 ))?;
3000 row_ids.push(row_id.try_into()?);
3001 }
3002 Ok(row_ids)
3003 };
3004 context.sql.transaction(trans_fn).await
3005}
3006
3007pub async fn send_text_msg(
3011 context: &Context,
3012 chat_id: ChatId,
3013 text_to_send: String,
3014) -> Result<MsgId> {
3015 ensure!(
3016 !chat_id.is_special(),
3017 "bad chat_id, can not be a special chat: {chat_id}"
3018 );
3019
3020 let mut msg = Message::new_text(text_to_send);
3021 send_msg(context, chat_id, &mut msg).await
3022}
3023
3024pub async fn send_edit_request(context: &Context, msg_id: MsgId, new_text: String) -> Result<()> {
3026 let mut original_msg = Message::load_from_db(context, msg_id).await?;
3027 ensure!(
3028 original_msg.from_id == ContactId::SELF,
3029 "Can edit only own messages"
3030 );
3031 ensure!(!original_msg.is_info(), "Cannot edit info messages");
3032 ensure!(!original_msg.has_html(), "Cannot edit HTML messages");
3033 ensure!(original_msg.viewtype != Viewtype::Call, "Cannot edit calls");
3034 ensure!(
3035 !original_msg.text.is_empty(), "Cannot add text"
3037 );
3038 ensure!(!new_text.trim().is_empty(), "Edited text cannot be empty");
3039 if original_msg.text == new_text {
3040 info!(context, "Text unchanged.");
3041 return Ok(());
3042 }
3043
3044 save_text_edit_to_db(context, &mut original_msg, &new_text).await?;
3045
3046 let mut edit_msg = Message::new_text(EDITED_PREFIX.to_owned() + &new_text); edit_msg.set_quote(context, Some(&original_msg)).await?; if original_msg.get_showpadlock() {
3049 edit_msg.param.set_int(Param::GuaranteeE2ee, 1);
3050 }
3051 edit_msg
3052 .param
3053 .set(Param::TextEditFor, original_msg.rfc724_mid);
3054 edit_msg.hidden = true;
3055 send_msg(context, original_msg.chat_id, &mut edit_msg).await?;
3056 Ok(())
3057}
3058
3059pub(crate) async fn save_text_edit_to_db(
3060 context: &Context,
3061 original_msg: &mut Message,
3062 new_text: &str,
3063) -> Result<()> {
3064 original_msg.param.set_int(Param::IsEdited, 1);
3065 context
3066 .sql
3067 .execute(
3068 "UPDATE msgs SET txt=?, txt_normalized=?, param=? WHERE id=?",
3069 (
3070 new_text,
3071 normalize_text(new_text),
3072 original_msg.param.to_string(),
3073 original_msg.id,
3074 ),
3075 )
3076 .await?;
3077 context.emit_msgs_changed(original_msg.chat_id, original_msg.id);
3078 Ok(())
3079}
3080
3081async fn donation_request_maybe(context: &Context) -> Result<()> {
3082 let secs_between_checks = 30 * 24 * 60 * 60;
3083 let now = time();
3084 let ts = context
3085 .get_config_i64(Config::DonationRequestNextCheck)
3086 .await?;
3087 if ts > now {
3088 return Ok(());
3089 }
3090 let msg_cnt = context.sql.count(
3091 "SELECT COUNT(*) FROM msgs WHERE state>=? AND hidden=0",
3092 (MessageState::OutDelivered,),
3093 );
3094 let ts = if ts == 0 || msg_cnt.await? < 100 {
3095 now.saturating_add(secs_between_checks)
3096 } else {
3097 let mut msg = Message::new_text(stock_str::donation_request(context));
3098 add_device_msg(context, None, Some(&mut msg)).await?;
3099 i64::MAX
3100 };
3101 context
3102 .set_config_internal(Config::DonationRequestNextCheck, Some(&ts.to_string()))
3103 .await
3104}
3105
3106#[derive(Debug)]
3108pub struct MessageListOptions {
3109 pub info_only: bool,
3111
3112 pub add_daymarker: bool,
3114}
3115
3116pub async fn get_chat_msgs(context: &Context, chat_id: ChatId) -> Result<Vec<ChatItem>> {
3118 get_chat_msgs_ex(
3119 context,
3120 chat_id,
3121 MessageListOptions {
3122 info_only: false,
3123 add_daymarker: false,
3124 },
3125 )
3126 .await
3127}
3128
3129#[expect(clippy::arithmetic_side_effects)]
3132pub async fn get_chat_msgs_ex(
3133 context: &Context,
3134 chat_id: ChatId,
3135 options: MessageListOptions,
3136) -> Result<Vec<ChatItem>> {
3137 let MessageListOptions {
3138 info_only,
3139 add_daymarker,
3140 } = options;
3141 let process_row = if info_only {
3143 |row: &rusqlite::Row| {
3144 let params = row.get::<_, String>("param")?;
3146 let (from_id, to_id) = (
3147 row.get::<_, ContactId>("from_id")?,
3148 row.get::<_, ContactId>("to_id")?,
3149 );
3150 let is_info_msg: bool = from_id == ContactId::INFO
3151 || to_id == ContactId::INFO
3152 || match Params::from_str(¶ms) {
3153 Ok(p) => {
3154 let cmd = p.get_cmd();
3155 cmd != SystemMessage::Unknown && cmd != SystemMessage::AutocryptSetupMessage
3156 }
3157 _ => false,
3158 };
3159
3160 Ok((
3161 row.get::<_, i64>("timestamp")?,
3162 row.get::<_, MsgId>("id")?,
3163 !is_info_msg,
3164 ))
3165 }
3166 } else {
3167 |row: &rusqlite::Row| {
3168 Ok((
3169 row.get::<_, i64>("timestamp")?,
3170 row.get::<_, MsgId>("id")?,
3171 false,
3172 ))
3173 }
3174 };
3175 let process_rows = |rows: rusqlite::AndThenRows<_>| {
3176 let mut sorted_rows = Vec::new();
3179 for row in rows {
3180 let (ts, curr_id, exclude_message): (i64, MsgId, bool) = row?;
3181 if !exclude_message {
3182 sorted_rows.push((ts, curr_id));
3183 }
3184 }
3185 sorted_rows.sort_unstable();
3186
3187 let mut ret = Vec::new();
3188 let mut last_day = 0;
3189 let cnv_to_local = gm2local_offset();
3190
3191 for (ts, curr_id) in sorted_rows {
3192 if add_daymarker {
3193 let curr_local_timestamp = ts + cnv_to_local;
3194 let secs_in_day = 86400;
3195 let curr_day = curr_local_timestamp / secs_in_day;
3196 if curr_day != last_day {
3197 ret.push(ChatItem::DayMarker {
3198 timestamp: curr_day * secs_in_day - cnv_to_local,
3199 });
3200 last_day = curr_day;
3201 }
3202 }
3203 ret.push(ChatItem::Message { msg_id: curr_id });
3204 }
3205 Ok(ret)
3206 };
3207
3208 let items = if info_only {
3209 context
3210 .sql
3211 .query_map(
3212 "SELECT m.id AS id, m.timestamp AS timestamp, m.param AS param, m.from_id AS from_id, m.to_id AS to_id
3214 FROM msgs m
3215 WHERE m.chat_id=?
3216 AND m.hidden=0
3217 AND (
3218 m.param GLOB '*\nS=*' OR param GLOB 'S=*'
3219 OR m.from_id == ?
3220 OR m.to_id == ?
3221 );",
3222 (chat_id, ContactId::INFO, ContactId::INFO),
3223 process_row,
3224 process_rows,
3225 )
3226 .await?
3227 } else {
3228 context
3229 .sql
3230 .query_map(
3231 "SELECT m.id AS id, m.timestamp AS timestamp
3232 FROM msgs m
3233 WHERE m.chat_id=?
3234 AND m.hidden=0;",
3235 (chat_id,),
3236 process_row,
3237 process_rows,
3238 )
3239 .await?
3240 };
3241 Ok(items)
3242}
3243
3244pub async fn marknoticed_all_chats(context: &Context) -> Result<()> {
3247 let list = context
3249 .sql
3250 .query_map_vec(
3251 "SELECT DISTINCT(c.id)
3252 FROM msgs m
3253 INNER JOIN chats c
3254 ON m.chat_id=c.id
3255 WHERE m.state=?
3256 AND m.hidden=0
3257 AND m.chat_id>9
3258 AND c.blocked=0;",
3259 (MessageState::InFresh,),
3260 |row| {
3261 let msg_id: ChatId = row.get(0)?;
3262 Ok(msg_id)
3263 },
3264 )
3265 .await?;
3266
3267 for chat_id in list {
3268 marknoticed_chat(context, chat_id).await?;
3269 }
3270
3271 Ok(())
3272}
3273
3274pub async fn marknoticed_chat(context: &Context, chat_id: ChatId) -> Result<()> {
3277 if chat_id.is_archived_link() {
3280 let chat_ids_in_archive = context
3281 .sql
3282 .query_map_vec(
3283 "SELECT DISTINCT(m.chat_id) FROM msgs m
3284 LEFT JOIN chats c ON m.chat_id=c.id
3285 WHERE m.state=10 AND m.hidden=0 AND m.chat_id>9 AND c.archived=1",
3286 (),
3287 |row| {
3288 let chat_id: ChatId = row.get(0)?;
3289 Ok(chat_id)
3290 },
3291 )
3292 .await?;
3293 if chat_ids_in_archive.is_empty() {
3294 return Ok(());
3295 }
3296
3297 context
3298 .sql
3299 .transaction(|transaction| {
3300 let mut stmt = transaction.prepare(
3301 "UPDATE msgs SET state=13 WHERE state=10 AND hidden=0 AND chat_id = ?",
3302 )?;
3303 for chat_id_in_archive in &chat_ids_in_archive {
3304 stmt.execute((chat_id_in_archive,))?;
3305 }
3306 Ok(())
3307 })
3308 .await?;
3309
3310 for chat_id_in_archive in chat_ids_in_archive {
3311 start_chat_ephemeral_timers(context, chat_id_in_archive).await?;
3312 context.emit_event(EventType::MsgsNoticed(chat_id_in_archive));
3313 chatlist_events::emit_chatlist_item_changed(context, chat_id_in_archive);
3314 }
3315 } else {
3316 start_chat_ephemeral_timers(context, chat_id).await?;
3317
3318 let noticed_msgs_count = context
3319 .sql
3320 .execute(
3321 "UPDATE msgs
3322 SET state=?
3323 WHERE state=?
3324 AND hidden=0
3325 AND chat_id=?;",
3326 (MessageState::InNoticed, MessageState::InFresh, chat_id),
3327 )
3328 .await?;
3329
3330 let hidden_messages = context
3333 .sql
3334 .query_map_vec(
3335 "SELECT id FROM msgs
3336 WHERE state=?
3337 AND hidden=1
3338 AND chat_id=?
3339 ORDER BY id LIMIT 100", (MessageState::InFresh, chat_id), |row| {
3342 let msg_id: MsgId = row.get(0)?;
3343 Ok(msg_id)
3344 },
3345 )
3346 .await?;
3347 message::markseen_msgs(context, hidden_messages).await?;
3348 if noticed_msgs_count == 0 {
3349 return Ok(());
3350 }
3351 }
3352
3353 context.emit_event(EventType::MsgsNoticed(chat_id));
3354 chatlist_events::emit_chatlist_item_changed(context, chat_id);
3355 context.on_archived_chats_maybe_noticed();
3356 Ok(())
3357}
3358
3359pub(crate) async fn mark_old_messages_as_noticed(
3366 context: &Context,
3367 mut msgs: Vec<ReceivedMsg>,
3368) -> Result<()> {
3369 if context.get_config_bool(Config::TeamProfile).await? {
3370 return Ok(());
3371 }
3372
3373 msgs.retain(|m| m.state.is_outgoing());
3374 if msgs.is_empty() {
3375 return Ok(());
3376 }
3377
3378 let mut msgs_by_chat: HashMap<ChatId, ReceivedMsg> = HashMap::new();
3379 for msg in msgs {
3380 let chat_id = msg.chat_id;
3381 if let Some(existing_msg) = msgs_by_chat.get(&chat_id) {
3382 if msg.sort_timestamp > existing_msg.sort_timestamp {
3383 msgs_by_chat.insert(chat_id, msg);
3384 }
3385 } else {
3386 msgs_by_chat.insert(chat_id, msg);
3387 }
3388 }
3389
3390 let changed_chats = context
3391 .sql
3392 .transaction(|transaction| {
3393 let mut changed_chats = Vec::new();
3394 for (_, msg) in msgs_by_chat {
3395 let changed_rows = transaction.execute(
3396 "UPDATE msgs
3397 SET state=?
3398 WHERE state=?
3399 AND hidden=0
3400 AND chat_id=?
3401 AND timestamp<=?;",
3402 (
3403 MessageState::InNoticed,
3404 MessageState::InFresh,
3405 msg.chat_id,
3406 msg.sort_timestamp,
3407 ),
3408 )?;
3409 if changed_rows > 0 {
3410 changed_chats.push(msg.chat_id);
3411 }
3412 }
3413 Ok(changed_chats)
3414 })
3415 .await?;
3416
3417 if !changed_chats.is_empty() {
3418 info!(
3419 context,
3420 "Marking chats as noticed because there are newer outgoing messages: {changed_chats:?}."
3421 );
3422 context.on_archived_chats_maybe_noticed();
3423 }
3424
3425 for c in changed_chats {
3426 start_chat_ephemeral_timers(context, c).await?;
3427 context.emit_event(EventType::MsgsNoticed(c));
3428 chatlist_events::emit_chatlist_item_changed(context, c);
3429 }
3430
3431 Ok(())
3432}
3433
3434pub async fn markfresh_chat(context: &Context, chat_id: ChatId) -> Result<()> {
3436 let affected_rows = context
3437 .sql
3438 .execute(
3439 "UPDATE msgs
3440 SET state=?1
3441 WHERE id=(SELECT id
3442 FROM msgs
3443 WHERE state IN (?1, ?2, ?3) AND hidden=0 AND chat_id=?4
3444 ORDER BY timestamp DESC, id DESC
3445 LIMIT 1)
3446 AND state!=?1",
3447 (
3448 MessageState::InFresh,
3449 MessageState::InNoticed,
3450 MessageState::InSeen,
3451 chat_id,
3452 ),
3453 )
3454 .await?;
3455
3456 if affected_rows == 0 {
3457 return Ok(());
3458 }
3459
3460 context.emit_msgs_changed_without_msg_id(chat_id);
3461 chatlist_events::emit_chatlist_item_changed(context, chat_id);
3462
3463 Ok(())
3464}
3465
3466pub async fn get_chat_media(
3473 context: &Context,
3474 chat_id: Option<ChatId>,
3475 msg_type: Viewtype,
3476 msg_type2: Viewtype,
3477 msg_type3: Viewtype,
3478) -> Result<Vec<MsgId>> {
3479 let list = if msg_type == Viewtype::Webxdc
3480 && msg_type2 == Viewtype::Unknown
3481 && msg_type3 == Viewtype::Unknown
3482 {
3483 context
3484 .sql
3485 .query_map_vec(
3486 "SELECT id
3487 FROM msgs
3488 WHERE (1=? OR chat_id=?)
3489 AND chat_id != ?
3490 AND type = ?
3491 AND hidden=0
3492 ORDER BY max(timestamp, timestamp_rcvd), id;",
3493 (
3494 chat_id.is_none(),
3495 chat_id.unwrap_or_else(|| ChatId::new(0)),
3496 DC_CHAT_ID_TRASH,
3497 Viewtype::Webxdc,
3498 ),
3499 |row| {
3500 let msg_id: MsgId = row.get(0)?;
3501 Ok(msg_id)
3502 },
3503 )
3504 .await?
3505 } else {
3506 context
3507 .sql
3508 .query_map_vec(
3509 "SELECT id
3510 FROM msgs
3511 WHERE (1=? OR chat_id=?)
3512 AND chat_id != ?
3513 AND type IN (?, ?, ?)
3514 AND hidden=0
3515 ORDER BY timestamp, id;",
3516 (
3517 chat_id.is_none(),
3518 chat_id.unwrap_or_else(|| ChatId::new(0)),
3519 DC_CHAT_ID_TRASH,
3520 msg_type,
3521 if msg_type2 != Viewtype::Unknown {
3522 msg_type2
3523 } else {
3524 msg_type
3525 },
3526 if msg_type3 != Viewtype::Unknown {
3527 msg_type3
3528 } else {
3529 msg_type
3530 },
3531 ),
3532 |row| {
3533 let msg_id: MsgId = row.get(0)?;
3534 Ok(msg_id)
3535 },
3536 )
3537 .await?
3538 };
3539 Ok(list)
3540}
3541
3542pub async fn get_chat_contacts(context: &Context, chat_id: ChatId) -> Result<Vec<ContactId>> {
3544 context
3547 .sql
3548 .query_map_vec(
3549 "SELECT cc.contact_id
3550 FROM chats_contacts cc
3551 LEFT JOIN contacts c
3552 ON c.id=cc.contact_id
3553 WHERE cc.chat_id=? AND cc.add_timestamp >= cc.remove_timestamp
3554 ORDER BY c.id=1, c.last_seen DESC, c.id DESC;",
3555 (chat_id,),
3556 |row| {
3557 let contact_id: ContactId = row.get(0)?;
3558 Ok(contact_id)
3559 },
3560 )
3561 .await
3562}
3563
3564pub async fn get_past_chat_contacts(context: &Context, chat_id: ChatId) -> Result<Vec<ContactId>> {
3568 let now = time();
3569 context
3570 .sql
3571 .query_map_vec(
3572 "SELECT cc.contact_id
3573 FROM chats_contacts cc
3574 LEFT JOIN contacts c
3575 ON c.id=cc.contact_id
3576 WHERE cc.chat_id=?
3577 AND cc.add_timestamp < cc.remove_timestamp
3578 AND ? < cc.remove_timestamp
3579 ORDER BY c.id=1, cc.remove_timestamp DESC, c.id DESC",
3580 (chat_id, now.saturating_sub(60 * 24 * 3600)),
3581 |row| {
3582 let contact_id: ContactId = row.get(0)?;
3583 Ok(contact_id)
3584 },
3585 )
3586 .await
3587}
3588
3589pub async fn create_group(context: &Context, name: &str) -> Result<ChatId> {
3591 create_group_ex(context, Sync, create_id(), name).await
3592}
3593
3594pub async fn create_group_unencrypted(context: &Context, name: &str) -> Result<ChatId> {
3596 create_group_ex(context, Sync, String::new(), name).await
3597}
3598
3599pub(crate) async fn create_group_ex(
3606 context: &Context,
3607 sync: sync::Sync,
3608 grpid: String,
3609 name: &str,
3610) -> Result<ChatId> {
3611 let mut chat_name = sanitize_single_line(name);
3612 if chat_name.is_empty() {
3613 error!(context, "Invalid chat name: {name}.");
3616 chat_name = "…".to_string();
3617 }
3618
3619 let timestamp = create_smeared_timestamp(context);
3620 let row_id = context
3621 .sql
3622 .insert(
3623 "INSERT INTO chats
3624 (type, name, name_normalized, grpid, param, created_timestamp)
3625 VALUES(?, ?, ?, ?, \'U=1\', ?)",
3626 (
3627 Chattype::Group,
3628 &chat_name,
3629 normalize_text(&chat_name),
3630 &grpid,
3631 timestamp,
3632 ),
3633 )
3634 .await?;
3635
3636 let chat_id = ChatId::new(u32::try_from(row_id)?);
3637 add_to_chat_contacts_table(context, timestamp, chat_id, &[ContactId::SELF]).await?;
3638
3639 context.emit_msgs_changed_without_ids();
3640 chatlist_events::emit_chatlist_changed(context);
3641 chatlist_events::emit_chatlist_item_changed(context, chat_id);
3642
3643 if !grpid.is_empty() {
3644 chat_id.add_e2ee_notice(context, timestamp).await?;
3646 }
3647
3648 if !context.get_config_bool(Config::Bot).await?
3649 && !context.get_config_bool(Config::SkipStartMessages).await?
3650 {
3651 let text = if !grpid.is_empty() {
3652 stock_str::new_group_send_first_message(context)
3654 } else {
3655 stock_str::chat_unencrypted_explanation(context)
3657 };
3658 chat_id.add_start_info_message(context, &text).await?;
3659 }
3660 if let (true, true) = (sync.into(), !grpid.is_empty()) {
3661 let id = SyncId::Grpid(grpid);
3662 let action = SyncAction::CreateGroupEncrypted(chat_name);
3663 self::sync(context, id, action).await.log_err(context).ok();
3664 }
3665 Ok(chat_id)
3666}
3667
3668pub async fn create_broadcast(context: &Context, chat_name: String) -> Result<ChatId> {
3684 let grpid = create_id();
3685 let secret = create_broadcast_secret();
3686 create_out_broadcast_ex(context, Sync, grpid, chat_name, secret).await
3687}
3688
3689const SQL_INSERT_BROADCAST_SECRET: &str =
3690 "INSERT INTO broadcast_secrets (chat_id, secret) VALUES (?, ?)
3691 ON CONFLICT(chat_id) DO UPDATE SET secret=excluded.secret";
3692
3693pub(crate) async fn create_out_broadcast_ex(
3694 context: &Context,
3695 sync: sync::Sync,
3696 grpid: String,
3697 chat_name: String,
3698 secret: String,
3699) -> Result<ChatId> {
3700 let chat_name = sanitize_single_line(&chat_name);
3701 if chat_name.is_empty() {
3702 bail!("Invalid broadcast channel name: {chat_name}.");
3703 }
3704
3705 let timestamp = create_smeared_timestamp(context);
3706 let trans_fn = |t: &mut rusqlite::Transaction| -> Result<ChatId> {
3707 let cnt: u32 = t.query_row(
3708 "SELECT COUNT(*) FROM chats WHERE grpid=?",
3709 (&grpid,),
3710 |row| row.get(0),
3711 )?;
3712 ensure!(cnt == 0, "{cnt} chats exist with grpid {grpid}");
3713
3714 t.execute(
3715 "INSERT INTO chats
3716 (type, name, name_normalized, grpid, created_timestamp)
3717 VALUES(?, ?, ?, ?, ?)",
3718 (
3719 Chattype::OutBroadcast,
3720 &chat_name,
3721 normalize_text(&chat_name),
3722 &grpid,
3723 timestamp,
3724 ),
3725 )?;
3726 let chat_id = ChatId::new(t.last_insert_rowid().try_into()?);
3727
3728 t.execute(SQL_INSERT_BROADCAST_SECRET, (chat_id, &secret))?;
3729 Ok(chat_id)
3730 };
3731 let chat_id = context.sql.transaction(trans_fn).await?;
3732 chat_id.add_e2ee_notice(context, timestamp).await?;
3733
3734 context.emit_msgs_changed_without_ids();
3735 chatlist_events::emit_chatlist_changed(context);
3736 chatlist_events::emit_chatlist_item_changed(context, chat_id);
3737
3738 if sync.into() {
3739 let id = SyncId::Grpid(grpid);
3740 let action = SyncAction::CreateOutBroadcast { chat_name, secret };
3741 self::sync(context, id, action).await.log_err(context).ok();
3742 }
3743
3744 Ok(chat_id)
3745}
3746
3747pub(crate) async fn load_broadcast_secret(
3748 context: &Context,
3749 chat_id: ChatId,
3750) -> Result<Option<String>> {
3751 context
3752 .sql
3753 .query_get_value(
3754 "SELECT secret FROM broadcast_secrets WHERE chat_id=?",
3755 (chat_id,),
3756 )
3757 .await
3758}
3759
3760pub(crate) async fn save_broadcast_secret(
3761 context: &Context,
3762 chat_id: ChatId,
3763 secret: &str,
3764) -> Result<()> {
3765 info!(context, "Saving broadcast secret for chat {chat_id}");
3766 context
3767 .sql
3768 .execute(SQL_INSERT_BROADCAST_SECRET, (chat_id, secret))
3769 .await?;
3770
3771 Ok(())
3772}
3773
3774pub(crate) async fn delete_broadcast_secret(context: &Context, chat_id: ChatId) -> Result<()> {
3775 info!(context, "Removing broadcast secret for chat {chat_id}");
3776 context
3777 .sql
3778 .execute("DELETE FROM broadcast_secrets WHERE chat_id=?", (chat_id,))
3779 .await?;
3780
3781 Ok(())
3782}
3783
3784pub(crate) async fn update_chat_contacts_table(
3786 context: &Context,
3787 timestamp: i64,
3788 id: ChatId,
3789 contacts: &BTreeSet<ContactId>,
3790) -> Result<()> {
3791 context
3792 .sql
3793 .transaction(move |transaction| {
3794 transaction.execute(
3798 "UPDATE chats_contacts
3799 SET remove_timestamp=MAX(add_timestamp+1, ?)
3800 WHERE chat_id=?",
3801 (timestamp, id),
3802 )?;
3803
3804 if !contacts.is_empty() {
3805 let mut statement = transaction.prepare(
3806 "INSERT INTO chats_contacts (chat_id, contact_id, add_timestamp)
3807 VALUES (?1, ?2, ?3)
3808 ON CONFLICT (chat_id, contact_id)
3809 DO UPDATE SET add_timestamp=remove_timestamp",
3810 )?;
3811
3812 for contact_id in contacts {
3813 statement.execute((id, contact_id, timestamp))?;
3817 }
3818 }
3819 Ok(())
3820 })
3821 .await?;
3822 Ok(())
3823}
3824
3825pub(crate) async fn add_to_chat_contacts_table(
3827 context: &Context,
3828 timestamp: i64,
3829 chat_id: ChatId,
3830 contact_ids: &[ContactId],
3831) -> Result<()> {
3832 context
3833 .sql
3834 .transaction(move |transaction| {
3835 let mut add_statement = transaction.prepare(
3836 "INSERT INTO chats_contacts (chat_id, contact_id, add_timestamp) VALUES(?1, ?2, ?3)
3837 ON CONFLICT (chat_id, contact_id)
3838 DO UPDATE SET add_timestamp=MAX(remove_timestamp, ?3)",
3839 )?;
3840
3841 for contact_id in contact_ids {
3842 add_statement.execute((chat_id, contact_id, timestamp))?;
3843 }
3844 Ok(())
3845 })
3846 .await?;
3847
3848 Ok(())
3849}
3850
3851pub(crate) async fn remove_from_chat_contacts_table(
3854 context: &Context,
3855 chat_id: ChatId,
3856 contact_id: ContactId,
3857) -> Result<()> {
3858 let now = time();
3859 context
3860 .sql
3861 .execute(
3862 "UPDATE chats_contacts
3863 SET remove_timestamp=MAX(add_timestamp+1, ?)
3864 WHERE chat_id=? AND contact_id=?",
3865 (now, chat_id, contact_id),
3866 )
3867 .await?;
3868 Ok(())
3869}
3870
3871pub(crate) async fn remove_from_chat_contacts_table_without_trace(
3879 context: &Context,
3880 chat_id: ChatId,
3881 contact_id: ContactId,
3882) -> Result<()> {
3883 context
3884 .sql
3885 .execute(
3886 "DELETE FROM chats_contacts
3887 WHERE chat_id=? AND contact_id=?",
3888 (chat_id, contact_id),
3889 )
3890 .await?;
3891
3892 Ok(())
3893}
3894
3895pub async fn add_contact_to_chat(
3898 context: &Context,
3899 chat_id: ChatId,
3900 contact_id: ContactId,
3901) -> Result<()> {
3902 add_contact_to_chat_ex(context, Sync, chat_id, contact_id, false).await?;
3903 Ok(())
3904}
3905
3906pub(crate) async fn add_contact_to_chat_ex(
3907 context: &Context,
3908 mut sync: sync::Sync,
3909 chat_id: ChatId,
3910 contact_id: ContactId,
3911 from_handshake: bool,
3912) -> Result<bool> {
3913 ensure!(!chat_id.is_special(), "can not add member to special chats");
3914 let contact = Contact::get_by_id(context, contact_id).await?;
3915 let mut msg = Message::new(Viewtype::default());
3916
3917 chat_id.reset_gossiped_timestamp(context).await?;
3918
3919 let mut chat = Chat::load_from_db(context, chat_id).await?;
3921 ensure!(
3922 chat.typ == Chattype::Group || (from_handshake && chat.typ == Chattype::OutBroadcast),
3923 "{chat_id} is not a group where one can add members",
3924 );
3925 ensure!(
3926 Contact::real_exists_by_id(context, contact_id).await? || contact_id == ContactId::SELF,
3927 "invalid contact_id {contact_id} for adding to group"
3928 );
3929 ensure!(
3930 chat.typ != Chattype::OutBroadcast || contact_id != ContactId::SELF,
3931 "Cannot add SELF to broadcast channel."
3932 );
3933 match chat.is_encrypted(context).await? {
3934 true => ensure!(
3935 contact.is_key_contact(),
3936 "Only key-contacts can be added to encrypted chats"
3937 ),
3938 false => ensure!(
3939 !contact.is_key_contact(),
3940 "Only address-contacts can be added to unencrypted chats"
3941 ),
3942 }
3943
3944 if !chat.is_self_in_chat(context).await? {
3945 context.emit_event(EventType::ErrorSelfNotInGroup(
3946 "Cannot add contact to group; self not in group.".into(),
3947 ));
3948 warn!(
3949 context,
3950 "Can not add contact because the account is not part of the group/broadcast."
3951 );
3952 return Ok(false);
3953 }
3954 if from_handshake && chat.param.get_int(Param::Unpromoted).unwrap_or_default() == 1 {
3955 let smeared_time = smeared_time(context);
3956 chat.param
3957 .remove(Param::Unpromoted)
3958 .set_i64(Param::GroupNameTimestamp, smeared_time)
3959 .set_i64(Param::GroupDescriptionTimestamp, smeared_time);
3960 chat.update_param(context).await?;
3961 }
3962 if context.is_self_addr(contact.get_addr()).await? {
3963 warn!(
3966 context,
3967 "Invalid attempt to add self e-mail address to group."
3968 );
3969 return Ok(false);
3970 }
3971
3972 if is_contact_in_chat(context, chat_id, contact_id).await? {
3973 if !from_handshake {
3974 return Ok(true);
3975 }
3976 } else {
3977 add_to_chat_contacts_table(context, time(), chat_id, &[contact_id]).await?;
3979 }
3980 if chat.is_promoted() {
3981 msg.viewtype = Viewtype::Text;
3982
3983 let contact_addr = contact.get_addr().to_lowercase();
3984 let added_by = if from_handshake && chat.typ == Chattype::OutBroadcast {
3985 ContactId::UNDEFINED
3990 } else {
3991 ContactId::SELF
3992 };
3993 msg.text = stock_str::msg_add_member_local(context, contact.id, added_by).await;
3994 msg.param.set_cmd(SystemMessage::MemberAddedToGroup);
3995 msg.param.set(Param::Arg, contact_addr);
3996 msg.param.set_int(Param::Arg2, from_handshake.into());
3997 let fingerprint = contact.fingerprint().map(|f| f.hex());
3998 msg.param.set_optional(Param::Arg4, fingerprint);
3999 msg.param
4000 .set_int(Param::ContactAddedRemoved, contact.id.to_u32() as i32);
4001 if chat.typ == Chattype::OutBroadcast {
4002 let secret = load_broadcast_secret(context, chat_id)
4003 .await?
4004 .context("Failed to find broadcast shared secret")?;
4005 msg.param.set(PARAM_BROADCAST_SECRET, secret);
4006 }
4007 send_msg(context, chat_id, &mut msg).await?;
4008
4009 sync = Nosync;
4010 }
4011 context.emit_event(EventType::ChatModified(chat_id));
4012 if sync.into() {
4013 chat.sync_contacts(context).await.log_err(context).ok();
4014 }
4015 if chat.typ == Chattype::OutBroadcast {
4016 resend_last_msgs(context, chat.id, &contact)
4017 .await
4018 .log_err(context)
4019 .ok();
4020 }
4021 Ok(true)
4022}
4023
4024async fn resend_last_msgs(context: &Context, chat_id: ChatId, to_contact: &Contact) -> Result<()> {
4025 let msgs: Vec<MsgId> = context
4026 .sql
4027 .query_map_vec(
4028 "
4029SELECT id
4030FROM msgs
4031WHERE chat_id=?
4032 AND hidden=0
4033 AND NOT ( -- Exclude info and system messages
4034 param GLOB '*\nS=*' OR param GLOB 'S=*'
4035 OR from_id=?
4036 OR to_id=?
4037 )
4038 AND type!=?
4039ORDER BY timestamp DESC, id DESC LIMIT ?",
4040 (
4041 chat_id,
4042 ContactId::INFO,
4043 ContactId::INFO,
4044 Viewtype::Webxdc,
4045 constants::N_MSGS_TO_NEW_BROADCAST_MEMBER,
4046 ),
4047 |row: &rusqlite::Row| Ok(row.get::<_, MsgId>(0)?),
4048 )
4049 .await?
4050 .into_iter()
4051 .rev()
4052 .collect();
4053 resend_msgs_ex(context, &msgs, to_contact.fingerprint()).await
4054}
4055
4056#[expect(clippy::arithmetic_side_effects)]
4062pub(crate) async fn shall_attach_selfavatar(context: &Context, chat_id: ChatId) -> Result<bool> {
4063 let timestamp_some_days_ago = time() - DC_RESEND_USER_AVATAR_DAYS * 24 * 60 * 60;
4064 let needs_attach = context
4065 .sql
4066 .query_map(
4067 "SELECT c.selfavatar_sent
4068 FROM chats_contacts cc
4069 LEFT JOIN contacts c ON c.id=cc.contact_id
4070 WHERE cc.chat_id=? AND cc.contact_id!=? AND cc.add_timestamp >= cc.remove_timestamp",
4071 (chat_id, ContactId::SELF),
4072 |row| {
4073 let selfavatar_sent: i64 = row.get(0)?;
4074 Ok(selfavatar_sent)
4075 },
4076 |rows| {
4077 let mut needs_attach = false;
4078 for row in rows {
4079 let selfavatar_sent = row?;
4080 if selfavatar_sent < timestamp_some_days_ago {
4081 needs_attach = true;
4082 }
4083 }
4084 Ok(needs_attach)
4085 },
4086 )
4087 .await?;
4088 Ok(needs_attach)
4089}
4090
4091#[derive(Debug, Copy, Clone, PartialEq, Eq, Serialize, Deserialize)]
4093pub enum MuteDuration {
4094 NotMuted,
4096
4097 Forever,
4099
4100 Until(std::time::SystemTime),
4102}
4103
4104impl rusqlite::types::ToSql for MuteDuration {
4105 fn to_sql(&self) -> rusqlite::Result<rusqlite::types::ToSqlOutput<'_>> {
4106 let duration: i64 = match &self {
4107 MuteDuration::NotMuted => 0,
4108 MuteDuration::Forever => -1,
4109 MuteDuration::Until(when) => {
4110 let duration = when
4111 .duration_since(SystemTime::UNIX_EPOCH)
4112 .map_err(|err| rusqlite::Error::ToSqlConversionFailure(Box::new(err)))?;
4113 i64::try_from(duration.as_secs())
4114 .map_err(|err| rusqlite::Error::ToSqlConversionFailure(Box::new(err)))?
4115 }
4116 };
4117 let val = rusqlite::types::Value::Integer(duration);
4118 let out = rusqlite::types::ToSqlOutput::Owned(val);
4119 Ok(out)
4120 }
4121}
4122
4123impl rusqlite::types::FromSql for MuteDuration {
4124 fn column_result(value: rusqlite::types::ValueRef) -> rusqlite::types::FromSqlResult<Self> {
4125 match i64::column_result(value)? {
4128 0 => Ok(MuteDuration::NotMuted),
4129 -1 => Ok(MuteDuration::Forever),
4130 n if n > 0 => match SystemTime::UNIX_EPOCH.checked_add(Duration::from_secs(n as u64)) {
4131 Some(t) => Ok(MuteDuration::Until(t)),
4132 None => Err(rusqlite::types::FromSqlError::OutOfRange(n)),
4133 },
4134 _ => Ok(MuteDuration::NotMuted),
4135 }
4136 }
4137}
4138
4139pub async fn set_muted(context: &Context, chat_id: ChatId, duration: MuteDuration) -> Result<()> {
4141 set_muted_ex(context, Sync, chat_id, duration).await
4142}
4143
4144pub(crate) async fn set_muted_ex(
4145 context: &Context,
4146 sync: sync::Sync,
4147 chat_id: ChatId,
4148 duration: MuteDuration,
4149) -> Result<()> {
4150 ensure!(!chat_id.is_special(), "Invalid chat ID");
4151 context
4152 .sql
4153 .execute(
4154 "UPDATE chats SET muted_until=? WHERE id=?;",
4155 (duration, chat_id),
4156 )
4157 .await
4158 .context(format!("Failed to set mute duration for {chat_id}"))?;
4159 context.emit_event(EventType::ChatModified(chat_id));
4160 chatlist_events::emit_chatlist_item_changed(context, chat_id);
4161 if sync.into() {
4162 let chat = Chat::load_from_db(context, chat_id).await?;
4163 chat.sync(context, SyncAction::SetMuted(duration))
4164 .await
4165 .log_err(context)
4166 .ok();
4167 }
4168 Ok(())
4169}
4170
4171pub async fn remove_contact_from_chat(
4173 context: &Context,
4174 chat_id: ChatId,
4175 contact_id: ContactId,
4176) -> Result<()> {
4177 ensure!(
4178 !chat_id.is_special(),
4179 "bad chat_id, can not be special chat: {chat_id}"
4180 );
4181 ensure!(
4182 !contact_id.is_special() || contact_id == ContactId::SELF,
4183 "Cannot remove special contact"
4184 );
4185
4186 let chat = Chat::load_from_db(context, chat_id).await?;
4187 if chat.typ == Chattype::InBroadcast {
4188 ensure!(
4189 contact_id == ContactId::SELF,
4190 "Cannot remove other member from incoming broadcast channel"
4191 );
4192 delete_broadcast_secret(context, chat_id).await?;
4193 }
4194
4195 ensure!(
4196 matches!(
4197 chat.typ,
4198 Chattype::Group | Chattype::OutBroadcast | Chattype::InBroadcast
4199 ),
4200 "Cannot remove members from non-group chats."
4201 );
4202
4203 if !chat.is_self_in_chat(context).await? {
4204 let err_msg =
4205 format!("Cannot remove contact {contact_id} from chat {chat_id}: self not in group.");
4206 context.emit_event(EventType::ErrorSelfNotInGroup(err_msg.clone()));
4207 bail!("{err_msg}");
4208 }
4209
4210 let mut sync = Nosync;
4211
4212 if chat.is_promoted() && chat.typ != Chattype::OutBroadcast {
4213 remove_from_chat_contacts_table(context, chat_id, contact_id).await?;
4214 } else {
4215 remove_from_chat_contacts_table_without_trace(context, chat_id, contact_id).await?;
4216 }
4217
4218 if let Some(contact) = Contact::get_by_id_optional(context, contact_id).await? {
4222 if chat.is_promoted() {
4223 let addr = contact.get_addr();
4224 let fingerprint = contact.fingerprint().map(|f| f.hex());
4225
4226 let res =
4227 send_member_removal_msg(context, &chat, contact_id, addr, fingerprint.as_deref())
4228 .await;
4229
4230 if contact_id == ContactId::SELF {
4231 res?;
4232 } else if let Err(e) = res {
4233 warn!(
4234 context,
4235 "remove_contact_from_chat({chat_id}, {contact_id}): send_msg() failed: {e:#}."
4236 );
4237 }
4238 } else {
4239 sync = Sync;
4240 }
4241 }
4242 context.emit_event(EventType::ChatModified(chat_id));
4243 if sync.into() {
4244 chat.sync_contacts(context).await.log_err(context).ok();
4245 }
4246
4247 Ok(())
4248}
4249
4250async fn send_member_removal_msg(
4251 context: &Context,
4252 chat: &Chat,
4253 contact_id: ContactId,
4254 addr: &str,
4255 fingerprint: Option<&str>,
4256) -> Result<MsgId> {
4257 let mut msg = Message::new(Viewtype::Text);
4258
4259 if contact_id == ContactId::SELF {
4260 if chat.typ == Chattype::InBroadcast {
4261 msg.text = stock_str::msg_you_left_broadcast(context);
4262 } else {
4263 msg.text = stock_str::msg_group_left_local(context, ContactId::SELF).await;
4264 }
4265 } else {
4266 msg.text = stock_str::msg_del_member_local(context, contact_id, ContactId::SELF).await;
4267 }
4268
4269 msg.param.set_cmd(SystemMessage::MemberRemovedFromGroup);
4270 msg.param.set(Param::Arg, addr.to_lowercase());
4271 msg.param.set_optional(Param::Arg4, fingerprint);
4272 msg.param
4273 .set(Param::ContactAddedRemoved, contact_id.to_u32());
4274
4275 send_msg(context, chat.id, &mut msg).await
4276}
4277
4278pub async fn set_chat_description(
4288 context: &Context,
4289 chat_id: ChatId,
4290 new_description: &str,
4291) -> Result<()> {
4292 set_chat_description_ex(context, Sync, chat_id, new_description).await
4293}
4294
4295async fn set_chat_description_ex(
4296 context: &Context,
4297 mut sync: sync::Sync,
4298 chat_id: ChatId,
4299 new_description: &str,
4300) -> Result<()> {
4301 let new_description = sanitize_bidi_characters(new_description.trim());
4302
4303 ensure!(!chat_id.is_special(), "Invalid chat ID");
4304
4305 let chat = Chat::load_from_db(context, chat_id).await?;
4306 ensure!(
4307 chat.typ == Chattype::Group || chat.typ == Chattype::OutBroadcast,
4308 "Can only set description for groups / broadcasts"
4309 );
4310 ensure!(
4311 !chat.grpid.is_empty(),
4312 "Cannot set description for ad hoc groups"
4313 );
4314 if !chat.is_self_in_chat(context).await? {
4315 context.emit_event(EventType::ErrorSelfNotInGroup(
4316 "Cannot set chat description; self not in group".into(),
4317 ));
4318 bail!("Cannot set chat description; self not in group");
4319 }
4320
4321 let old_description = get_chat_description(context, chat_id).await?;
4322 if old_description == new_description {
4323 return Ok(());
4324 }
4325
4326 context
4327 .sql
4328 .execute(
4329 "INSERT OR REPLACE INTO chats_descriptions(chat_id, description) VALUES(?, ?)",
4330 (chat_id, &new_description),
4331 )
4332 .await?;
4333
4334 if chat.is_promoted() {
4335 let mut msg = Message::new(Viewtype::Text);
4336 msg.text = stock_str::msg_chat_description_changed(context, ContactId::SELF).await;
4337 msg.param.set_cmd(SystemMessage::GroupDescriptionChanged);
4338
4339 msg.id = send_msg(context, chat_id, &mut msg).await?;
4340 context.emit_msgs_changed(chat_id, msg.id);
4341 sync = Nosync;
4342 }
4343 context.emit_event(EventType::ChatModified(chat_id));
4344
4345 if sync.into() {
4346 chat.sync(context, SyncAction::SetDescription(new_description))
4347 .await
4348 .log_err(context)
4349 .ok();
4350 }
4351
4352 Ok(())
4353}
4354
4355pub async fn get_chat_description(context: &Context, chat_id: ChatId) -> Result<String> {
4360 let description = context
4361 .sql
4362 .query_get_value(
4363 "SELECT description FROM chats_descriptions WHERE chat_id=?",
4364 (chat_id,),
4365 )
4366 .await?
4367 .unwrap_or_default();
4368 Ok(description)
4369}
4370
4371pub async fn set_chat_name(context: &Context, chat_id: ChatId, new_name: &str) -> Result<()> {
4379 rename_ex(context, Sync, chat_id, new_name).await
4380}
4381
4382async fn rename_ex(
4383 context: &Context,
4384 mut sync: sync::Sync,
4385 chat_id: ChatId,
4386 new_name: &str,
4387) -> Result<()> {
4388 let new_name = sanitize_single_line(new_name);
4389 let mut success = false;
4391
4392 ensure!(!new_name.is_empty(), "Invalid name");
4393 ensure!(!chat_id.is_special(), "Invalid chat ID");
4394
4395 let chat = Chat::load_from_db(context, chat_id).await?;
4396 let mut msg = Message::new(Viewtype::default());
4397
4398 if chat.typ == Chattype::Group
4399 || chat.typ == Chattype::Mailinglist
4400 || chat.typ == Chattype::OutBroadcast
4401 {
4402 if chat.name == new_name {
4403 success = true;
4404 } else if !chat.is_self_in_chat(context).await? {
4405 context.emit_event(EventType::ErrorSelfNotInGroup(
4406 "Cannot set chat name; self not in group".into(),
4407 ));
4408 } else {
4409 context
4410 .sql
4411 .execute(
4412 "UPDATE chats SET name=?, name_normalized=? WHERE id=?",
4413 (&new_name, normalize_text(&new_name), chat_id),
4414 )
4415 .await?;
4416 if chat.is_promoted()
4417 && !chat.is_mailing_list()
4418 && sanitize_single_line(&chat.name) != new_name
4419 {
4420 msg.viewtype = Viewtype::Text;
4421 msg.text = if chat.typ == Chattype::OutBroadcast {
4422 stock_str::msg_broadcast_name_changed(context, &chat.name, &new_name)
4423 } else {
4424 stock_str::msg_grp_name(context, &chat.name, &new_name, ContactId::SELF).await
4425 };
4426 msg.param.set_cmd(SystemMessage::GroupNameChanged);
4427 if !chat.name.is_empty() {
4428 msg.param.set(Param::Arg, &chat.name);
4429 }
4430 msg.id = send_msg(context, chat_id, &mut msg).await?;
4431 context.emit_msgs_changed(chat_id, msg.id);
4432 sync = Nosync;
4433 }
4434 context.emit_event(EventType::ChatModified(chat_id));
4435 chatlist_events::emit_chatlist_item_changed(context, chat_id);
4436 success = true;
4437 }
4438 }
4439
4440 if !success {
4441 bail!("Failed to set name");
4442 }
4443 if sync.into() && chat.name != new_name {
4444 let sync_name = new_name.to_string();
4445 chat.sync(context, SyncAction::Rename(sync_name))
4446 .await
4447 .log_err(context)
4448 .ok();
4449 }
4450 Ok(())
4451}
4452
4453pub async fn set_chat_profile_image(
4459 context: &Context,
4460 chat_id: ChatId,
4461 new_image: &str, ) -> Result<()> {
4463 ensure!(!chat_id.is_special(), "Invalid chat ID");
4464 let mut chat = Chat::load_from_db(context, chat_id).await?;
4465 ensure!(
4466 chat.typ == Chattype::Group || chat.typ == Chattype::OutBroadcast,
4467 "Can only set profile image for groups / broadcasts"
4468 );
4469 ensure!(
4470 !chat.grpid.is_empty(),
4471 "Cannot set profile image for ad hoc groups"
4472 );
4473 if !chat.is_self_in_chat(context).await? {
4475 context.emit_event(EventType::ErrorSelfNotInGroup(
4476 "Cannot set chat profile image; self not in group.".into(),
4477 ));
4478 bail!("Failed to set profile image");
4479 }
4480 let mut msg = Message::new(Viewtype::Text);
4481 msg.param
4482 .set_int(Param::Cmd, SystemMessage::GroupImageChanged as i32);
4483 if new_image.is_empty() {
4484 chat.param.remove(Param::ProfileImage);
4485 msg.param.remove(Param::Arg);
4486 msg.text = if chat.typ == Chattype::OutBroadcast {
4487 stock_str::msg_broadcast_img_changed(context)
4488 } else {
4489 stock_str::msg_grp_img_deleted(context, ContactId::SELF).await
4490 };
4491 } else {
4492 let mut image_blob = BlobObject::create_and_deduplicate(
4493 context,
4494 Path::new(new_image),
4495 Path::new(new_image),
4496 )?;
4497 image_blob.recode_to_avatar_size(context).await?;
4498 chat.param.set(Param::ProfileImage, image_blob.as_name());
4499 msg.param.set(Param::Arg, image_blob.as_name());
4500 msg.text = if chat.typ == Chattype::OutBroadcast {
4501 stock_str::msg_broadcast_img_changed(context)
4502 } else {
4503 stock_str::msg_grp_img_changed(context, ContactId::SELF).await
4504 };
4505 }
4506 chat.update_param(context).await?;
4507 if chat.is_promoted() {
4508 msg.id = send_msg(context, chat_id, &mut msg).await?;
4509 context.emit_msgs_changed(chat_id, msg.id);
4510 }
4511 context.emit_event(EventType::ChatModified(chat_id));
4512 chatlist_events::emit_chatlist_item_changed(context, chat_id);
4513 Ok(())
4514}
4515
4516pub async fn forward_msgs(context: &Context, msg_ids: &[MsgId], chat_id: ChatId) -> Result<()> {
4518 forward_msgs_2ctx(context, msg_ids, context, chat_id).await
4519}
4520
4521#[expect(clippy::arithmetic_side_effects)]
4523pub async fn forward_msgs_2ctx(
4524 ctx_src: &Context,
4525 msg_ids: &[MsgId],
4526 ctx_dst: &Context,
4527 chat_id: ChatId,
4528) -> Result<()> {
4529 ensure!(!msg_ids.is_empty(), "empty msgs_ids: nothing to forward");
4530 ensure!(!chat_id.is_special(), "can not forward to special chat");
4531
4532 let mut created_msgs: Vec<MsgId> = Vec::new();
4533 let mut curr_timestamp: i64;
4534
4535 chat_id
4536 .unarchive_if_not_muted(ctx_dst, MessageState::Undefined)
4537 .await?;
4538 let mut chat = Chat::load_from_db(ctx_dst, chat_id).await?;
4539 if let Some(reason) = chat.why_cant_send(ctx_dst).await? {
4540 bail!("cannot send to {chat_id}: {reason}");
4541 }
4542 curr_timestamp = create_smeared_timestamps(ctx_dst, msg_ids.len());
4543 let mut msgs = Vec::with_capacity(msg_ids.len());
4544 for id in msg_ids {
4545 let ts: i64 = ctx_src
4546 .sql
4547 .query_get_value("SELECT timestamp FROM msgs WHERE id=?", (id,))
4548 .await?
4549 .with_context(|| format!("No message {id}"))?;
4550 msgs.push((ts, *id));
4551 }
4552 msgs.sort_unstable();
4553 for (_, id) in msgs {
4554 let src_msg_id: MsgId = id;
4555 let mut msg = Message::load_from_db(ctx_src, src_msg_id).await?;
4556 if msg.state == MessageState::OutDraft {
4557 bail!("cannot forward drafts.");
4558 }
4559
4560 let mut param = msg.param;
4561 msg.param = Params::new();
4562
4563 if msg.get_viewtype() != Viewtype::Sticker {
4564 let forwarded_msg_id = match ctx_src.blobdir == ctx_dst.blobdir {
4565 true => src_msg_id,
4566 false => MsgId::new_unset(),
4567 };
4568 msg.param
4569 .set_int(Param::Forwarded, forwarded_msg_id.to_u32() as i32);
4570 }
4571
4572 if msg.get_viewtype() == Viewtype::Call {
4573 msg.viewtype = Viewtype::Text;
4574 }
4575 msg.text += &msg.additional_text;
4576
4577 let param = &mut param;
4578
4579 if ctx_src.blobdir == ctx_dst.blobdir {
4582 msg.param.steal(param, Param::File);
4583 } else if let Some(src_path) = param.get_file_path(ctx_src)? {
4584 let new_blob = BlobObject::create_and_deduplicate(ctx_dst, &src_path, &src_path)
4585 .context("Failed to copy blob file to destination account")?;
4586 msg.param.set(Param::File, new_blob.as_name());
4587 }
4588 msg.param.steal(param, Param::Filename);
4589 msg.param.steal(param, Param::Width);
4590 msg.param.steal(param, Param::Height);
4591 msg.param.steal(param, Param::Duration);
4592 msg.param.steal(param, Param::MimeType);
4593 msg.param.steal(param, Param::ProtectQuote);
4594 msg.param.steal(param, Param::Quote);
4595 msg.param.steal(param, Param::Summary1);
4596 if msg.has_html() {
4597 msg.set_html(src_msg_id.get_html(ctx_src).await?);
4598 }
4599 msg.in_reply_to = None;
4600
4601 msg.subject = "".to_string();
4603
4604 msg.state = MessageState::OutPending;
4605 msg.rfc724_mid = create_outgoing_rfc724_mid();
4606 msg.timestamp_sort = curr_timestamp;
4607 chat.prepare_msg_raw(ctx_dst, &mut msg, None).await?;
4608
4609 curr_timestamp += 1;
4610 if !create_send_msg_jobs(ctx_dst, &mut msg).await?.is_empty() {
4611 ctx_dst.scheduler.interrupt_smtp().await;
4612 }
4613 created_msgs.push(msg.id);
4614 }
4615 for msg_id in created_msgs {
4616 ctx_dst.emit_msgs_changed(chat_id, msg_id);
4617 }
4618 Ok(())
4619}
4620
4621pub async fn save_msgs(context: &Context, msg_ids: &[MsgId]) -> Result<()> {
4624 let mut msgs = Vec::with_capacity(msg_ids.len());
4625 for id in msg_ids {
4626 let ts: i64 = context
4627 .sql
4628 .query_get_value("SELECT timestamp FROM msgs WHERE id=?", (id,))
4629 .await?
4630 .with_context(|| format!("No message {id}"))?;
4631 msgs.push((ts, *id));
4632 }
4633 msgs.sort_unstable();
4634 for (_, src_msg_id) in msgs {
4635 let dest_rfc724_mid = create_outgoing_rfc724_mid();
4636 let src_rfc724_mid = save_copy_in_self_talk(context, src_msg_id, &dest_rfc724_mid).await?;
4637 context
4638 .add_sync_item(SyncData::SaveMessage {
4639 src: src_rfc724_mid,
4640 dest: dest_rfc724_mid,
4641 })
4642 .await?;
4643 }
4644 context.scheduler.interrupt_smtp().await;
4645 Ok(())
4646}
4647
4648pub(crate) async fn save_copy_in_self_talk(
4654 context: &Context,
4655 src_msg_id: MsgId,
4656 dest_rfc724_mid: &String,
4657) -> Result<String> {
4658 let dest_chat_id = ChatId::create_for_contact(context, ContactId::SELF).await?;
4659 let mut msg = Message::load_from_db(context, src_msg_id).await?;
4660 msg.param.remove(Param::Cmd);
4661 msg.param.remove(Param::WebxdcDocument);
4662 msg.param.remove(Param::WebxdcDocumentTimestamp);
4663 msg.param.remove(Param::WebxdcSummary);
4664 msg.param.remove(Param::WebxdcSummaryTimestamp);
4665 msg.param.remove(Param::PostMessageFileBytes);
4666 msg.param.remove(Param::PostMessageViewtype);
4667
4668 msg.text += &msg.additional_text;
4669
4670 if !msg.original_msg_id.is_unset() {
4671 bail!("message already saved.");
4672 }
4673
4674 let copy_fields = "from_id, to_id, timestamp_rcvd, type,
4675 mime_modified, mime_headers, mime_compressed, mime_in_reply_to, subject, msgrmsg";
4676 let row_id = context
4677 .sql
4678 .insert(
4679 &format!(
4680 "INSERT INTO msgs ({copy_fields},
4681 timestamp_sent,
4682 txt, chat_id, rfc724_mid, state, timestamp, param, starred)
4683 SELECT {copy_fields},
4684 -- Outgoing messages on originating device
4685 -- have timestamp_sent == 0.
4686 -- We copy sort timestamp instead
4687 -- so UIs display the same timestamp
4688 -- for saved and original message.
4689 IIF(timestamp_sent == 0, timestamp, timestamp_sent),
4690 ?, ?, ?, ?, ?, ?, ?
4691 FROM msgs WHERE id=?;"
4692 ),
4693 (
4694 msg.text,
4695 dest_chat_id,
4696 dest_rfc724_mid,
4697 if msg.from_id == ContactId::SELF {
4698 MessageState::OutDelivered
4699 } else {
4700 MessageState::InSeen
4701 },
4702 create_smeared_timestamp(context),
4703 msg.param.to_string(),
4704 src_msg_id,
4705 src_msg_id,
4706 ),
4707 )
4708 .await?;
4709 let dest_msg_id = MsgId::new(row_id.try_into()?);
4710
4711 context.emit_msgs_changed(msg.chat_id, src_msg_id);
4712 context.emit_msgs_changed(dest_chat_id, dest_msg_id);
4713 chatlist_events::emit_chatlist_changed(context);
4714 chatlist_events::emit_chatlist_item_changed(context, dest_chat_id);
4715
4716 Ok(msg.rfc724_mid)
4717}
4718
4719pub async fn resend_msgs(context: &Context, msg_ids: &[MsgId]) -> Result<()> {
4723 resend_msgs_ex(context, msg_ids, None).await
4724}
4725
4726pub(crate) async fn resend_msgs_ex(
4734 context: &Context,
4735 msg_ids: &[MsgId],
4736 to_fingerprint: Option<Fingerprint>,
4737) -> Result<()> {
4738 let to_fingerprint = to_fingerprint.map(|f| f.hex());
4739 let mut msgs: Vec<Message> = Vec::new();
4740 for msg_id in msg_ids {
4741 let msg = Message::load_from_db(context, *msg_id).await?;
4742 ensure!(
4743 msg.from_id == ContactId::SELF,
4744 "can resend only own messages"
4745 );
4746 ensure!(!msg.is_info(), "cannot resend info messages");
4747 msgs.push(msg)
4748 }
4749
4750 for mut msg in msgs {
4751 match msg.get_state() {
4752 MessageState::OutPending
4754 | MessageState::OutFailed
4755 | MessageState::OutDelivered
4756 | MessageState::OutMdnRcvd => {
4757 if to_fingerprint.is_none() {
4760 message::update_msg_state(context, msg.id, MessageState::OutPending).await?;
4761 }
4762 }
4763 msg_state => bail!("Unexpected message state {msg_state}"),
4764 }
4765 if let Some(to_fingerprint) = &to_fingerprint {
4766 msg.param.set(Param::Arg4, to_fingerprint.clone());
4767 }
4768 if create_send_msg_jobs(context, &mut msg).await?.is_empty() {
4769 continue;
4770 }
4771
4772 context.emit_event(EventType::MsgsChanged {
4776 chat_id: msg.chat_id,
4777 msg_id: msg.id,
4778 });
4779 chatlist_events::emit_chatlist_item_changed(context, msg.chat_id);
4782
4783 if msg.viewtype == Viewtype::Webxdc {
4784 let conn_fn = |conn: &mut rusqlite::Connection| {
4785 let range = conn.query_row(
4786 "SELECT IFNULL(min(id), 1), IFNULL(max(id), 0) \
4787 FROM msgs_status_updates WHERE msg_id=?",
4788 (msg.id,),
4789 |row| {
4790 let min_id: StatusUpdateSerial = row.get(0)?;
4791 let max_id: StatusUpdateSerial = row.get(1)?;
4792 Ok((min_id, max_id))
4793 },
4794 )?;
4795 if range.0 > range.1 {
4796 return Ok(());
4797 };
4798 conn.execute(
4802 "INSERT INTO smtp_status_updates (msg_id, first_serial, last_serial, descr) \
4803 VALUES(?, ?, ?, '') \
4804 ON CONFLICT(msg_id) \
4805 DO UPDATE SET first_serial=min(first_serial - 1, excluded.first_serial)",
4806 (msg.id, range.0, range.1),
4807 )?;
4808 Ok(())
4809 };
4810 context.sql.call_write(conn_fn).await?;
4811 }
4812 context.scheduler.interrupt_smtp().await;
4813 }
4814 Ok(())
4815}
4816
4817pub(crate) async fn get_chat_cnt(context: &Context) -> Result<usize> {
4818 if context.sql.is_open().await {
4819 let count = context
4821 .sql
4822 .count("SELECT COUNT(*) FROM chats WHERE id>9 AND blocked=0;", ())
4823 .await?;
4824 Ok(count)
4825 } else {
4826 Ok(0)
4827 }
4828}
4829
4830pub(crate) async fn get_chat_id_by_grpid(
4832 context: &Context,
4833 grpid: &str,
4834) -> Result<Option<(ChatId, Blocked)>> {
4835 context
4836 .sql
4837 .query_row_optional(
4838 "SELECT id, blocked FROM chats WHERE grpid=?;",
4839 (grpid,),
4840 |row| {
4841 let chat_id = row.get::<_, ChatId>(0)?;
4842
4843 let b = row.get::<_, Option<Blocked>>(1)?.unwrap_or_default();
4844 Ok((chat_id, b))
4845 },
4846 )
4847 .await
4848}
4849
4850#[expect(clippy::arithmetic_side_effects)]
4855pub async fn add_device_msg_with_importance(
4856 context: &Context,
4857 label: Option<&str>,
4858 msg: Option<&mut Message>,
4859 important: bool,
4860) -> Result<MsgId> {
4861 ensure!(
4862 label.is_some() || msg.is_some(),
4863 "device-messages need label, msg or both"
4864 );
4865 let mut chat_id = ChatId::new(0);
4866 let mut msg_id = MsgId::new_unset();
4867
4868 if let Some(label) = label
4869 && was_device_msg_ever_added(context, label).await?
4870 {
4871 info!(context, "Device-message {label} already added.");
4872 return Ok(msg_id);
4873 }
4874
4875 if let Some(msg) = msg {
4876 chat_id = ChatId::get_for_contact(context, ContactId::DEVICE).await?;
4877
4878 let rfc724_mid = create_outgoing_rfc724_mid();
4879 let timestamp_sent = create_smeared_timestamp(context);
4880
4881 msg.timestamp_sort = timestamp_sent;
4884 if let Some(last_msg_time) = chat_id.get_timestamp(context).await?
4885 && msg.timestamp_sort <= last_msg_time
4886 {
4887 msg.timestamp_sort = last_msg_time + 1;
4888 }
4889 prepare_msg_blob(context, msg).await?;
4890 let state = MessageState::InFresh;
4891 let row_id = context
4892 .sql
4893 .insert(
4894 "INSERT INTO msgs (
4895 chat_id,
4896 from_id,
4897 to_id,
4898 timestamp,
4899 timestamp_sent,
4900 timestamp_rcvd,
4901 type,state,
4902 txt,
4903 txt_normalized,
4904 param,
4905 rfc724_mid)
4906 VALUES (?,?,?,?,?,?,?,?,?,?,?,?);",
4907 (
4908 chat_id,
4909 ContactId::DEVICE,
4910 ContactId::SELF,
4911 msg.timestamp_sort,
4912 timestamp_sent,
4913 timestamp_sent, msg.viewtype,
4915 state,
4916 &msg.text,
4917 normalize_text(&msg.text),
4918 msg.param.to_string(),
4919 rfc724_mid,
4920 ),
4921 )
4922 .await?;
4923 context.new_msgs_notify.notify_one();
4924
4925 msg_id = MsgId::new(u32::try_from(row_id)?);
4926 if !msg.hidden {
4927 chat_id.unarchive_if_not_muted(context, state).await?;
4928 }
4929 }
4930
4931 if let Some(label) = label {
4932 context
4933 .sql
4934 .execute("INSERT INTO devmsglabels (label) VALUES (?);", (label,))
4935 .await?;
4936 }
4937
4938 if !msg_id.is_unset() {
4939 chat_id.emit_msg_event(context, msg_id, important);
4940 }
4941
4942 Ok(msg_id)
4943}
4944
4945pub async fn add_device_msg(
4947 context: &Context,
4948 label: Option<&str>,
4949 msg: Option<&mut Message>,
4950) -> Result<MsgId> {
4951 add_device_msg_with_importance(context, label, msg, false).await
4952}
4953
4954pub async fn was_device_msg_ever_added(context: &Context, label: &str) -> Result<bool> {
4956 ensure!(!label.is_empty(), "empty label");
4957 let exists = context
4958 .sql
4959 .exists(
4960 "SELECT COUNT(label) FROM devmsglabels WHERE label=?",
4961 (label,),
4962 )
4963 .await?;
4964
4965 Ok(exists)
4966}
4967
4968pub(crate) async fn delete_and_reset_all_device_msgs(context: &Context) -> Result<()> {
4976 context
4977 .sql
4978 .execute("DELETE FROM msgs WHERE from_id=?;", (ContactId::DEVICE,))
4979 .await?;
4980 context.sql.execute("DELETE FROM devmsglabels;", ()).await?;
4981
4982 context
4984 .sql
4985 .execute(
4986 r#"INSERT INTO devmsglabels (label) VALUES ("core-welcome-image"), ("core-welcome")"#,
4987 (),
4988 )
4989 .await?;
4990 context
4991 .set_config_internal(Config::QuotaExceeding, None)
4992 .await?;
4993 Ok(())
4994}
4995
4996#[expect(clippy::too_many_arguments)]
5001pub(crate) async fn add_info_msg_with_cmd(
5002 context: &Context,
5003 chat_id: ChatId,
5004 text: &str,
5005 cmd: SystemMessage,
5006 timestamp_sort: Option<i64>,
5009 timestamp_sent_rcvd: i64,
5011 parent: Option<&Message>,
5012 from_id: Option<ContactId>,
5013 added_removed_id: Option<ContactId>,
5014) -> Result<MsgId> {
5015 let rfc724_mid = create_outgoing_rfc724_mid();
5016 let ephemeral_timer = chat_id.get_ephemeral_timer(context).await?;
5017
5018 let mut param = Params::new();
5019 if cmd != SystemMessage::Unknown {
5020 param.set_cmd(cmd);
5021 }
5022 if let Some(contact_id) = added_removed_id {
5023 param.set(Param::ContactAddedRemoved, contact_id.to_u32().to_string());
5024 }
5025
5026 let timestamp_sort = if let Some(ts) = timestamp_sort {
5027 ts
5028 } else {
5029 let sort_to_bottom = true;
5030 chat_id
5031 .calc_sort_timestamp(context, smeared_time(context), sort_to_bottom)
5032 .await?
5033 };
5034
5035 let row_id =
5036 context.sql.insert(
5037 "INSERT INTO msgs (chat_id,from_id,to_id,timestamp,timestamp_sent,timestamp_rcvd,type,state,txt,txt_normalized,rfc724_mid,ephemeral_timer,param,mime_in_reply_to)
5038 VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?);",
5039 (
5040 chat_id,
5041 from_id.unwrap_or(ContactId::INFO),
5042 ContactId::INFO,
5043 timestamp_sort,
5044 timestamp_sent_rcvd,
5045 timestamp_sent_rcvd,
5046 Viewtype::Text,
5047 MessageState::InNoticed,
5048 text,
5049 normalize_text(text),
5050 rfc724_mid,
5051 ephemeral_timer,
5052 param.to_string(),
5053 parent.map(|msg|msg.rfc724_mid.clone()).unwrap_or_default()
5054 )
5055 ).await?;
5056 context.new_msgs_notify.notify_one();
5057
5058 let msg_id = MsgId::new(row_id.try_into()?);
5059 context.emit_msgs_changed(chat_id, msg_id);
5060
5061 Ok(msg_id)
5062}
5063
5064pub(crate) async fn add_info_msg(context: &Context, chat_id: ChatId, text: &str) -> Result<MsgId> {
5066 add_info_msg_with_cmd(
5067 context,
5068 chat_id,
5069 text,
5070 SystemMessage::Unknown,
5071 None,
5072 time(),
5073 None,
5074 None,
5075 None,
5076 )
5077 .await
5078}
5079
5080pub(crate) async fn update_msg_text_and_timestamp(
5081 context: &Context,
5082 chat_id: ChatId,
5083 msg_id: MsgId,
5084 text: &str,
5085 timestamp: i64,
5086) -> Result<()> {
5087 context
5088 .sql
5089 .execute(
5090 "UPDATE msgs SET txt=?, txt_normalized=?, timestamp=? WHERE id=?;",
5091 (text, normalize_text(text), timestamp, msg_id),
5092 )
5093 .await?;
5094 context.emit_msgs_changed(chat_id, msg_id);
5095 Ok(())
5096}
5097
5098async fn set_contacts_by_addrs(context: &Context, id: ChatId, addrs: &[String]) -> Result<()> {
5100 let chat = Chat::load_from_db(context, id).await?;
5101 ensure!(
5102 !chat.is_encrypted(context).await?,
5103 "Cannot add address-contacts to encrypted chat {id}"
5104 );
5105 ensure!(
5106 chat.typ == Chattype::OutBroadcast,
5107 "{id} is not a broadcast list",
5108 );
5109 let mut contacts = BTreeSet::new();
5110 for addr in addrs {
5111 let contact_addr = ContactAddress::new(addr)?;
5112 let contact = Contact::add_or_lookup(context, "", &contact_addr, Origin::Hidden)
5113 .await?
5114 .0;
5115 contacts.insert(contact);
5116 }
5117 let contacts_old = BTreeSet::<ContactId>::from_iter(get_chat_contacts(context, id).await?);
5118 if contacts == contacts_old {
5119 return Ok(());
5120 }
5121 context
5122 .sql
5123 .transaction(move |transaction| {
5124 transaction.execute("DELETE FROM chats_contacts WHERE chat_id=?", (id,))?;
5125
5126 let mut statement = transaction
5129 .prepare("INSERT INTO chats_contacts (chat_id, contact_id) VALUES (?, ?)")?;
5130 for contact_id in &contacts {
5131 statement.execute((id, contact_id))?;
5132 }
5133 Ok(())
5134 })
5135 .await?;
5136 context.emit_event(EventType::ChatModified(id));
5137 Ok(())
5138}
5139
5140async fn set_contacts_by_fingerprints(
5144 context: &Context,
5145 id: ChatId,
5146 fingerprint_addrs: &[(String, String)],
5147) -> Result<()> {
5148 let chat = Chat::load_from_db(context, id).await?;
5149 ensure!(
5150 chat.is_encrypted(context).await?,
5151 "Cannot add key-contacts to unencrypted chat {id}"
5152 );
5153 ensure!(
5154 matches!(chat.typ, Chattype::Group | Chattype::OutBroadcast),
5155 "{id} is not a group or broadcast",
5156 );
5157 let mut contacts = BTreeSet::new();
5158 for (fingerprint, addr) in fingerprint_addrs {
5159 let contact = Contact::add_or_lookup_ex(context, "", addr, fingerprint, Origin::Hidden)
5160 .await?
5161 .0;
5162 contacts.insert(contact);
5163 }
5164 let contacts_old = BTreeSet::<ContactId>::from_iter(get_chat_contacts(context, id).await?);
5165 if contacts == contacts_old {
5166 return Ok(());
5167 }
5168 let broadcast_contacts_added = context
5169 .sql
5170 .transaction(move |transaction| {
5171 if chat.typ != Chattype::OutBroadcast {
5177 transaction.execute("DELETE FROM chats_contacts WHERE chat_id=?", (id,))?;
5178 }
5179
5180 let mut statement = transaction.prepare(
5183 "INSERT OR IGNORE INTO chats_contacts (chat_id, contact_id) VALUES (?, ?)",
5184 )?;
5185 let mut broadcast_contacts_added = Vec::new();
5186 for contact_id in &contacts {
5187 if statement.execute((id, contact_id))? > 0 && chat.typ == Chattype::OutBroadcast {
5188 broadcast_contacts_added.push(*contact_id);
5189 }
5190 }
5191 Ok(broadcast_contacts_added)
5192 })
5193 .await?;
5194 let timestamp = smeared_time(context);
5195 for added_id in broadcast_contacts_added {
5196 let msg = stock_str::msg_add_member_local(context, added_id, ContactId::UNDEFINED).await;
5197 add_info_msg_with_cmd(
5198 context,
5199 id,
5200 &msg,
5201 SystemMessage::MemberAddedToGroup,
5202 Some(timestamp),
5203 timestamp,
5204 None,
5205 Some(ContactId::SELF),
5206 Some(added_id),
5207 )
5208 .await?;
5209 }
5210 context.emit_event(EventType::ChatModified(id));
5211 Ok(())
5212}
5213
5214#[derive(Debug, Serialize, Deserialize, PartialEq)]
5216pub(crate) enum SyncId {
5217 ContactAddr(String),
5219
5220 ContactFingerprint(String),
5222
5223 Grpid(String),
5224 Msgids(Vec<String>),
5226
5227 Device,
5229}
5230
5231#[derive(Debug, Serialize, Deserialize, PartialEq)]
5233pub(crate) enum SyncAction {
5234 Block,
5235 Unblock,
5236 Accept,
5237 SetVisibility(ChatVisibility),
5238 SetMuted(MuteDuration),
5239 CreateOutBroadcast {
5241 chat_name: String,
5242 secret: String,
5243 },
5244 CreateGroupEncrypted(String),
5246 Rename(String),
5247 SetContacts(Vec<String>),
5249 SetPgpContacts(Vec<(String, String)>),
5253 SetDescription(String),
5254 Delete,
5255}
5256
5257impl Context {
5258 pub(crate) async fn sync_alter_chat(&self, id: &SyncId, action: &SyncAction) -> Result<()> {
5260 let chat_id = match id {
5261 SyncId::ContactAddr(addr) => {
5262 if let SyncAction::Rename(to) = action {
5263 Contact::create_ex(self, Nosync, to, addr).await?;
5264 return Ok(());
5265 }
5266 let addr = ContactAddress::new(addr).context("Invalid address")?;
5267 let (contact_id, _) =
5268 Contact::add_or_lookup(self, "", &addr, Origin::Hidden).await?;
5269 match action {
5270 SyncAction::Block => {
5271 return contact::set_blocked(self, Nosync, contact_id, true).await;
5272 }
5273 SyncAction::Unblock => {
5274 return contact::set_blocked(self, Nosync, contact_id, false).await;
5275 }
5276 _ => (),
5277 }
5278 ChatIdBlocked::get_for_contact(self, contact_id, Blocked::Request)
5281 .await?
5282 .id
5283 }
5284 SyncId::ContactFingerprint(fingerprint) => {
5285 let name = "";
5286 let addr = "";
5287 let (contact_id, _) =
5288 Contact::add_or_lookup_ex(self, name, addr, fingerprint, Origin::Hidden)
5289 .await?;
5290 match action {
5291 SyncAction::Rename(to) => {
5292 contact_id.set_name_ex(self, Nosync, to).await?;
5293 self.emit_event(EventType::ContactsChanged(Some(contact_id)));
5294 return Ok(());
5295 }
5296 SyncAction::Block => {
5297 return contact::set_blocked(self, Nosync, contact_id, true).await;
5298 }
5299 SyncAction::Unblock => {
5300 return contact::set_blocked(self, Nosync, contact_id, false).await;
5301 }
5302 _ => (),
5303 }
5304 ChatIdBlocked::get_for_contact(self, contact_id, Blocked::Request)
5305 .await?
5306 .id
5307 }
5308 SyncId::Grpid(grpid) => {
5309 match action {
5310 SyncAction::CreateOutBroadcast { chat_name, secret } => {
5311 create_out_broadcast_ex(
5312 self,
5313 Nosync,
5314 grpid.to_string(),
5315 chat_name.clone(),
5316 secret.to_string(),
5317 )
5318 .await?;
5319 return Ok(());
5320 }
5321 SyncAction::CreateGroupEncrypted(name) => {
5322 create_group_ex(self, Nosync, grpid.clone(), name).await?;
5323 return Ok(());
5324 }
5325 _ => {}
5326 }
5327 get_chat_id_by_grpid(self, grpid)
5328 .await?
5329 .with_context(|| format!("No chat for grpid '{grpid}'"))?
5330 .0
5331 }
5332 SyncId::Msgids(msgids) => {
5333 let msg = message::get_by_rfc724_mids(self, msgids)
5334 .await?
5335 .with_context(|| format!("No message found for Message-IDs {msgids:?}"))?;
5336 ChatId::lookup_by_message(&msg)
5337 .with_context(|| format!("No chat found for Message-IDs {msgids:?}"))?
5338 }
5339 SyncId::Device => ChatId::get_for_contact(self, ContactId::DEVICE).await?,
5340 };
5341 match action {
5342 SyncAction::Block => chat_id.block_ex(self, Nosync).await,
5343 SyncAction::Unblock => chat_id.unblock_ex(self, Nosync).await,
5344 SyncAction::Accept => chat_id.accept_ex(self, Nosync).await,
5345 SyncAction::SetVisibility(v) => chat_id.set_visibility_ex(self, Nosync, *v).await,
5346 SyncAction::SetMuted(duration) => set_muted_ex(self, Nosync, chat_id, *duration).await,
5347 SyncAction::CreateOutBroadcast { .. } | SyncAction::CreateGroupEncrypted(..) => {
5348 Err(anyhow!("sync_alter_chat({id:?}, {action:?}): Bad request."))
5350 }
5351 SyncAction::Rename(to) => rename_ex(self, Nosync, chat_id, to).await,
5352 SyncAction::SetDescription(to) => {
5353 set_chat_description_ex(self, Nosync, chat_id, to).await
5354 }
5355 SyncAction::SetContacts(addrs) => set_contacts_by_addrs(self, chat_id, addrs).await,
5356 SyncAction::SetPgpContacts(fingerprint_addrs) => {
5357 set_contacts_by_fingerprints(self, chat_id, fingerprint_addrs).await
5358 }
5359 SyncAction::Delete => chat_id.delete_ex(self, Nosync).await,
5360 }
5361 }
5362
5363 pub(crate) fn on_archived_chats_maybe_noticed(&self) {
5368 self.emit_msgs_changed_without_msg_id(DC_CHAT_ID_ARCHIVED_LINK);
5369 }
5370}
5371
5372#[cfg(test)]
5373mod chat_tests;