server.js

   1 'use strict'
   2
   3 // ----------- Node modules -----------
   4 const bodyParser = require('body-parser')
   5 const express = require('express')
   6 const expressValidator = require('express-validator')
   7 const http = require('http')
   8 const morgan = require('morgan')
   9 const path = require('path')
  10 const TrackerServer = require('bittorrent-tracker').Server
  11 const WebSocketServer = require('ws').Server
  12
  13 // Create our main app
  14 const app = express()
  15
  16 // ----------- Checker -----------
  17 const checker = require('./server/initializers/checker')
  18
  19 const miss = checker.checkConfig()
  20 if (miss.length !== 0) {
  21   throw new Error('Miss some configurations keys : ' + miss)
  22 }
  23
  24 // ----------- PeerTube modules -----------
  25 const config = require('config')
  26 const constants = require('./server/initializers/constants')
  27 const customValidators = require('./server/helpers/customValidators')
  28 const database = require('./server/initializers/database')
  29 const installer = require('./server/initializers/installer')
  30 const logger = require('./server/helpers/logger')
  31 const poolRequests = require('./server/lib/poolRequests')
  32 const routes = require('./server/controllers')
  33 const utils = require('./server/helpers/utils')
  34 const videos = require('./server/lib/videos')
  35 const webtorrent = require('./server/lib/webtorrent')
  36
  37 // Get configurations
  38 const port = config.get('listen.port')
  39
  40 // ----------- Database -----------
  41 database.connect()
  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 // API routes
  68 const api_route = '/api/' + constants.API_VERSION
  69 app.use(api_route, routes.api)
  70
  71 // Static files
  72 app.use('/app', express.static(path.join(__dirname, '/client'), { maxAge: 0 }))
  73 // 404 for static files not found
  74 app.use('/app/*', function (req, res, next) {
  75   res.sendStatus(404)
  76 })
  77
  78 // Client application
  79 app.use('/*', function (req, res, next) {
  80   res.sendFile(path.join(__dirname, 'client/index.html'))
  81 })
  82
  83 // ----------- Tracker -----------
  84
  85 const 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 const server = http.createServer(app)
 101 const 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   const err = new Error('Not Found')
 111   err.status = 404
 112   next(err)
 113 })
 114
 115 app.use(function (err, req, res, next) {
 116   logger.error(err)
 117   res.sendStatus(err.status || 500)
 118 })
 119
 120 installer.installApplication(function (err) {
 121   if (err) throw err
 122
 123   // Create/activate the webtorrent module
 124   webtorrent.create(function () {
 125     function cleanForExit () {
 126       utils.cleanForExit(webtorrent.app)
 127     }
 128
 129     function exitGracefullyOnSignal () {
 130       process.exit(-1)
 131     }
 132
 133     process.on('exit', cleanForExit)
 134     process.on('SIGINT', exitGracefullyOnSignal)
 135     process.on('SIGTERM', exitGracefullyOnSignal)
 136
 137     // ----------- Make the server listening -----------
 138     server.listen(port, function () {
 139       // Activate the pool requests
 140       poolRequests.activate()
 141
 142       videos.seedAllExisting(function () {
 143         logger.info('Seeded all the videos')
 144         logger.info('Server listening on port %d', port)
 145         app.emit('ready')
 146       })
 147     })
 148   })
 149 })
 150
 151 module.exports = app