@prototypevip/baileys 0.0.2 → 0.0.3

package/lib/Store/prototype-store.js

@@ -380,6 +380,33 @@ export const makePrototypeStore = (config = {}) => {
 
   const presencesCache = {};
 
+  const makeMessagesStore = () =>
+    new Proxy(
+      makeOrderedDictionary((m) => {
+        const participant = m.key.participant
+          ? resolveJID(m.key.participant, store.signalRepository?.lidMapping)
+          : m.key.fromMe
+            ? "me"
+            : "";
+        return participant ? `${participant}:${m.key.id}` : m.key.id;
+      }),
+      {
+        get(target, prop) {
+          if (prop in target) return target[prop];
+          return target.get(prop);
+        },
+        set(target, prop, value) {
+          target.upsert(value);
+          return true;
+        },
+        deleteProperty(target, prop) {
+          const item = target.get(prop);
+          if (item) return target.remove(item);
+          return false;
+        },
+      },
+    );
+
   const store = {
     signalRepository: config.signalRepository,
     chats: new Proxy(chatsCache, {
@@ -389,34 +416,7 @@ export const makePrototypeStore = (config = {}) => {
         if (!target[jid]) {
           target[jid] = {
             id: jid,
-            messages: new Proxy(
-              makeOrderedDictionary((m) => {
-                const participant = m.key.participant
-                  ? resolveJID(
-                      m.key.participant,
-                      store.signalRepository?.lidMapping,
-                    )
-                  : m.key.fromMe
-                    ? "me"
-                    : "";
-                return participant ? `${participant}:${m.key.id}` : m.key.id;
-              }),
-              {
-                get(target, prop) {
-                  if (prop in target) return target[prop];
-                  return target.get(prop);
-                },
-                set(target, prop, value) {
-                  target.upsert(value);
-                  return true;
-                },
-                deleteProperty(target, prop) {
-                  const item = target.get(prop);
-                  if (item) return target.remove(item);
-                  return false;
-                },
-              },
-            ),
+            messages: makeMessagesStore(),
             get isGroup() {
               return jid.endsWith("@g.us");
             },
@@ -788,7 +788,7 @@ export const makePrototypeStore = (config = {}) => {
             store.signalRepository?.lidMapping,
           );
           if (chatsCache[jid]) {
-            chatsCache[jid].messages = {};
+            chatsCache[jid].messages = makeMessagesStore();
             if (db)
               pendingUpdates.set(`del:m:${jid}`, {
                 type: "delete",
@@ -954,5 +954,22 @@ export const makePrototypeStore = (config = {}) => {
     },
   };
 
+  store.close = () => {
+    if (flushInterval) clearInterval(flushInterval);
+    if (cleanInterval) clearInterval(cleanInterval);
+    try {
+      _flush();
+    } catch {
+      // ignore flush errors on close
+    }
+    if (db) {
+      try {
+        db.close();
+      } catch {
+        // ignore close errors
+      }
+    }
+  };
+
   return store;
 };

package/lib/Utils/index.js

@@ -26,4 +26,5 @@ export * from "./cstoken-utils.js";
 export * from "./use-prototype-auth-state.js";
 export * from "./sticker-utils.js";
 export * from "./rich-message-utils.js";
+export * from "./message-builder.js";
 export * from "./constants.js";