PeerTube/server/models/video/video-blacklist.ts

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 listForApi: BlacklistedVideoMethods.ListForApi
let loadByVideoId: BlacklistedVideoMethods.LoadByVideoId

export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
  BlacklistedVideo = sequelize.define<BlacklistedVideoInstance, BlacklistedVideoAttributes>('BlacklistedVideo',
    {},
    {
      indexes: [
        {
          fields: [ 'videoId' ],
          unique: true
        }
      ]
    }
  )

  const classMethods = [
    associate,

    listForApi,
    loadByVideoId
  ]
  const instanceMethods = [
    toFormattedJSON
  ]
  addMethodsToModel(BlacklistedVideo, classMethods, instanceMethods)

  return BlacklistedVideo
}

// ------------------------------ METHODS ------------------------------

toFormattedJSON = function (this: BlacklistedVideoInstance) {
  let video: VideoInstance

  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
  }
}

// ------------------------------ STATICS ------------------------------

function associate (models) {
  BlacklistedVideo.belongsTo(models.Video, {
    foreignKey: {
      name: 'videoId',
      allowNull: false
    },
    onDelete: 'CASCADE'
  })
}

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 BlacklistedVideo.findAndCountAll(query).then(({ rows, count }) => {
    return {
      data: rows,
      total: count
    }
  })
}

loadByVideoId = function (id: number) {
  const query = {
    where: {
      videoId: id
    }
  }

  return BlacklistedVideo.findOne(query)
}