Rename webtorrent files (process/node)

[github/Chocobozzz/PeerTube.git] / models / poolRequests.js

;(function () {
  'use strict'

  var mongoose = require('mongoose')

  var logger = require('../helpers/logger')

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

  var poolRequestsSchema = mongoose.Schema({
    type: String,
    id: String, // Special id to find duplicates (video created we want to remove...)
    request: mongoose.Schema.Types.Mixed
  })
  var PoolRequestsDB = mongoose.model('poolRequests', poolRequestsSchema)

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

  var PoolRequests = {
    create: create,
    findById: findById,
    list: list,
    removeRequestById: removeRequestById,
    removeRequests: removeRequests
  }

  function create (id, type, request, callback) {
    PoolRequestsDB.create({ id: id, type: type, request: request }, callback)
  }

  function findById (id, callback) {
    PoolRequestsDB.findOne({ id: id }, callback)
  }

  function list (callback) {
    PoolRequestsDB.find({}, { _id: 1, type: 1, request: 1 }, callback)
  }

  function removeRequestById (id, callback) {
    PoolRequestsDB.remove({ id: id }, callback)
  }

  function removeRequests (ids) {
    PoolRequestsDB.remove({ _id: { $in: ids } }, function (err) {
      if (err) {
        logger.error('Cannot remove requests from the pool requests database.', { error: err })
        return // Abort
      }

      logger.info('Pool requests flushed.')
    })
  }

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

  module.exports = PoolRequests
})()