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'
16 async function updateActorAvatarFile (
17 accountOrChannel: MAccountDefault | MChannelDefault,
18 avatarPhysicalFile: Express.Multer.File
20 const extension = extname(avatarPhysicalFile.filename)
21 const avatarName = uuidv4() + extension
22 const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName)
23 await processImage(avatarPhysicalFile.path, destination, AVATARS_SIZE)
25 return retryTransactionWrapper(() => {
26 return sequelizeTypescript.transaction(async t => {
33 const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarInfo, t)
34 await updatedActor.save({ transaction: t })
36 await sendUpdateActor(accountOrChannel, t)
38 return updatedActor.Avatar
43 async function deleteActorAvatarFile (
44 accountOrChannel: MAccountDefault | MChannelDefault
46 return retryTransactionWrapper(() => {
47 return sequelizeTypescript.transaction(async t => {
48 const updatedActor = await deleteActorAvatarInstance(accountOrChannel.Actor, t)
49 await updatedActor.save({ transaction: t })
51 await sendUpdateActor(accountOrChannel, t)
53 return updatedActor.Avatar
58 type DownloadImageQueueTask = { fileUrl: string, filename: string }
60 const downloadImageQueue = queue<DownloadImageQueueTask, Error>((task, cb) => {
61 downloadImage(task.fileUrl, CONFIG.STORAGE.AVATARS_DIR, task.filename, AVATARS_SIZE)
63 .catch(err => cb(err))
64 }, QUEUE_CONCURRENCY.AVATAR_PROCESS_IMAGE)
66 function pushAvatarProcessInQueue (task: DownloadImageQueueTask) {
67 return new Promise((res, rej) => {
68 downloadImageQueue.push(task, err => {
69 if (err) return rej(err)
76 // Unsafe so could returns paths that does not exist anymore
77 const avatarPathUnsafeCache = new LRUCache<string, string>({ max: LRU_CACHE.AVATAR_STATIC.MAX_SIZE })
80 avatarPathUnsafeCache,
81 updateActorAvatarFile,
82 deleteActorAvatarFile,
83 pushAvatarProcessInQueue