Overleaf/services/track-changes/app/js/PackManager.js

/* eslint-disable
    no-unused-vars,
*/
// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
 * decaffeinate suggestions:
 * DS101: Remove unnecessary use of Array.from
 * DS102: Remove unnecessary code created because of implicit returns
 * DS205: Consider reworking code to avoid use of IIFEs
 * DS207: Consider shorter variations of null checks
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
let PackManager
const async = require('async')
const _ = require('underscore')
const Bson = require('bson')
const BSON = new Bson()
const { db, ObjectId } = require('./mongodb')
const logger = require('@overleaf/logger')
const LockManager = require('./LockManager')
const MongoAWS = require('./MongoAWS')
const Metrics = require('@overleaf/metrics')
const ProjectIterator = require('./ProjectIterator')
const DocIterator = require('./DocIterator')
const Settings = require('@overleaf/settings')
const util = require('util')
const keys = Settings.redis.lock.key_schema

// Sharejs operations are stored in a 'pack' object
//
//  e.g.  a single sharejs update looks like
//
//   {
//     "doc_id" : 549dae9e0a2a615c0c7f0c98,
//     "project_id" : 549dae9c0a2a615c0c7f0c8c,
//     "op" : [ {"p" : 6981,	"d" : "?"	} ],
//     "meta" : {	"user_id" : 52933..., "start_ts" : 1422310693931,	"end_ts" : 1422310693931 },
//     "v" : 17082
//   }
//
//  and a pack looks like this
//
//   {
//     "doc_id" : 549dae9e0a2a615c0c7f0c98,
//     "project_id" : 549dae9c0a2a615c0c7f0c8c,
//     "pack" : [ U1, U2, U3, ...., UN],
//     "meta" : {	"user_id" : 52933..., "start_ts" : 1422310693931,	"end_ts" : 1422310693931 },
//     "v" : 17082
//     "v_end" : ...
//   }
//
//  where U1, U2, U3, .... are single updates stripped of their
//  doc_id and project_id fields (which are the same for all the
//  updates in the pack).
//
//  The pack itself has v and meta fields, this makes it possible to
//  treat packs and single updates in a similar way.
//
//  The v field of the pack itself is from the first entry U1, the
//  v_end field from UN.  The meta.end_ts field of the pack itself is
//  from the last entry UN, the meta.start_ts field from U1.

const DAYS = 24 * 3600 * 1000 // one day in milliseconds

module.exports = PackManager = {
  MAX_SIZE: 1024 * 1024, // make these configurable parameters
  MAX_COUNT: 1024,

  insertCompressedUpdates(
    projectId,
    docId,
    lastUpdate,
    newUpdates,
    temporary,
    callback
  ) {
    if (callback == null) {
      callback = function () {}
    }
    if (newUpdates.length === 0) {
      return callback()
    }

    // never append permanent ops to a pack that will expire
    if (
      (lastUpdate != null ? lastUpdate.expiresAt : undefined) != null &&
      !temporary
    ) {
      lastUpdate = null
    }

    const updatesToFlush = []
    const updatesRemaining = newUpdates.slice()

    let n = (lastUpdate != null ? lastUpdate.n : undefined) || 0
    let sz = (lastUpdate != null ? lastUpdate.sz : undefined) || 0

    while (
      updatesRemaining.length &&
      n < PackManager.MAX_COUNT &&
      sz < PackManager.MAX_SIZE
    ) {
      const nextUpdate = updatesRemaining[0]
      const nextUpdateSize = BSON.calculateObjectSize(nextUpdate)
      if (nextUpdateSize + sz > PackManager.MAX_SIZE && n > 0) {
        break
      }
      n++
      sz += nextUpdateSize
      updatesToFlush.push(updatesRemaining.shift())
    }

    return PackManager.flushCompressedUpdates(
      projectId,
      docId,
      lastUpdate,
      updatesToFlush,
      temporary,
      function (error) {
        if (error != null) {
          return callback(error)
        }
        return PackManager.insertCompressedUpdates(
          projectId,
          docId,
          null,
          updatesRemaining,
          temporary,
          callback
        )
      }
    )
  },

  flushCompressedUpdates(
    projectId,
    docId,
    lastUpdate,
    newUpdates,
    temporary,
    callback
  ) {
    if (callback == null) {
      callback = function () {}
    }
    if (newUpdates.length === 0) {
      return callback()
    }

    let canAppend = false
    // check if it is safe to append to an existing pack
    if (lastUpdate != null) {
      if (!temporary && lastUpdate.expiresAt == null) {
        // permanent pack appends to permanent pack
        canAppend = true
      }
      const age =
        Date.now() -
        (lastUpdate.meta != null ? lastUpdate.meta.start_ts : undefined)
      if (temporary && lastUpdate.expiresAt != null && age < 1 * DAYS) {
        // temporary pack appends to temporary pack if same day
        canAppend = true
      }
    }

    if (canAppend) {
      return PackManager.appendUpdatesToExistingPack(
        projectId,
        docId,
        lastUpdate,
        newUpdates,
        temporary,
        callback
      )
    } else {
      return PackManager.insertUpdatesIntoNewPack(
        projectId,
        docId,
        newUpdates,
        temporary,
        callback
      )
    }
  },

  insertUpdatesIntoNewPack(projectId, docId, newUpdates, temporary, callback) {
    if (callback == null) {
      callback = function () {}
    }
    const first = newUpdates[0]
    const last = newUpdates[newUpdates.length - 1]
    const n = newUpdates.length
    const sz = BSON.calculateObjectSize(newUpdates)
    const newPack = {
      project_id: ObjectId(projectId.toString()),
      doc_id: ObjectId(docId.toString()),
      pack: newUpdates,
      n,
      sz,
      meta: {
        start_ts: first.meta.start_ts,
        end_ts: last.meta.end_ts,
      },
      v: first.v,
      v_end: last.v,
      temporary,
    }
    if (temporary) {
      newPack.expiresAt = new Date(Date.now() + 7 * DAYS)
      newPack.last_checked = new Date(Date.now() + 30 * DAYS) // never check temporary packs
    }
    logger.debug(
      { projectId, docId, newUpdates },
      'inserting updates into new pack'
    )
    return db.docHistory.insertOne(newPack, function (err) {
      if (err != null) {
        return callback(err)
      }
      Metrics.inc(`insert-pack-${temporary ? 'temporary' : 'permanent'}`)
      if (temporary) {
        return callback()
      } else {
        return PackManager.updateIndex(projectId, docId, callback)
      }
    })
  },

  appendUpdatesToExistingPack(
    projectId,
    docId,
    lastUpdate,
    newUpdates,
    temporary,
    callback
  ) {
    if (callback == null) {
      callback = function () {}
    }
    const first = newUpdates[0]
    const last = newUpdates[newUpdates.length - 1]
    const n = newUpdates.length
    const sz = BSON.calculateObjectSize(newUpdates)
    const query = {
      _id: lastUpdate._id,
      project_id: ObjectId(projectId.toString()),
      doc_id: ObjectId(docId.toString()),
      pack: { $exists: true },
    }
    const update = {
      $push: {
        pack: { $each: newUpdates },
      },
      $inc: {
        n,
        sz,
      },
      $set: {
        'meta.end_ts': last.meta.end_ts,
        v_end: last.v,
      },
    }
    if (lastUpdate.expiresAt && temporary) {
      update.$set.expiresAt = new Date(Date.now() + 7 * DAYS)
    }
    logger.debug(
      { projectId, docId, lastUpdate, newUpdates },
      'appending updates to existing pack'
    )
    Metrics.inc(`append-pack-${temporary ? 'temporary' : 'permanent'}`)
    return db.docHistory.updateOne(query, update, callback)
  },

  // Retrieve all changes for a document

  getOpsByVersionRange(projectId, docId, fromVersion, toVersion, callback) {
    if (callback == null) {
      callback = function () {}
    }
    return PackManager.loadPacksByVersionRange(
      projectId,
      docId,
      fromVersion,
      toVersion,
      function (error) {
        if (error) return callback(error)
        const query = { doc_id: ObjectId(docId.toString()) }
        if (toVersion != null) {
          query.v = { $lte: toVersion }
        }
        if (fromVersion != null) {
          query.v_end = { $gte: fromVersion }
        }
        // console.log "query:", query
        return db.docHistory
          .find(query)
          .sort({ v: -1 })
          .toArray(function (err, result) {
            if (err != null) {
              return callback(err)
            }
            // console.log "getOpsByVersionRange:", err, result
            const updates = []
            const opInRange = function (op, from, to) {
              if (fromVersion != null && op.v < fromVersion) {
                return false
              }
              if (toVersion != null && op.v > toVersion) {
                return false
              }
              return true
            }
            for (const docHistory of Array.from(result)) {
              // console.log 'adding', docHistory.pack
              for (const op of Array.from(docHistory.pack.reverse())) {
                if (opInRange(op, fromVersion, toVersion)) {
                  op.project_id = docHistory.project_id
                  op.doc_id = docHistory.doc_id
                  // console.log "added op", op.v, fromVersion, toVersion
                  updates.push(op)
                }
              }
            }
            return callback(null, updates)
          })
      }
    )
  },

  loadPacksByVersionRange(projectId, docId, fromVersion, toVersion, callback) {
    return PackManager.getIndex(docId, function (err, indexResult) {
      let pack
      if (err != null) {
        return callback(err)
      }
      const indexPacks =
        (indexResult != null ? indexResult.packs : undefined) || []
      const packInRange = function (pack, from, to) {
        if (fromVersion != null && pack.v_end < fromVersion) {
          return false
        }
        if (toVersion != null && pack.v > toVersion) {
          return false
        }
        return true
      }
      const neededIds = (() => {
        const result = []
        for (pack of Array.from(indexPacks)) {
          if (packInRange(pack, fromVersion, toVersion)) {
            result.push(pack._id)
          }
        }
        return result
      })()
      if (neededIds.length) {
        return PackManager.fetchPacksIfNeeded(
          projectId,
          docId,
          neededIds,
          callback
        )
      } else {
        return callback()
      }
    })
  },

  fetchPacksIfNeeded(projectId, docId, packIds, callback) {
    let id
    return db.docHistory
      .find({ _id: { $in: packIds.map(ObjectId) } }, { projection: { _id: 1 } })
      .toArray(function (err, loadedPacks) {
        if (err != null) {
          return callback(err)
        }
        const allPackIds = (() => {
          const result1 = []
          for (id of Array.from(packIds)) {
            result1.push(id.toString())
          }
          return result1
        })()
        const loadedPackIds = Array.from(loadedPacks).map(pack =>
          pack._id.toString()
        )
        const packIdsToFetch = _.difference(allPackIds, loadedPackIds)
        logger.debug(
          { projectId, docId, loadedPackIds, allPackIds, packIdsToFetch },
          'analysed packs'
        )
        if (packIdsToFetch.length === 0) {
          return callback()
        }
        return async.eachLimit(
          packIdsToFetch,
          4,
          (packId, cb) => MongoAWS.unArchivePack(projectId, docId, packId, cb),
          function (err) {
            if (err != null) {
              return callback(err)
            }
            logger.debug({ projectId, docId }, 'done unarchiving')
            return callback()
          }
        )
      })
  },

  findAllDocsInProject(projectId, callback) {
    const docIdSet = new Set()
    async.series(
      [
        cb => {
          db.docHistory
            .find(
              { project_id: ObjectId(projectId) },
              { projection: { pack: false } }
            )
            .toArray((err, packs) => {
              if (err) return callback(err)
              packs.forEach(pack => {
                docIdSet.add(pack.doc_id.toString())
              })
              return cb()
            })
        },
        cb => {
          db.docHistoryIndex
            .find({ project_id: ObjectId(projectId) })
            .toArray((err, indexes) => {
              if (err) return callback(err)
              indexes.forEach(index => {
                docIdSet.add(index._id.toString())
              })
              return cb()
            })
        },
      ],
      err => {
        if (err) return callback(err)
        callback(null, [...docIdSet])
      }
    )
  },

  // rewrite any query using doc_id to use _id instead
  // (because docHistoryIndex uses the doc_id)

  _rewriteQueryForIndex(query) {
    const indexQuery = _.omit(query, 'doc_id')
    if ('doc_id' in query) {
      indexQuery._id = query.doc_id
    }
    return indexQuery
  },

  // Retrieve all changes across a project

  _findPacks(query, sortKeys, callback) {
    // get all the docHistory Entries
    return db.docHistory
      .find(query, { projection: { pack: false } })
      .sort(sortKeys)
      .toArray(function (err, packs) {
        let pack
        if (err != null) {
          return callback(err)
        }
        const allPacks = []
        const seenIds = {}
        for (pack of Array.from(packs)) {
          allPacks.push(pack)
          seenIds[pack._id] = true
        }
        const indexQuery = PackManager._rewriteQueryForIndex(query)
        return db.docHistoryIndex
          .find(indexQuery)
          .toArray(function (err, indexes) {
            if (err != null) {
              return callback(err)
            }
            for (const index of Array.from(indexes)) {
              for (pack of Array.from(index.packs)) {
                if (!seenIds[pack._id]) {
                  pack.project_id = index.project_id
                  pack.doc_id = index._id
                  pack.fromIndex = true
                  allPacks.push(pack)
                  seenIds[pack._id] = true
                }
              }
            }
            return callback(null, allPacks)
          })
      })
  },

  makeProjectIterator(projectId, before, callback) {
    PackManager._findPacks(
      { project_id: ObjectId(projectId) },
      { 'meta.end_ts': -1 },
      function (err, allPacks) {
        if (err) return callback(err)
        callback(
          null,
          new ProjectIterator(allPacks, before, PackManager.getPackById)
        )
      }
    )
  },

  makeDocIterator(docId, callback) {
    PackManager._findPacks(
      { doc_id: ObjectId(docId) },
      { v: -1 },
      function (err, allPacks) {
        if (err) return callback(err)
        callback(null, new DocIterator(allPacks, PackManager.getPackById))
      }
    )
  },

  getPackById(projectId, docId, packId, callback) {
    return db.docHistory.findOne({ _id: packId }, function (err, pack) {
      if (err != null) {
        return callback(err)
      }
      if (pack == null) {
        return MongoAWS.unArchivePack(projectId, docId, packId, callback)
      } else if (pack.expiresAt != null && pack.temporary === false) {
        // we only need to touch the TTL when listing the changes in the project
        // because diffs on individual documents are always done after that
        return PackManager.increaseTTL(pack, callback)
        // only do this for cached packs, not temporary ones to avoid older packs
        // being kept longer than newer ones (which messes up the last update version)
      } else {
        return callback(null, pack)
      }
    })
  },

  increaseTTL(pack, callback) {
    if (pack.expiresAt < new Date(Date.now() + 6 * DAYS)) {
      // update cache expiry since we are using this pack
      return db.docHistory.updateOne(
        { _id: pack._id },
        { $set: { expiresAt: new Date(Date.now() + 7 * DAYS) } },
        err => callback(err, pack)
      )
    } else {
      return callback(null, pack)
    }
  },

  // Manage docHistoryIndex collection

  getIndex(docId, callback) {
    return db.docHistoryIndex.findOne(
      { _id: ObjectId(docId.toString()) },
      callback
    )
  },

  getPackFromIndex(docId, packId, callback) {
    return db.docHistoryIndex.findOne(
      { _id: ObjectId(docId.toString()), 'packs._id': packId },
      { projection: { 'packs.$': 1 } },
      callback
    )
  },

  getLastPackFromIndex(docId, callback) {
    return db.docHistoryIndex.findOne(
      { _id: ObjectId(docId.toString()) },
      { projection: { packs: { $slice: -1 } } },
      function (err, indexPack) {
        if (err != null) {
          return callback(err)
        }
        if (indexPack == null) {
          return callback()
        }
        return callback(null, indexPack[0])
      }
    )
  },

  getIndexWithKeys(docId, callback) {
    return PackManager.getIndex(docId, function (err, index) {
      if (err != null) {
        return callback(err)
      }
      if (index == null) {
        return callback()
      }
      for (const pack of Array.from(
        (index != null ? index.packs : undefined) || []
      )) {
        index[pack._id] = pack
      }
      return callback(null, index)
    })
  },

  initialiseIndex(projectId, docId, callback) {
    return PackManager.findCompletedPacks(
      projectId,
      docId,
      function (err, packs) {
        // console.log 'err', err, 'packs', packs, packs?.length
        if (err != null) {
          return callback(err)
        }
        if (packs == null) {
          return callback()
        }
        return PackManager.insertPacksIntoIndexWithLock(
          projectId,
          docId,
          packs,
          callback
        )
      }
    )
  },

  updateIndex(projectId, docId, callback) {
    // find all packs prior to current pack
    return PackManager.findUnindexedPacks(
      projectId,
      docId,
      function (err, newPacks) {
        if (err != null) {
          return callback(err)
        }
        if (newPacks == null || newPacks.length === 0) {
          return callback()
        }
        return PackManager.insertPacksIntoIndexWithLock(
          projectId,
          docId,
          newPacks,
          function (err) {
            if (err != null) {
              return callback(err)
            }
            logger.debug(
              { projectId, docId, newPacks },
              'added new packs to index'
            )
            return callback()
          }
        )
      }
    )
  },

  findCompletedPacks(projectId, docId, callback) {
    const query = {
      doc_id: ObjectId(docId.toString()),
      expiresAt: { $exists: false },
    }
    return db.docHistory
      .find(query, { projection: { pack: false } })
      .sort({ v: 1 })
      .toArray(function (err, packs) {
        if (err != null) {
          return callback(err)
        }
        if (packs == null) {
          return callback()
        }
        if (!(packs != null ? packs.length : undefined)) {
          return callback()
        }
        const last = packs.pop() // discard the last pack, if it's still in progress
        if (last.finalised) {
          packs.push(last)
        } // it's finalised so we push it back to archive it
        return callback(null, packs)
      })
  },

  findPacks(projectId, docId, callback) {
    const query = {
      doc_id: ObjectId(docId.toString()),
      expiresAt: { $exists: false },
    }
    return db.docHistory
      .find(query, { projection: { pack: false } })
      .sort({ v: 1 })
      .toArray(function (err, packs) {
        if (err != null) {
          return callback(err)
        }
        if (packs == null) {
          return callback()
        }
        if (!(packs != null ? packs.length : undefined)) {
          return callback()
        }
        return callback(null, packs)
      })
  },

  findUnindexedPacks(projectId, docId, callback) {
    return PackManager.getIndexWithKeys(docId, function (err, indexResult) {
      if (err != null) {
        return callback(err)
      }
      return PackManager.findCompletedPacks(
        projectId,
        docId,
        function (err, historyPacks) {
          let pack
          if (err != null) {
            return callback(err)
          }
          if (historyPacks == null) {
            return callback()
          }
          // select only the new packs not already in the index
          let newPacks = (() => {
            const result = []
            for (pack of Array.from(historyPacks)) {
              if (
                (indexResult != null ? indexResult[pack._id] : undefined) ==
                null
              ) {
                result.push(pack)
              }
            }
            return result
          })()
          newPacks = (() => {
            const result1 = []
            for (pack of Array.from(newPacks)) {
              result1.push(
                _.omit(
                  pack,
                  'doc_id',
                  'project_id',
                  'n',
                  'sz',
                  'last_checked',
                  'finalised'
                )
              )
            }
            return result1
          })()
          if (newPacks.length) {
            logger.debug(
              { projectId, docId, n: newPacks.length },
              'found new packs'
            )
          }
          return callback(null, newPacks)
        }
      )
    })
  },

  insertPacksIntoIndexWithLock(projectId, docId, newPacks, callback) {
    return LockManager.runWithLock(
      keys.historyIndexLock({ doc_id: docId }),
      releaseLock =>
        PackManager._insertPacksIntoIndex(
          projectId,
          docId,
          newPacks,
          releaseLock
        ),
      callback
    )
  },

  _insertPacksIntoIndex(projectId, docId, newPacks, callback) {
    return db.docHistoryIndex.updateOne(
      { _id: ObjectId(docId.toString()) },
      {
        $setOnInsert: { project_id: ObjectId(projectId.toString()) },
        $push: {
          packs: { $each: newPacks, $sort: { v: 1 } },
        },
      },
      {
        upsert: true,
      },
      callback
    )
  },

  // Archiving packs to S3

  archivePack(projectId, docId, packId, callback) {
    const clearFlagOnError = function (err, cb) {
      if (err != null) {
        // clear the inS3 flag on error
        return PackManager.clearPackAsArchiveInProgress(
          projectId,
          docId,
          packId,
          function (err2) {
            if (err2 != null) {
              return cb(err2)
            }
            return cb(err)
          }
        )
      } else {
        return cb()
      }
    }
    return async.series(
      [
        cb =>
          PackManager.checkArchiveNotInProgress(projectId, docId, packId, cb),
        cb =>
          PackManager.markPackAsArchiveInProgress(projectId, docId, packId, cb),
        cb =>
          MongoAWS.archivePack(projectId, docId, packId, err =>
            clearFlagOnError(err, cb)
          ),
        cb =>
          PackManager.checkArchivedPack(projectId, docId, packId, err =>
            clearFlagOnError(err, cb)
          ),
        cb => PackManager.markPackAsArchived(projectId, docId, packId, cb),
        cb =>
          PackManager.setTTLOnArchivedPack(projectId, docId, packId, callback),
      ],
      callback
    )
  },

  checkArchivedPack(projectId, docId, packId, callback) {
    return db.docHistory.findOne({ _id: packId }, function (err, pack) {
      if (err != null) {
        return callback(err)
      }
      if (pack == null) {
        return callback(new Error('pack not found'))
      }
      return MongoAWS.readArchivedPack(
        projectId,
        docId,
        packId,
        function (err, result) {
          if (err) return callback(err)
          delete result.last_checked
          delete pack.last_checked
          // need to compare ids as ObjectIds with .equals()
          for (const key of ['_id', 'project_id', 'doc_id']) {
            if (result[key].equals(pack[key])) {
              result[key] = pack[key]
            }
          }
          for (let i = 0; i < result.pack.length; i++) {
            const op = result.pack[i]
            if (op._id != null && op._id.equals(pack.pack[i]._id)) {
              op._id = pack.pack[i]._id
            }
          }
          if (_.isEqual(pack, result)) {
            return callback()
          } else {
            logger.err(
              {
                pack,
                result,
                jsondiff: JSON.stringify(pack) === JSON.stringify(result),
              },
              'difference when comparing packs'
            )
            return callback(
              new Error('pack retrieved from s3 does not match pack in mongo')
            )
          }
        }
      )
    })
  },
  // Extra methods to test archive/unarchive for a doc_id

  pushOldPacks(projectId, docId, callback) {
    return PackManager.findPacks(projectId, docId, function (err, packs) {
      if (err != null) {
        return callback(err)
      }
      if (!(packs != null ? packs.length : undefined)) {
        return callback()
      }
      return PackManager.processOldPack(
        projectId,
        docId,
        packs[0]._id,
        callback
      )
    })
  },

  pullOldPacks(projectId, docId, callback) {
    return PackManager.loadPacksByVersionRange(
      projectId,
      docId,
      null,
      null,
      callback
    )
  },

  // Processing old packs via worker

  processOldPack(projectId, docId, packId, callback) {
    const markAsChecked = err =>
      PackManager.markPackAsChecked(projectId, docId, packId, function (err2) {
        if (err2 != null) {
          return callback(err2)
        }
        return callback(err)
      })
    logger.debug({ projectId, docId }, 'processing old packs')
    return db.docHistory.findOne({ _id: packId }, function (err, pack) {
      if (err != null) {
        return markAsChecked(err)
      }
      if (pack == null) {
        return markAsChecked()
      }
      if (pack.expiresAt != null) {
        return callback()
      } // return directly
      return PackManager.finaliseIfNeeded(
        projectId,
        docId,
        pack._id,
        pack,
        function (err) {
          if (err != null) {
            return markAsChecked(err)
          }
          return PackManager.updateIndexIfNeeded(
            projectId,
            docId,
            function (err) {
              if (err != null) {
                return markAsChecked(err)
              }
              return PackManager.findUnarchivedPacks(
                projectId,
                docId,
                function (err, unarchivedPacks) {
                  if (err != null) {
                    return markAsChecked(err)
                  }
                  if (
                    !(unarchivedPacks != null
                      ? unarchivedPacks.length
                      : undefined)
                  ) {
                    logger.debug(
                      { projectId, docId },
                      'no packs need archiving'
                    )
                    return markAsChecked()
                  }
                  return async.eachSeries(
                    unarchivedPacks,
                    (pack, cb) =>
                      PackManager.archivePack(projectId, docId, pack._id, cb),
                    function (err) {
                      if (err != null) {
                        return markAsChecked(err)
                      }
                      logger.debug({ projectId, docId }, 'done processing')
                      return markAsChecked()
                    }
                  )
                }
              )
            }
          )
        }
      )
    })
  },

  finaliseIfNeeded(projectId, docId, packId, pack, callback) {
    const sz = pack.sz / (1024 * 1024) // in fractions of a megabyte
    const n = pack.n / 1024 // in fraction of 1024 ops
    const age = (Date.now() - pack.meta.end_ts) / DAYS
    if (age < 30) {
      // always keep if less than 1 month old
      logger.debug({ projectId, docId, packId, age }, 'less than 30 days old')
      return callback()
    }
    // compute an archiving threshold which decreases for each month of age
    const archiveThreshold = 30 / age
    if (sz > archiveThreshold || n > archiveThreshold || age > 90) {
      logger.debug(
        { projectId, docId, packId, age, archiveThreshold, sz, n },
        'meets archive threshold'
      )
      return PackManager.markPackAsFinalisedWithLock(
        projectId,
        docId,
        packId,
        callback
      )
    } else {
      logger.debug(
        { projectId, docId, packId, age, archiveThreshold, sz, n },
        'does not meet archive threshold'
      )
      return callback()
    }
  },

  markPackAsFinalisedWithLock(projectId, docId, packId, callback) {
    return LockManager.runWithLock(
      keys.historyLock({ doc_id: docId }),
      releaseLock =>
        PackManager._markPackAsFinalised(projectId, docId, packId, releaseLock),
      callback
    )
  },

  _markPackAsFinalised(projectId, docId, packId, callback) {
    logger.debug({ projectId, docId, packId }, 'marking pack as finalised')
    return db.docHistory.updateOne(
      { _id: packId },
      { $set: { finalised: true } },
      callback
    )
  },

  updateIndexIfNeeded(projectId, docId, callback) {
    logger.debug({ projectId, docId }, 'archiving old packs')
    return PackManager.getIndexWithKeys(docId, function (err, index) {
      if (err != null) {
        return callback(err)
      }
      if (index == null) {
        return PackManager.initialiseIndex(projectId, docId, callback)
      } else {
        return PackManager.updateIndex(projectId, docId, callback)
      }
    })
  },

  markPackAsChecked(projectId, docId, packId, callback) {
    logger.debug({ projectId, docId, packId }, 'marking pack as checked')
    return db.docHistory.updateOne(
      { _id: packId },
      { $currentDate: { last_checked: true } },
      callback
    )
  },

  findUnarchivedPacks(projectId, docId, callback) {
    return PackManager.getIndex(docId, function (err, indexResult) {
      if (err != null) {
        return callback(err)
      }
      const indexPacks =
        (indexResult != null ? indexResult.packs : undefined) || []
      const unArchivedPacks = (() => {
        const result = []
        for (const pack of Array.from(indexPacks)) {
          if (pack.inS3 == null) {
            result.push(pack)
          }
        }
        return result
      })()
      if (unArchivedPacks.length) {
        logger.debug(
          { projectId, docId, n: unArchivedPacks.length },
          'find unarchived packs'
        )
      }
      return callback(null, unArchivedPacks)
    })
  },

  // Archive locking flags

  checkArchiveNotInProgress(projectId, docId, packId, callback) {
    logger.debug(
      { projectId, docId, packId },
      'checking if archive in progress'
    )
    return PackManager.getPackFromIndex(docId, packId, function (err, result) {
      if (err != null) {
        return callback(err)
      }
      if (result == null) {
        return callback(new Error('pack not found in index'))
      }
      if (result.inS3) {
        return callback(new Error('pack archiving already done'))
      } else if (result.inS3 != null) {
        return callback(new Error('pack archiving already in progress'))
      } else {
        return callback()
      }
    })
  },

  markPackAsArchiveInProgress(projectId, docId, packId, callback) {
    logger.debug(
      { projectId, docId },
      'marking pack as archive in progress status'
    )
    return db.docHistoryIndex.findOneAndUpdate(
      {
        _id: ObjectId(docId.toString()),
        packs: { $elemMatch: { _id: packId, inS3: { $exists: false } } },
      },
      { $set: { 'packs.$.inS3': false } },
      { projection: { 'packs.$': 1 } },
      function (err, result) {
        if (err != null) {
          return callback(err)
        }
        if (!result.value) {
          return callback(new Error('archive is already in progress'))
        }
        logger.debug(
          { projectId, docId, packId },
          'marked as archive in progress'
        )
        return callback()
      }
    )
  },

  clearPackAsArchiveInProgress(projectId, docId, packId, callback) {
    logger.debug(
      { projectId, docId, packId },
      'clearing as archive in progress'
    )
    return db.docHistoryIndex.updateOne(
      {
        _id: ObjectId(docId.toString()),
        packs: { $elemMatch: { _id: packId, inS3: false } },
      },
      { $unset: { 'packs.$.inS3': true } },
      callback
    )
  },

  markPackAsArchived(projectId, docId, packId, callback) {
    logger.debug({ projectId, docId, packId }, 'marking pack as archived')
    return db.docHistoryIndex.findOneAndUpdate(
      {
        _id: ObjectId(docId.toString()),
        packs: { $elemMatch: { _id: packId, inS3: false } },
      },
      { $set: { 'packs.$.inS3': true } },
      { projection: { 'packs.$': 1 } },
      function (err, result) {
        if (err != null) {
          return callback(err)
        }
        if (!result.value) {
          return callback(new Error('archive is not marked as progress'))
        }
        logger.debug({ projectId, docId, packId }, 'marked as archived')
        return callback()
      }
    )
  },

  setTTLOnArchivedPack(projectId, docId, packId, callback) {
    return db.docHistory.updateOne(
      { _id: packId },
      { $set: { expiresAt: new Date(Date.now() + 1 * DAYS) } },
      function (err) {
        if (err) {
          return callback(err)
        }
        logger.debug({ projectId, docId, packId }, 'set expiry on pack')
        return callback()
      }
    )
  },
}

module.exports.promises = {
  getOpsByVersionRange: util.promisify(PackManager.getOpsByVersionRange),
  findAllDocsInProject: util.promisify(PackManager.findAllDocsInProject),
  makeDocIterator: util.promisify(PackManager.makeDocIterator),
}

//	_getOneDayInFutureWithRandomDelay: ->
//		thirtyMins = 1000 * 60 * 30
//		randomThirtyMinMax = Math.ceil(Math.random() * thirtyMins)
//		return new Date(Date.now() + randomThirtyMinMax + 1*DAYS)