server.js

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