-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',
- {},
+import { BelongsTo, Column, CreatedAt, ForeignKey, Model, Table, UpdatedAt } from 'sequelize-typescript'
+import { SortType } from '../../helpers/utils'
+import { getSortOnModel } from '../utils'
+import { VideoModel } from './video'
+
+@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)
+})
+export class VideoBlacklistModel extends Model<VideoBlacklistModel> {
- return BlacklistedVideo
-}
-
-// ------------------------------ METHODS ------------------------------
+ @CreatedAt
+ createdAt: Date
-toFormattedJSON = function (this: BlacklistedVideoInstance) {
- let video: VideoInstance
+ @UpdatedAt
+ updatedAt: Date
- video = this.Video
+ @ForeignKey(() => VideoModel)
+ @Column
+ videoId: number
- 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
- }
-}
-
-// ------------------------------ STATICS ------------------------------
-
-function associate (models) {
- BlacklistedVideo.belongsTo(models.Video, {
+ @BelongsTo(() => VideoModel, {
foreignKey: {
- name: 'videoId',
allowNull: false
},
- onDelete: 'CASCADE'
+ onDelete: 'cascade'
})
-}
-
-countTotal = function () {
- return BlacklistedVideo.count()
-}
-
-list = function () {
- return BlacklistedVideo.findAll()
-}
+ Video: VideoModel
+
+ static listForApi (start: number, count: number, sort: SortType) {
+ const query = {
+ offset: start,
+ limit: count,
+ order: [ getSortOnModel(sort.sortModel, sort.sortValue) ],
+ include: [ { model: VideoModel } ]
+ }
-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 } ]
+ return VideoBlacklistModel.findAndCountAll(query)
+ .then(({ rows, count }) => {
+ return {
+ data: rows,
+ total: count
+ }
+ })
}
- return BlacklistedVideo.findAndCountAll(query).then(({ rows, count }) => {
- return {
- data: rows,
- total: count
+ static loadByVideoId (id: number) {
+ const query = {
+ where: {
+ videoId: id
+ }
}
- })
-}
-loadById = function (id: number) {
- return BlacklistedVideo.findById(id)
-}
+ return VideoBlacklistModel.findOne(query)
+ }
-loadByVideoId = function (id: number) {
- const query = {
- where: {
- videoId: id
+ toFormattedJSON () {
+ const video = this.Video
+
+ 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
}
}
-
- return BlacklistedVideo.findOne(query)
}