]>
Commit | Line | Data |
---|---|---|
1 | ;(function () { | |
2 | 'use strict' | |
3 | ||
4 | // ----------- Node modules ----------- | |
5 | var bodyParser = require('body-parser') | |
6 | var express = require('express') | |
7 | var expressValidator = require('express-validator') | |
8 | var http = require('http') | |
9 | var morgan = require('morgan') | |
10 | var path = require('path') | |
11 | var TrackerServer = require('bittorrent-tracker').Server | |
12 | var WebSocketServer = require('ws').Server | |
13 | ||
14 | // Create our main app | |
15 | var app = express() | |
16 | ||
17 | // ----------- Checker ----------- | |
18 | var checker = require('./initializers/checker') | |
19 | ||
20 | var miss = checker.checkConfig() | |
21 | if (miss.length !== 0) { | |
22 | // Do not use logger module | |
23 | console.error('Miss some configurations keys.', { miss: miss }) | |
24 | process.exit(0) | |
25 | } | |
26 | ||
27 | checker.createDirectoriesIfNotExist() | |
28 | ||
29 | // ----------- PeerTube modules ----------- | |
30 | var config = require('config') | |
31 | var constants = require('./initializers/constants') | |
32 | var customValidators = require('./helpers/customValidators') | |
33 | var logger = require('./helpers/logger') | |
34 | var poolRequests = require('./lib/poolRequests') | |
35 | var routes = require('./controllers') | |
36 | var utils = require('./helpers/utils') | |
37 | var videos = require('./models/videos') | |
38 | var webtorrent = require('./lib/webTorrentNode') | |
39 | ||
40 | // Get configurations | |
41 | var port = config.get('listen.port') | |
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 | // Static files | |
68 | app.use(express.static(path.join(__dirname, '/public'), { maxAge: 0 })) | |
69 | ||
70 | // Jade template from ./views directory | |
71 | app.set('views', path.join(__dirname, '/views')) | |
72 | app.set('view engine', 'jade') | |
73 | ||
74 | // API routes | |
75 | var api_route = '/api/' + constants.API_VERSION | |
76 | app.use(api_route, routes.api) | |
77 | ||
78 | // Views routes | |
79 | app.use('/', routes.views) | |
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 | // Prod : no stacktraces leaked to user | |
114 | if (process.env.NODE_ENV === 'production') { | |
115 | app.use(function (err, req, res, next) { | |
116 | logger.error('Error : ' + err.message, { error: err }) | |
117 | res.status(err.status || 500) | |
118 | res.render('error', { | |
119 | message: err.message, | |
120 | error: {} | |
121 | }) | |
122 | }) | |
123 | } else { | |
124 | app.use(function (err, req, res, next) { | |
125 | logger.error('Error : ' + err.message, { error: err }) | |
126 | res.status(err.status || 500) | |
127 | res.render('error', { | |
128 | message: err.message, | |
129 | error: err | |
130 | }) | |
131 | }) | |
132 | } | |
133 | ||
134 | // ----------- Create the certificates if they don't already exist ----------- | |
135 | utils.createCertsIfNotExist(function (err) { | |
136 | if (err) throw err | |
137 | // Create/activate the webtorrent module | |
138 | webtorrent.create(function () { | |
139 | function cleanForExit () { | |
140 | utils.cleanForExit(webtorrent.app) | |
141 | } | |
142 | ||
143 | function exitGracefullyOnSignal () { | |
144 | process.exit() | |
145 | } | |
146 | ||
147 | process.on('exit', cleanForExit) | |
148 | process.on('SIGINT', exitGracefullyOnSignal) | |
149 | process.on('SIGTERM', exitGracefullyOnSignal) | |
150 | ||
151 | // ----------- Make the server listening ----------- | |
152 | server.listen(port, function () { | |
153 | // Activate the pool requests | |
154 | poolRequests.activate() | |
155 | ||
156 | videos.seedAll(function () { | |
157 | logger.info('Seeded all the videos') | |
158 | logger.info('Server listening on port %d', port) | |
159 | app.emit('ready') | |
160 | }) | |
161 | }) | |
162 | }) | |
163 | }) | |
164 | ||
165 | module.exports = app | |
166 | })() |