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 updateLocalActorAvatarFile (
17 accountOrChannel: MAccountDefault | MChannelDefault,
18 avatarPhysicalFile: Express.Multer.File
20 const extension = extname(avatarPhysicalFile.filename)
22 const avatarName = uuidv4() + extension
23 const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName)
24 await processImage(avatarPhysicalFile.path, destination, AVATARS_SIZE)
26 return retryTransactionWrapper(() => {
27 return sequelizeTypescript.transaction(async t => {
34 const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarInfo, t)
35 await updatedActor.save({ transaction: t })
37 await sendUpdateActor(accountOrChannel, t)
39 return updatedActor.Avatar
44 async function deleteLocalActorAvatarFile (
45 accountOrChannel: MAccountDefault | MChannelDefault
47 return retryTransactionWrapper(() => {
48 return sequelizeTypescript.transaction(async t => {
49 const updatedActor = await deleteActorAvatarInstance(accountOrChannel.Actor, t)
50 await updatedActor.save({ transaction: t })
52 await sendUpdateActor(accountOrChannel, t)
54 return updatedActor.Avatar
59 type DownloadImageQueueTask = { fileUrl: string, filename: string }
61 const downloadImageQueue = queue<DownloadImageQueueTask, Error>((task, cb) => {
62 downloadImage(task.fileUrl, CONFIG.STORAGE.AVATARS_DIR, task.filename, AVATARS_SIZE)
64 .catch(err => cb(err))
65 }, QUEUE_CONCURRENCY.AVATAR_PROCESS_IMAGE)
67 function pushAvatarProcessInQueue (task: DownloadImageQueueTask) {
68 return new Promise<void>((res, rej) => {
69 downloadImageQueue.push(task, err => {
70 if (err) return rej(err)
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 })
81 avatarPathUnsafeCache,
82 updateLocalActorAvatarFile,
83 deleteLocalActorAvatarFile,
84 pushAvatarProcessInQueue