PeerTube/server/models/pod.js

207 lines
4.1 KiB
JavaScript
Raw Normal View History

'use strict'
const map = require('lodash/map')
2016-03-16 16:29:27 -05:00
const constants = require('../initializers/constants')
2016-12-28 08:49:23 -06:00
const customPodsValidators = require('../helpers/custom-validators').pods
// ---------------------------------------------------------------------------
2016-12-11 14:50:51 -06:00
module.exports = function (sequelize, DataTypes) {
const Pod = sequelize.define('Pod',
{
host: {
2016-12-28 08:49:23 -06:00
type: DataTypes.STRING,
allowNull: false,
validate: {
isHost: function (value) {
const res = customPodsValidators.isHostValid(value)
if (res === false) throw new Error('Host not valid.')
}
}
2016-12-11 14:50:51 -06:00
},
publicKey: {
2016-12-28 08:49:23 -06:00
type: DataTypes.STRING(5000),
allowNull: false
2016-12-11 14:50:51 -06:00
},
score: {
type: DataTypes.INTEGER,
2016-12-28 08:49:23 -06:00
defaultValue: constants.FRIEND_SCORE.BASE,
allowNull: false,
validate: {
isInt: true,
max: constants.FRIEND_SCORE.MAX
}
2017-02-16 12:19:56 -06:00
},
email: {
type: DataTypes.STRING(400),
allowNull: false
2016-12-11 14:50:51 -06:00
}
},
{
2016-12-29 02:33:28 -06:00
indexes: [
{
2017-02-16 12:24:34 -06:00
fields: [ 'host' ],
unique: true
2016-12-29 02:33:28 -06:00
},
{
fields: [ 'score' ]
}
],
2016-12-11 14:50:51 -06:00
classMethods: {
associate,
countAll,
incrementScores,
list,
listAllIds,
2017-01-10 15:24:42 -06:00
listRandomPodIdsWithRequest,
2016-12-11 14:50:51 -06:00
listBadPods,
load,
loadByHost,
removeAll
},
instanceMethods: {
toFormatedJSON
}
}
)
return Pod
}
// ------------------------------ METHODS ------------------------------
function toFormatedJSON () {
const json = {
2016-12-11 14:50:51 -06:00
id: this.id,
host: this.host,
2017-02-16 12:19:56 -06:00
email: this.email,
score: this.score,
2016-12-11 14:50:51 -06:00
createdAt: this.createdAt
}
return json
}
// ------------------------------ Statics ------------------------------
2016-12-11 14:50:51 -06:00
function associate (models) {
this.belongsToMany(models.Request, {
foreignKey: 'podId',
through: models.RequestToPod,
2016-12-24 09:59:17 -06:00
onDelete: 'cascade'
2016-12-11 14:50:51 -06:00
})
}
function countAll (callback) {
2016-12-11 14:50:51 -06:00
return this.count().asCallback(callback)
}
function incrementScores (ids, value, callback) {
if (!callback) callback = function () {}
2016-12-11 14:50:51 -06:00
const update = {
score: this.sequelize.literal('score +' + value)
}
2016-12-28 08:49:23 -06:00
const options = {
2016-12-11 14:50:51 -06:00
where: {
id: {
$in: ids
}
2016-12-28 08:49:23 -06:00
},
// In this case score is a literal and not an integer so we do not validate it
validate: false
2016-12-11 14:50:51 -06:00
}
2016-12-28 08:49:23 -06:00
return this.update(update, options).asCallback(callback)
}
function list (callback) {
2016-12-11 14:50:51 -06:00
return this.findAll().asCallback(callback)
}
2015-06-09 10:41:40 -05:00
function listAllIds (transaction, callback) {
if (!callback) {
callback = transaction
transaction = null
}
2016-12-11 14:50:51 -06:00
const query = {
attributes: [ 'id' ]
}
if (transaction) query.transaction = transaction
2016-12-11 14:50:51 -06:00
return this.findAll(query).asCallback(function (err, pods) {
if (err) return callback(err)
2016-12-11 14:50:51 -06:00
return callback(null, map(pods, 'id'))
})
}
2017-01-10 15:24:42 -06:00
function listRandomPodIdsWithRequest (limit, callback) {
const self = this
self.count().asCallback(function (err, count) {
if (err) return callback(err)
// Optimization...
if (count === 0) return callback(null, [])
let start = Math.floor(Math.random() * count) - limit
if (start < 0) start = 0
const query = {
attributes: [ 'id' ],
order: [
[ 'id', 'ASC' ]
],
offset: start,
limit: limit,
where: {
id: {
$in: [
this.sequelize.literal('SELECT "podId" FROM "RequestToPods"')
]
}
}
}
return this.findAll(query).asCallback(function (err, pods) {
if (err) return callback(err)
return callback(null, map(pods, 'id'))
})
})
}
function listBadPods (callback) {
2016-12-11 14:50:51 -06:00
const query = {
where: {
score: { $lte: 0 }
}
}
return this.findAll(query).asCallback(callback)
}
2015-06-09 10:41:40 -05:00
function load (id, callback) {
2016-12-11 14:50:51 -06:00
return this.findById(id).asCallback(callback)
}
2016-01-31 04:23:52 -06:00
function loadByHost (host, callback) {
2016-12-11 14:50:51 -06:00
const query = {
where: {
host: host
}
}
return this.findOne(query).asCallback(callback)
}
2016-01-31 04:23:52 -06:00
function removeAll (callback) {
2016-12-11 14:50:51 -06:00
return this.destroy().asCallback(callback)
}