Add like/dislike system for videos

[github/Chocobozzz/PeerTube.git] / server / models / user-video-rate.js

'use strict'

/*
  User rates per video.

*/

const values = require('lodash/values')

const constants = require('../initializers/constants')

// ---------------------------------------------------------------------------

module.exports = function (sequelize, DataTypes) {
  const UserVideoRate = sequelize.define('UserVideoRate',
    {
      type: {
        type: DataTypes.ENUM(values(constants.VIDEO_RATE_TYPES)),
        allowNull: false
      }
    },
    {
      indexes: [
        {
          fields: [ 'videoId', 'userId', 'type' ],
          unique: true
        }
      ],
      classMethods: {
        associate,

        load
      }
    }
  )

  return UserVideoRate
}

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

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

  this.belongsTo(models.User, {
    foreignKey: {
      name: 'userId',
      allowNull: false
    },
    onDelete: 'CASCADE'
  })
}

function load (userId, videoId, transaction, callback) {
  if (!callback) {
    callback = transaction
    transaction = null
  }

  const query = {
    where: {
      userId,
      videoId
    }
  }

  const options = {}
  if (transaction) options.transaction = transaction

  return this.findOne(query, options).asCallback(callback)
}