import { remove } from 'fs-extra' import { join } from 'path' import { AfterDestroy, AllowNull, BelongsTo, Column, CreatedAt, Default, ForeignKey, Is, Model, Table, UpdatedAt } from 'sequelize-typescript' import { MActorImage, MActorImageFormattable } from '@server/types/models' import { getLowercaseExtension } from '@shared/core-utils' import { ActivityIconObject, ActorImageType } from '@shared/models' import { AttributesOnly } from '@shared/typescript-utils' import { ActorImage } from '../../../shared/models/actors/actor-image.model' import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc' import { logger } from '../../helpers/logger' import { CONFIG } from '../../initializers/config' import { LAZY_STATIC_PATHS, MIMETYPES, WEBSERVER } from '../../initializers/constants' import { buildSQLAttributes, throwIfNotValid } from '../shared' import { ActorModel } from './actor' @Table({ tableName: 'actorImage', indexes: [ { fields: [ 'filename' ], unique: true }, { fields: [ 'actorId', 'type', 'width' ], unique: true } ] }) export class ActorImageModel extends Model>> { @AllowNull(false) @Column filename: string @AllowNull(true) @Default(null) @Column height: number @AllowNull(true) @Default(null) @Column width: number @AllowNull(true) @Is('ActorImageFileUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'fileUrl', true)) @Column fileUrl: string @AllowNull(false) @Column onDisk: boolean @AllowNull(false) @Column type: ActorImageType @CreatedAt createdAt: Date @UpdatedAt updatedAt: Date @ForeignKey(() => ActorModel) @Column actorId: number @BelongsTo(() => ActorModel, { foreignKey: { allowNull: false }, onDelete: 'CASCADE' }) Actor: ActorModel @AfterDestroy static removeFilesAndSendDelete (instance: ActorImageModel) { logger.info('Removing actor image file %s.', instance.filename) // Don't block the transaction instance.removeImage() .catch(err => logger.error('Cannot remove actor image file %s.', instance.filename, { err })) } // --------------------------------------------------------------------------- static getSQLAttributes (tableName: string, aliasPrefix = '') { return buildSQLAttributes({ model: this, tableName, aliasPrefix }) } // --------------------------------------------------------------------------- static loadByName (filename: string) { const query = { where: { filename } } return ActorImageModel.findOne(query) } static getImageUrl (image: MActorImage) { if (!image) return undefined return WEBSERVER.URL + image.getStaticPath() } toFormattedJSON (this: MActorImageFormattable): ActorImage { return { width: this.width, path: this.getStaticPath(), createdAt: this.createdAt, updatedAt: this.updatedAt } } toActivityPubObject (): ActivityIconObject { const extension = getLowercaseExtension(this.filename) return { type: 'Image', mediaType: MIMETYPES.IMAGE.EXT_MIMETYPE[extension], height: this.height, width: this.width, url: ActorImageModel.getImageUrl(this) } } getStaticPath () { switch (this.type) { case ActorImageType.AVATAR: return join(LAZY_STATIC_PATHS.AVATARS, this.filename) case ActorImageType.BANNER: return join(LAZY_STATIC_PATHS.BANNERS, this.filename) default: throw new Error('Unknown actor image type: ' + this.type) } } getPath () { return join(CONFIG.STORAGE.ACTOR_IMAGES, this.filename) } removeImage () { const imagePath = join(CONFIG.STORAGE.ACTOR_IMAGES, this.filename) return remove(imagePath) } isOwned () { return !this.fileUrl } }