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 // ----------- Database -----------
  17 const config = require('config')
  18 const constants = require('./server/initializers/constants')
  19 const database = require('./server/initializers/database')
  20 const logger = require('./server/helpers/logger')
  21
  22 database.connect()
  23
  24 // ----------- Checker -----------
  25 const checker = require('./server/initializers/checker')
  26
  27 const miss = checker.checkConfig()
  28 if (miss.length !== 0) {
  29   throw new Error('Miss some configurations keys : ' + miss)
  30 }
  31
  32 // ----------- PeerTube modules -----------
  33 const customValidators = require('./server/helpers/custom-validators')
  34 const installer = require('./server/initializers/installer')
  35 const mongoose = require('mongoose')
  36 const routes = require('./server/controllers')
  37 const Request = mongoose.model('Request')
  38
  39 // Get configurations
  40 const port = config.get('listen.port')
  41
  42 // ----------- Command line -----------
  43
  44 // ----------- App -----------
  45
  46 // For the logger
  47 app.use(morgan('combined', { stream: logger.stream }))
  48 // For body requests
  49 app.use(bodyParser.json())
  50 app.use(bodyParser.urlencoded({ extended: false }))
  51 // Validate some params for the API
  52 app.use(expressValidator({
  53   customValidators: customValidators
  54 }))
  55
  56 // ----------- Views, routes and static files -----------
  57
  58 // Catch sefaults
  59 require('segfault-handler').registerHandler()
  60
  61 // API routes
  62 const apiRoute = '/api/' + constants.API_VERSION
  63 app.use(apiRoute, routes.api)
  64
  65 // Static files
  66 app.use('/client', express.static(path.join(__dirname, '/client/dist'), { maxAge: 0 }))
  67 // 404 for static files not found
  68 app.use('/client/*', function (req, res, next) {
  69   res.sendStatus(404)
  70 })
  71
  72 const torrentsPhysicalPath = path.join(__dirname, config.get('storage.torrents'))
  73 app.use(constants.STATIC_PATHS.TORRENTS, express.static(torrentsPhysicalPath, { maxAge: 0 }))
  74
  75 // Uploads path for webseeding
  76 const uploadsPhysicalPath = path.join(__dirname, config.get('storage.uploads'))
  77 app.use(constants.STATIC_PATHS.WEBSEED, express.static(uploadsPhysicalPath, { maxAge: 0 }))
  78
  79 // Thumbnails path for express
  80 const thumbnailsPhysicalPath = path.join(__dirname, config.get('storage.thumbnails'))
  81 app.use(constants.STATIC_PATHS.THUMBNAILS, express.static(thumbnailsPhysicalPath, { maxAge: 0 }))
  82
  83 // Client application
  84 app.use('/*', function (req, res, next) {
  85   res.sendFile(path.join(__dirname, 'client/dist/index.html'))
  86 })
  87
  88 // ----------- Tracker -----------
  89
  90 const trackerServer = new TrackerServer({
  91   http: false,
  92   udp: false,
  93   ws: false,
  94   dht: false
  95 })
  96
  97 trackerServer.on('error', function (err) {
  98   logger.error(err)
  99 })
 100
 101 trackerServer.on('warning', function (err) {
 102   logger.error(err)
 103 })
 104
 105 const server = http.createServer(app)
 106 const wss = new WebSocketServer({server: server, path: '/tracker/socket'})
 107 wss.on('connection', function (ws) {
 108   trackerServer.onWebSocketConnection(ws)
 109 })
 110
 111 // ----------- Errors -----------
 112
 113 // Catch 404 and forward to error handler
 114 app.use(function (req, res, next) {
 115   const err = new Error('Not Found')
 116   err.status = 404
 117   next(err)
 118 })
 119
 120 app.use(function (err, req, res, next) {
 121   logger.error(err)
 122   res.sendStatus(err.status || 500)
 123 })
 124
 125 installer.installApplication(function (err) {
 126   if (err) throw err
 127
 128   // ----------- Make the server listening -----------
 129   server.listen(port, function () {
 130     // Activate the pool requests
 131     Request.activate()
 132
 133     logger.info('Seeded all the videos')
 134     logger.info('Server listening on port %d', port)
 135     app.emit('ready')
 136   })
 137 })
 138
 139 module.exports = app