server.js

const http = require('http')
const express = require('express')
const WebSocket = require('ws')
const bodyParser = require('body-parser')
const moment = require('moment')
const {v4: uuid} = require('uuid')

const app = express()
const {log, conf, loadIPdatabase} = require('./utils.js')
const openvpn = require('./openvpn.js')
const db = require('./database.js')

// HTTP authentication
if (conf.get('username') && conf.get('username').length)
  app.use((req, res, next) => {
    // Parse login and password from headers
    const b64auth = (req.headers.authorization || '').split(' ')[1] || ''
    const [login, password] = Buffer.from(b64auth, 'base64').toString().split(':')
    // Verify login and password are set and correct
    if (!login || !password || login !== conf.get('username') || password !== conf.get('password')) {
      res.set('WWW-Authenticate', 'Basic realm="401"')
      res.status(401).send('Authentication required.')
    } else
      next()
  })

app.get('/', (_, res) => res.sendFile(`${__dirname}/dist/index.html`))
app.get('/cfg', (_, res) => res.json(conf.get('web')))
app.use('/', express.static(`${__dirname}/dist`))
app.use(bodyParser.json())

let cityLookup
const clientUpdates = new Map()
const clients = new Map()
const servers = conf.get('servers') || []
const broadcast = data => clients.forEach(ws => ws.send(JSON.stringify(data)))
let logMutex = Promise.resolve()
const logEvent = event => {
  const logic = data => new Promise(resolve => {
    db.Log.findOne({
      where: {
        server: data.server,
        node: data.node,
        timestamp: {[db.op.between]: [data.timestamp - 60, data.timestamp + 60]}
      },
      order: [['timestamp', 'DESC']]
    }).then(entry => {
      if (!entry)
        db.Log.create(data).then(() => {
          broadcast(data)
          resolve()
        })
      else {
        if ((entry.event === 'disconnect' || entry.event === 'reconnect') && data.event === 'connect') {
          entry.event = 'reconnect'
          data.event = 'reconnect'
          entry.timestamp = moment().unix()
          entry.save().then(resolve)
        } else
          db.Log.create(data).then(resolve)
        broadcast(data)
      }
    })
  })
  logMutex.then(() => logMutex = logic(event))
}
const clientToEntry = client => {
  const loc = cityLookup(client.pub)
  client.node = client.cn
  if (client.node === undefined) {
    client.node = "Unknown_" + client.pub;
  }
  if (loc) {
    client.country_code = loc.country.iso_code
    client.country_name = loc.country.names.en
    client.lat = loc.location.latitude
    client.lon = loc.location.longitude
  }
  return client
}

const validateNumber = n => Number.isFinite(parseFloat(n, 10))

app.get('/time', (_, res) => res.json({time: moment().unix()}))
app.get('/servers', (_, res) => res.json(servers.map((server, idx) => ({name: server.name, id: idx}))))

const validateServer = (req, res, next) => {
  const serverId = req.params.id || req.params[0]
  if (!validateNumber(serverId))
    return res.sendStatus(400)
  if (!servers[serverId])
    return res.sendStatus(404)
  next()
}
app.post('/server/:id/disconnect/:cid', validateServer, (req, res) => {
  if (!validateNumber(req.params.cid))
    return res.sendStatus(400)
  const cid = parseInt(req.params.cid, 10)
  servers[req.params.id].vpnclient.disconnect(cid)
  res.sendStatus(200)
})
app.get('/entries/:id', validateServer, (req, res) => {
  const customSort = items => {
    if (!items)
      return []
    const keys = new Map()
    items.forEach(itm => keys.set(itm.node, `${itm.country_name}${itm.node}${itm.pub}`))
    items.sort((a, b) => {
      if (keys.get(a.node) < keys.get(b.node))
        return -1
      if (keys.get(a.node) > keys.get(b.node))
        return 1
      return 0
    })
    return items
  }
  res.json(customSort(servers[req.params.id].entries))
})
// /log/:id/size/:search
app.get(/\/log\/([0-9]*)\/size\/(.*)/, validateServer, (req, res) => {
  const needle = `%${(req.params[1].trim() || '')}%`
  db.Log.count({
    where: {
      server: req.params[0], node: {[db.op.like]: needle}
    }
  }).then(size => res.json({value: size}))
})
// /log/:id/:page/:size/:search
app.get(/\/log\/([0-9]*)\/([0-9]*)\/([-0-9]*)\/(.*)/, validateServer, (req, res) => {
  if (!validateNumber(req.params[1]) || !validateNumber(req.params[2]))
    return res.sendStatus(400)
  const needle = `%${(req.params[3].trim() || '')}%`
  const page = parseInt(req.params[1], 10)
  const size = parseInt(req.params[2], 10)
  const query = db.Log.findAll({
    where: {server: req.params[0], node: {[db.op.like]: needle}}, offset: (page - 1) * size, limit: size, order: [['timestamp', 'DESC']]
  })
  query.then(data => res.json(data))
})

const httpServer = http.createServer(app)
const wss = new WebSocket.Server({server: httpServer})

wss.on('connection', ws => {
  const id = uuid()
  clients.set(id, ws)
  ws.on('close', () => clients.delete(id))
  ws.on('error', err => {
    log(err)
    clients.delete(id)
  })
})

Promise.all([loadIPdatabase(), db.init()]).then(results => {
  [cityLookup] = results
  setInterval(() => {
    broadcast(Array.from(clientUpdates.values()))
    clientUpdates.clear()
  }, 5000)
  servers.forEach((server, idx) => {
    server.entries = []
    const client = new openvpn(server.host, server.man_port, server.man_pwd, server.netmask)
    client.getClients().then(clts => server.entries = clts.map(clientToEntry))
    client.on('client-connect', cl => {
      const entry = clientToEntry(cl)
      server.entries = server.entries.filter(itm => itm.node !== entry.node)
      logEvent({
        server: idx, event: 'connect', timestamp: moment().unix(), ...entry
      })
      server.entries.push(entry)
    })
    client.on('client-disconnect', cl => {
      const entry = Object.assign(clientToEntry(cl), {event: 'disconnect'})
      const oLen = server.entries.length
      server.entries = server.entries.filter(itm => itm.node !== entry.node)
      if (oLen !== server.entries.length)
        logEvent({
          server: idx, event: 'disconnect', timestamp: moment().unix(), ...entry
        })
    })
    client.on('client-update', cl => {
      const entry = Object.assign(clientToEntry(cl), {server: idx, event: 'update', timestamp: moment().unix()})
      clientUpdates.set(entry.cid, entry)
    })
    server.vpnclient = client
  })
  httpServer.listen({host: conf.get('bind'), port: conf.get('port'), exclusive: true})
})