mirror of
https://github.com/EvolutionAPI/evolution-manager.git
synced 2025-12-24 21:27:44 -06:00
feat: migrate-to-mongo 0.4.6
This commit is contained in:
44
lib/api/migrateToMongo/migrators/chats.js
Normal file
44
lib/api/migrateToMongo/migrators/chats.js
Normal file
@@ -0,0 +1,44 @@
|
||||
const FOLDER_PATH = [{
|
||||
path: "/store/chats",
|
||||
key: "chats"
|
||||
}]
|
||||
|
||||
const fs = require("fs")
|
||||
const path = require("path")
|
||||
|
||||
|
||||
module.exports = async (instanceName, options, progressBars, conn) => {
|
||||
var files = []
|
||||
|
||||
FOLDER_PATH.forEach(folder => {
|
||||
files = files.concat(getFiles(instanceName, folder))
|
||||
})
|
||||
const progress = progressBars.create(files.length, 0)
|
||||
progress.update({ process: 'Chats' })
|
||||
for (const file of files) {
|
||||
const collectionName = file.key
|
||||
const collection = conn.collection(collectionName)
|
||||
|
||||
const data = JSON.parse(fs.readFileSync(file.path, 'utf8'))
|
||||
data._id = file.path.split('\\').pop().split('.')[0]
|
||||
await collection.findOneAndUpdate({ _id: data._id }, { $set: data }, { upsert: true })
|
||||
|
||||
progress.increment()
|
||||
}
|
||||
}
|
||||
|
||||
function getFiles(instanceName, opts) {
|
||||
var files = []
|
||||
const folder = opts.path || opts
|
||||
const folderPath = path.join(process.cwd(), folder)
|
||||
|
||||
fs.readdirSync(folderPath).forEach(file => {
|
||||
const filePath = path.join(folderPath, file)
|
||||
if (fs.statSync(filePath).isDirectory()) {
|
||||
files = files.concat(getFiles(instanceName, { ...opts, path: `${folder}/${file}` }))
|
||||
} else if (file.includes(instanceName) || folder.includes(instanceName)) {
|
||||
files.push({ ...opts, path: filePath })
|
||||
}
|
||||
})
|
||||
return files
|
||||
}
|
||||
Reference in New Issue
Block a user