var mongoose = require('mongoose')
  , db = require('../lib/database')
  , Tx = require('../models/tx')  
  , Address = require('../models/address')  
  , settings = require('../lib/settings');


var COUNT = 5000; //number of blocks to index

function exit() {  
  mongoose.disconnect();
  process.exit(0);
}

var dbString = 'mongodb://' + settings.dbsettings.user;
dbString = dbString + ':' + settings.dbsettings.password;
dbString = dbString + '@' + settings.dbsettings.address;
dbString = dbString + ':' + settings.dbsettings.port;
dbString = dbString + "/IQUIDUS-BENCHMARK";

mongoose.connect(dbString, { useNewUrlParser: true, useCreateIndex: true, useUnifiedTopology: true }, function(err) {
  if (err) {
    console.log('Unable to connect to database: %s', dbString);
    console.log('Aborting');
    exit();
  }
  Tx.remove({}, function(err) { 
    Address.remove({}, function(err2) { 
      var s_timer = new Date().getTime();
      db.update_tx_db(settings.coin, 1, COUNT, settings.update_timeout, function(){
        var e_timer = new Date().getTime();
        Tx.count({}, function(txerr, txcount){
          Address.count({}, function(aerr, acount){
            var stats = {
              tx_count: txcount,
              address_count: acount,
              seconds: (e_timer - s_timer)/1000,
            };
            console.log(stats);
            exit();
          });
        });
      });
    });
  });
});