feat: Add emoji and alot of other fixes

toju-app/src/app/domains/attachment/application/services/attachment-manager.service.ts

@@ -44,7 +44,11 @@ export class AttachmentManagerService {
     effect(() => {
       if (this.database.isReady() && !this.isDatabaseInitialised) {
         this.isDatabaseInitialised = true;
-        void this.persistence.initFromDatabase();
+        void this.persistence.initFromDatabase().then(() => {
+          if (this.watchedRoomId) {
+            void this.restoreLocalAttachmentsForRoom(this.watchedRoomId);
+          }
+        });
       }
     });
 
@@ -56,12 +60,14 @@ export class AttachmentManagerService {
       this.watchedRoomId = this.extractWatchedRoomId(event.urlAfterRedirects || event.url);
 
       if (this.watchedRoomId) {
+        void this.restoreLocalAttachmentsForRoom(this.watchedRoomId);
         void this.requestAutoDownloadsForRoom(this.watchedRoomId);
       }
     });
 
     this.webrtc.onPeerConnected.subscribe(() => {
       if (this.watchedRoomId) {
+        void this.restoreLocalAttachmentsForRoom(this.watchedRoomId);
         void this.requestAutoDownloadsForRoom(this.watchedRoomId);
       }
     });
@@ -110,11 +116,11 @@ export class AttachmentManagerService {
     return this.transfer.getAttachmentMetasForMessages(messageIds);
   }
 
-  registerSyncedAttachments(
+  async registerSyncedAttachments(
     attachmentMap: Record<string, AttachmentMeta[]>,
     messageRoomIds?: Record<string, string>
-  ): void {
-    this.transfer.registerSyncedAttachments(attachmentMap, messageRoomIds);
+  ): Promise<void> {
+    await this.transfer.registerSyncedAttachments(attachmentMap, messageRoomIds);
 
     for (const [messageId, attachments] of Object.entries(attachmentMap)) {
       for (const attachment of attachments) {
@@ -123,20 +129,20 @@ export class AttachmentManagerService {
     }
   }
 
-  requestFromAnyPeer(messageId: string, attachment: Attachment): void {
-    this.transfer.requestFromAnyPeer(messageId, attachment);
+  requestFromAnyPeer(messageId: string, attachment: Attachment): Promise<void> {
+    return this.transfer.requestFromAnyPeer(messageId, attachment);
   }
 
   handleFileNotFound(payload: FileNotFoundPayload): void {
     this.transfer.handleFileNotFound(payload);
   }
 
-  requestImageFromAnyPeer(messageId: string, attachment: Attachment): void {
-    this.transfer.requestImageFromAnyPeer(messageId, attachment);
+  requestImageFromAnyPeer(messageId: string, attachment: Attachment): Promise<void> {
+    return this.transfer.requestImageFromAnyPeer(messageId, attachment);
   }
 
-  requestFile(messageId: string, attachment: Attachment): void {
-    this.transfer.requestFile(messageId, attachment);
+  requestFile(messageId: string, attachment: Attachment): Promise<void> {
+    return this.transfer.requestFile(messageId, attachment);
   }
 
   async publishAttachments(
@@ -180,11 +186,66 @@ export class AttachmentManagerService {
     await this.transfer.fulfillRequestWithFile(messageId, fileId, targetPeerId, file);
   }
 
+  private async restoreLocalAttachmentsForRoom(roomId: string): Promise<void> {
+    if (!this.isRoomWatched(roomId)) {
+      return;
+    }
+
+    await this.persistence.whenReady();
+
+    const messageIds = await this.collectMessageIdsForRoom(roomId);
+    let hasChanges = false;
+
+    for (const messageId of messageIds) {
+      for (const attachment of this.runtimeStore.getAttachmentsForMessage(messageId)) {
+        if (await this.persistence.tryRestoreAttachmentFromLocal(attachment)) {
+          hasChanges = true;
+        }
+      }
+    }
+
+    if (hasChanges) {
+      this.runtimeStore.touch();
+    }
+  }
+
+  private async collectMessageIdsForRoom(roomId: string): Promise<string[]> {
+    if (isDirectMessageAttachmentRoomId(roomId)) {
+      const messageIds: string[] = [];
+
+      for (const [messageId] of this.runtimeStore.getAttachmentEntries()) {
+        const attachmentRoomId = await this.persistence.resolveMessageRoomId(messageId);
+
+        if (attachmentRoomId === roomId) {
+          messageIds.push(messageId);
+        }
+      }
+
+      return messageIds;
+    }
+
+    if (!this.database.isReady()) {
+      return Array.from(this.runtimeStore.getAttachmentEntries())
+        .filter(([messageId]) => this.runtimeStore.getMessageRoomId(messageId) === roomId)
+        .map(([messageId]) => messageId);
+    }
+
+    const messages = await this.database.getMessages(roomId, 500, 0);
+
+    for (const message of messages) {
+      this.runtimeStore.rememberMessageRoom(message.id, message.roomId);
+    }
+
+    return messages.map((message) => message.id);
+  }
+
   private async runAutoDownloadsForRoom(roomId: string): Promise<void> {
     if (!this.isRoomWatched(roomId)) {
       return;
     }
 
+    await this.restoreLocalAttachmentsForRoom(roomId);
+
     if (isDirectMessageAttachmentRoomId(roomId)) {
       await this.requestAutoDownloadsForRuntimeRoom(roomId);
       return;
@@ -242,7 +303,7 @@ export class AttachmentManagerService {
       if (this.transfer.hasPendingRequest(messageId, attachment.id))
         continue;
 
-      this.transfer.requestFromAnyPeer(messageId, attachment);
+      void this.transfer.requestFromAnyPeer(messageId, attachment);
     }
   }