server/lib/avatar.ts

   1 import 'multer'
   2 import { sendUpdateActor } from './activitypub/send'
   3 import { AVATARS_SIZE, CONFIG, sequelizeTypescript } from '../initializers'
   4 import { updateActorAvatarInstance } from './activitypub'
   5 import { processImage } from '../helpers/image-utils'
   6 import { AccountModel } from '../models/account/account'
   7 import { VideoChannelModel } from '../models/video/video-channel'
   8 import { extname, join } from 'path'
   9 import { retryTransactionWrapper } from '../helpers/database-utils'
  10
  11 async function updateActorAvatarFile (avatarPhysicalFile: Express.Multer.File, accountOrChannel: AccountModel | VideoChannelModel) {
  12   const extension = extname(avatarPhysicalFile.filename)
  13   const avatarName = accountOrChannel.Actor.uuid + extension
  14   const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName)
  15   await processImage(avatarPhysicalFile, destination, AVATARS_SIZE)
  16
  17   return retryTransactionWrapper(() => {
  18     return sequelizeTypescript.transaction(async t => {
  19       const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarName, t)
  20       await updatedActor.save({ transaction: t })
  21
  22       await sendUpdateActor(accountOrChannel, t)
  23
  24       return updatedActor.Avatar
  25     })
  26   })
  27 }
  28
  29 export {
  30   updateActorAvatarFile
  31 }