mirror of https://github.com/Chocobozzz/PeerTube
82 lines
1.9 KiB
TypeScript
82 lines
1.9 KiB
TypeScript
import { join } from 'path'
|
|
import { AfterDestroy, AllowNull, Column, CreatedAt, Is, Model, Table, UpdatedAt } from 'sequelize-typescript'
|
|
import { Avatar } from '../../../shared/models/avatars/avatar.model'
|
|
import { LAZY_STATIC_PATHS } from '../../initializers/constants'
|
|
import { logger } from '../../helpers/logger'
|
|
import { remove } from 'fs-extra'
|
|
import { CONFIG } from '../../initializers/config'
|
|
import { throwIfNotValid } from '../utils'
|
|
import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc'
|
|
import { MAvatarFormattable } from '@server/types/models'
|
|
|
|
@Table({
|
|
tableName: 'avatar',
|
|
indexes: [
|
|
{
|
|
fields: [ 'filename' ],
|
|
unique: true
|
|
}
|
|
]
|
|
})
|
|
export class AvatarModel extends Model {
|
|
|
|
@AllowNull(false)
|
|
@Column
|
|
filename: string
|
|
|
|
@AllowNull(true)
|
|
@Is('AvatarFileUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'fileUrl', true))
|
|
@Column
|
|
fileUrl: string
|
|
|
|
@AllowNull(false)
|
|
@Column
|
|
onDisk: boolean
|
|
|
|
@CreatedAt
|
|
createdAt: Date
|
|
|
|
@UpdatedAt
|
|
updatedAt: Date
|
|
|
|
@AfterDestroy
|
|
static removeFilesAndSendDelete (instance: AvatarModel) {
|
|
logger.info('Removing avatar file %s.', instance.filename)
|
|
|
|
// Don't block the transaction
|
|
instance.removeAvatar()
|
|
.catch(err => logger.error('Cannot remove avatar file %s.', instance.filename, err))
|
|
}
|
|
|
|
static loadByName (filename: string) {
|
|
const query = {
|
|
where: {
|
|
filename
|
|
}
|
|
}
|
|
|
|
return AvatarModel.findOne(query)
|
|
}
|
|
|
|
toFormattedJSON (this: MAvatarFormattable): Avatar {
|
|
return {
|
|
path: this.getStaticPath(),
|
|
createdAt: this.createdAt,
|
|
updatedAt: this.updatedAt
|
|
}
|
|
}
|
|
|
|
getStaticPath () {
|
|
return join(LAZY_STATIC_PATHS.AVATARS, this.filename)
|
|
}
|
|
|
|
getPath () {
|
|
return join(CONFIG.STORAGE.AVATARS_DIR, this.filename)
|
|
}
|
|
|
|
removeAvatar () {
|
|
const avatarPath = join(CONFIG.STORAGE.AVATARS_DIR, this.filename)
|
|
return remove(avatarPath)
|
|
}
|
|
}
|