filesafe/scripts/thumbs.js

const { stripIndents } = require('./_utils')
const config = require('./../config')
const fs = require('fs')
const path = require('path')
const utils = require('./../controllers/utilsController')

const thumbs = {
  mode: null,
  force: null,
  verbose: null,
  cfcache: null
}

thumbs.mayGenerateThumb = extname => {
  return ([1, 3].includes(thumbs.mode) && utils.imageExtensions.includes(extname)) ||
    ([2, 3].includes(thumbs.mode) && utils.videoExtensions.includes(extname))
}

thumbs.getFiles = directory => {
  return new Promise((resolve, reject) => {
    fs.readdir(directory, async (error, names) => {
      if (error) return reject(error)
      const files = []
      await Promise.all(names.map(name => {
        return new Promise((resolve, reject) => {
          fs.lstat(path.join(directory, name), (error, stats) => {
            if (error) return reject(error)
            if (stats.isFile() && !name.startsWith('.')) files.push(name)
            resolve()
          })
        })
      }))
      resolve(files)
    })
  })
}

thumbs.do = async () => {
  const location = process.argv[1].replace(process.cwd() + '/', '')
  const args = process.argv.slice(2)

  thumbs.mode = parseInt(args[0])
  thumbs.force = parseInt(args[1] || 0)
  thumbs.verbose = parseInt(args[2] || 0)
  thumbs.cfcache = parseInt(args[3] || 0)
  if (![1, 2, 3].includes(thumbs.mode) ||
    ![0, 1].includes(thumbs.force) ||
    ![0, 1].includes(thumbs.verbose) ||
    args.includes('--help') ||
    args.includes('-h'))
    return console.log(stripIndents(`
      Generate thumbnails.

      Usage  :\nnode ${location} <mode=1|2|3> [force=0|1] [verbose=0|1] [cfcache=0|1]

      mode   : 1 = images only, 2 = videos only, 3 = both images and videos
      force  : 0 = no force (default), 1 = overwrite existing thumbnails
      verbose: 0 = only print missing thumbs (default), 1 = print all
      cfcache: 0 = do not clear cloudflare cache (default), 1 = clear cloudflare cache
    `))

  const uploadsDir = path.resolve(config.uploads.folder)
  const thumbsDir = path.join(uploadsDir, 'thumbs')
  const _uploads = await thumbs.getFiles(uploadsDir)

  let _thumbs = await thumbs.getFiles(thumbsDir)
  _thumbs = _thumbs.map(_thumb => {
    const extname = path.extname(_thumb)
    return _thumb.slice(0, -extname.length)
  })

  const succeeded = []
  let error = 0
  let skipped = 0
  await new Promise((resolve, reject) => {
    const generate = async i => {
      const _upload = _uploads[i]
      if (!_upload) return resolve()

      const extname = path.extname(_upload)
      const basename = _upload.slice(0, -extname.length)

      if (_thumbs.includes(basename) && !thumbs.force) {
        if (thumbs.verbose) console.log(`${_upload}: thumb exists.`)
        skipped++
      } else if (!thumbs.mayGenerateThumb(extname)) {
        if (thumbs.verbose) console.log(`${_upload}: extension skipped.`)
        skipped++
      } else {
        const start = Date.now()
        const generated = await utils.generateThumbs(_upload, thumbs.force)
        console.log(`${_upload}: ${(Date.now() - start) / 1000}s: ${generated ? 'OK' : 'ERROR'}`)
        generated ? succeeded.push(_upload) : error++
      }
      return generate(i + 1)
    }
    return generate(0)
  })
  console.log(`Success: ${succeeded.length}\nError: ${error}\nSkipped: ${skipped}`)

  if (thumbs.cfcache && succeeded.length) {
    console.log('Purging Cloudflare\'s cache...')
    const results = await utils.purgeCloudflareCache(succeeded.map(name => {
      const extname = utils.extname(name)
      return `thumbs/${name.slice(0, -extname.length)}.png`
    }), true, false)
    for (let i = 0; i < results.length; i++) {
      if (results[i].errors.length)
        results[i].errors.forEach(error => console.error(`CF: ${error}`))
      console.log(`Status [${i}]: ${results[i].success ? 'OK' : 'ERROR'}`)
    }
  }
}

thumbs.do()