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