Fix scheduled messages

src/components/chat/bubbleGroups.ts

@@ -94,6 +94,10 @@ class BubbleGroup {
     return this.items[this.items.length - 1];
   }
 
+  get lastTimestamp() {
+    return this.lastItem.timestamp;
+  }
+
   get lastMid() {
     return this.lastItem.mid;
   }
@@ -143,33 +147,11 @@ class BubbleGroup {
 
   insertItem(item: GroupItem) {
     const {items} = this;
-    const {timestamp, mid} = item;
+    insertInDescendSortedArray(items, item, this.groups.sortGroupItemsKey);
-    if(this.chat.type === 'scheduled') {
-      let foundMidOnSameTimestamp = 0;
-      let i = 0, length = items.length;
-      for(; i < length; ++i) {
-        const {timestamp: _timestamp, mid: _mid} = items[i];
-  
-        if(timestamp < _timestamp) {
-          break;
-        } else if(timestamp === _timestamp) {
-          foundMidOnSameTimestamp = _mid;
-        } 
-        
-        if(foundMidOnSameTimestamp && mid < foundMidOnSameTimestamp) {
-          break;
-        }
-      }
-
-      items.splice(i, 0, item);
-    } else {
-      // insertInDescendSortedArray(items, item, 'mid');
-      insertInDescendSortedArray(items, item, 'groupMid');
-    }
 
     item.group = this;
     if(items.length === 1) {
-      insertInDescendSortedArray(this.groups.groups, this, 'lastMid');
+      this.groups.insertGroup(this);
     }
   }
 
@@ -231,7 +213,7 @@ class BubbleGroup {
       return;
     }
 
-    const dateContainer = this.chat.bubbles.getDateContainerByTimestamp(this.firstTimestamp);
+    const dateContainer = this.chat.bubbles.getDateContainerByTimestamp(this.dateTimestamp / 1000);
     // const idx = this.groups.indexOf(group);
     const dateGroups = this.groups.groups.filter((_group) => _group.dateTimestamp === this.dateTimestamp);
     const dateGroupsLength = dateGroups.length;
@@ -276,9 +258,14 @@ export default class BubbleGroups {
   private itemsMap: Map<HTMLElement, GroupItem> = new Map();
   public groups: Array<BubbleGroup> = []; // descend sorted
   private newGroupDiff = 121; // * 121 in scheduled messages
+  private sortItemsKey: Extract<keyof GroupItem, 'timestamp' | 'mid'>;
+  private sortGroupsKey: Extract<keyof BubbleGroup, 'lastMid' | 'lastTimestamp'>;
+  public sortGroupItemsKey: Extract<keyof GroupItem, 'groupMid' | 'timestamp'>;
 
   constructor(private chat: Chat) {
-
+    this.sortItemsKey = chat.type === 'scheduled' ? 'timestamp' : 'mid';
+    this.sortGroupsKey = chat.type === 'scheduled' ? 'lastTimestamp' : 'lastMid';
+    this.sortGroupItemsKey = /* chat.type === 'scheduled' ? 'timestamp' :  */'groupMid';
   }
 
   removeItem(item: GroupItem) {
@@ -369,7 +356,7 @@ export default class BubbleGroups {
     // item.group.insertItem(item);
     
     indexOfAndSplice(this.itemsArr, item);
-    insertInDescendSortedArray(this.itemsArr, item, 'mid');
+    this.insertItemToArray(item, this.itemsArr);
   }
 
   changeItemBubble(item: GroupItem, bubble: HTMLElement) {
@@ -405,7 +392,7 @@ export default class BubbleGroups {
 
   findGroupSiblingByItem(item: GroupItem, items: GroupItem[]) {
     items = items.slice();
-    const idx = insertInDescendSortedArray(items, item, 'mid');
+    const idx = this.insertItemToArray(item, items);
     // return this.findGroupSiblingInSiblings(item, this.getSiblingsAtIndex(idx, items));
     return this.findGroupSiblingInItems(item, items, idx);
   }
@@ -436,8 +423,16 @@ export default class BubbleGroups {
     this.addItemToCache(item);
   }
 
+  insertItemToArray(item: GroupItem, array: GroupItem[]) {
+    return insertInDescendSortedArray(array, item, this.sortItemsKey);
+  }
+
+  insertGroup(group: BubbleGroup) {
+    return insertInDescendSortedArray(this.groups, group, this.sortGroupsKey);
+  }
+
   addItemToCache(item: GroupItem) {
-    insertInDescendSortedArray(this.itemsArr, item, 'mid');
+    this.insertItemToArray(item, this.itemsArr);
     this.itemsMap.set(item.bubble, item);
   }
 
@@ -463,9 +458,10 @@ export default class BubbleGroups {
     const {dateTimestamp} = this.chat.bubbles.getDateForDateContainer(timestamp);
     const item: GroupItem = {
       mid, 
-      groupMid: mid, 
+      groupMid: this.chat.type === 'scheduled' ? +`${(timestamp * 1000 - dateTimestamp) / 1000}.${mid}` : mid, 
       fromId: this.getMessageFromId(message), 
       bubble, 
+      // timestamp: this.chat.type === 'scheduled' ? +`${(timestamp * 1000 - dateTimestamp) / 1000}.${mid}` : timestamp, 
       timestamp,
       dateTimestamp, 
       mounted: false, 
@@ -505,7 +501,8 @@ export default class BubbleGroups {
   }
 
   prepareForGrouping(bubble: HTMLElement, message: MyMessage) {
-    if(this.getItemByBubble(bubble)) {
+    const foundItem = this.getItemByBubble(bubble);
+    if(foundItem) { // should happen only on edit
       debugger;
       return;
     }

src/components/chat/bubbles.ts

@@ -284,7 +284,7 @@ export default class ChatBubbles {
 
     // will call when sent for update pos
     this.listenerSetter.add(rootScope)('history_update', async({storageKey, mid, message}) => {
-      if(this.chat.messagesStorageKey !== storageKey) {
+      if(this.chat.messagesStorageKey !== storageKey || this.chat.type === 'scheduled') {
         return;
       }
 
@@ -741,7 +741,8 @@ export default class ChatBubbles {
       });
     }
 
-    if(!IS_MOBILE && this.chat.type !== 'pinned') {
+    if(this.chat.type !== 'pinned' && this.chat.type !== 'scheduled') {
+      if(!IS_MOBILE) {
         this.listenerSetter.add(container)('dblclick', async(e) => {
           if(this.chat.selection.isSelecting || 
             !(await this.chat.canSend())) {
@@ -762,9 +763,7 @@ export default class ChatBubbles {
             this.chat.input.initMessageReply(mid);
           }
         });
-    }
+      } else if(IS_TOUCH_SUPPORTED) {
-
-    if(IS_TOUCH_SUPPORTED) {
         const className = 'is-gesturing-reply';
         const MAX = 64;
         const replyAfter = MAX * .75;
@@ -832,6 +831,7 @@ export default class ChatBubbles {
         });
       }
     }
+  }
 
   public constructPeerHelpers() {
     // will call when message is sent (only 1)
@@ -2121,14 +2121,15 @@ export default class ChatBubbles {
     this.deleteEmptyDateGroups();
 
     if(!ignoreOnScroll) {
-      this.onScroll();
+      this.scrollable.onScroll();
+      // this.onScroll();
     }
   }
 
   private setTopPadding(middleware = this.getMiddleware()) {
     let isPaddingNeeded = false;
     let setPaddingTo: HTMLElement;
-    if(!this.isTopPaddingSet) {
+    if(!this.isTopPaddingSet && this.chat.type !== 'scheduled') {
       const {clientHeight, scrollHeight} = this.scrollable.container;
       isPaddingNeeded = clientHeight === scrollHeight;
       /* const firstEl = this.chatInner.firstElementChild as HTMLElement;
@@ -3236,6 +3237,20 @@ export default class ChatBubbles {
   }/*  & {
     unmountIfFound?: boolean
   } */>) {
+    let modifiedGroups: typeof groups;
+
+    if(this.chat.type === 'scheduled') {
+      modifiedGroups = new Set();
+      items.forEach(({bubble, message}) => {
+        const item = this.bubbleGroups.getItemByBubble(bubble);
+        const group = item?.group;
+        if(group && item.message.date !== message.date) {
+          this.bubbleGroups.removeItem(item);
+          modifiedGroups.add(group);
+        }
+      });
+    }
+    
     items.forEach(({bubble, message}) => {
       this.bubbleGroups.prepareForGrouping(bubble, message);
     });
@@ -3245,11 +3260,17 @@ export default class ChatBubbles {
     const avatarPromises = Array.from(groups).map((group) => {
       if(group.avatar) return;
       const firstItem = group.firstItem;
-      if(this.chat.isAvatarNeeded(firstItem.message)) {
+      if(firstItem && this.chat.isAvatarNeeded(firstItem.message)) {
         return group.createAvatar(firstItem.message);
       }
     }).filter(Boolean);
 
+    if(modifiedGroups) {
+      for(const group of modifiedGroups) {
+        groups.add(group);
+      }
+    }
+
     return {
       groups: [...groups], 
       avatarPromises

src/components/chat/contextMenu.ts

@@ -39,6 +39,7 @@ import { attachContextMenuListener } from "../../helpers/dom/attachContextMenuLi
 import filterAsync from "../../helpers/array/filterAsync";
 import appDownloadManager from "../../lib/appManagers/appDownloadManager";
 import { SERVICE_PEER_ID } from "../../lib/mtproto/mtproto_config";
+import { MessagesStorageKey } from "../../lib/appManagers/appMessagesManager";
 
 export default class ChatContextMenu {
   private buttons: (ButtonMenuItemOptions & {verify: () => boolean | Promise<boolean>, notDirect?: () => boolean, withSelection?: true, isSponsored?: true})[];
@@ -340,8 +341,9 @@ export default class ChatContextMenu {
         }
 
         for(const [peerId, mids] of this.chat.selection.selectedMids) {
+          const storageKey: MessagesStorageKey = `${peerId}_${this.chat.type === 'scheduled' ? 'scheduled' : 'history'}`;
           for(const mid of mids) {
-            const message = (await this.managers.appMessagesManager.getMessageByPeer(peerId, mid)) as Message.message;
+            const message = (await this.managers.appMessagesManager.getMessageFromStorage(storageKey, mid)) as Message.message;
             if(!!message.message) {
               return true;
             }

src/components/chat/input.ts

@@ -2384,7 +2384,11 @@ export default class ChatInput {
         clearDraft: true
       });
 
+      if(this.chat.type === 'scheduled') {
+        this.onMessageSent(true);
+      } else {
         this.onMessageSent(false, false);
+      }
       // this.onMessageSent();
     }