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/requestsScheduler')
  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 // Catch sefaults
  60 require('segfault-handler').registerHandler()
  61
  62 // API routes
  63 const apiRoute = '/api/' + constants.API_VERSION
  64 app.use(apiRoute, routes.api)
  65
  66 // Static files
  67 app.use('/client', express.static(path.join(__dirname, '/client/dist'), { maxAge: 0 }))
  68 // 404 for static files not found
  69 app.use('/client/*', function (req, res, next) {
  70   res.sendStatus(404)
  71 })
  72
  73 // Thumbnails path for express
  74 const thumbnailsPhysicalPath = path.join(__dirname, config.get('storage.thumbnails'))
  75 app.use(constants.THUMBNAILS_STATIC_PATH, express.static(thumbnailsPhysicalPath, { maxAge: 0 }))
  76
  77 // Client application
  78 app.use('/*', function (req, res, next) {
  79   res.sendFile(path.join(__dirname, 'client/dist/index.html'))
  80 })
  81
  82 // ----------- Tracker -----------
  83
  84 const trackerServer = new TrackerServer({
  85   http: false,
  86   udp: false,
  87   ws: false,
  88   dht: false
  89 })
  90
  91 trackerServer.on('error', function (err) {
  92   logger.error(err)
  93 })
  94
  95 trackerServer.on('warning', function (err) {
  96   logger.error(err)
  97 })
  98
  99 const server = http.createServer(app)
 100 const wss = new WebSocketServer({server: server, path: '/tracker/socket'})
 101 wss.on('connection', function (ws) {
 102   trackerServer.onWebSocketConnection(ws)
 103 })
 104
 105 // ----------- Errors -----------
 106
 107 // Catch 404 and forward to error handler
 108 app.use(function (req, res, next) {
 109   const err = new Error('Not Found')
 110   err.status = 404
 111   next(err)
 112 })
 113
 114 app.use(function (err, req, res, next) {
 115   logger.error(err)
 116   res.sendStatus(err.status || 500)
 117 })
 118
 119 installer.installApplication(function (err) {
 120   if (err) throw err
 121
 122   // Create/activate the webtorrent module
 123   webtorrent.create(function () {
 124     function cleanForExit () {
 125       utils.cleanForExit(webtorrent.app)
 126     }
 127
 128     function exitGracefullyOnSignal () {
 129       process.exit(-1)
 130     }
 131
 132     process.on('exit', cleanForExit)
 133     process.on('SIGINT', exitGracefullyOnSignal)
 134     process.on('SIGTERM', exitGracefullyOnSignal)
 135
 136     // ----------- Make the server listening -----------
 137     server.listen(port, function () {
 138       // Activate the pool requests
 139       poolRequests.activate()
 140
 141       videos.seedAllExisting(function () {
 142         logger.info('Seeded all the videos')
 143         logger.info('Server listening on port %d', port)
 144         app.emit('ready')
 145       })
 146     })
 147   })
 148 })
 149
 150 module.exports = app