server/models/video/video-blacklist.ts

   1 import * as Sequelize from 'sequelize'
   2
   3 import { addMethodsToModel, getSort } from '../utils'
   4 import {
   5   BlacklistedVideoInstance,
   6   BlacklistedVideoAttributes,
   7
   8   BlacklistedVideoMethods
   9 } from './video-blacklist-interface'
  10
  11 let BlacklistedVideo: Sequelize.Model<BlacklistedVideoInstance, BlacklistedVideoAttributes>
  12 let toFormatedJSON: BlacklistedVideoMethods.ToFormatedJSON
  13 let countTotal: BlacklistedVideoMethods.CountTotal
  14 let list: BlacklistedVideoMethods.List
  15 let listForApi: BlacklistedVideoMethods.ListForApi
  16 let loadById: BlacklistedVideoMethods.LoadById
  17 let loadByVideoId: BlacklistedVideoMethods.LoadByVideoId
  18
  19 export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) {
  20   BlacklistedVideo = sequelize.define<BlacklistedVideoInstance, BlacklistedVideoAttributes>('BlacklistedVideo',
  21     {},
  22     {
  23       indexes: [
  24         {
  25           fields: [ 'videoId' ],
  26           unique: true
  27         }
  28       ]
  29     }
  30   )
  31
  32   const classMethods = [
  33     associate,
  34
  35     countTotal,
  36     list,
  37     listForApi,
  38     loadById,
  39     loadByVideoId
  40   ]
  41   const instanceMethods = [
  42     toFormatedJSON
  43   ]
  44   addMethodsToModel(BlacklistedVideo, classMethods, instanceMethods)
  45
  46   return BlacklistedVideo
  47 }
  48
  49 // ------------------------------ METHODS ------------------------------
  50
  51 toFormatedJSON = function (this: BlacklistedVideoInstance) {
  52   return {
  53     id: this.id,
  54     videoId: this.videoId,
  55     createdAt: this.createdAt
  56   }
  57 }
  58
  59 // ------------------------------ STATICS ------------------------------
  60
  61 function associate (models) {
  62   BlacklistedVideo.belongsTo(models.Video, {
  63     foreignKey: {
  64       name: 'videoId',
  65       allowNull: false
  66     },
  67     onDelete: 'CASCADE'
  68   })
  69 }
  70
  71 countTotal = function () {
  72   return BlacklistedVideo.count()
  73 }
  74
  75 list = function () {
  76   return BlacklistedVideo.findAll()
  77 }
  78
  79 listForApi = function (start: number, count: number, sort: string) {
  80   const query = {
  81     offset: start,
  82     limit: count,
  83     order: [ getSort(sort) ]
  84   }
  85
  86   return BlacklistedVideo.findAndCountAll(query).then(({ rows, count }) => {
  87     return {
  88       data: rows,
  89       total: count
  90     }
  91   })
  92 }
  93
  94 loadById = function (id: number) {
  95   return BlacklistedVideo.findById(id)
  96 }
  97
  98 loadByVideoId = function (id: number) {
  99   const query = {
 100     where: {
 101       videoId: id
 102     }
 103   }
 104
 105   return BlacklistedVideo.findOne(query)
 106 }