143 lines
3.3 KiB
TypeScript
143 lines
3.3 KiB
TypeScript
import * as Sequelize from 'sequelize'
|
|
|
|
import { CONFIG } from '../../initializers'
|
|
import { isVideoAbuseReasonValid } from '../../helpers'
|
|
|
|
import { addMethodsToModel, getSort } from '../utils'
|
|
import {
|
|
VideoAbuseInstance,
|
|
VideoAbuseAttributes,
|
|
|
|
VideoAbuseMethods
|
|
} from './video-abuse-interface'
|
|
import { VideoAbuseObject } from '../../../shared/models/activitypub/objects/video-abuse-object'
|
|
|
|
let VideoAbuse: Sequelize.Model<VideoAbuseInstance, VideoAbuseAttributes>
|
|
let toFormattedJSON: VideoAbuseMethods.ToFormattedJSON
|
|
let listForApi: VideoAbuseMethods.ListForApi
|
|
let toActivityPubObject: VideoAbuseMethods.ToActivityPubObject
|
|
|
|
export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
|
|
VideoAbuse = sequelize.define<VideoAbuseInstance, VideoAbuseAttributes>('VideoAbuse',
|
|
{
|
|
reason: {
|
|
type: DataTypes.STRING,
|
|
allowNull: false,
|
|
validate: {
|
|
reasonValid: value => {
|
|
const res = isVideoAbuseReasonValid(value)
|
|
if (res === false) throw new Error('Video abuse reason is not valid.')
|
|
}
|
|
}
|
|
}
|
|
},
|
|
{
|
|
indexes: [
|
|
{
|
|
fields: [ 'videoId' ]
|
|
},
|
|
{
|
|
fields: [ 'reporterAccountId' ]
|
|
}
|
|
]
|
|
}
|
|
)
|
|
|
|
const classMethods = [
|
|
associate,
|
|
|
|
listForApi
|
|
]
|
|
const instanceMethods = [
|
|
toFormattedJSON,
|
|
toActivityPubObject
|
|
]
|
|
addMethodsToModel(VideoAbuse, classMethods, instanceMethods)
|
|
|
|
return VideoAbuse
|
|
}
|
|
|
|
// ------------------------------ METHODS ------------------------------
|
|
|
|
toFormattedJSON = function (this: VideoAbuseInstance) {
|
|
let reporterServerHost
|
|
|
|
if (this.Account.Server) {
|
|
reporterServerHost = this.Account.Server.host
|
|
} else {
|
|
// It means it's our video
|
|
reporterServerHost = CONFIG.WEBSERVER.HOST
|
|
}
|
|
|
|
const json = {
|
|
id: this.id,
|
|
reason: this.reason,
|
|
reporterUsername: this.Account.name,
|
|
reporterServerHost,
|
|
videoId: this.Video.id,
|
|
videoUUID: this.Video.uuid,
|
|
videoName: this.Video.name,
|
|
createdAt: this.createdAt
|
|
}
|
|
|
|
return json
|
|
}
|
|
|
|
toActivityPubObject = function (this: VideoAbuseInstance) {
|
|
const videoAbuseObject: VideoAbuseObject = {
|
|
type: 'Flag' as 'Flag',
|
|
content: this.reason,
|
|
object: this.Video.url
|
|
}
|
|
|
|
return videoAbuseObject
|
|
}
|
|
|
|
// ------------------------------ STATICS ------------------------------
|
|
|
|
function associate (models) {
|
|
VideoAbuse.belongsTo(models.Account, {
|
|
foreignKey: {
|
|
name: 'reporterAccountId',
|
|
allowNull: true
|
|
},
|
|
onDelete: 'CASCADE'
|
|
})
|
|
|
|
VideoAbuse.belongsTo(models.Video, {
|
|
foreignKey: {
|
|
name: 'videoId',
|
|
allowNull: false
|
|
},
|
|
onDelete: 'CASCADE'
|
|
})
|
|
}
|
|
|
|
listForApi = function (start: number, count: number, sort: string) {
|
|
const query = {
|
|
offset: start,
|
|
limit: count,
|
|
order: [ getSort(sort) ],
|
|
include: [
|
|
{
|
|
model: VideoAbuse['sequelize'].models.Account,
|
|
required: true,
|
|
include: [
|
|
{
|
|
model: VideoAbuse['sequelize'].models.Server,
|
|
required: false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
model: VideoAbuse['sequelize'].models.Video,
|
|
required: true
|
|
}
|
|
]
|
|
}
|
|
|
|
return VideoAbuse.findAndCountAll(query).then(({ rows, count }) => {
|
|
return { total: count, data: rows }
|
|
})
|
|
}
|