]>
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' |
4bbfc6c6 C |
4 | import { updateActorAvatarInstance } from './activitypub' |
5 | import { processImage } from '../helpers/image-utils' | |
4bbfc6c6 | 6 | import { extname, join } from 'path' |
4a534352 | 7 | import { retryTransactionWrapper } from '../helpers/database-utils' |
ecf3f060 | 8 | import * as uuidv4 from 'uuid/v4' |
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' | |
0283eaac | 14 | import { MAccountDefault, MChannelDefault } from '../typings/models' |
4bbfc6c6 | 15 | |
453e83ea C |
16 | async function updateActorAvatarFile ( |
17 | avatarPhysicalFile: Express.Multer.File, | |
0283eaac | 18 | accountOrChannel: MAccountDefault | MChannelDefault |
453e83ea | 19 | ) { |
4bbfc6c6 | 20 | const extension = extname(avatarPhysicalFile.filename) |
ecf3f060 | 21 | const avatarName = uuidv4() + extension |
4bbfc6c6 | 22 | const destination = join(CONFIG.STORAGE.AVATARS_DIR, avatarName) |
2fb5b3a5 | 23 | await processImage(avatarPhysicalFile.path, destination, AVATARS_SIZE) |
4bbfc6c6 | 24 | |
4a534352 C |
25 | return retryTransactionWrapper(() => { |
26 | return sequelizeTypescript.transaction(async t => { | |
557b13ae C |
27 | const avatarInfo = { |
28 | name: avatarName, | |
29 | fileUrl: null, | |
30 | onDisk: true | |
31 | } | |
32 | ||
33 | const updatedActor = await updateActorAvatarInstance(accountOrChannel.Actor, avatarInfo, t) | |
4a534352 | 34 | await updatedActor.save({ transaction: t }) |
4bbfc6c6 | 35 | |
4a534352 | 36 | await sendUpdateActor(accountOrChannel, t) |
4bbfc6c6 | 37 | |
4a534352 C |
38 | return updatedActor.Avatar |
39 | }) | |
4bbfc6c6 C |
40 | }) |
41 | } | |
42 | ||
557b13ae C |
43 | type DownloadImageQueueTask = { fileUrl: string, filename: string } |
44 | ||
45 | const downloadImageQueue = queue<DownloadImageQueueTask, Error>((task, cb) => { | |
46 | downloadImage(task.fileUrl, CONFIG.STORAGE.AVATARS_DIR, task.filename, AVATARS_SIZE) | |
47 | .then(() => cb()) | |
48 | .catch(err => cb(err)) | |
49 | }, QUEUE_CONCURRENCY.AVATAR_PROCESS_IMAGE) | |
50 | ||
51 | function pushAvatarProcessInQueue (task: DownloadImageQueueTask) { | |
52 | return new Promise((res, rej) => { | |
53 | downloadImageQueue.push(task, err => { | |
54 | if (err) return rej(err) | |
55 | ||
56 | return res() | |
57 | }) | |
58 | }) | |
59 | } | |
60 | ||
61 | // Unsafe so could returns paths that does not exist anymore | |
62 | const avatarPathUnsafeCache = new LRUCache<string, string>({ max: LRU_CACHE.AVATAR_STATIC.MAX_SIZE }) | |
63 | ||
4bbfc6c6 | 64 | export { |
557b13ae C |
65 | avatarPathUnsafeCache, |
66 | updateActorAvatarFile, | |
67 | pushAvatarProcessInQueue | |
4bbfc6c6 | 68 | } |