wikijs-fork/server/master.js

285 lines
7.9 KiB
JavaScript
Raw Normal View History

const autoload = require('auto-load')
const bodyParser = require('body-parser')
const compression = require('compression')
const cookieParser = require('cookie-parser')
const cors = require('cors')
const express = require('express')
2019-04-21 06:04:00 +00:00
const session = require('express-session')
const favicon = require('serve-favicon')
const fs = require('fs-extra')
const http = require('http')
2019-01-30 06:30:05 +00:00
const https = require('https')
const path = require('path')
2019-02-10 19:02:26 +00:00
const _ = require('lodash')
const { ApolloServer } = require('apollo-server-express')
/* global WIKI */
2017-07-29 04:11:22 +00:00
2017-12-17 04:41:16 +00:00
module.exports = async () => {
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// Load core modules
2017-07-29 21:33:08 +00:00
// ----------------------------------------
WIKI.auth = require('./core/auth').init()
WIKI.lang = require('./core/localization').init()
WIKI.mail = require('./core/mail').init()
2019-01-13 20:37:45 +00:00
WIKI.system = require('./core/system').init()
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// Load middlewares
2017-07-29 21:33:08 +00:00
// ----------------------------------------
var mw = autoload(path.join(WIKI.SERVERPATH, '/middlewares'))
var ctrl = autoload(path.join(WIKI.SERVERPATH, '/controllers'))
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// Define Express App
// ----------------------------------------
const app = express()
WIKI.app = app
2017-07-29 21:33:08 +00:00
app.use(compression())
// ----------------------------------------
// Security
// ----------------------------------------
app.use(mw.security)
app.use(cors(WIKI.config.cors))
app.options('*', cors(WIKI.config.cors))
if (WIKI.config.trustProxy) {
app.enable('trust proxy')
}
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// Public Assets
// ----------------------------------------
app.use(favicon(path.join(WIKI.ROOTPATH, 'assets', 'favicon.ico')))
app.use(express.static(path.join(WIKI.ROOTPATH, 'assets'), {
2017-07-29 21:33:08 +00:00
index: false,
maxAge: '7d'
}))
// ----------------------------------------
// Passport Authentication
// ----------------------------------------
app.use(cookieParser())
2019-04-21 06:04:00 +00:00
app.use(session({
secret: WIKI.config.sessionSecret,
resave: false,
saveUninitialized: false
}))
app.use(WIKI.auth.passport.initialize())
2019-01-07 03:03:34 +00:00
app.use(WIKI.auth.authenticate)
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// SEO
// ----------------------------------------
app.use(mw.seo)
// ----------------------------------------
// View Engine Setup
// ----------------------------------------
2017-07-29 04:11:22 +00:00
app.set('views', path.join(WIKI.SERVERPATH, 'views'))
2017-07-29 21:33:08 +00:00
app.set('view engine', 'pug')
2017-07-29 04:11:22 +00:00
2017-07-29 21:33:08 +00:00
app.use(bodyParser.json({ limit: '1mb' }))
app.use(bodyParser.urlencoded({ extended: false, limit: '1mb' }))
2017-07-29 04:11:22 +00:00
// ----------------------------------------
// Localization
// ----------------------------------------
WIKI.lang.attachMiddleware(app)
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// View accessible data
// ----------------------------------------
2017-07-29 04:11:22 +00:00
app.locals.basedir = WIKI.ROOTPATH
2017-07-29 21:33:08 +00:00
app.locals._ = require('lodash')
app.locals.moment = require('moment')
2018-06-11 03:23:09 +00:00
app.locals.moment.locale(WIKI.config.lang.code)
app.locals.config = WIKI.config
2018-12-24 22:38:34 +00:00
app.locals.pageMeta = {
title: '',
description: WIKI.config.description,
image: '',
url: '/'
}
2017-07-29 04:11:22 +00:00
2018-01-27 22:39:55 +00:00
// ----------------------------------------
// HMR (Dev Mode Only)
// ----------------------------------------
if (global.DEV) {
2018-02-03 21:48:25 +00:00
app.use(global.WP_DEV.devMiddleware)
app.use(global.WP_DEV.hotMiddleware)
2018-01-27 22:39:55 +00:00
}
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// Apollo Server (GraphQL)
// ----------------------------------------
const graphqlSchema = require('./graph')
const apolloServer = new ApolloServer({
...graphqlSchema,
context: ({ req, res }) => ({ req, res }),
subscriptions: {
onConnect: (connectionParams, webSocket) => {
},
path: '/graphql-subscriptions'
}
})
app.use('/graphql', mw.upload)
apolloServer.applyMiddleware({ app })
// ----------------------------------------
// Routing
2017-07-29 21:33:08 +00:00
// ----------------------------------------
2017-07-29 04:11:22 +00:00
2017-07-29 21:33:08 +00:00
app.use('/', ctrl.auth)
2019-01-07 03:03:34 +00:00
app.use('/', ctrl.common)
2017-07-29 04:11:22 +00:00
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// Error handling
// ----------------------------------------
2017-07-29 04:11:22 +00:00
app.use((req, res, next) => {
2017-07-29 21:33:08 +00:00
var err = new Error('Not Found')
err.status = 404
next(err)
})
2017-07-29 04:11:22 +00:00
app.use((err, req, res, next) => {
2017-07-29 21:33:08 +00:00
res.status(err.status || 500)
2019-01-25 22:30:31 +00:00
_.set(res.locals, 'pageMeta.title', 'Error')
2017-07-29 21:33:08 +00:00
res.render('error', {
message: err.message,
error: WIKI.IS_DEBUG ? err : {}
2017-07-29 21:33:08 +00:00
})
2017-07-29 04:11:22 +00:00
})
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// HTTP/S server
2017-07-29 21:33:08 +00:00
// ----------------------------------------
2018-01-27 22:39:55 +00:00
let srvConnections = {}
app.set('port', WIKI.config.port)
2019-01-30 06:30:05 +00:00
if (WIKI.config.ssl.enabled) {
WIKI.logger.info(`HTTPS Server on port: [ ${WIKI.config.port} ]`)
const tlsOpts = {}
try {
if (WIKI.config.ssl.format === 'pem') {
tlsOpts.key = fs.readFileSync(WIKI.config.ssl.key)
tlsOpts.cert = fs.readFileSync(WIKI.config.ssl.cert)
} else {
tlsOpts.pfx = fs.readFileSync(WIKI.config.ssl.pfx)
}
if (!_.isEmpty(WIKI.config.ssl.passphrase)) {
tlsOpts.passphrase = WIKI.config.ssl.passphrase
}
if (!_.isEmpty(WIKI.config.ssl.dhparam)) {
tlsOpts.dhparam = WIKI.config.ssl.dhparam
}
} catch (err) {
WIKI.logger.error('Failed to setup HTTPS server parameters:')
WIKI.logger.error(err)
return process.exit(1)
}
WIKI.server = https.createServer(tlsOpts, app)
// HTTP Redirect Server
if (WIKI.config.ssl.redirectNonSSLPort) {
WIKI.serverAlt = http.createServer((req, res) => {
res.writeHead(301, { 'Location': 'https://' + req.headers['host'] + req.url })
res.end()
})
}
2019-01-30 06:30:05 +00:00
} else {
WIKI.logger.info(`HTTP Server on port: [ ${WIKI.config.port} ]`)
WIKI.server = http.createServer(app)
}
apolloServer.installSubscriptionHandlers(WIKI.server)
2017-07-29 21:33:08 +00:00
2018-09-01 03:42:14 +00:00
WIKI.server.listen(WIKI.config.port, WIKI.config.bindIP)
WIKI.server.on('error', (error) => {
2017-07-29 21:33:08 +00:00
if (error.syscall !== 'listen') {
2017-07-29 04:11:22 +00:00
throw error
2017-07-29 21:33:08 +00:00
}
// handle specific listen errors with friendly messages
switch (error.code) {
case 'EACCES':
WIKI.logger.error('Listening on port ' + WIKI.config.port + ' requires elevated privileges!')
2017-07-29 21:33:08 +00:00
return process.exit(1)
case 'EADDRINUSE':
WIKI.logger.error('Port ' + WIKI.config.port + ' is already in use!')
2017-07-29 21:33:08 +00:00
return process.exit(1)
default:
throw error
}
})
2017-07-29 04:11:22 +00:00
WIKI.server.on('connection', conn => {
2018-01-27 22:39:55 +00:00
let key = `${conn.remoteAddress}:${conn.remotePort}`
srvConnections[key] = conn
conn.on('close', function() {
delete srvConnections[key]
})
})
WIKI.server.on('listening', () => {
2019-01-30 06:30:05 +00:00
if (WIKI.config.ssl.enabled) {
WIKI.logger.info('HTTPS Server: [ RUNNING ]')
// Start HTTP Redirect Server
if (WIKI.config.ssl.redirectNonSSLPort) {
WIKI.serverAlt.listen(WIKI.config.ssl.redirectNonSSLPort, WIKI.config.bindIP)
WIKI.serverAlt.on('error', (error) => {
if (error.syscall !== 'listen') {
throw error
}
switch (error.code) {
case 'EACCES':
WIKI.logger.error('(HTTP Redirect) Listening on port ' + WIKI.config.port + ' requires elevated privileges!')
return process.exit(1)
case 'EADDRINUSE':
WIKI.logger.error('(HTTP Redirect) Port ' + WIKI.config.port + ' is already in use!')
return process.exit(1)
default:
throw error
}
})
WIKI.serverAlt.on('listening', () => {
WIKI.logger.info('HTTP Server: [ RUNNING in redirect mode ]')
})
}
2019-01-30 06:30:05 +00:00
} else {
WIKI.logger.info('HTTP Server: [ RUNNING ]')
}
2017-07-29 21:33:08 +00:00
})
2017-07-29 04:11:22 +00:00
WIKI.server.destroy = (cb) => {
WIKI.server.close(cb)
2018-01-27 22:39:55 +00:00
for (let key in srvConnections) {
srvConnections[key].destroy()
}
if (WIKI.config.ssl.enabled && WIKI.config.ssl.redirectNonSSLPort) {
WIKI.serverAlt.close(cb)
}
2018-01-27 22:39:55 +00:00
}
2017-07-29 21:33:08 +00:00
return true
2017-10-08 02:44:35 +00:00
}