1 files changed, 61 insertions, 0 deletions

diff --git a/initializers/database.js b/initializers/database.js
new file mode 100644
index 000000000..4570d3739
--- /dev/null
+++ b/initializers/database.js
@@ -0,0 +1,61 @@
+;(function () {
+  'use strict'
+
+  var config = require('config')
+  var mongoose = require('mongoose')
+
+  var constants = require('./constants')
+  var logger = require('../helpers/logger')
+
+  var dbname = 'peertube' + config.get('database.suffix')
+  var host = config.get('database.host')
+  var port = config.get('database.port')
+
+  // ----------- Videos -----------
+  var videosSchema = mongoose.Schema({
+    name: String,
+    namePath: String,
+    description: String,
+    magnetUri: String,
+    podUrl: String
+  })
+
+  var VideosDB = mongoose.model('videos', videosSchema)
+
+  // ----------- Pods -----------
+  var podsSchema = mongoose.Schema({
+    url: 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)
+  mongoose.connection.on('error', function () {
+    logger.error('Mongodb connection error.')
+    process.exit(0)
+  })
+
+  mongoose.connection.on('open', function () {
+    logger.info('Connected to mongodb.')
+  })
+
+  // ----------- Export -----------
+  module.exports = {
+    VideosDB: VideosDB,
+    PodsDB: PodsDB,
+    PoolRequestsDB: PoolRequestsDB
+  }
+})()