From f07486f0f6ed0cf3006c97d48db4ee87c1e321fd Mon Sep 17 00:00:00 2001
From: Chris Kleeschulte <chrisk@bitpay.com>
Date: Thu, 2 Feb 2017 11:38:29 -0500
Subject: [PATCH] Added mempool database encodings.

---
 lib/services/mempool/encoding.js | 213 ++++---------------------------
 1 file changed, 27 insertions(+), 186 deletions(-)

diff --git a/lib/services/mempool/encoding.js b/lib/services/mempool/encoding.js
index 1b127516..b793497c 100644
--- a/lib/services/mempool/encoding.js
+++ b/lib/services/mempool/encoding.js
@@ -23,9 +23,9 @@ Encoding.prototype.encodeMempoolAddressIndexKey = function(address, txid) {
   return Buffer.concat(buffers);
 };
 
-Encoding.prototype.encodeMempoolAddressIndexValue = function(buffer) {
+Encoding.prototype.decodeMempoolAddressIndexKey = function(buffer) {
   var reader = new BufferReader(buffer);
-  reader.read(1);
+  reader.read(3);
 
   var addressSize = reader.readUInt8();
   var address = reader.read(addressSize).toString('utf8');
@@ -37,197 +37,38 @@ Encoding.prototype.encodeMempoolAddressIndexValue = function(buffer) {
   };
 };
 
-Encoding.prototype.encodeMempoolAddressIndexValue = function(address, txid) {
-};
-Encoding.prototype.encodeWalletTransactionValue = function(txid) {
-  return new Buffer(txid, 'hex');
-};
-
-Encoding.prototype.decodeWalletTransactionValue = function(buffer) {
-  return buffer.toString('hex');
-};
-
-Encoding.prototype.encodeWalletUtxoKey = function(walletId, txid, outputIndex) {
-  var buffers = [this.servicePrefix];
-
-  var walletIdSizeBuffer = new Buffer(1);
-  walletIdSizeBuffer.writeUInt8(walletId.length);
-  var walletIdBuffer = new Buffer(walletId, 'utf8');
-
-  buffers.push(walletIdSizeBuffer);
-  buffers.push(walletIdBuffer);
-
-  if(txid) {
-    var txidBuffer = new Buffer(txid, 'hex');
-    buffers.push(txidBuffer);
-  }
-
-  if(outputIndex !== undefined) {
-    var outputIndexBuffer = new Buffer(4);
-    outputIndexBuffer.writeUInt32BE(outputIndex);
-    buffers.push(outputIndexBuffer);
-  }
-
-  return Buffer.concat(buffers);
-};
-
-Encoding.prototype.decodeWalletUtxoKey = function(buffer) {
-  var reader = new BufferReader(buffer);
-  reader.read(1);
-
-  var walletIdSize = reader.readUInt8();
-  var walletId = reader.read(walletIdSize).toString('utf8');
-  var txid = reader.read(32).toString('hex');
-  var outputIndex = reader.readUInt32BE();
-  return {
-    walletId: walletId,
-    txid: txid,
-    outputIndex: outputIndex
-  };
-};
-
-Encoding.prototype.encodeWalletUtxoValue = function(height, satoshis, scriptBuffer) {
-  var heightBuffer = new Buffer(4);
-  heightBuffer.writeUInt32BE(height);
-  var satoshisBuffer = new Buffer(8);
-  satoshisBuffer.writeDoubleBE(satoshis);
-  return Buffer.concat([height, satoshisBuffer, scriptBuffer]);
-};
-
-Encoding.prototype.decodeWalletUtxoValue = function(buffer) {
-  var reader = new BufferReader(buffer);
-  var height = reader.readUInt32BE();
-  var satoshis = reader.readDoubleBE();
-  var scriptBuffer = reader.read(buffer.length - 12);
-  return {
-    height: height,
-    satoshis: satoshis,
-    script: scriptBuffer
-  };
-};
-
-Encoding.prototype.encodeWalletUtxoSatoshisKey = function(walletId, satoshis, txid, outputIndex) {
-  var buffers = [this.servicePrefix];
-
-  var walletIdSizeBuffer = new Buffer(1);
-  walletIdSizeBuffer.writeUInt8(walletId.length);
-  var walletIdBuffer = new Buffer(walletId, 'utf8');
-
-  buffers.push(walletIdSizeBuffer);
-  buffers.push(walletIdBuffer);
-
-  if(satoshis !== undefined) {
-    var satoshisBuffer = new Buffer(8);
-    satoshisBuffer.writeUInt32BE(satoshis);
-    buffers.push(satoshisBuffer);
-  }
-
-  if(txid) {
-    var txidBuffer = new Buffer(txid, 'hex');
-    buffers.push(txidBuffer);
-  }
-
-  if(outputIndex !== undefined) {
-    var outputIndexBuffer = new Buffer(4);
-    outputIndexBuffer.writeUInt32BE(outputIndex);
-    buffers.push(outputIndexBuffer);
-  }
-
-  return Buffer.concat(buffers);
-};
-
-Encoding.prototype.decodeWalletUtxoSatoshisKey = function(buffer) {
-  var reader = new BufferReader(buffer);
-  reader.read(1);
-
-  var walletIdSizeBuffer = reader.readUInt8();
-  var walletId = reader.read(walletIdSizeBuffer).toString('utf8');
-  var satoshis = reader.readDoubleBE();
-  var txid = reader.read(32).toString('hex');
-  var outputIndex = reader.readUInt32BE();
-  return {
-    walletId: walletId,
-    satoshis: satoshis,
-    txid: txid,
-    outputIndex: outputIndex
-  };
-};
-
-Encoding.prototype.encodeWalletUtxoSatoshisValue = function(height, scriptBuffer) {
-  var heightBuffer = new Buffer(4);
-  heightBuffer.writeUInt32BE(height);
-  return Buffer.concat([height, scriptBuffer]);
-};
-
-Encoding.prototype.decodeWalletUtxoSatoshisValue = function(buffer) {
-  var reader = new BufferReader(buffer);
-  var height = reader.readUInt32BE();
-  var scriptBuffer = reader.read(buffer.length - 4);
-  return {
-    height: height,
-    script: scriptBuffer
-  };
-};
-
-Encoding.prototype.encodeWalletAddressesKey = function(walletId) {
-  var prefix = new Buffer('00', 'hex');
-  var walletIdBuffer = new Buffer(walletId, 'hex');
-  return Buffer.concat([this.servicePrefix, prefix, walletIdBuffer]);
-};
-
-Encoding.prototype.decodeWalletAddressesKey = function(buffer) {
-  return buffer.slice(3).toString('hex');
-};
-
-Encoding.prototype.encodeWalletAddressesValue = function(addresses) {
-  var bufferList = [];
-  var addressesLengthBuffer = new Buffer(4);
-  addressesLengthBuffer.writeUInt32BE(addresses.length);
-  bufferList.push(addressesLengthBuffer);
-  for(var i = 0; i < addresses.length; i++) {
-    var addressSizeBuffer = new Buffer(1);
-    addressSizeBuffer.writeUInt8(addresses[i].length);
-    bufferList.push(addressSizeBuffer);
-    bufferList.push(new Buffer(addresses[i], 'utf8'));
-  }
-
-  return Buffer.concat(bufferList);
-};
-
-Encoding.prototype.decodeWalletAddressesValue = function(buffer) {
-  var reader = new BufferReader(buffer);
-  var addressesLength = reader.readUInt32BE();
-  var addresses = [];
-  var addressSize = 0;
-  for(var i = 0; i < addressesLength.length; i++) {
-    addressSize = reader.readUInt8(addressSize);
-    addresses.push(reader.read(addressSize).toString('utf8'));
-  }
-
-  return addresses;
-};
-
-Encoding.prototype.encodeWalletBalanceKey = function(walletId) {
+Encoding.prototype.encodeMempoolTransactionKey = function(txid) {
   var prefix = new Buffer('01', 'hex');
-  var walletIdBuffer = new Buffer(walletId, 'hex');
-  return Buffer.concat([this.servicePrefix, prefix, walletIdBuffer]);
+  var buffers = [this.servicePrefix, prefix];
+  var txidBuffer = new Buffer(txid, 'hex');
+  buffers.push(txidBuffer);
+  return Buffer.concat(buffers);
 };
 
-Encoding.prototype.decodeWalletBalanceKey = function(buffer) {
-  return buffer.slice(3).toString('hex');
+Encoding.prototype.decodeMempoolTransactionKey = function(buffer) {
+  return buffer.slice(4).toString('hex');
 };
 
-Encoding.prototype.encodeWalletBalanceValue = function(balance) {
-  var balanceBuffer = new Buffer(8);
-  balanceBuffer.writeUInt32BE(balance);
-  return balanceBuffer;
+Encoding.prototype.encodeMempoolTransactionValue = function(transaction) {
+  var inputValues = transaction.__inputValues || [];
+  var inputValuesBuffer = new Buffer(8 * inputValues.length);
+  for(var i = 0; i < inputValues.length; i++) {
+    inputValuesBuffer.writeDoubleBE(inputValues[i], i * 8);
+  }
+  var inputValuesLengthBuffer = new Buffer(2);
+  inputValuesLengthBuffer.writeUInt16BE(inputValues.length * 8);
+  return new Buffer.concat([inputValuesLengthBuffer, inputValuesBuffer, transaction.toBuffer()]);
 };
 
-Encoding.prototype.decodeWalletBalanceValue = function(buffer) {
-  var reader = new BufferReader(buffer);
-  var balance = reader.readDoubleBE();
-
-  return balance;
+Encoding.prototype.decodeMempoolTransactionValue = function(buffer) {
+  var inputValues = [];
+  var inputValuesLength = buffer.readUInt16BE();
+  for(var i = 0; i < inputValuesLength / 8; i++) {
+    inputValues.push(buffer.readDoubleBE(i * 8 + 2));
+  }
+  var transaction = new bitcore.Transaction(buffer.slice(inputValues.length * 8 + 2));
+  transaction.__inputValues = inputValues;
+  return transaction;
 };
 
 module.exports = Encoding;