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