overleaf/services/clsi/app/js/ProjectPersistenceManager.js

// 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
 * DS207: Consider shorter variations of null checks
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
let ProjectPersistenceManager
const UrlCache = require('./UrlCache')
const CompileManager = require('./CompileManager')
const async = require('async')
const logger = require('@overleaf/logger')
const oneDay = 24 * 60 * 60 * 1000
const Settings = require('@overleaf/settings')
const diskusage = require('diskusage')
const { callbackify } = require('util')
const Path = require('path')
const fs = require('fs')

// projectId -> timestamp mapping.
const LAST_ACCESS = new Map()

async function refreshExpiryTimeout() {
  const paths = [
    Settings.path.compilesDir,
    Settings.path.outputDir,
    Settings.path.clsiCacheDir,
  ]
  for (const path of paths) {
    try {
      const stats = await diskusage.check(path)
      const lowDisk = stats.available / stats.total < 0.1

      const lowerExpiry = ProjectPersistenceManager.EXPIRY_TIMEOUT * 0.9
      if (lowDisk && Settings.project_cache_length_ms / 2 < lowerExpiry) {
        logger.warn(
          {
            stats,
            newExpiryTimeoutInDays: (lowerExpiry / oneDay).toFixed(2),
          },
          'disk running low on space, modifying EXPIRY_TIMEOUT'
        )
        ProjectPersistenceManager.EXPIRY_TIMEOUT = lowerExpiry
        break
      }
    } catch (err) {
      logger.err({ err, path }, 'error getting disk usage')
    }
  }
}

module.exports = ProjectPersistenceManager = {
  EXPIRY_TIMEOUT: Settings.project_cache_length_ms || oneDay * 2.5,

  promises: {
    refreshExpiryTimeout,
  },

  refreshExpiryTimeout: callbackify(refreshExpiryTimeout),

  init() {
    fs.readdir(Settings.path.compilesDir, (err, dirs) => {
      if (err) {
        logger.warn({ err }, 'cannot get project listing')
        dirs = []
      }

      async.eachLimit(
        dirs,
        10,
        (projectAndUserId, cb) => {
          const compileDir = Path.join(
            Settings.path.compilesDir,
            projectAndUserId
          )
          const projectId = projectAndUserId.slice(0, 24)
          fs.stat(compileDir, (err, stats) => {
            if (err) {
              // Schedule for immediate cleanup
              LAST_ACCESS.set(projectId, 0)
            } else {
              // Cleanup eventually.
              LAST_ACCESS.set(projectId, stats.mtime.getTime())
            }
            cb()
          })
        },
        () => {
          setInterval(() => {
            ProjectPersistenceManager.refreshExpiryTimeout(() => {
              ProjectPersistenceManager.clearExpiredProjects(err => {
                if (err) {
                  logger.error({ err }, 'clearing expired projects failed')
                }
              })
            })
          }, 10 * 60 * 1000)
        }
      )
    })
  },

  markProjectAsJustAccessed(projectId, callback) {
    LAST_ACCESS.set(projectId, Date.now())
    callback()
  },

  clearExpiredProjects(callback) {
    if (callback == null) {
      callback = function () {}
    }
    return ProjectPersistenceManager._findExpiredProjectIds(function (
      error,
      projectIds
    ) {
      if (error != null) {
        return callback(error)
      }
      logger.debug({ projectIds }, 'clearing expired projects')
      const jobs = Array.from(projectIds || []).map(projectId =>
        (
          projectId => callback =>
            ProjectPersistenceManager.clearProjectFromCache(
              projectId,
              { reason: 'expired' },
              function (err) {
                if (err != null) {
                  logger.error({ err, projectId }, 'error clearing project')
                }
                return callback()
              }
            )
        )(projectId)
      )
      return async.series(jobs, function (error) {
        if (error != null) {
          return callback(error)
        }
        return CompileManager.clearExpiredProjects(
          ProjectPersistenceManager.EXPIRY_TIMEOUT,
          error => callback(error)
        )
      })
    })
  }, // ignore any errors from deleting directories

  clearProject(projectId, userId, callback) {
    if (callback == null) {
      callback = function () {}
    }
    logger.debug({ projectId, userId }, 'clearing project for user')
    return CompileManager.clearProject(projectId, userId, function (error) {
      if (error != null) {
        return callback(error)
      }
      return ProjectPersistenceManager.clearProjectFromCache(
        projectId,
        { reason: 'cleared' },
        function (error) {
          if (error != null) {
            return callback(error)
          }
          return callback()
        }
      )
    })
  },

  clearProjectFromCache(projectId, options, callback) {
    if (callback == null) {
      callback = function () {}
    }
    logger.debug({ projectId }, 'clearing project from cache')
    return UrlCache.clearProject(projectId, options, function (error) {
      if (error != null) {
        logger.err({ error, projectId }, 'error clearing project from cache')
        return callback(error)
      }
      return ProjectPersistenceManager._clearProjectFromDatabase(
        projectId,
        function (error) {
          if (error != null) {
            logger.err(
              { error, projectId },
              'error clearing project from database'
            )
          }
          return callback(error)
        }
      )
    })
  },

  _clearProjectFromDatabase(projectId, callback) {
    LAST_ACCESS.delete(projectId)
    callback()
  },

  _findExpiredProjectIds(callback) {
    const expiredFrom = Date.now() - ProjectPersistenceManager.EXPIRY_TIMEOUT
    const expiredProjectsIds = []
    for (const [projectId, lastAccess] of LAST_ACCESS.entries()) {
      if (lastAccess < expiredFrom) {
        expiredProjectsIds.push(projectId)
      }
    }
    // ^ may be a fairly busy loop, continue detached.
    setTimeout(() => callback(null, expiredProjectsIds), 0)
  },
}

logger.debug(
  { EXPIRY_TIMEOUT: ProjectPersistenceManager.EXPIRY_TIMEOUT },
  'project assets kept timeout'
)