]>
Commit | Line | Data |
---|---|---|
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 | } |