lamassu-server/lib/machine-loader.js

const _ = require('lodash/fp')
const axios = require('axios')

const logger = require('./logger')
const db = require('./db')
const pairing = require('./pairing')
const notifier = require('./notifier')
const dbm = require('./postgresql_interface')
const configManager = require('./new-config-manager')
const settingsLoader = require('./new-settings-loader')

module.exports = {getMachineName, getMachines, getMachine, getMachineNames, setMachine}

function getMachines () {
  return db.any('select * from devices where display=TRUE order by created')
    .then(rr => rr.map(r => ({
      deviceId: r.device_id,
      cashbox: r.cashbox,
      cassette1: r.cassette1,
      cassette2: r.cassette2,
      version: r.version,
      model: r.model,
      pairedAt: new Date(r.created),
      lastPing: new Date(r.last_online),
      name: r.name,
      // TODO: we shall start using this JSON field at some point
      // location: r.location,
      paired: r.paired
    })))
}

function getConfig (defaultConfig) {
  if (defaultConfig) return Promise.resolve(defaultConfig)

  return settingsLoader.loadLatest().config
}

function getMachineNames (config) {
  const fullyFunctionalStatus = {label: 'Fully functional', type: 'success'}
  const unresponsiveStatus = {label: 'Unresponsive', type: 'error'}
  const stuckStatus = {label: 'Stuck', type: 'error'}
  
  return Promise.all([getMachines(), getConfig(config)])
    .then(([machines, config]) => Promise.all(
      [machines, notifier.checkPings(machines), dbm.machineEvents(), config]
    ))
    .then(([machines, pings, events, config]) => {
      const getStatus = (ping, stuck) => {
        if (ping && ping.age) return unresponsiveStatus

        if (stuck && stuck.age) return stuckStatus
        
        return fullyFunctionalStatus
      }

      const addName = r => {
        const cashOutConfig = configManager.getCashOut(r.deviceId, config)

        const cashOut = !!cashOutConfig.active
        
        const statuses = [
          getStatus(
            _.first(pings[r.deviceId]),
            _.first(notifier.checkStuckScreen(events, r.name))
          )
        ]

        return _.assign(r, {cashOut, statuses})
      }

      return _.map(addName, machines)
    })
}

/**
 * Given the machine id, get the machine name
 *
 * @name getMachineName
 * @function
 * @async
 *
 * @param {string} machineId machine id
 * @returns {string} machine name
 */
function getMachineName (machineId) {
  const sql = 'select * from devices where device_id=$1'
  return db.oneOrNone(sql, [machineId])
    .then(it => it.name)
}

function getMachine (machineId) {
  const sql = 'select * from devices where device_id=$1'
  return db.oneOrNone(sql, [machineId]).then(res => _.mapKeys(_.camelCase)(res))
}

function renameMachine (rec) {
  const sql = 'update devices set name=$1 where device_id=$2'
  return db.none(sql, [rec.newName, rec.deviceId])
}

function resetCashOutBills (rec) {
  const sql = `
  update devices set cassette1=$1, cassette2=$2 where device_id=$3;
  update notifications set read = 't' where device_id = $3 AND type = 'fiatBalance' AND read = 'f';
  `
  return db.none(sql, [rec.cassettes[0], rec.cassettes[1], rec.deviceId])
}

function emptyCashInBills (rec) {
  const sql = 'update devices set cashbox=0 where device_id=$1'
  return db.none(sql, [rec.deviceId])
}

function setCassetteBills (rec) {
  const sql = 'update devices set cashbox=$1, cassette1=$2, cassette2=$3 where device_id=$4'
  return db.none(sql, [rec.cashbox, rec.cassettes[0], rec.cassettes[1], rec.deviceId])
}

function unpair (rec) {
  return pairing.unpair(rec.deviceId)
}

function reboot (rec) {
  return axios.post(`http://localhost:3030/reboot?device_id=${rec.deviceId}`)
}

function shutdown (rec) {
  return axios.post(`http://localhost:3030/shutdown?device_id=${rec.deviceId}`)
}

function restartServices (rec) {
  return axios.post(`http://localhost:3030/restartServices?device_id=${rec.deviceId}`)
}

function setMachine (rec) {
  switch (rec.action) {
    case 'rename': return renameMachine(rec)
    case 'emptyCashInBills': return emptyCashInBills(rec)
    case 'resetCashOutBills': return resetCashOutBills(rec)
    case 'setCassetteBills': return setCassetteBills(rec)
    case 'unpair': return unpair(rec)
    case 'reboot': return reboot(rec)
    case 'shutdown': return shutdown(rec)
    case 'restartServices': return restartServices(rec)
    default: throw new Error('No such action: ' + rec.action)
  }
}