-import * as Sequelize from 'sequelize'
-
-import { SortType } from '../../helpers'
-import { addMethodsToModel, getSortOnModel } from '../utils'
-import { VideoInstance } from './video-interface'
import {
- BlacklistedVideoInstance,
- BlacklistedVideoAttributes,
-
- BlacklistedVideoMethods
-} from './video-blacklist-interface'
-
-let BlacklistedVideo: Sequelize.Model<BlacklistedVideoInstance, BlacklistedVideoAttributes>
-let toFormattedJSON: BlacklistedVideoMethods.ToFormattedJSON
-let countTotal: BlacklistedVideoMethods.CountTotal
-let list: BlacklistedVideoMethods.List
-let listForApi: BlacklistedVideoMethods.ListForApi
-let loadById: BlacklistedVideoMethods.LoadById
-let loadByVideoId: BlacklistedVideoMethods.LoadByVideoId
-
-export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
- BlacklistedVideo = sequelize.define<BlacklistedVideoInstance, BlacklistedVideoAttributes>('BlacklistedVideo',
- {},
+ AfterCreate,
+ AfterDestroy,
+ AllowNull,
+ BelongsTo,
+ Column,
+ CreatedAt,
+ DataType,
+ ForeignKey,
+ Is,
+ Model,
+ Table,
+ UpdatedAt
+} from 'sequelize-typescript'
+import { getSortOnModel, SortType, throwIfNotValid } from '../utils'
+import { VideoModel } from './video'
+import { isVideoBlacklistReasonValid } from '../../helpers/custom-validators/video-blacklist'
+import { Emailer } from '../../lib/emailer'
+import { VideoBlacklist } from '../../../shared/models/videos'
+import { CONSTRAINTS_FIELDS } from '../../initializers'
+
+@Table({
+ tableName: 'videoBlacklist',
+ indexes: [
{
- indexes: [
- {
- fields: [ 'videoId' ],
- unique: true
- }
- ]
+ fields: [ 'videoId' ],
+ unique: true
}
- )
-
- const classMethods = [
- associate,
-
- countTotal,
- list,
- listForApi,
- loadById,
- loadByVideoId
]
- const instanceMethods = [
- toFormattedJSON
- ]
- addMethodsToModel(BlacklistedVideo, classMethods, instanceMethods)
-
- return BlacklistedVideo
-}
-
-// ------------------------------ METHODS ------------------------------
+})
+export class VideoBlacklistModel extends Model<VideoBlacklistModel> {
-toFormattedJSON = function (this: BlacklistedVideoInstance) {
- let video: VideoInstance
+ @AllowNull(true)
+ @Is('VideoBlacklistReason', value => throwIfNotValid(value, isVideoBlacklistReasonValid, 'reason'))
+ @Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEO_BLACKLIST.REASON.max))
+ reason: string
- video = this.Video
+ @CreatedAt
+ createdAt: Date
- return {
- id: this.id,
- videoId: this.videoId,
- createdAt: this.createdAt,
- updatedAt: this.updatedAt,
- name: video.name,
- uuid: video.uuid,
- description: video.description,
- duration: video.duration,
- views: video.views,
- likes: video.likes,
- dislikes: video.dislikes,
- nsfw: video.nsfw
- }
-}
+ @UpdatedAt
+ updatedAt: Date
-// ------------------------------ STATICS ------------------------------
+ @ForeignKey(() => VideoModel)
+ @Column
+ videoId: number
-function associate (models) {
- BlacklistedVideo.belongsTo(models.Video, {
+ @BelongsTo(() => VideoModel, {
foreignKey: {
- name: 'videoId',
allowNull: false
},
- onDelete: 'CASCADE'
+ onDelete: 'cascade'
})
-}
-
-countTotal = function () {
- return BlacklistedVideo.count()
-}
+ Video: VideoModel
-list = function () {
- return BlacklistedVideo.findAll()
-}
+ @AfterCreate
+ static sendBlacklistEmailNotification (instance: VideoBlacklistModel) {
+ return Emailer.Instance.addVideoBlacklistReportJob(instance.videoId, instance.reason)
+ }
-listForApi = function (start: number, count: number, sort: SortType) {
- const query = {
- offset: start,
- limit: count,
- order: [ getSortOnModel(sort.sortModel, sort.sortValue) ],
- include: [ { model: BlacklistedVideo['sequelize'].models.Video } ]
+ @AfterDestroy
+ static sendUnblacklistEmailNotification (instance: VideoBlacklistModel) {
+ return Emailer.Instance.addVideoUnblacklistReportJob(instance.videoId)
}
- return BlacklistedVideo.findAndCountAll(query).then(({ rows, count }) => {
- return {
- data: rows,
- total: count
+ static listForApi (start: number, count: number, sort: SortType) {
+ const query = {
+ offset: start,
+ limit: count,
+ order: getSortOnModel(sort.sortModel, sort.sortValue),
+ include: [
+ {
+ model: VideoModel,
+ required: true
+ }
+ ]
}
- })
-}
-loadById = function (id: number) {
- return BlacklistedVideo.findById(id)
-}
+ return VideoBlacklistModel.findAndCountAll(query)
+ .then(({ rows, count }) => {
+ return {
+ data: rows,
+ total: count
+ }
+ })
+ }
-loadByVideoId = function (id: number) {
- const query = {
- where: {
- videoId: id
+ static loadByVideoId (id: number) {
+ const query = {
+ where: {
+ videoId: id
+ }
}
+
+ return VideoBlacklistModel.findOne(query)
}
- return BlacklistedVideo.findOne(query)
+ toFormattedJSON (): VideoBlacklist {
+ const video = this.Video
+
+ return {
+ id: this.id,
+ createdAt: this.createdAt,
+ updatedAt: this.updatedAt,
+ reason: this.reason,
+
+ video: {
+ id: video.id,
+ name: video.name,
+ uuid: video.uuid,
+ description: video.description,
+ duration: video.duration,
+ views: video.views,
+ likes: video.likes,
+ dislikes: video.dislikes,
+ nsfw: video.nsfw
+ }
+ }
+ }
}