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, '/client'), { maxAge: 0 }))
  71
  72 // API routes
  73 var api_route = '/api/' + constants.API_VERSION
  74 app.use(api_route, routes.api)
  75
  76 // Client application
  77 app.use('/*', function (req, res, next) {
  78   res.sendFile(path.join(__dirname, 'client/index.html'))
  79 })
  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 app.use(function (err, req, res, next) {
 114   logger.error(err)
 115   res.sendStatus(err.status || 500)
 116 })
 117
 118 // ----------- Create the certificates if they don't already exist -----------
 119 peertubeCrypto.createCertsIfNotExist(function (err) {
 120   if (err) throw err
 121   // Create/activate the webtorrent module
 122   webtorrent.create(function () {
 123     function cleanForExit () {
 124       utils.cleanForExit(webtorrent.app)
 125     }
 126
 127     function exitGracefullyOnSignal () {
 128       process.exit(-1)
 129     }
 130
 131     process.on('exit', cleanForExit)
 132     process.on('SIGINT', exitGracefullyOnSignal)
 133     process.on('SIGTERM', exitGracefullyOnSignal)
 134
 135     // ----------- Make the server listening -----------
 136     server.listen(port, function () {
 137       // Activate the pool requests
 138       poolRequests.activate()
 139
 140       videos.seedAllExisting(function () {
 141         logger.info('Seeded all the videos')
 142         logger.info('Server listening on port %d', port)
 143         app.emit('ready')
 144       })
 145     })
 146   })
 147 })
 148
 149 module.exports = app