cache jids

Author: clairton

src/services/client_baileys.ts

@@ -215,11 +215,11 @@ class ClientBaileys implements Client {
   private async toJid(phoneNumber: string) {
     const bindJid: string = phoneNumberToJid(phoneNumber)
     if (isIndividualJid(bindJid)) {
-      const results = await this.sock?.onWhatsApp(bindJid)
-      const result = results && results[0]
-      if (result && result.exists) {
-        console.debug(`${bindJid} exists on WhatsApp, as jid: ${result.jid}`)
-        return result.jid
+      // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+      const jid = await this.store?.dataStore?.getJid(bindJid, this.sock!)
+      if (jid) {
+        console.debug(`${bindJid} exists on WhatsApp, as jid: ${jid}`)
+        return jid
       } else {
         console.warn(`${bindJid} not exists on WhatsApp`)
         return ''

src/services/data_store.ts

@@ -1,4 +1,4 @@
-import { makeInMemoryStore, WAMessage, WAMessageKey } from '@adiwajshing/baileys'
+import { makeInMemoryStore, WAMessage, WAMessageKey, WASocket } from '@adiwajshing/baileys'
 
 export const dataStores: Map<string, DataStore> = new Map()
 
@@ -9,6 +9,7 @@ export interface getDataStore {
 export type DataStore = ReturnType<typeof makeInMemoryStore> & {
   loadKey: (id: string) => WAMessageKey | undefined
   setKey: (id: string, key: WAMessageKey) => void
+  getJid: (phone: string, sock: Partial<WASocket>) => Promise<string>
   setMessage: (id: string, message: WAMessage) => void
   saveMedia: (waMessage: WAMessage) => Promise<void>
   cleanSession: () => Promise<void>

src/services/data_store_file.ts

@@ -9,6 +9,7 @@ import {
   WAMessageUpdate,
   MessageUpsertType,
   WAMessageKey,
+  WASocket,
 } from '@adiwajshing/baileys'
 import makeOrderedDictionary from '@adiwajshing/baileys/lib/Store/make-ordered-dictionary'
 import { waMessageID } from '@adiwajshing/baileys/lib/Store/make-in-memory-store'
@@ -97,19 +98,22 @@ const saveMedia = async (phone: string, waMessage: WAMessage) => {
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 const dataStoreFile = (phone: string, config: any): DataStore => {
   const keys: Map<string, proto.IMessageKey> = new Map()
+  const jids: Map<string, string> = new Map()
   const store = makeInMemoryStore(config)
   const dataStore = store as DataStore
   const { bind, toJSON, fromJSON } = store
   store.toJSON = () => {
     return {
       ...toJSON(),
       keys: keys.values(),
+      jids,
     }
   }
   store.fromJSON = (json) => {
     fromJSON(json)
     const jsonData = json as {
       keys: proto.IMessageKey[]
+      jids: Map<string, string>
       chats: Chat[]
       contacts: { [id: string]: Contact }
       messages: { [id: string]: WAMessage[] }
@@ -147,6 +151,20 @@ const dataStoreFile = (phone: string, config: any): DataStore => {
   dataStore.setKey = (id: string, key: WAMessageKey) => {
     return keys.set(id, key)
   }
+  dataStore.getJid = async (phoneOrJid: string, sock: Partial<WASocket>) => {
+    if (!jids.has(phoneOrJid)) {
+      // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+      const results = await sock.onWhatsApp!(phoneOrJid)
+      const result = results && results[0]
+      if (result && result.exists) {
+        console.debug(`${phoneOrJid} exists on WhatsApp, as jid: ${result.jid}`)
+        jids.set(phoneOrJid, result.jid)
+      } else {
+        console.warn(`${phoneOrJid} not exists on WhatsApp`)
+      }
+    }
+    return jids.get(phoneOrJid) || ''
+  }
   dataStore.setMessage = (id: string, message: WAMessage) => {
     if (!store.messages[id]) {
       store.messages[id] = makeOrderedDictionary(waMessageID)