overleaf/services/web/scripts/add_multiple_emails.js
Alasdair Smith 659242b457 Merge pull request #1037 from sharelatex/as-decaffeinate
Decaffeinate frontend

GitOrigin-RevId: 1c8c53dedecfe55f9936a13408df17b852f996de
2018-11-05 10:19:25 +00:00

64 lines
1.4 KiB
JavaScript

const mongojs = require('../app/js/infrastructure/mongojs')
const { db } = mongojs
const async = require('async')
const minilist = require('minimist')
const updateUser = function(user, callback) {
console.log(`Updating user ${user._id}`)
const update = {
$set: {
emails: [
{
email: user.email,
createdAt: new Date()
}
]
}
}
db.users.update({ _id: user._id }, update, callback)
}
const updateUsers = (users, callback) =>
async.eachLimit(users, ASYNC_LIMIT, updateUser, function(error) {
if (error) {
callback(error)
return
}
counter += users.length
console.log(`${counter} users updated`)
loopForUsers(callback)
})
var loopForUsers = callback =>
db.users
.find({ emails: { $exists: false } }, { email: 1 })
.limit(FETCH_LIMIT, function(error, users) {
if (error) {
callback(error)
return
}
if (users.length === 0) {
console.log(`DONE (${counter} users updated)`)
return callback()
}
updateUsers(users, callback)
})
var counter = 0
var run = () =>
loopForUsers(function(error) {
if (error) {
throw error
}
process.exit()
})
let FETCH_LIMIT, ASYNC_LIMIT
var setup = function() {
let args = minilist(process.argv.slice(2))
FETCH_LIMIT = args.fetch ? args.fetch : 100
ASYNC_LIMIT = args.async ? args.async : 10
}
setup()
run()