|  | @@ -2,7 +2,6 @@
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  // ===========================================
 | 
	
		
			
				|  |  |  // Wiki.js
 | 
	
		
			
				|  |  | -// 1.0.1
 | 
	
		
			
				|  |  |  // Licensed under AGPLv3
 | 
	
		
			
				|  |  |  // ===========================================
 | 
	
		
			
				|  |  |  
 | 
	
	
		
			
				|  | @@ -28,251 +27,28 @@ wiki.data = appconf.data
 | 
	
		
			
				|  |  |  // Load Winston
 | 
	
		
			
				|  |  |  // ----------------------------------------
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -wiki.logger = require('./modules/logger')(wiki.IS_DEBUG, 'SERVER')
 | 
	
		
			
				|  |  | -wiki.logger.info('Wiki.js is initializing...')
 | 
	
		
			
				|  |  | +wiki.logger = require('./modules/logger')()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  // ----------------------------------------
 | 
	
		
			
				|  |  | -// Load global modules
 | 
	
		
			
				|  |  | +// Start Cluster
 | 
	
		
			
				|  |  |  // ----------------------------------------
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -wiki.disk = require('./modules/disk').init()
 | 
	
		
			
				|  |  | -wiki.db = require('./modules/db').init()
 | 
	
		
			
				|  |  | -wiki.entries = require('./modules/entries').init()
 | 
	
		
			
				|  |  | -wiki.git = require('./modules/git').init(false)
 | 
	
		
			
				|  |  | -wiki.lang = require('i18next')
 | 
	
		
			
				|  |  | -wiki.mark = require('./modules/markdown')
 | 
	
		
			
				|  |  | -wiki.redis = require('./modules/redis').init()
 | 
	
		
			
				|  |  | -wiki.search = require('./modules/search').init()
 | 
	
		
			
				|  |  | -wiki.upl = require('./modules/uploads').init()
 | 
	
		
			
				|  |  | +const cluster = require('cluster')
 | 
	
		
			
				|  |  | +const numCPUs = require('os').cpus().length
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Load modules
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -const autoload = require('auto-load')
 | 
	
		
			
				|  |  | -const bodyParser = require('body-parser')
 | 
	
		
			
				|  |  | -const compression = require('compression')
 | 
	
		
			
				|  |  | -const cookieParser = require('cookie-parser')
 | 
	
		
			
				|  |  | -const express = require('express')
 | 
	
		
			
				|  |  | -const favicon = require('serve-favicon')
 | 
	
		
			
				|  |  | -const flash = require('connect-flash')
 | 
	
		
			
				|  |  | -const fork = require('child_process').fork
 | 
	
		
			
				|  |  | -const http = require('http')
 | 
	
		
			
				|  |  | -const i18nBackend = require('i18next-node-fs-backend')
 | 
	
		
			
				|  |  | -const passport = require('passport')
 | 
	
		
			
				|  |  | -const passportSocketIo = require('passport.socketio')
 | 
	
		
			
				|  |  | -const session = require('express-session')
 | 
	
		
			
				|  |  | -const SessionRedisStore = require('connect-redis')(session)
 | 
	
		
			
				|  |  | -const graceful = require('node-graceful')
 | 
	
		
			
				|  |  | -const socketio = require('socket.io')
 | 
	
		
			
				|  |  | -const graphqlApollo = require('apollo-server-express')
 | 
	
		
			
				|  |  | -const graphqlSchema = require('./modules/graphql')
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -var mw = autoload(path.join(wiki.SERVERPATH, '/middlewares'))
 | 
	
		
			
				|  |  | -var ctrl = autoload(path.join(wiki.SERVERPATH, '/controllers'))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Define Express App
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -const app = express()
 | 
	
		
			
				|  |  | -wiki.app = app
 | 
	
		
			
				|  |  | -app.use(compression())
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Security
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use(mw.security)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Public Assets
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use(favicon(path.join(wiki.ROOTPATH, 'assets', 'favicon.ico')))
 | 
	
		
			
				|  |  | -app.use(express.static(path.join(wiki.ROOTPATH, 'assets'), {
 | 
	
		
			
				|  |  | -  index: false,
 | 
	
		
			
				|  |  | -  maxAge: '7d'
 | 
	
		
			
				|  |  | -}))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Passport Authentication
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -require('./modules/auth')(passport)
 | 
	
		
			
				|  |  | -wiki.rights = require('./modules/rights')
 | 
	
		
			
				|  |  | -wiki.rights.init()
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -let sessionStore = new SessionRedisStore({
 | 
	
		
			
				|  |  | -  client: wiki.redis
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use(cookieParser())
 | 
	
		
			
				|  |  | -app.use(session({
 | 
	
		
			
				|  |  | -  name: 'wikijs.sid',
 | 
	
		
			
				|  |  | -  store: sessionStore,
 | 
	
		
			
				|  |  | -  secret: wiki.config.sessionSecret,
 | 
	
		
			
				|  |  | -  resave: false,
 | 
	
		
			
				|  |  | -  saveUninitialized: false
 | 
	
		
			
				|  |  | -}))
 | 
	
		
			
				|  |  | -app.use(flash())
 | 
	
		
			
				|  |  | -app.use(passport.initialize())
 | 
	
		
			
				|  |  | -app.use(passport.session())
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// SEO
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use(mw.seo)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Localization Engine
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -wiki.lang.use(i18nBackend).init({
 | 
	
		
			
				|  |  | -  load: 'languageOnly',
 | 
	
		
			
				|  |  | -  ns: ['common', 'admin', 'auth', 'errors', 'git'],
 | 
	
		
			
				|  |  | -  defaultNS: 'common',
 | 
	
		
			
				|  |  | -  saveMissing: false,
 | 
	
		
			
				|  |  | -  preload: [wiki.config.lang],
 | 
	
		
			
				|  |  | -  lng: wiki.config.lang,
 | 
	
		
			
				|  |  | -  fallbackLng: 'en',
 | 
	
		
			
				|  |  | -  backend: {
 | 
	
		
			
				|  |  | -    loadPath: path.join(wiki.SERVERPATH, 'locales/{{lng}}/{{ns}}.json')
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// View Engine Setup
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.set('views', path.join(wiki.SERVERPATH, 'views'))
 | 
	
		
			
				|  |  | -app.set('view engine', 'pug')
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use(bodyParser.json({ limit: '1mb' }))
 | 
	
		
			
				|  |  | -app.use(bodyParser.urlencoded({ extended: false, limit: '1mb' }))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// View accessible data
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.locals._ = require('lodash')
 | 
	
		
			
				|  |  | -app.locals.t = wiki.lang.t.bind(wiki.lang)
 | 
	
		
			
				|  |  | -app.locals.moment = require('moment')
 | 
	
		
			
				|  |  | -app.locals.moment.locale(wiki.config.lang)
 | 
	
		
			
				|  |  | -app.locals.appconfig = wiki.config
 | 
	
		
			
				|  |  | -app.use(mw.flash)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Controllers
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use('/', ctrl.auth)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use('/graphql', graphqlApollo.graphqlExpress({ schema: graphqlSchema }))
 | 
	
		
			
				|  |  | -app.use('/graphiql', graphqlApollo.graphiqlExpress({ endpointURL: '/graphql' }))
 | 
	
		
			
				|  |  | -app.use('/uploads', mw.auth, ctrl.uploads)
 | 
	
		
			
				|  |  | -app.use('/admin', mw.auth, ctrl.admin)
 | 
	
		
			
				|  |  | -app.use('/', mw.auth, ctrl.pages)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Error handling
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use(function (req, res, next) {
 | 
	
		
			
				|  |  | -  var err = new Error('Not Found')
 | 
	
		
			
				|  |  | -  err.status = 404
 | 
	
		
			
				|  |  | -  next(err)
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.use(function (err, req, res, next) {
 | 
	
		
			
				|  |  | -  res.status(err.status || 500)
 | 
	
		
			
				|  |  | -  res.render('error', {
 | 
	
		
			
				|  |  | -    message: err.message,
 | 
	
		
			
				|  |  | -    error: wiki.IS_DEBUG ? err : {}
 | 
	
		
			
				|  |  | -  })
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Start HTTP server
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -wiki.logger.info('Starting HTTP/WS server on port ' + wiki.config.port + '...')
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -app.set('port', wiki.config.port)
 | 
	
		
			
				|  |  | -var server = http.createServer(app)
 | 
	
		
			
				|  |  | -var io = socketio(server)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -server.listen(wiki.config.port)
 | 
	
		
			
				|  |  | -server.on('error', (error) => {
 | 
	
		
			
				|  |  | -  if (error.syscall !== 'listen') {
 | 
	
		
			
				|  |  | -    throw error
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  // handle specific listen errors with friendly messages
 | 
	
		
			
				|  |  | -  switch (error.code) {
 | 
	
		
			
				|  |  | -    case 'EACCES':
 | 
	
		
			
				|  |  | -      wiki.logger.error('Listening on port ' + wiki.config.port + ' requires elevated privileges!')
 | 
	
		
			
				|  |  | -      return process.exit(1)
 | 
	
		
			
				|  |  | -    case 'EADDRINUSE':
 | 
	
		
			
				|  |  | -      wiki.logger.error('Port ' + wiki.config.port + ' is already in use!')
 | 
	
		
			
				|  |  | -      return process.exit(1)
 | 
	
		
			
				|  |  | -    default:
 | 
	
		
			
				|  |  | -      throw error
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | +if (cluster.isMaster) {
 | 
	
		
			
				|  |  | +  wiki.logger.info('Wiki.js is initializing...')
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -server.on('listening', () => {
 | 
	
		
			
				|  |  | -  wiki.logger.info('HTTP/WS server started successfully! [RUNNING]')
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// WebSocket
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | +  require('./master')
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -io.use(passportSocketIo.authorize({
 | 
	
		
			
				|  |  | -  key: 'wikijs.sid',
 | 
	
		
			
				|  |  | -  store: sessionStore,
 | 
	
		
			
				|  |  | -  secret: wiki.config.sessionSecret,
 | 
	
		
			
				|  |  | -  cookieParser,
 | 
	
		
			
				|  |  | -  success: (data, accept) => {
 | 
	
		
			
				|  |  | -    accept()
 | 
	
		
			
				|  |  | -  },
 | 
	
		
			
				|  |  | -  fail: (data, message, error, accept) => {
 | 
	
		
			
				|  |  | -    accept()
 | 
	
		
			
				|  |  | +  for (let i = 0; i < numCPUs; i++) {
 | 
	
		
			
				|  |  | +    cluster.fork()
 | 
	
		
			
				|  |  |    }
 | 
	
		
			
				|  |  | -}))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -io.on('connection', ctrl.ws)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Start child processes
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -let bgAgent = fork(path.join(wiki.SERVERPATH, 'agent.js'))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -bgAgent.on('message', m => {
 | 
	
		
			
				|  |  | -  if (!m.action) {
 | 
	
		
			
				|  |  | -    return
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  switch (m.action) {
 | 
	
		
			
				|  |  | -    case 'searchAdd':
 | 
	
		
			
				|  |  | -      wiki.search.add(m.content)
 | 
	
		
			
				|  |  | -      break
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -// Graceful shutdown
 | 
	
		
			
				|  |  | -// ----------------------------------------
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -graceful.on('exit', () => {
 | 
	
		
			
				|  |  | -  wiki.logger.info('- SHUTTING DOWN - Terminating Background Agent...')
 | 
	
		
			
				|  |  | -  bgAgent.kill()
 | 
	
		
			
				|  |  | -  wiki.logger.info('- SHUTTING DOWN - Performing git sync...')
 | 
	
		
			
				|  |  | -  return global.git.resync().then(() => {
 | 
	
		
			
				|  |  | -    wiki.logger.info('- SHUTTING DOWN - Git sync successful. Now safe to exit.')
 | 
	
		
			
				|  |  | -    process.exit()
 | 
	
		
			
				|  |  | +  cluster.on('exit', (worker, code, signal) => {
 | 
	
		
			
				|  |  | +    wiki.logger.info(`Worker #${worker.id} died.`)
 | 
	
		
			
				|  |  |    })
 | 
	
		
			
				|  |  | -})
 | 
	
		
			
				|  |  | +} else {
 | 
	
		
			
				|  |  | +  wiki.logger.info(`Background Worker #${cluster.worker.id} is starting...`)
 | 
	
		
			
				|  |  | +  // require('./worker')
 | 
	
		
			
				|  |  | +}
 |