]>
Commit | Line | Data |
---|---|---|
4bbfc6c6 | 1 | import 'multer' |
4bbfc6c6 | 2 | import { sendUpdateActor } from './activitypub/send' |
557b13ae | 3 | import { AVATARS_SIZE, LRU_CACHE, QUEUE_CONCURRENCY } from '../initializers/constants' |
1ea7da81 | 4 | import { updateActorAvatarInstance, deleteActorAvatarInstance } from './activitypub/actor' |
4bbfc6c6 | 5 | import { processImage } from '../helpers/image-utils' |
4bbfc6c6 | 6 | import { extname, join } from 'path' |
4a534352 | 7 | import { retryTransactionWrapper } from '../helpers/database-utils' |
bdd428a6 | 8 | import { v4 as uuidv4 } from 'uuid' |
6dd9de95 | 9 | import { CONFIG } from '../initializers/config' |
74dc3bca | 10 | import { sequelizeTypescript } from '../initializers/database' |
557b13ae C |
11 | import * as LRUCache from 'lru-cache' |
12 | import { queue } from 'async' | |
13 | import { downloadImage } from '../helpers/requests' | |
26d6bf65 | 14 | import { MAccountDefault, MChannelDefault } from '../types/models' |
4bbfc6c6 | 15 | |
e08ff02a | 16 | async function updateLocalActorAvatarFile ( |
1ea7da81 RK |
17 | accountOrChannel: MAccountDefault | MChannelDefault, |
18 | avatarPhysicalFile: Express.Multer.File | |
453e83ea | 19 | ) { |
4bbfc6c6 | 20 | const extension = extname(avatarPhysicalFile.filename) |
e08ff02a | 21 | |
ecf3f060 | 22 | const avatarName = uuidv4() + extension |
4bbfc6c6 | 23 | const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName) |
2fb5b3a5 | 24 | await processImage(avatarPhysicalFile.path, destination, AVATARS_SIZE) |
4bbfc6c6 | 25 | |
4a534352 C |
26 | return retryTransactionWrapper(() => { |
27 | return sequelizeTypescript.transaction(async t => { | |
557b13ae C |
28 | const avatarInfo = { |
29 | name: avatarName, | |
30 | fileUrl: null, | |
31 | onDisk: true | |
32 | } | |
33 | ||
34 | const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarInfo, t) | |
4a534352 | 35 | await updatedActor.save({ transaction: t }) |
4bbfc6c6 | 36 | |
4a534352 | 37 | await sendUpdateActor(accountOrChannel, t) |
4bbfc6c6 | 38 | |
4a534352 C |
39 | return updatedActor.Avatar |
40 | }) | |
4bbfc6c6 C |
41 | }) |
42 | } | |
43 | ||
e08ff02a | 44 | async function deleteLocalActorAvatarFile ( |
1ea7da81 RK |
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 | ||
557b13ae C |
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) { | |
ba5a8d89 | 68 | return new Promise<void>((res, rej) => { |
557b13ae C |
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 | ||
4bbfc6c6 | 80 | export { |
557b13ae | 81 | avatarPathUnsafeCache, |
e08ff02a C |
82 | updateLocalActorAvatarFile, |
83 | deleteLocalActorAvatarFile, | |
557b13ae | 84 | pushAvatarProcessInQueue |
4bbfc6c6 | 85 | } |