@hansaka02/baileys 7.3.2 → 7.3.6

package/lib/WABinary/jid-utils.js

@@ -14,39 +14,76 @@ const STORIES_JID = 'status@broadcast'
 
 const META_AI_JID = '13135550002@c.us'
 
+const WAJIDDomains = {
+  WHATSAPP: 0,
+  LID: 1,
+  HOSTED: 128,
+  HOSTED_LID: 129
+}
+
+const getServerFromDomainType = (initialServer, domainType) => {
+    switch (domainType) {
+        case WAJIDDomains.LID:
+            return 'lid'
+        case WAJIDDomains.HOSTED:
+            return 'hosted'
+        case WAJIDDomains.HOSTED_LID:
+            return 'hosted.lid'
+        case WAJIDDomains.WHATSAPP:
+        default:
+            return initialServer
+    }
+}
+
 const jidEncode = (user, server, device, agent) => {
     return `${user || ''}${!!agent ? `_${agent}` : ''}${!!device ? `:${device}` : ''}@${server}`
 }
 
 const jidDecode = (jid) => {
+    // todo: investigate how to implement hosted ids in this case
     const sepIdx = typeof jid === 'string' ? jid.indexOf('@') : -1
+    
     if (sepIdx < 0) {
         return undefined
     }
+    
     const server = jid.slice(sepIdx + 1)
     const userCombined = jid.slice(0, sepIdx)
     const [userAgent, device] = userCombined.split(':')
-    const user = userAgent.split('_')[0]
+    const [user, agent] = userAgent.split('_')
+    
+    let domainType = WAJIDDomains.WHATSAPP
+    
+    if (server === 'lid') {
+        domainType = WAJIDDomains.LID
+    }
+    else if (server === 'hosted') {
+        domainType = WAJIDDomains.HOSTED
+    }
+    else if (server === 'hosted.lid') {
+        domainType = WAJIDDomains.HOSTED_LID
+    }
+    else if (agent) {
+        domainType = parseInt(agent)
+    }
     return {
         server: server,
-        user,
-        domainType: server === 'lid' ? 1 : 0,
+        user: user,
+        domainType,
         device: device ? +device : undefined
     }
 }
 
 /** is the jid a user */
-const areJidsSameUser = (jid1, jid2) => {
-    return jidDecode(jid1)?.user === jidDecode(jid2)?.user
-}
+const areJidsSameUser = (jid1, jid2) => jidDecode(jid1)?.user === jidDecode(jid2)?.user
 
 /** is the jid Meta AI */
 const isJidMetaAI = (jid) => jid?.endsWith('@bot')
 
-/** is the jid a user */
-const isJidUser = (jid) => jid?.endsWith('@s.whatsapp.net')
+/** is the jid a PN user */
+const isPnUser = (jid) => jid?.endsWith('@s.whatsapp.net')
 
-/** is the lid a user */
+/** is the jid a LID */
 const isLidUser = (jid) => jid?.endsWith('@lid')
 
 /** is the jid a broadcast */
@@ -61,6 +98,12 @@ const isJidStatusBroadcast = (jid) => jid === 'status@broadcast'
 /** is the jid a newsletter */
 const isJidNewsletter = (jid) => jid?.endsWith('@newsletter')
 
+/** is the jid a hosted PN */
+const isHostedPnUser = (jid) => jid?.endsWith('@hosted')
+
+/** is the jid a hosted LID */
+const isHostedLidUser = (jid) => jid?.endsWith('@hosted.lid')
+
 /** is the jid a bot */
 const botRegexp = /^1313555\d{4}$|^131655500\d{2}$/
 const isJidBot = (jid) => (jid && botRegexp.test(jid.split('@')[0]) && jid.endsWith('@c.us'))
@@ -88,17 +131,21 @@ module.exports = {
   PSA_WID, 
   STORIES_JID, 
   META_AI_JID, 
+  WAJIDDomains, 
   jidEncode, 
   jidDecode, 
   areJidsSameUser, 
   isJidMetaAI, 
-  isJidUser, 
+  isPnUser, 
   isLidUser, 
   isJidBroadcast, 
   isJidGroup, 
   isJidStatusBroadcast, 
   isJidNewsletter, 
+  isHostedPnUser, 
+  isHostedLidUser, 
   isJidBot, 
   transferDevice, 
-  jidNormalizedUser
+  jidNormalizedUser, 
+  getServerFromDomainType
 }