filesafe/scripts/thumbs.js

const fs = require('fs')
const path = require('path')
const utils = require('./../controllers/utilsController')

const thumbs = {
  mode: null,
  force: 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.stat(path.join(directory, name), (error, stat) => {
            if (error) { return reject(error) }
            if (stat.isFile() && !name.startsWith('.')) { files.push(name) }
            resolve()
          })
        })
      }))
      resolve(files)
    })
  })
}

thumbs.do = async () => {
  const args = process.argv.slice(2)

  thumbs.mode = parseInt(args[0])
  thumbs.force = parseInt(args[1])
  if ((isNaN(thumbs.mode) || ![1, 2, 3].includes(thumbs.mode)) ||
  (!isNaN(thumbs.force) && ![0, 1].includes(thumbs.force))) {
    console.log('Usage:\nyarn thumbs <mode=1|2|3> [force=0|1]\n')
    console.log('mode : 1 = images only, 2 = videos only, 3 = both images and videos')
    console.log('force: 0 = no force (default), 1 = overwrite existing thumbnails')
    return
  }

  const uploadsDir = path.join(__dirname, '..', 'uploads')
  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)
  })

  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) {
        console.log(`${_upload}: thumb exists.`)
      } else if (!thumbs.mayGenerateThumb(extname)) {
        console.log(`${_upload}: extension skipped.`)
      } else {
        const generated = await utils.generateThumbs(_upload, thumbs.force)
        console.log(`${_upload}: ${String(generated)}`)
      }
      generate(i + 1)
    }
    generate(0)
  })
}

thumbs.do()