@prototypevip/baileys 0.0.2 → 0.0.4

package/lib/Store/prototype-store.js

@@ -285,7 +285,12 @@ export const makePrototypeStore = (config = {}) => {
               }
             }
           } catch (err) {
-            console.error("PrototypeStore Transaction Step Error:", err);
+            const code = err?.code || err?.message || "";
+            if (String(code).includes("READONLY") || String(code).includes("SQLITE_READONLY")) {
+              console.error("PrototypeStore Transaction Step Error (readonly DB - check file path/permissions/volume):", err.message || err);
+            } else {
+              console.error("PrototypeStore Transaction Step Error:", err);
+            }
           }
         }
       })();
@@ -380,6 +385,49 @@ export const makePrototypeStore = (config = {}) => {
 
   const presencesCache = {};
 
+
+  const safeAssignChat = (existing, incoming) => {
+    if (!existing || !incoming) return existing;
+    const savedMessages = existing.messages;
+    Object.assign(existing, incoming, { isChats: true });
+    // Never let chat events wipe the ordered messages store
+    if (
+      !existing.messages ||
+      typeof existing.messages.upsert !== "function"
+    ) {
+      existing.messages = savedMessages || makeMessagesStore();
+    }
+    if (!existing.id) existing.id = incoming.id || existing.id;
+    return existing;
+  };
+
+  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,38 +437,17 @@ 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");
             },
           };
+        } else if (
+          !target[jid].messages ||
+          typeof target[jid].messages.upsert !== "function"
+        ) {
+          // Recover if messages was overwritten by Object.assign / chat updates
+          target[jid].messages = makeMessagesStore();
         }
         return target[jid];
       },
@@ -606,7 +633,7 @@ export const makePrototypeStore = (config = {}) => {
           if (!id || id === "status@broadcast") continue;
 
           const existingChat = store.chats[id];
-          Object.assign(existingChat, chat, { isChats: true });
+          safeAssignChat(existingChat, chat);
 
           if (id.endsWith("@g.us")) {
             const current = groupsCache[id] || {};
@@ -731,16 +758,26 @@ export const makePrototypeStore = (config = {}) => {
               store.signalRepository?.lidMapping,
             );
             const existingChat = store.chats[id];
-            Object.assign(existingChat, chat, { isChats: true });
+            safeAssignChat(existingChat, chat);
           }
         },
       );
 
       listen("messages.upsert", async ({ messages }) => {
         for (let msg of messages) {
-          const chatId = msg.key.remoteJid;
+          const rawChatId = msg?.key?.remoteJid;
+          if (!rawChatId) continue;
+          const chatId = await resolveJIDAsync(
+            rawChatId,
+            store.signalRepository?.lidMapping,
+          );
           if (!chatId) continue;
-          store.chats[chatId].messages.upsert(cloneM(msg));
+          const chat = store.chats[chatId];
+          if (!chat) continue;
+          if (!chat.messages || typeof chat.messages.upsert !== "function") {
+            chat.messages = makeMessagesStore();
+          }
+          chat.messages.upsert(cloneM(msg));
 
           if (db) {
             const msgKey = getMessageKey(
@@ -788,7 +825,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 +991,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";