1 files changed, 177 insertions, 0 deletions

diff --git a/src/pods.js b/src/pods.js
new file mode 100644
index 000000000..9d6539685
--- /dev/null
+++ b/src/pods.js
@@ -0,0 +1,177 @@
+;(function () {
+  'use strict'
+
+  var fs = require('fs')
+  var config = require('config')
+  var async = require('async')
+  var request = require('request')
+
+  var logger = require('./logger')
+  var utils = require('./utils')
+  var PodsDB = require('./database').PodsDB
+
+  var pods = {}
+  var http = config.get('webserver.https') ? 'https' : 'http'
+  var host = config.get('webserver.host')
+  var port = config.get('webserver.port')
+
+  // ----------- Private functions -----------
+
+  function getForeignPodsList (url, callback) {
+    var path = '/api/pods'
+
+    request.get(url + path, function (err, response, body) {
+      if (err) throw err
+      callback(JSON.parse(body))
+    })
+  }
+
+  // ----------- Public functions -----------
+  pods.list = function (callback) {
+    PodsDB.find(function (err, pods_list) {
+      if (err) {
+        logger.error('Cannot get the list of the pods.', { error: err })
+        return callback(err)
+      }
+
+      return callback(null, pods_list)
+    })
+  }
+
+  // { url }
+  pods.add = function (data, callback) {
+    logger.info('Adding pod: %s', data.url)
+
+    var params = {
+      url: data.url,
+      publicKey: data.publicKey
+    }
+
+    PodsDB.create(params, function (err, pod) {
+      if (err) {
+        logger.error('Cannot insert the pod.', { error: err })
+        return callback(err)
+      }
+
+      fs.readFile(utils.certDir + 'petube.pub', 'utf8', function (err, cert) {
+        if (err) {
+          logger.error('Cannot read cert file.', { error: err })
+          return callback(err)
+        }
+
+        return callback(null, { cert: cert })
+      })
+    })
+  }
+
+  // { path, data }
+  pods.makeSecureRequest = function (data, callback) {
+    PodsDB.find({}, { url: 1, publicKey: 1 }).exec(function (err, urls) {
+      if (err) {
+        logger.error('Cannot get the list of the pods.', { error: err })
+        return callback(err)
+      }
+
+      logger.debug('Make multiple requests.')
+      utils.makeMultipleRetryRequest(
+        { encrypt: true, sign: true, method: data.method, path: data.path, data: data.data },
+
+        urls,
+
+        function (err, response, body, url) {
+          if (err || response.statusCode !== 200) {
+            logger.error('Error sending secure request to %s/%s pod.', url, data.path, { error: err })
+          }
+        },
+
+        function (err) {
+          if (err) {
+            logger.error('There was some errors when sending the video meta data.', { error: err })
+            return callback(err)
+          }
+
+          logger.debug('Finished')
+          callback(null)
+        }
+      )
+    })
+  }
+
+  pods.makeFriends = function (callback) {
+    logger.debug('Read public key...')
+    fs.readFile(utils.certDir + 'petube.pub', 'utf8', function (err, cert) {
+      if (err) {
+        logger.error('Cannot read public cert.', { error: err })
+        return callback(err)
+      }
+
+      var urls = config.get('network.friends')
+      var pods_score = {}
+
+      async.each(urls, function (url, callback) {
+        // Always add a trust pod
+        pods_score[url] = Infinity
+
+        getForeignPodsList(url, function (foreign_pods_list) {
+          if (foreign_pods_list.length === 0) return callback()
+
+          async.each(foreign_pods_list, function (foreign_pod, callback) {
+            var foreign_url = foreign_pod.url
+            if (pods_score[foreign_url]) pods_score[foreign_url]++
+            else pods_score[foreign_url] = 1
+            callback()
+          }, callback)
+        })
+      }, function () {
+        logger.debug('Pods score', { pods_score: pods_score })
+
+        // Build the list of pods to add
+        // Only add a pod if it exists in more than a half base pods
+        var pods_list = []
+        var base_score = urls.length / 2
+        Object.keys(pods_score).forEach(function (pod) {
+          if (pods_score[pod] > base_score) pods_list.push({ url: pod })
+        })
+
+        logger.debug('Pods that we keep', { pods: pods_list })
+
+        var data = {
+          url: http + '://' + host + ':' + port,
+          publicKey: cert
+        }
+
+        logger.debug('Make requests...')
+
+        utils.makeMultipleRetryRequest(
+          { method: 'POST', path: '/api/pods/', data: data },
+
+          pods_list,
+
+          function eachRequest (err, response, body, url) {
+            if (!err && response.statusCode === 200) {
+              pods.add({ url: url, publicKey: body.cert }, function (err) {
+                if (err) {
+                  logger.error('Error with adding %s pod.', url, { error: err })
+                }
+              })
+            } else {
+              logger.error('Error with adding %s pod.', url)
+            }
+          },
+
+          function endRequests (err) {
+            if (err) {
+              logger.error('There was some errors when we wanted to make friends.', { error: err })
+              return callback(err)
+            }
+
+            logger.debug('Finished')
+            callback(null)
+          }
+        )
+      })
+    })
+  }
+
+  module.exports = pods
+})()