insight-ui-zcash/lib/Sync.js

'use strict';

require('classtool');


function spec() {
  var mongoose        = require('mongoose');
  var config          = require('../config/config');
  var sockets         = require('../app/controllers/socket.js');
  var BlockDb         = require('./BlockDb').class();
  var TransactionDb   = require('./TransactionDb').class();
  var async           = require('async');


  function Sync() {
    this.blockDb = new BlockDb();
    this.txDb    = new TransactionDb();
  }

  Sync.prototype.init = function(opts, cb) {
    var self = this;

    self.opts = opts;

    if (!(opts && opts.skipDbConnection)) {

      if (mongoose.connection.readyState !== 1) {
        mongoose.connect(config.db, function(err) {
          if (err) {
            console.log('CRITICAL ERROR: connecting to mongoDB:',err);
            return (err);
          }
        });
      }

      self.db = mongoose.connection;

      self.db.on('error', function(err) {
          console.log('MongoDB ERROR:' + err);
          return cb(err);
      });

      self.db.on('disconnect', function(err) {
          console.log('MongoDB disconnect:' + err);
          return cb(err);
      });

      return self.db.once('open', function(err) {
        return cb(err);
      });
    }
    else return cb();
  };

  Sync.prototype.close = function() {
    if ( this.db && this.db.readyState ) {
      this.db.close();
    }
  };


  Sync.prototype.destroy = function(next) {
    var self = this;
    async.series([
      function(b) { try {self.blockDb.drop(b);} catch (e) { return b(); } },
      function(b) { try {self.TransactionDb.drop(b);} catch (e) { return b(); } },
    ], next);
  };


  Sync.prototype.hasBlock = function(hash, cb) {
    var self = this;

    return self.blockDb.has(hash, cb);
  };
 
  Sync.prototype.countNotOrphan = function(hash, cb) {
    var self = this;
    return self.blockDb.countNotOrphan(hash, cb);
  };
 

  Sync.prototype.storeBlock = function(block, cb) {
    var self = this;

    self.txDb.createFromBlock(block, function(err, insertedTxs, updateAddrs) {
      if (err) return cb(err);

      self.blockDb.add(block, function(err){
        if (err) return cb(err);
        self._handleBroadcast(block, insertedTxs, updateAddrs);
        return cb();
      });
    });
  };

  Sync.prototype.checkOrphan = function(fromBlock, toBlock, c) {
    var self = this;

    var hash = fromBlock;

    var co = 0;
    var limit = 10;
    var cont = 1;

    async.whilst(
      function () {
        if (++co > limit) {
console.log('[Sync.js.109] WARN: Reach reog depth limit'); //TODO
        }
        return cont && hash && hash !== toBlock && co < limit;
      },
      function (w_c) {
        //check with RPC if the block is mainchain
        self.blockDb.fromHashWithInfo(hash, function (err, info) {

          if (!info) {
            console.log('[Sync.js.107:hash:ORPHAN]',hash); //TODO
            self.txDb.setOrphan(hash, function(err) {
              if (err) return w_c(err);
              self.blockDb.setOrphan(hash, function(err, prevHash){
                hash = prevHash;
                return w_c(err);
              });
            });
          }
          else {
            console.log('[Sync.js.107:hash:NOT ORPHAN]',hash); //TODO
            cont = 0;
            return w_c();
          }
        });
      },
      function (err) {
        return c(err);
      }
    );
  };

  Sync.prototype._handleBroadcast = function(hash, inserted_txs, updated_addrs) {
    var self = this;

    if (hash && self.opts.broadcast_blocks) {
      sockets.broadcast_block({hash: hash});
    }

    if (inserted_txs && self.opts.broadcast_txs) {
      inserted_txs.forEach(function(tx) {
        sockets.broadcast_tx(tx);
      });
    }

    if (updated_addrs && self.opts.broadcast_addresses) {
      updated_addrs.forEach(function(addr, txs){
        txs.forEach(function(addr, t){
          sockets.broadcast_address_tx(addr, {'txid': t});

        });
      });
    }
  };



  Sync.prototype.storeTxs = function(txs, cb) {
    var self = this;

    // TODO
    self.txDb.createFromTxs(txs, function(err, inserted_txs, updated_addrs) {
      if (err) return cb(err);

      self._handleBroadcast(null, inserted_txs, updated_addrs);
      return cb(err);
    });
  };
  return Sync;
}
module.defineClass(spec);