Update node modules

[github/Chocobozzz/PeerTube.git] / src / database.js

diff --git a/src/database.js b/src/database.js
index 020bfd961202b623e0022a247c861753ac93bfff..e03176b31e9a0f6fc16be6457b060c94a5985ac1 100644 (file)
--- a/src/database.js
+++ b/src/database.js
@@ -4,6 +4,7 @@
   var config = require('config')
   var mongoose = require('mongoose')
 
+  var constants = require('./constants')
   var logger = require('./logger')
 
   var dbname = 'peertube' + config.get('database.suffix')
@@ -24,11 +25,21 @@
   // ----------- Pods -----------
   var podsSchema = mongoose.Schema({
     url: String,
-    publicKey: String
+    publicKey: String,
+    score: { type: Number, max: constants.FRIEND_BASE_SCORE }
   })
 
   var PodsDB = mongoose.model('pods', podsSchema)
 
+  // ----------- PoolRequests -----------
+  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)
+
   // ----------- Connection -----------
 
   mongoose.connect('mongodb://' + host + ':' + port + '/' + dbname)
@@ -44,6 +55,7 @@
   // ----------- Export -----------
   module.exports = {
     VideosDB: VideosDB,
-    PodsDB: PodsDB
+    PodsDB: PodsDB,
+    PoolRequestsDB: PoolRequestsDB
   }
 })()