server/lib/avatar.ts

   1 import 'multer'
   2 import { sendUpdateActor } from './activitypub/send'
   3 import { AVATARS_SIZE } from '../initializers/constants'
   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 import * as uuidv4 from 'uuid/v4'
  11 import { CONFIG } from '../initializers/config'
  12 import { sequelizeTypescript } from '../initializers/database'
  13
  14 async function updateActorAvatarFile (avatarPhysicalFile: Express.Multer.File, accountOrChannel: AccountModel | VideoChannelModel) {
  15   const extension = extname(avatarPhysicalFile.filename)
  16   const avatarName = uuidv4() + extension
  17   const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName)
  18   await processImage(avatarPhysicalFile.path, destination, AVATARS_SIZE)
  19
  20   return retryTransactionWrapper(() => {
  21     return sequelizeTypescript.transaction(async t => {
  22       const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarName, t)
  23       await updatedActor.save({ transaction: t })
  24
  25       await sendUpdateActor(accountOrChannel, t)
  26
  27       return updatedActor.Avatar
  28     })
  29   })
  30 }
  31
  32 export {
  33   updateActorAvatarFile
  34 }