wikijs-fork/server/controllers/admin.js

'use strict'

/* global wiki */

var express = require('express')
var router = express.Router()
const Promise = require('bluebird')
const validator = require('validator')
const _ = require('lodash')
const axios = require('axios')
const path = require('path')
const fs = Promise.promisifyAll(require('fs-extra'))
const os = require('os')
const filesize = require('filesize.js')

/**
 * Admin
 */
router.get('/', (req, res) => {
  res.redirect('/admin/profile')
})

router.get('/profile', (req, res) => {
  if (res.locals.isGuest) {
    return res.render('error-forbidden')
  }

  res.render('pages/admin/profile', { adminTab: 'profile' })
})

router.post('/profile', (req, res) => {
  if (res.locals.isGuest) {
    return res.render('error-forbidden')
  }

  return wiki.db.User.findById(req.user.id).then((usr) => {
    usr.name = _.trim(req.body.name)
    if (usr.provider === 'local' && req.body.password !== '********') {
      let nPwd = _.trim(req.body.password)
      if (nPwd.length < 6) {
        return Promise.reject(new Error('New Password too short!'))
      } else {
        return wiki.db.User.hashPassword(nPwd).then((pwd) => {
          usr.password = pwd
          return usr.save()
        })
      }
    } else {
      return usr.save()
    }
  }).then(() => {
    return res.json({ msg: 'OK' })
  }).catch((err) => {
    res.status(400).json({ msg: err.message })
  })
})

router.get('/stats', (req, res) => {
  if (res.locals.isGuest) {
    return res.render('error-forbidden')
  }

  Promise.all([
    wiki.db.Entry.count(),
    wiki.db.UplFile.count(),
    wiki.db.User.count()
  ]).spread((totalEntries, totalUploads, totalUsers) => {
    return res.render('pages/admin/stats', {
      totalEntries, totalUploads, totalUsers, adminTab: 'stats'
    }) || true
  }).catch((err) => {
    throw err
  })
})

router.get('/users', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.render('error-forbidden')
  }

  wiki.db.User.find({})
    .select('-password -rights')
    .sort('name email')
    .exec().then((usrs) => {
      res.render('pages/admin/users', { adminTab: 'users', usrs })
    })
})

router.get('/users/:id', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.render('error-forbidden')
  }

  if (!validator.isMongoId(req.params.id)) {
    return res.render('error-forbidden')
  }

  wiki.db.User.findById(req.params.id)
    .select('-password -providerId')
    .exec().then((usr) => {
      let usrOpts = {
        canChangeEmail: (usr.email !== 'guest' && usr.provider === 'local' && usr.email !== req.app.locals.appconfig.admin),
        canChangeName: (usr.email !== 'guest'),
        canChangePassword: (usr.email !== 'guest' && usr.provider === 'local'),
        canChangeRole: (usr.email !== 'guest' && !(usr.provider === 'local' && usr.email === req.app.locals.appconfig.admin)),
        canBeDeleted: (usr.email !== 'guest' && !(usr.provider === 'local' && usr.email === req.app.locals.appconfig.admin))
      }

      res.render('pages/admin/users-edit', { adminTab: 'users', usr, usrOpts })
    }).catch(err => { // eslint-disable-line handle-callback-err
      return res.status(404).end() || true
    })
})

/**
 * Create / Authorize a new user
 */
router.post('/users/create', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.status(401).json({ msg: 'Unauthorized' })
  }

  let nUsr = {
    email: _.toLower(_.trim(req.body.email)),
    provider: _.trim(req.body.provider),
    password: req.body.password,
    name: _.trim(req.body.name)
  }

  if (!validator.isEmail(nUsr.email)) {
    return res.status(400).json({ msg: 'Invalid email address' })
  } else if (!validator.isIn(nUsr.provider, ['local', 'google', 'windowslive', 'facebook', 'github', 'slack'])) {
    return res.status(400).json({ msg: 'Invalid provider' })
  } else if (nUsr.provider === 'local' && !validator.isLength(nUsr.password, { min: 6 })) {
    return res.status(400).json({ msg: 'Password too short or missing' })
  } else if (nUsr.provider === 'local' && !validator.isLength(nUsr.name, { min: 2 })) {
    return res.status(400).json({ msg: 'Name is missing' })
  }

  wiki.db.User.findOne({ email: nUsr.email, provider: nUsr.provider }).then(exUsr => {
    if (exUsr) {
      return res.status(400).json({ msg: 'User already exists!' }) || true
    }

    let pwdGen = (nUsr.provider === 'local') ? wiki.db.User.hashPassword(nUsr.password) : Promise.resolve(true)
    return pwdGen.then(nPwd => {
      if (nUsr.provider !== 'local') {
        nUsr.password = ''
        nUsr.name = '-- pending --'
      } else {
        nUsr.password = nPwd
      }

      nUsr.rights = [{
        role: 'read',
        path: '/',
        exact: false,
        deny: false
      }]

      return wiki.db.User.create(nUsr).then(() => {
        return res.json({ ok: true })
      })
    }).catch(err => {
      wiki.logger.warn(err)
      return res.status(500).json({ msg: err })
    })
  }).catch(err => {
    wiki.logger.warn(err)
    return res.status(500).json({ msg: err })
  })
})

router.post('/users/:id', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.status(401).json({ msg: wiki.lang.t('errors:unauthorized') })
  }

  if (!validator.isMongoId(req.params.id)) {
    return res.status(400).json({ msg: wiki.lang.t('errors:invaliduserid') })
  }

  return wiki.db.User.findById(req.params.id).then((usr) => {
    usr.name = _.trim(req.body.name)
    usr.rights = JSON.parse(req.body.rights)
    if (usr.provider === 'local' && req.body.password !== '********') {
      let nPwd = _.trim(req.body.password)
      if (nPwd.length < 6) {
        return Promise.reject(new Error(wiki.lang.t('errors:newpasswordtooshort')))
      } else {
        return wiki.db.User.hashPassword(nPwd).then((pwd) => {
          usr.password = pwd
          return usr.save()
        })
      }
    } else {
      return usr.save()
    }
  }).then((usr) => {
    // Update guest rights for future requests
    if (usr.provider === 'local' && usr.email === 'guest') {
      wiki.rights.guest = usr
    }
    return usr
  }).then(() => {
    return res.json({ msg: 'OK' })
  }).catch((err) => {
    res.status(400).json({ msg: err.message })
  })
})

/**
 * Delete / Deauthorize a user
 */
router.delete('/users/:id', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.status(401).json({ msg: wiki.lang.t('errors:unauthorized') })
  }

  if (!validator.isMongoId(req.params.id)) {
    return res.status(400).json({ msg: wiki.lang.t('errors:invaliduserid') })
  }

  return wiki.db.User.findByIdAndRemove(req.params.id).then(() => {
    return res.json({ ok: true })
  }).catch((err) => {
    res.status(500).json({ ok: false, msg: err.message })
  })
})

router.get('/settings', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.render('error-forbidden')
  }
  res.render('pages/admin/settings', { adminTab: 'settings' })
})

router.get('/system', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.render('error-forbidden')
  }

  let hostInfo = {
    cpus: os.cpus(),
    hostname: os.hostname(),
    nodeversion: process.version,
    os: `${os.type()} (${os.platform()}) ${os.release()} ${os.arch()}`,
    totalmem: filesize(os.totalmem()),
    cwd: process.cwd()
  }

  fs.readJsonAsync(path.join(wiki.ROOTPATH, 'package.json')).then(packageObj => {
    axios.get('https://api.github.com/repos/Requarks/wiki/releases/latest').then(resp => {
      let sysversion = {
        current: 'v' + packageObj.version,
        latest: resp.data.tag_name,
        latestPublishedAt: resp.data.published_at
      }

      res.render('pages/admin/system', { adminTab: 'system', hostInfo, sysversion })
    }).catch(err => {
      wiki.logger.warn(err)
      res.render('pages/admin/system', { adminTab: 'system', hostInfo, sysversion: { current: 'v' + packageObj.version } })
    })
  })
})

router.post('/system/install', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.render('error-forbidden')
  }

  // let sysLib = require(path.join(ROOTPATH, 'libs/system.js'))
  // sysLib.install('v1.0-beta.7')
  res.status(400).send('Sorry, Upgrade/Re-Install via the web UI is not yet ready. You must use the npm upgrade method in the meantime.').end()
})

router.get('/theme', (req, res) => {
  if (!res.locals.rights.manage) {
    return res.render('error-forbidden')
  }
  res.render('pages/admin/theme', { adminTab: 'theme' })
})

router.post('/theme', (req, res) => {
  if (res.locals.isGuest) {
    return res.render('error-forbidden')
  }

  if (!validator.isIn(req.body.primary, wiki.data.colors)) {
    return res.status(406).json({ msg: 'Primary color is invalid.' })
  } else if (!validator.isIn(req.body.alt, wiki.data.colors)) {
    return res.status(406).json({ msg: 'Alternate color is invalid.' })
  } else if (!validator.isIn(req.body.footer, wiki.data.colors)) {
    return res.status(406).json({ msg: 'Footer color is invalid.' })
  }

  wiki.config.theme.primary = req.body.primary
  wiki.config.theme.alt = req.body.alt
  wiki.config.theme.footer = req.body.footer
  wiki.config.theme.code.dark = req.body.codedark === 'true'
  wiki.config.theme.code.colorize = req.body.codecolorize === 'true'

  return res.json({ msg: 'OK' })
})

module.exports = router