server.js

   1 ;(function () {
   2   'use strict'
   3
   4   // ----------- Node modules -----------
   5   var bodyParser = require('body-parser')
   6   var express = require('express')
   7   var expressValidator = require('express-validator')
   8   var http = require('http')
   9   var morgan = require('morgan')
  10   var multer = require('multer')
  11   var path = require('path')
  12   var TrackerServer = require('bittorrent-tracker').Server
  13   var WebSocketServer = require('ws').Server
  14
  15   // Create our main app
  16   var app = express()
  17
  18   // ----------- Checker -----------
  19   var checker = require('./src/checker')
  20
  21   var miss = checker.checkConfig()
  22   if (miss.length !== 0) {
  23     // Do not use logger module
  24     console.error('Miss some configurations keys.', { miss: miss })
  25     process.exit(0)
  26   }
  27
  28   checker.createDirectoriesIfNotExist()
  29
  30   // ----------- Constants -----------
  31   var utils = require('./src/utils')
  32
  33   global.API_VERSION = 'v1'
  34   global.FRIEND_BASE_SCORE = utils.isTestInstance() ? 20 : 100
  35
  36   // ----------- PeerTube modules -----------
  37   var config = require('config')
  38   var customValidators = require('./src/customValidators')
  39   var logger = require('./src/logger')
  40   var poolRequests = require('./src/poolRequests')
  41   var routes = require('./routes')
  42   var videos = require('./src/videos')
  43   var webtorrent = require('./src/webTorrentNode')
  44
  45   // Get configurations
  46   var port = config.get('listen.port')
  47   var uploads = config.get('storage.uploads')
  48
  49   // ----------- Command line -----------
  50
  51   // ----------- App -----------
  52
  53   // For the logger
  54   app.use(morgan('combined', { stream: logger.stream }))
  55   // For body requests
  56   app.use(bodyParser.json())
  57   // For POST file requests
  58   app.use(multer({ dest: uploads }))
  59   app.use(bodyParser.urlencoded({ extended: false }))
  60   // Validate some params for the API
  61   app.use(expressValidator({
  62     customValidators: customValidators
  63   }))
  64
  65   // ----------- Views, routes and static files -----------
  66
  67   // Livereload
  68   app.use(require('connect-livereload')({
  69     port: 35729
  70   }))
  71
  72   // Catch sefaults
  73   require('segfault-handler').registerHandler()
  74
  75   // Static files
  76   app.use(express.static(path.join(__dirname, '/public'), { maxAge: 0 }))
  77
  78   // Jade template from ./views directory
  79   app.set('views', path.join(__dirname, '/views'))
  80   app.set('view engine', 'jade')
  81
  82   // API routes
  83   var api_route = '/api/' + global.API_VERSION
  84   app.use(api_route, routes.api)
  85
  86   // Views routes
  87   app.use('/', routes.views)
  88
  89   // ----------- Tracker -----------
  90
  91   var trackerServer = new TrackerServer({
  92     http: false,
  93     udp: false,
  94     ws: false,
  95     dht: false
  96   })
  97
  98   trackerServer.on('error', function (err) {
  99     logger.error(err)
 100   })
 101
 102   trackerServer.on('warning', function (err) {
 103     logger.error(err)
 104   })
 105
 106   var server = http.createServer(app)
 107   var wss = new WebSocketServer({server: server, path: '/tracker/socket'})
 108   wss.on('connection', function (ws) {
 109     trackerServer.onWebSocketConnection(ws)
 110   })
 111
 112   // ----------- Errors -----------
 113
 114   // Catch 404 and forward to error handler
 115   app.use(function (req, res, next) {
 116     var err = new Error('Not Found')
 117     err.status = 404
 118     next(err)
 119   })
 120
 121   // Prod : no stacktraces leaked to user
 122   if (process.env.NODE_ENV === 'production') {
 123     app.use(function (err, req, res, next) {
 124       logger.error('Error : ' + err.message, { error: err })
 125       res.status(err.status || 500)
 126       res.render('error', {
 127         message: err.message,
 128         error: {}
 129       })
 130     })
 131   } else {
 132     app.use(function (err, req, res, next) {
 133       logger.error('Error : ' + err.message, { error: err })
 134       res.status(err.status || 500)
 135       res.render('error', {
 136         message: err.message,
 137         error: err
 138       })
 139     })
 140   }
 141
 142   // ----------- Create the certificates if they don't already exist -----------
 143   utils.createCertsIfNotExist(function (err) {
 144     if (err) throw err
 145     // Create/activate the webtorrent module
 146     webtorrent.create(function () {
 147       function cleanForExit () {
 148         utils.cleanForExit(webtorrent.app)
 149       }
 150
 151       function exitGracefullyOnSignal () {
 152         process.exit()
 153       }
 154
 155       process.on('exit', cleanForExit)
 156       process.on('SIGINT', exitGracefullyOnSignal)
 157       process.on('SIGTERM', exitGracefullyOnSignal)
 158
 159       // ----------- Make the server listening -----------
 160       server.listen(port, function () {
 161         // Activate the pool requests
 162         poolRequests.activate()
 163
 164         videos.seedAll(function () {
 165           logger.info('Seeded all the videos')
 166           logger.info('Server listening on port %d', port)
 167           app.emit('ready')
 168         })
 169       })
 170     })
 171   })
 172
 173   module.exports = app
 174 })()