server/lib/avatar.ts

   1 import 'multer'
   2 import { sendUpdateActor } from './activitypub/send'
   3 import { AVATARS_SIZE, LRU_CACHE, QUEUE_CONCURRENCY } from '../initializers/constants'
   4 import { updateActorAvatarInstance, deleteActorAvatarInstance } from './activitypub/actor'
   5 import { processImage } from '../helpers/image-utils'
   6 import { extname, join } from 'path'
   7 import { retryTransactionWrapper } from '../helpers/database-utils'
   8 import { v4 as uuidv4 } from 'uuid'
   9 import { CONFIG } from '../initializers/config'
  10 import { sequelizeTypescript } from '../initializers/database'
  11 import * as LRUCache from 'lru-cache'
  12 import { queue } from 'async'
  13 import { downloadImage } from '../helpers/requests'
  14 import { MAccountDefault, MChannelDefault } from '../types/models'
  15
  16 async function updateLocalActorAvatarFile (
  17   accountOrChannel: MAccountDefault | MChannelDefault,
  18   avatarPhysicalFile: Express.Multer.File
  19 ) {
  20   const extension = extname(avatarPhysicalFile.filename)
  21
  22   const avatarName = uuidv4() + extension
  23   const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName)
  24   await processImage(avatarPhysicalFile.path, destination, AVATARS_SIZE)
  25
  26   return retryTransactionWrapper(() => {
  27     return sequelizeTypescript.transaction(async t => {
  28       const avatarInfo = {
  29         name: avatarName,
  30         fileUrl: null,
  31         onDisk: true
  32       }
  33
  34       const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarInfo, t)
  35       await updatedActor.save({ transaction: t })
  36
  37       await sendUpdateActor(accountOrChannel, t)
  38
  39       return updatedActor.Avatar
  40     })
  41   })
  42 }
  43
  44 async function deleteLocalActorAvatarFile (
  45   accountOrChannel: MAccountDefault | MChannelDefault
  46 ) {
  47   return retryTransactionWrapper(() => {
  48     return sequelizeTypescript.transaction(async t => {
  49       const updatedActor = await deleteActorAvatarInstance(accountOrChannel.Actor, t)
  50       await updatedActor.save({ transaction: t })
  51
  52       await sendUpdateActor(accountOrChannel, t)
  53
  54       return updatedActor.Avatar
  55     })
  56   })
  57 }
  58
  59 type DownloadImageQueueTask = { fileUrl: string, filename: string }
  60
  61 const downloadImageQueue = queue<DownloadImageQueueTask, Error>((task, cb) => {
  62   downloadImage(task.fileUrl, CONFIG.STORAGE.AVATARS_DIR, task.filename, AVATARS_SIZE)
  63     .then(() => cb())
  64     .catch(err => cb(err))
  65 }, QUEUE_CONCURRENCY.AVATAR_PROCESS_IMAGE)
  66
  67 function pushAvatarProcessInQueue (task: DownloadImageQueueTask) {
  68   return new Promise<void>((res, rej) => {
  69     downloadImageQueue.push(task, err => {
  70       if (err) return rej(err)
  71
  72       return res()
  73     })
  74   })
  75 }
  76
  77 // Unsafe so could returns paths that does not exist anymore
  78 const avatarPathUnsafeCache = new LRUCache<string, string>({ max: LRU_CACHE.AVATAR_STATIC.MAX_SIZE })
  79
  80 export {
  81   avatarPathUnsafeCache,
  82   updateLocalActorAvatarFile,
  83   deleteLocalActorAvatarFile,
  84   pushAvatarProcessInQueue
  85 }