-import * as Sequelize from 'sequelize'
-
-import { CONFIG } from '../../initializers'
-import { isVideoAbuseReasonValid } from '../../helpers'
-
-import { addMethodsToModel, getSort } from '../utils'
+import { AllowNull, BelongsTo, Column, CreatedAt, DataType, Default, ForeignKey, Is, Model, Table, UpdatedAt } from 'sequelize-typescript'
+import { VideoAbuseObject } from '../../../shared/models/activitypub/objects'
+import { VideoAbuse } from '../../../shared/models/videos'
import {
- VideoAbuseInstance,
- VideoAbuseAttributes,
-
- VideoAbuseMethods
-} from './video-abuse-interface'
-
-let VideoAbuse: Sequelize.Model<VideoAbuseInstance, VideoAbuseAttributes>
-let toFormattedJSON: VideoAbuseMethods.ToFormattedJSON
-let listForApi: VideoAbuseMethods.ListForApi
-
-export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
- VideoAbuse = sequelize.define<VideoAbuseInstance, VideoAbuseAttributes>('VideoAbuse',
+ isVideoAbuseModerationCommentValid,
+ isVideoAbuseReasonValid,
+ isVideoAbuseStateValid
+} from '../../helpers/custom-validators/video-abuses'
+import { AccountModel } from '../account/account'
+import { getSort, throwIfNotValid } from '../utils'
+import { VideoModel } from './video'
+import { VideoAbuseState } from '../../../shared'
+import { CONSTRAINTS_FIELDS, VIDEO_ABUSE_STATES } from '../../initializers/constants'
+import { MVideoAbuse, MVideoAbuseFormattable, MVideoAbuseVideo } from '../../typings/models'
+import * as Bluebird from 'bluebird'
+
+@Table({
+ tableName: 'videoAbuse',
+ indexes: [
{
- 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.')
- }
- }
- }
+ fields: [ 'videoId' ]
},
{
- indexes: [
- {
- fields: [ 'videoId' ]
- },
- {
- fields: [ 'reporterAccountId' ]
- }
- ]
+ fields: [ 'reporterAccountId' ]
}
- )
-
- const classMethods = [
- associate,
-
- listForApi
- ]
- const instanceMethods = [
- toFormattedJSON
]
- 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
-}
-
-// ------------------------------ STATICS ------------------------------
-
-function associate (models) {
- VideoAbuse.belongsTo(models.Account, {
+})
+export class VideoAbuseModel extends Model<VideoAbuseModel> {
+
+ @AllowNull(false)
+ @Default(null)
+ @Is('VideoAbuseReason', value => throwIfNotValid(value, isVideoAbuseReasonValid, 'reason'))
+ @Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEO_ABUSES.REASON.max))
+ reason: string
+
+ @AllowNull(false)
+ @Default(null)
+ @Is('VideoAbuseState', value => throwIfNotValid(value, isVideoAbuseStateValid, 'state'))
+ @Column
+ state: VideoAbuseState
+
+ @AllowNull(true)
+ @Default(null)
+ @Is('VideoAbuseModerationComment', value => throwIfNotValid(value, isVideoAbuseModerationCommentValid, 'moderationComment', true))
+ @Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEO_ABUSES.MODERATION_COMMENT.max))
+ moderationComment: string
+
+ @CreatedAt
+ createdAt: Date
+
+ @UpdatedAt
+ updatedAt: Date
+
+ @ForeignKey(() => AccountModel)
+ @Column
+ reporterAccountId: number
+
+ @BelongsTo(() => AccountModel, {
foreignKey: {
- name: 'reporterAccountId',
- allowNull: true
+ allowNull: false
},
- onDelete: 'CASCADE'
+ onDelete: 'cascade'
})
+ Account: AccountModel
- VideoAbuse.belongsTo(models.Video, {
+ @ForeignKey(() => VideoModel)
+ @Column
+ videoId: number
+
+ @BelongsTo(() => VideoModel, {
foreignKey: {
- name: 'videoId',
allowNull: false
},
- onDelete: 'CASCADE'
+ onDelete: 'cascade'
})
-}
+ Video: VideoModel
-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
+ static loadByIdAndVideoId (id: number, videoId: number): Bluebird<MVideoAbuse> {
+ const query = {
+ where: {
+ id,
+ videoId
}
- ]
+ }
+ return VideoAbuseModel.findOne(query)
}
- return VideoAbuse.findAndCountAll(query).then(({ rows, count }) => {
- return { total: count, data: rows }
- })
+ static listForApi (start: number, count: number, sort: string) {
+ const query = {
+ offset: start,
+ limit: count,
+ order: getSort(sort),
+ include: [
+ {
+ model: AccountModel,
+ required: true
+ },
+ {
+ model: VideoModel,
+ required: true
+ }
+ ]
+ }
+
+ return VideoAbuseModel.findAndCountAll(query)
+ .then(({ rows, count }) => {
+ return { total: count, data: rows }
+ })
+ }
+
+ toFormattedJSON (this: MVideoAbuseFormattable): VideoAbuse {
+ return {
+ id: this.id,
+ reason: this.reason,
+ reporterAccount: this.Account.toFormattedJSON(),
+ state: {
+ id: this.state,
+ label: VideoAbuseModel.getStateLabel(this.state)
+ },
+ moderationComment: this.moderationComment,
+ video: {
+ id: this.Video.id,
+ uuid: this.Video.uuid,
+ name: this.Video.name
+ },
+ createdAt: this.createdAt
+ }
+ }
+
+ toActivityPubObject (this: MVideoAbuseVideo): VideoAbuseObject {
+ return {
+ type: 'Flag' as 'Flag',
+ content: this.reason,
+ object: this.Video.url
+ }
+ }
+
+ private static getStateLabel (id: number) {
+ return VIDEO_ABUSE_STATES[id] || 'Unknown'
+ }
}
projects / oweals / peertube.git / blobdiff