]>
Commit | Line | Data |
---|---|---|
1 | import { remove } from 'fs-extra' | |
2 | import { join } from 'path' | |
3 | import { | |
4 | AfterDestroy, | |
5 | AllowNull, | |
6 | BelongsTo, | |
7 | Column, | |
8 | CreatedAt, | |
9 | Default, | |
10 | ForeignKey, | |
11 | Is, | |
12 | Model, | |
13 | Table, | |
14 | UpdatedAt | |
15 | } from 'sequelize-typescript' | |
16 | import { MActorImage, MActorImageFormattable } from '@server/types/models' | |
17 | import { getLowercaseExtension } from '@shared/core-utils' | |
18 | import { ActivityIconObject, ActorImageType } from '@shared/models' | |
19 | import { AttributesOnly } from '@shared/typescript-utils' | |
20 | import { ActorImage } from '../../../shared/models/actors/actor-image.model' | |
21 | import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc' | |
22 | import { logger } from '../../helpers/logger' | |
23 | import { CONFIG } from '../../initializers/config' | |
24 | import { LAZY_STATIC_PATHS, MIMETYPES, WEBSERVER } from '../../initializers/constants' | |
25 | import { throwIfNotValid } from '../utils' | |
26 | import { ActorModel } from './actor' | |
27 | ||
28 | @Table({ | |
29 | tableName: 'actorImage', | |
30 | indexes: [ | |
31 | { | |
32 | fields: [ 'filename' ], | |
33 | unique: true | |
34 | }, | |
35 | { | |
36 | fields: [ 'actorId', 'type', 'width' ], | |
37 | unique: true | |
38 | } | |
39 | ] | |
40 | }) | |
41 | export class ActorImageModel extends Model<Partial<AttributesOnly<ActorImageModel>>> { | |
42 | ||
43 | @AllowNull(false) | |
44 | @Column | |
45 | filename: string | |
46 | ||
47 | @AllowNull(true) | |
48 | @Default(null) | |
49 | @Column | |
50 | height: number | |
51 | ||
52 | @AllowNull(true) | |
53 | @Default(null) | |
54 | @Column | |
55 | width: number | |
56 | ||
57 | @AllowNull(true) | |
58 | @Is('ActorImageFileUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'fileUrl', true)) | |
59 | @Column | |
60 | fileUrl: string | |
61 | ||
62 | @AllowNull(false) | |
63 | @Column | |
64 | onDisk: boolean | |
65 | ||
66 | @AllowNull(false) | |
67 | @Column | |
68 | type: ActorImageType | |
69 | ||
70 | @CreatedAt | |
71 | createdAt: Date | |
72 | ||
73 | @UpdatedAt | |
74 | updatedAt: Date | |
75 | ||
76 | @ForeignKey(() => ActorModel) | |
77 | @Column | |
78 | actorId: number | |
79 | ||
80 | @BelongsTo(() => ActorModel, { | |
81 | foreignKey: { | |
82 | allowNull: false | |
83 | }, | |
84 | onDelete: 'CASCADE' | |
85 | }) | |
86 | Actor: ActorModel | |
87 | ||
88 | @AfterDestroy | |
89 | static removeFilesAndSendDelete (instance: ActorImageModel) { | |
90 | logger.info('Removing actor image file %s.', instance.filename) | |
91 | ||
92 | // Don't block the transaction | |
93 | instance.removeImage() | |
94 | .catch(err => logger.error('Cannot remove actor image file %s.', instance.filename, err)) | |
95 | } | |
96 | ||
97 | static loadByName (filename: string) { | |
98 | const query = { | |
99 | where: { | |
100 | filename | |
101 | } | |
102 | } | |
103 | ||
104 | return ActorImageModel.findOne(query) | |
105 | } | |
106 | ||
107 | static getImageUrl (image: MActorImage) { | |
108 | if (!image) return undefined | |
109 | ||
110 | return WEBSERVER.URL + image.getStaticPath() | |
111 | } | |
112 | ||
113 | toFormattedJSON (this: MActorImageFormattable): ActorImage { | |
114 | return { | |
115 | width: this.width, | |
116 | path: this.getStaticPath(), | |
117 | createdAt: this.createdAt, | |
118 | updatedAt: this.updatedAt | |
119 | } | |
120 | } | |
121 | ||
122 | toActivityPubObject (): ActivityIconObject { | |
123 | const extension = getLowercaseExtension(this.filename) | |
124 | ||
125 | return { | |
126 | type: 'Image', | |
127 | mediaType: MIMETYPES.IMAGE.EXT_MIMETYPE[extension], | |
128 | height: this.height, | |
129 | width: this.width, | |
130 | url: ActorImageModel.getImageUrl(this) | |
131 | } | |
132 | } | |
133 | ||
134 | getStaticPath () { | |
135 | switch (this.type) { | |
136 | case ActorImageType.AVATAR: | |
137 | return join(LAZY_STATIC_PATHS.AVATARS, this.filename) | |
138 | ||
139 | case ActorImageType.BANNER: | |
140 | return join(LAZY_STATIC_PATHS.BANNERS, this.filename) | |
141 | } | |
142 | } | |
143 | ||
144 | getPath () { | |
145 | return join(CONFIG.STORAGE.ACTOR_IMAGES, this.filename) | |
146 | } | |
147 | ||
148 | removeImage () { | |
149 | const imagePath = join(CONFIG.STORAGE.ACTOR_IMAGES, this.filename) | |
150 | return remove(imagePath) | |
151 | } | |
152 | ||
153 | isOwned () { | |
154 | return !this.fileUrl | |
155 | } | |
156 | } |