Merge pull request #20256 from overleaf/jdt-enable-writefull-unset

Differentiate between unset and disabled Writefull

GitOrigin-RevId: 3cf8f12ede851dab5a8067bdbcddba6c69870573

services/web/scripts/split_writefull_disabled_from_unset.js

@@ -0,0 +1,59 @@
+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)
+    })
+}