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('./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('./initializers/constants')
  29 var customValidators = require('./helpers/customValidators')
  30 var database = require('./initializers/database')
  31 var logger = require('./helpers/logger')
  32 var peertubeCrypto = require('./helpers/peertubeCrypto')
  33 var poolRequests = require('./lib/poolRequests')
  34 var routes = require('./controllers')
  35 var utils = require('./helpers/utils')
  36 var videos = require('./lib/videos')
  37 var webtorrent = require('./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, '/public'), { maxAge: 0 }))
  71
  72 // Jade template from ./views directory
  73 app.set('views', path.join(__dirname, '/views'))
  74 app.set('view engine', 'jade')
  75
  76 // API routes
  77 var api_route = '/api/' + constants.API_VERSION
  78 app.use(api_route, routes.api)
  79
  80 // Views routes
  81 app.use('/', routes.views)
  82
  83 // ----------- Tracker -----------
  84
  85 var trackerServer = new TrackerServer({
  86   http: false,
  87   udp: false,
  88   ws: false,
  89   dht: false
  90 })
  91
  92 trackerServer.on('error', function (err) {
  93   logger.error(err)
  94 })
  95
  96 trackerServer.on('warning', function (err) {
  97   logger.error(err)
  98 })
  99
 100 var server = http.createServer(app)
 101 var wss = new WebSocketServer({server: server, path: '/tracker/socket'})
 102 wss.on('connection', function (ws) {
 103   trackerServer.onWebSocketConnection(ws)
 104 })
 105
 106 // ----------- Errors -----------
 107
 108 // Catch 404 and forward to error handler
 109 app.use(function (req, res, next) {
 110   var err = new Error('Not Found')
 111   err.status = 404
 112   next(err)
 113 })
 114
 115 // Prod : no stacktraces leaked to user
 116 if (process.env.NODE_ENV === 'production') {
 117   app.use(function (err, req, res, next) {
 118     logger.error(err)
 119     res.status(err.status || 500)
 120     res.render('error', {
 121       message: err.message,
 122       error: {}
 123     })
 124   })
 125 } else {
 126   app.use(function (err, req, res, next) {
 127     logger.error(err)
 128     res.status(err.status || 500)
 129     res.render('error', {
 130       message: err.message,
 131       error: err
 132     })
 133   })
 134 }
 135
 136 // ----------- Create the certificates if they don't already exist -----------
 137 peertubeCrypto.createCertsIfNotExist(function (err) {
 138   if (err) throw err
 139   // Create/activate the webtorrent module
 140   webtorrent.create(function () {
 141     function cleanForExit () {
 142       utils.cleanForExit(webtorrent.app)
 143     }
 144
 145     function exitGracefullyOnSignal () {
 146       process.exit(-1)
 147     }
 148
 149     process.on('exit', cleanForExit)
 150     process.on('SIGINT', exitGracefullyOnSignal)
 151     process.on('SIGTERM', exitGracefullyOnSignal)
 152
 153     // ----------- Make the server listening -----------
 154     server.listen(port, function () {
 155       // Activate the pool requests
 156       poolRequests.activate()
 157
 158       videos.seedAllExisting(function () {
 159         logger.info('Seeded all the videos')
 160         logger.info('Server listening on port %d', port)
 161         app.emit('ready')
 162       })
 163     })
 164   })
 165 })
 166
 167 module.exports = app