const _ = require('lodash/fp') const pgp = require('pg-promise')() const db = require('./db') const BN = require('./bn') const plugins = require('./plugins') const logger = require('./logger') const mapValuesWithKey = _.mapValues.convert({cap: false}) module.exports = {post, monitorPending} const PENDING_INTERVAL = '1 day' const MAX_PENDING = 10 function post (tx, pi) { const TransactionMode = pgp.txMode.TransactionMode const isolationLevel = pgp.txMode.isolationLevel const tmSRD = new TransactionMode({tiLevel: isolationLevel.serializable}) function transaction (t) { const sql = 'select * from cash_in_txs where id=$1' const sql2 = 'select * from bills where cash_in_txs_id=$1' return t.oneOrNone(sql, [tx.id]) .then(row => { return t.any(sql2, [tx.id]) .then(billRows => { const oldTx = toObj(row) return preProcess(oldTx, tx, pi) .then(preProcessedTx => upsert(oldTx, preProcessedTx)) .then(vector => { return insertNewBills(billRows, tx) .then(newBills => _.concat(vector, [newBills])) }) }) }) } transaction.txMode = tmSRD return db.tx(transaction) .then(txVector => { const [oldTx,, newBills] = txVector const oldBills = oldTx ? oldTx.bills : [] return postProcess(txVector, pi) .then(changes => update(oldTx, changes)) .then(tx => _.merge({bills: _.concat(oldBills, newBills)}, tx)) }) } function nilEqual (a, b) { if (_.isNil(a) && _.isNil(b)) return true return undefined } function isMonotonic (oldField, newField, fieldKey) { if (_.isNil(newField)) return false if (_.isBoolean(oldField)) return oldField === newField || !oldField if (oldField.isBigNumber) return oldField.lte(newField) if (_.isNumber(oldField)) return oldField <= newField throw new Error(`Unexpected value: ${oldField}`) } function ensureRatchet (oldField, newField, fieldKey) { const monotonic = ['cryptoAtoms', 'fiat', 'send', 'sendConfirmed', 'operatorCompleted', 'timedout'] const free = ['sendPending', 'error', 'errorCode'] if (_.isNil(oldField)) return true if (_.includes(fieldKey, monotonic)) return isMonotonic(oldField, newField, fieldKey) if (_.includes(fieldKey, free)) { if (_.isNil(newField)) return false return true } logger.error('This field [%s] should never change', fieldKey) logger.error('oldField: %j', oldField) logger.error('newField: %j', newField) throw new Error(`This field [${fieldKey}] should never change`) } function diff (oldTx, newTx) { let updatedTx = {} if (!oldTx) throw new Error('oldTx must not be null') if (!newTx) throw new Error('newTx must not be null') _.forEach(fieldKey => { const oldField = oldTx[fieldKey] const newField = newTx[fieldKey] if (fieldKey === 'bills') return if (_.isEqualWith(nilEqual, oldField, newField)) return if (!ensureRatchet(oldField, newField, fieldKey)) { logger.warn('Value from lamassu-machine would violate ratchet [%s]', fieldKey) logger.warn('Old tx: %j', oldTx) logger.warn('New tx: %j', newTx) return } updatedTx[fieldKey] = newField }, _.keys(newTx)) return updatedTx } function toObj (row) { if (!row) return null const keys = _.keys(row) let newObj = {} keys.forEach(key => { const objKey = _.camelCase(key) if (key === 'crypto_atoms' || key === 'fiat') { newObj[objKey] = BN(row[key]) return } if (key === 'device_time') { newObj[objKey] = parseInt(row[key], 10) return } newObj[objKey] = row[key] }) newObj.direction = 'cashIn' return newObj } function convertBigNumFields (obj) { const convert = (value, key) => _.includes(key, ['cryptoAtoms', 'fiat']) ? value.toString() : value const convertKey = key => _.includes(key, ['cryptoAtoms', 'fiat']) ? key + '#' : key return _.mapKeys(convertKey, mapValuesWithKey(convert, obj)) } function pullNewBills (billRows, tx) { if (_.isEmpty(tx.bills)) return [] const toBill = _.mapKeys(_.camelCase) const bills = _.map(toBill, billRows) return _.differenceBy(_.get('id'), tx.bills, bills) } const massage = _.flow(_.omit(['direction', 'bills']), convertBigNumFields, _.mapKeys(_.snakeCase)) function insertNewBills (billRows, tx) { const bills = pullNewBills(billRows, tx) if (_.isEmpty(bills)) return Promise.resolve([]) const dbBills = _.map(massage, bills) const columns = _.keys(dbBills[0]) const sql = pgp.helpers.insert(dbBills, columns, 'bills') return db.none(sql) .then(() => bills) } function upsert (oldTx, newTx) { if (!oldTx) { return insert(newTx) .then(tx => [oldTx, tx]) } return update(oldTx, diff(oldTx, newTx)) .then(tx => [oldTx, tx]) } function insert (tx) { const dbTx = massage(tx) const sql = pgp.helpers.insert(dbTx, null, 'cash_in_txs') + ' returning *' return db.one(sql) .then(toObj) } function update (tx, changes) { if (_.isEmpty(changes)) return Promise.resolve(tx) const dbChanges = massage(changes) const sql = pgp.helpers.update(dbChanges, null, 'cash_in_txs') + pgp.as.format(' where id=$1', [tx.id]) + ' returning *' return db.one(sql) .then(toObj) } function registerTrades (pi, txVector) { const newBills = _.last(txVector) _.forEach(bill => pi.buy(bill), newBills) } function logAction (rec, tx) { const action = { tx_id: tx.id, action: tx.sendCompleted ? 'sendCoins' : 'sendCoinsError', error: rec.error, error_code: rec.errorCode, tx_hash: rec.txHash } const sql = pgp.helpers.insert(action, null, 'cash_in_actions') return db.none(sql) .then(_.constant(rec)) } function postProcess (txVector, pi) { const [oldTx, newTx] = txVector registerTrades(pi, txVector) const isClearToSend = newTx.send && !oldTx.sendPending && !oldTx.sendConfirmed if (isClearToSend) { return pi.sendCoins(newTx) .then(txHash => ({ txHash, sendConfirmed: true, sendTime: 'now()^', sendPending: false, error: null, errorCode: null })) .catch(err => ({ sendTime: 'now()^', error: err.message, errorCode: err.name, sendPending: false })) .then(r => logAction(r, newTx)) } return Promise.resolve({}) } function preProcess (oldTx, newTx, pi) { return new Promise(resolve => { if (!oldTx) return resolve(newTx) if (newTx.send && !oldTx.send) return resolve(_.set('sendPending', true, newTx)) return resolve(newTx) }) } function monitorPending (settings) { const sql = `select * from cash_in_txs where created > now() - interval $1 and send and not send_confirmed and not send_pending and not operator_completed order by created limit $2` const processPending = row => { const tx = toObj(row) const pi = plugins(settings, tx.deviceId) return post(tx, pi) .catch(logger.error) } return db.any(sql, [PENDING_INTERVAL, MAX_PENDING]) .then(rows => Promise.all(_.map(processPending, rows))) .catch(logger.error) }