New directory organization

3 files changed, 143 insertions, 0 deletions

diff --git a/initializers/checker.js b/initializers/checker.js
new file mode 100644
index 000000000..7a3a53616
--- /dev/null
+++ b/initializers/checker.js
@@ -0,0 +1,45 @@
+;(function () {
+  'use strict'
+
+  var config = require('config')
+  var mkdirp = require('mkdirp')
+
+  var checker = {}
+
+  // Check the config files
+  checker.checkConfig = function () {
+    var required = [ 'listen.port',
+      'webserver.https', 'webserver.host', 'webserver.port',
+      'database.host', 'database.port', 'database.suffix',
+      'storage.certs', 'storage.uploads', 'storage.logs',
+      'network.friends' ]
+    var miss = []
+
+    for (var key of required) {
+      if (!config.has(key)) {
+        miss.push(key)
+      }
+    }
+
+    return miss
+  }
+
+  // Create directories for the storage if it doesn't exist
+  checker.createDirectoriesIfNotExist = function () {
+    var storages = config.get('storage')
+
+    for (var key of Object.keys(storages)) {
+      var path = storages[key]
+      try {
+        mkdirp.sync(__dirname + '/../' + path)
+      } catch (error) {
+        // Do not use logger
+        console.error('Cannot create ' + path + ':' + error)
+        process.exit(0)
+      }
+    }
+  }
+
+  // ----------- Export -----------
+  module.exports = checker
+})()
diff --git a/initializers/constants.js b/initializers/constants.js
new file mode 100644
index 000000000..00b713961
--- /dev/null
+++ b/initializers/constants.js
@@ -0,0 +1,37 @@
+;(function () {
+  'use strict'
+
+  var constants = {}
+
+  function isTestInstance () {
+    return (process.env.NODE_ENV === 'test')
+  }
+
+  // API version of our pod
+  constants.API_VERSION = 'v1'
+
+  // Score a pod has when we create it as a friend
+  constants.FRIEND_BASE_SCORE = 100
+
+  // Time to wait between requests to the friends
+  constants.INTERVAL = 60000
+
+  // Number of points we add/remove from a friend after a successful/bad request
+  constants.PODS_SCORE = {
+    MALUS: -10,
+    BONUS: 10
+  }
+
+  // Number of retries we make for the make retry requests (to friends...)
+  constants.REQUEST_RETRIES = 10
+
+  // Special constants for a test instance
+  if (isTestInstance() === true) {
+    constants.FRIEND_BASE_SCORE = 20
+    constants.INTERVAL = 10000
+    constants.REQUEST_RETRIES = 2
+  }
+
+  // ----------- Export -----------
+  module.exports = constants
+})()
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
+  }
+})()