@vreden/meta/lib/Utils/use-mongo-file-auth-state.js

Baileys is a lightweight JavaScript library for interacting with the WhatsApp Web API using WebSocket.

"use strict"

Object.defineProperty(exports, "__esModule", { value: true })

const WAProto_1 = require("../../WAProto")
const auth_utils_1 = require("./auth-utils")
const generics_1 = require("./generics")

/*
code from amiruldev readjusted by @irull2nd, don't delete WM!
*/
const useMongoFileAuthState = async (collection) => {
  const writeData = (data,id) => {
    const informationToStore = JSON.parse(
      JSON.stringify(data, generics_1.BufferJSON.replacer)
    )
    const update = {
      $set: {
        ...informationToStore,
      },
    }

    return collection.updateOne({_id: id},update, {upsert: true})
  }

  const readData = async (id) => {
    try {
      const data = JSON.stringify(await collection.findOne({_id: id}))
      return JSON.parse(data, generics_1.BufferJSON.reviver)
    } catch (err) {
      console.log(err)
    }
  }

  const removeData = async (id) => {
    try{
      await collection.deleteOne({_id: id})
    }catch(err){
      console.log('error',err)
    }
  }

  const creds = (await readData('creds')) || auth_utils_1.initAuthCreds()

   return{
    state:{
      creds,
      keys: {
        get: async (type,ids)=> {
          const data = {}
          await Promise
          .all(
            ids.map(async (id) => {
              let value = await readData(`${type}-${id}`)
              if(type === "app-state-sync-key"){
                value = WAProto_1.proto.Message.AppStateSyncKeyData.fromObject(data)
              }
              data[id] = value
            })
          )
          return data
        },
        set: async (data) => {
          const tasks = []
          for (const category of Object.keys(data)){
            for (const id of Object.keys(data[category])){
              const value = data[category][id]
              const key = `${category}-${id}`
              tasks.push(value? writeData(value,key) : removeData(key))
            }
          }
          await Promise.all(tasks)
        },
      },
    },
    saveCreds: () => {
      return writeData(creds, "creds")
    }
  }
}

module.exports = {
  useMongoFileAuthState
}