overleaf/services/web/scripts/split_writefull_disabled_from_unset.js

60 lines
1.5 KiB
JavaScript
Raw Normal View History

const { db, waitForDb } = require('../app/src/infrastructure/mongodb')
const { batchedUpdate } = require('./helpers/batchedUpdate')
const { ObjectId } = require('mongodb-legacy')
const fs = require('fs')
const CHUNK_SIZE = 1000
// Function to chunk the array
function chunkArray(array, size) {
const result = []
for (let i = 0; i < array.length; i += size) {
result.push(array.slice(i, i + size))
}
return result
}
async function main() {
// search for file of users who already explicitly opted out first
const optOutPath = process.argv[2]
const optedOutFile = fs.readFileSync(optOutPath, 'utf8')
let optedOutList = optedOutFile
optedOutList = optedOutFile.split('\n').map(id => new ObjectId(id))
console.log(`preserving opt-outs of ${optedOutList.length} users`)
await waitForDb()
// update all applicable user models
await batchedUpdate(
'users',
{ 'writefull.enabled': false }, // and is false
{ $set: { 'writefull.enabled': null } }
)
const chunks = chunkArray(optedOutList, CHUNK_SIZE)
// then reset any explicit false back to being false
// Iterate over each chunk and perform the query
for (const chunkedIds of chunks) {
console.log('batch update started')
await db.users.updateMany(
{ _id: { $in: chunkedIds } },
{ $set: { 'writefull.enabled': false } }
)
console.log('batch completed')
}
}
module.exports = main
if (require.main === module) {
main()
.then(() => {
process.exit(0)
})
.catch(error => {
console.error({ error })
process.exit(1)
})
}