wikijs-fork/server/master.js

213 lines
5.6 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')
const favicon = require('serve-favicon')
const http = require('http')
const path = require('path')
const session = require('express-session')
const SessionRedisStore = require('connect-redis')(session)
const { ApolloServer } = require('apollo-server-express')
2018-03-05 22:27:30 +00:00
// const oauth2orize = require('oauth2orize')
/* 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()
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))
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'
}))
// ----------------------------------------
// OAuth2 Server
// ----------------------------------------
2018-03-05 22:27:30 +00:00
// const OAuth2Server = oauth2orize.createServer()
2017-07-29 21:33:08 +00:00
// ----------------------------------------
// Passport Authentication
// ----------------------------------------
let sessionStore = new SessionRedisStore({
client: WIKI.redis
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(cookieParser())
app.use(session({
name: 'wikijs.sid',
store: sessionStore,
2018-05-28 18:46:55 +00:00
secret: WIKI.config.sessionSecret,
2017-07-29 21:33:08 +00:00
resave: false,
saveUninitialized: false
}))
app.use(WIKI.auth.passport.initialize())
app.use(WIKI.auth.passport.session())
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
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 })
})
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)
2017-07-29 04:11:22 +00:00
app.use('/', mw.auth, 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)
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 server
2017-07-29 21:33:08 +00:00
// ----------------------------------------
2018-01-27 22:39:55 +00:00
let srvConnections = {}
WIKI.logger.info(`HTTP Server on port: [ ${WIKI.config.port} ]`)
2017-07-29 21:33:08 +00:00
app.set('port', WIKI.config.port)
WIKI.server = http.createServer(app)
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', () => {
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()
}
}
2017-07-29 21:33:08 +00:00
return true
2017-10-08 02:44:35 +00:00
}