]>
Commit | Line | Data |
---|---|---|
d0800f76 | 1 | import { remove } from 'fs-extra' |
41fb13c3 | 2 | import LRUCache from 'lru-cache' |
ea54cd04 | 3 | import { join } from 'path' |
0b6f5316 | 4 | import { Transaction } from 'sequelize/types' |
136d7efd | 5 | import { ActorModel } from '@server/models/actor/actor' |
0628157f C |
6 | import { getLowercaseExtension } from '@shared/core-utils' |
7 | import { buildUUID } from '@shared/extra-utils' | |
136d7efd | 8 | import { ActivityPubActorType, ActorImageType } from '@shared/models' |
f4796856 | 9 | import { retryTransactionWrapper } from '../helpers/database-utils' |
6dd9de95 | 10 | import { CONFIG } from '../initializers/config' |
c53853ca | 11 | import { ACTOR_IMAGES_SIZE, LRU_CACHE, WEBSERVER } from '../initializers/constants' |
74dc3bca | 12 | import { sequelizeTypescript } from '../initializers/database' |
136d7efd | 13 | import { MAccountDefault, MActor, MChannelDefault } from '../types/models' |
d0800f76 | 14 | import { deleteActorImages, updateActorImages } from './activitypub/actors' |
f4796856 | 15 | import { sendUpdateActor } from './activitypub/send' |
3a54605d | 16 | import { downloadImageFromWorker, processImageFromWorker } from './worker/parent-process' |
4bbfc6c6 | 17 | |
136d7efd C |
18 | function buildActorInstance (type: ActivityPubActorType, url: string, preferredUsername: string) { |
19 | return new ActorModel({ | |
20 | type, | |
21 | url, | |
22 | preferredUsername, | |
23 | publicKey: null, | |
24 | privateKey: null, | |
25 | followersCount: 0, | |
26 | followingCount: 0, | |
27 | inboxUrl: url + '/inbox', | |
28 | outboxUrl: url + '/outbox', | |
29 | sharedInboxUrl: WEBSERVER.URL + '/inbox', | |
30 | followersUrl: url + '/followers', | |
31 | followingUrl: url + '/following' | |
32 | }) as MActor | |
33 | } | |
34 | ||
d0800f76 | 35 | async function updateLocalActorImageFiles ( |
1ea7da81 | 36 | accountOrChannel: MAccountDefault | MChannelDefault, |
2cb03dc1 C |
37 | imagePhysicalFile: Express.Multer.File, |
38 | type: ActorImageType | |
453e83ea | 39 | ) { |
d0800f76 | 40 | const processImageSize = async (imageSize: { width: number, height: number }) => { |
41 | const extension = getLowercaseExtension(imagePhysicalFile.filename) | |
42 | ||
43 | const imageName = buildUUID() + extension | |
44 | const destination = join(CONFIG.STORAGE.ACTOR_IMAGES, imageName) | |
3a54605d | 45 | await processImageFromWorker({ path: imagePhysicalFile.path, destination, newSize: imageSize, keepOriginal: true }) |
d0800f76 | 46 | |
47 | return { | |
48 | imageName, | |
49 | imageSize | |
50 | } | |
51 | } | |
52 | ||
53 | const processedImages = await Promise.all(ACTOR_IMAGES_SIZE[type].map(processImageSize)) | |
54 | await remove(imagePhysicalFile.path) | |
55 | ||
56 | return retryTransactionWrapper(() => sequelizeTypescript.transaction(async t => { | |
57 | const actorImagesInfo = processedImages.map(({ imageName, imageSize }) => ({ | |
58 | name: imageName, | |
59 | fileUrl: null, | |
60 | height: imageSize.height, | |
61 | width: imageSize.width, | |
62 | onDisk: true | |
63 | })) | |
64 | ||
65 | const updatedActor = await updateActorImages(accountOrChannel.Actor, type, actorImagesInfo, t) | |
66 | await updatedActor.save({ transaction: t }) | |
67 | ||
68 | await sendUpdateActor(accountOrChannel, t) | |
69 | ||
70 | return type === ActorImageType.AVATAR | |
71 | ? updatedActor.Avatars | |
72 | : updatedActor.Banners | |
73 | })) | |
4bbfc6c6 C |
74 | } |
75 | ||
2cb03dc1 | 76 | async function deleteLocalActorImageFile (accountOrChannel: MAccountDefault | MChannelDefault, type: ActorImageType) { |
1ea7da81 RK |
77 | return retryTransactionWrapper(() => { |
78 | return sequelizeTypescript.transaction(async t => { | |
d0800f76 | 79 | const updatedActor = await deleteActorImages(accountOrChannel.Actor, type, t) |
1ea7da81 RK |
80 | await updatedActor.save({ transaction: t }) |
81 | ||
82 | await sendUpdateActor(accountOrChannel, t) | |
83 | ||
d0800f76 | 84 | return updatedActor.Avatars |
1ea7da81 RK |
85 | }) |
86 | }) | |
87 | } | |
88 | ||
c53853ca C |
89 | // --------------------------------------------------------------------------- |
90 | ||
0b6f5316 C |
91 | async function findAvailableLocalActorName (baseActorName: string, transaction?: Transaction) { |
92 | let actor = await ActorModel.loadLocalByName(baseActorName, transaction) | |
93 | if (!actor) return baseActorName | |
94 | ||
95 | for (let i = 1; i < 30; i++) { | |
96 | const name = `${baseActorName}-${i}` | |
97 | ||
98 | actor = await ActorModel.loadLocalByName(name, transaction) | |
99 | if (!actor) return name | |
100 | } | |
101 | ||
102 | throw new Error('Cannot find available actor local name (too much iterations).') | |
103 | } | |
104 | ||
105 | // --------------------------------------------------------------------------- | |
106 | ||
c53853ca | 107 | function downloadActorImageFromWorker (options: { |
d0800f76 | 108 | fileUrl: string |
109 | filename: string | |
110 | type: ActorImageType | |
111 | size: typeof ACTOR_IMAGES_SIZE[ActorImageType][0] | |
c53853ca C |
112 | }) { |
113 | const downloaderOptions = { | |
114 | url: options.fileUrl, | |
115 | destDir: CONFIG.STORAGE.ACTOR_IMAGES, | |
116 | destName: options.filename, | |
117 | size: options.size | |
118 | } | |
557b13ae | 119 | |
c53853ca | 120 | return downloadImageFromWorker(downloaderOptions) |
557b13ae C |
121 | } |
122 | ||
123 | // Unsafe so could returns paths that does not exist anymore | |
f4796856 | 124 | const actorImagePathUnsafeCache = new LRUCache<string, string>({ max: LRU_CACHE.ACTOR_IMAGE_STATIC.MAX_SIZE }) |
557b13ae | 125 | |
4bbfc6c6 | 126 | export { |
f4796856 | 127 | actorImagePathUnsafeCache, |
d0800f76 | 128 | updateLocalActorImageFiles, |
0b6f5316 | 129 | findAvailableLocalActorName, |
c53853ca | 130 | downloadActorImageFromWorker, |
2cb03dc1 | 131 | deleteLocalActorImageFile, |
c53853ca | 132 | downloadImageFromWorker, |
136d7efd | 133 | buildActorInstance |
4bbfc6c6 | 134 | } |