import { db, waitForDb } from '../app/src/infrastructure/mongodb.js'
import { fileURLToPath } from 'url'

async function updateStringDates() {
  await waitForDb()
  const users = db.users.find({
    splitTests: { $exists: true },
  })

  let user
  let count = 0
  while ((user = await users.next())) {
    count += 1
    if (count % 10000 === 0) {
      console.log(`processed ${count} users...`)
    }

    const splitTests = user.splitTests
    for (const splitTestKey of Object.keys(splitTests)) {
      for (const variantIndex in splitTests[splitTestKey]) {
        splitTests[splitTestKey][variantIndex].assignedAt = new Date(
          splitTests[splitTestKey][variantIndex].assignedAt
        )
      }
    }

    await db.users.updateOne(
      {
        _id: user._id,
      },
      { $set: { splitTests } }
    )
  }
  console.log(`Updated ${count} assignedAt strings to dates!`)
}

if (fileURLToPath(import.meta.url) === process.argv[1]) {
  try {
    await updateStringDates()
    process.exit(0)
  } catch (error) {
    console.error(error)
    process.exit(1)
  }
}

export default updateStringDates