import _ from 'lodash' import { db } from '../app/src/infrastructure/mongodb.js' import { batchedUpdate } from '@overleaf/mongo-utils/batchedUpdate.js' import { promiseMapWithLimit } from '@overleaf/promise-utils' import { fileURLToPath } from 'node:url' 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( db.projects, { [FIELD]: false }, { $set: { [FIELD]: [] }, } ) console.error('Done, with first part for field:', FIELD) } if (STAGE.includes('SECOND')) { await batchedUpdate( db.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