mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-29 02:13:35 -05:00
26f3f3e2e2
Migrate scripts folder to esm 1/x GitOrigin-RevId: 4a4bc9a161f144fdb40ce3f2a0a9313b36c6df81
95 lines
2.3 KiB
JavaScript
95 lines
2.3 KiB
JavaScript
import _ from 'lodash'
|
|
import { db } from '../app/src/infrastructure/mongodb.js'
|
|
import BatchedUpdateModule from './helpers/batchedUpdate.mjs'
|
|
import { promiseMapWithLimit } from '@overleaf/promise-utils'
|
|
import { fileURLToPath } from 'url'
|
|
|
|
const { batchedUpdate } = BatchedUpdateModule
|
|
|
|
const WRITE_CONCURRENCY = parseInt(process.env.WRITE_CONCURRENCY, 10) || 10
|
|
|
|
// $ node scripts/convert_archived_state.mjs FIRST,SECOND
|
|
|
|
async function main(STAGE) {
|
|
for (const FIELD of ['archived', 'trashed']) {
|
|
if (STAGE.includes('FIRST')) {
|
|
await batchedUpdate(
|
|
'projects',
|
|
{ [FIELD]: false },
|
|
{
|
|
$set: { [FIELD]: [] },
|
|
}
|
|
)
|
|
|
|
console.error('Done, with first part for field:', FIELD)
|
|
}
|
|
|
|
if (STAGE.includes('SECOND')) {
|
|
await batchedUpdate(
|
|
'projects',
|
|
{ [FIELD]: true },
|
|
async function performUpdate(nextBatch) {
|
|
await promiseMapWithLimit(
|
|
WRITE_CONCURRENCY,
|
|
nextBatch,
|
|
async project => {
|
|
try {
|
|
await upgradeFieldToArray({ project, FIELD })
|
|
} catch (err) {
|
|
console.error(project._id, err)
|
|
throw err
|
|
}
|
|
}
|
|
)
|
|
},
|
|
{
|
|
_id: 1,
|
|
owner_ref: 1,
|
|
collaberator_refs: 1,
|
|
readOnly_refs: 1,
|
|
tokenAccessReadAndWrite_refs: 1,
|
|
tokenAccessReadOnly_refs: 1,
|
|
}
|
|
)
|
|
|
|
console.error('Done, with second part for field:', FIELD)
|
|
}
|
|
}
|
|
}
|
|
|
|
async function upgradeFieldToArray({ project, FIELD }) {
|
|
return db.projects.updateOne(
|
|
{ _id: project._id },
|
|
{
|
|
$set: { [FIELD]: getAllUserIds(project) },
|
|
}
|
|
)
|
|
}
|
|
|
|
function getAllUserIds(project) {
|
|
return _.unionWith(
|
|
[project.owner_ref],
|
|
project.collaberator_refs,
|
|
project.readOnly_refs,
|
|
project.tokenAccessReadAndWrite_refs,
|
|
project.tokenAccessReadOnly_refs,
|
|
_objectIdEquals
|
|
)
|
|
}
|
|
|
|
function _objectIdEquals(firstVal, secondVal) {
|
|
// For use as a comparator for unionWith
|
|
return firstVal.toString() === secondVal.toString()
|
|
}
|
|
|
|
if (fileURLToPath(import.meta.url) === process.argv[1]) {
|
|
try {
|
|
await main(process.argv.pop())
|
|
process.exit(0)
|
|
} catch (error) {
|
|
console.error({ error })
|
|
process.exit(1)
|
|
}
|
|
}
|
|
|
|
export default main
|