]>
Commit | Line | Data |
---|---|---|
1 | import { minBy } from 'lodash' | |
2 | import { join } from 'path' | |
3 | import { getImageSize, processImage } from '@server/helpers/image-utils' | |
4 | import { CONFIG } from '@server/initializers/config' | |
5 | import { ACTOR_IMAGES_SIZE } from '@server/initializers/constants' | |
6 | import { updateActorImages } from '@server/lib/activitypub/actors' | |
7 | import { sendUpdateActor } from '@server/lib/activitypub/send' | |
8 | import { getBiggestActorImage } from '@server/lib/actor-image' | |
9 | import { JobQueue } from '@server/lib/job-queue' | |
10 | import { AccountModel } from '@server/models/account/account' | |
11 | import { ActorModel } from '@server/models/actor/actor' | |
12 | import { VideoChannelModel } from '@server/models/video/video-channel' | |
13 | import { MAccountDefault, MActorDefault, MChannelDefault } from '@server/types/models' | |
14 | import { getLowercaseExtension } from '@shared/core-utils' | |
15 | import { buildUUID } from '@shared/extra-utils' | |
16 | import { ActorImageType } from '@shared/models' | |
17 | import { initDatabaseModels } from '../../server/initializers/database' | |
18 | ||
19 | run() | |
20 | .then(() => process.exit(0)) | |
21 | .catch(err => { | |
22 | console.error(err) | |
23 | process.exit(-1) | |
24 | }) | |
25 | ||
26 | async function run () { | |
27 | console.log('Generate avatar miniatures from existing avatars.') | |
28 | ||
29 | await initDatabaseModels(true) | |
30 | JobQueue.Instance.init(true) | |
31 | ||
32 | const accounts: AccountModel[] = await AccountModel.findAll({ | |
33 | include: [ | |
34 | { | |
35 | model: ActorModel, | |
36 | required: true, | |
37 | where: { | |
38 | serverId: null | |
39 | } | |
40 | }, | |
41 | { | |
42 | model: VideoChannelModel, | |
43 | include: [ | |
44 | { | |
45 | model: AccountModel | |
46 | } | |
47 | ] | |
48 | } | |
49 | ] | |
50 | }) | |
51 | ||
52 | for (const account of accounts) { | |
53 | try { | |
54 | await fillAvatarSizeIfNeeded(account) | |
55 | await generateSmallerAvatarIfNeeded(account) | |
56 | } catch (err) { | |
57 | console.error(`Cannot process account avatar ${account.name}`, err) | |
58 | } | |
59 | ||
60 | for (const videoChannel of account.VideoChannels) { | |
61 | try { | |
62 | await fillAvatarSizeIfNeeded(videoChannel) | |
63 | await generateSmallerAvatarIfNeeded(videoChannel) | |
64 | } catch (err) { | |
65 | console.error(`Cannot process channel avatar ${videoChannel.name}`, err) | |
66 | } | |
67 | } | |
68 | } | |
69 | ||
70 | console.log('Generation finished!') | |
71 | } | |
72 | ||
73 | async function fillAvatarSizeIfNeeded (accountOrChannel: MAccountDefault | MChannelDefault) { | |
74 | const avatars = accountOrChannel.Actor.Avatars | |
75 | ||
76 | for (const avatar of avatars) { | |
77 | if (avatar.width && avatar.height) continue | |
78 | ||
79 | console.log('Filling size of avatars of %s.', accountOrChannel.name) | |
80 | ||
81 | const { width, height } = await getImageSize(join(CONFIG.STORAGE.ACTOR_IMAGES, avatar.filename)) | |
82 | avatar.width = width | |
83 | avatar.height = height | |
84 | ||
85 | await avatar.save() | |
86 | } | |
87 | } | |
88 | ||
89 | async function generateSmallerAvatarIfNeeded (accountOrChannel: MAccountDefault | MChannelDefault) { | |
90 | const avatars = accountOrChannel.Actor.Avatars | |
91 | if (avatars.length !== 1) { | |
92 | return | |
93 | } | |
94 | ||
95 | console.log(`Processing ${accountOrChannel.name}.`) | |
96 | ||
97 | await generateSmallerAvatar(accountOrChannel.Actor) | |
98 | accountOrChannel.Actor = Object.assign(accountOrChannel.Actor, { Server: null }) | |
99 | ||
100 | return sendUpdateActor(accountOrChannel, undefined) | |
101 | } | |
102 | ||
103 | async function generateSmallerAvatar (actor: MActorDefault) { | |
104 | const bigAvatar = getBiggestActorImage(actor.Avatars) | |
105 | ||
106 | const imageSize = minBy(ACTOR_IMAGES_SIZE[ActorImageType.AVATAR], 'width') | |
107 | const sourceFilename = bigAvatar.filename | |
108 | ||
109 | const newImageName = buildUUID() + getLowercaseExtension(sourceFilename) | |
110 | const source = join(CONFIG.STORAGE.ACTOR_IMAGES, sourceFilename) | |
111 | const destination = join(CONFIG.STORAGE.ACTOR_IMAGES, newImageName) | |
112 | ||
113 | await processImage({ path: source, destination, newSize: imageSize, keepOriginal: true }) | |
114 | ||
115 | const actorImageInfo = { | |
116 | name: newImageName, | |
117 | fileUrl: null, | |
118 | height: imageSize.height, | |
119 | width: imageSize.width, | |
120 | onDisk: true | |
121 | } | |
122 | ||
123 | await updateActorImages(actor, ActorImageType.AVATAR, [ actorImageInfo ], undefined) | |
124 | } |