refactor

3 files changed, 261 insertions, 0 deletions

diff --git a/routes/bitcoin/fee.js b/routes/bitcoin/fee.js
new file mode 100644
index 00000000..259f2329
--- /dev/null
+++ b/routes/bitcoin/fee.js
@@ -0,0 +1,25 @@
+module.exports = async (req, res) => {
+  let { user } = req;
+  let { address, amount, feeRate } = req.body;
+  let tx, fee;
+
+  try {
+    amount = parseInt(amount);
+
+    tx = await bc.createRawTransaction([], {
+      [address]: (amount / SATS).toFixed(8)
+    });
+
+    let params = {
+      subtractFeeFromOutputs: amount === user.balance ? [0] : []
+    } 
+
+    if (feeRate) params["feeRate"] = (parseInt(feeRate) / SATS * 1000).toFixed(8);
+
+    tx = await bc.fundRawTransaction(tx, params);
+    res.send({ tx });
+  } catch (e) {
+    l.error("bitcoin fee estimation error", e);
+    return res.status(500).send(e.message);
+  }
+};
diff --git a/routes/bitcoin/receive.js b/routes/bitcoin/receive.js
new file mode 100644
index 00000000..b5ab4298
--- /dev/null
+++ b/routes/bitcoin/receive.js
@@ -0,0 +1,143 @@
+const reverse = require("buffer-reverse");
+const zmq = require("zeromq");
+const { Op } = require("sequelize");
+
+const bitcoin = require("bitcoinjs-lib");
+
+const zmqRawBlock = zmq.socket("sub");
+zmqRawBlock.connect(config.bitcoin.zmqrawblock);
+zmqRawBlock.subscribe("rawblock");
+
+const zmqRawTx = zmq.socket("sub");
+zmqRawTx.connect(config.bitcoin.zmqrawtx);
+zmqRawTx.subscribe("rawtx");
+
+let NETWORK = bitcoin.networks[config.bitcoin.network === "mainnet" ? "bitcoin" : config.bitcoin.network];
+
+zmqRawTx.on("message", async (topic, message, sequence) => {
+  const hex = message.toString("hex");
+  let tx = bitcoin.Transaction.fromHex(message);
+  let hash = reverse(tx.getHash()).toString("hex");
+
+  if (payments.includes(hash)) return;
+
+  Promise.all(
+    tx.outs.map(async o => {
+      const { value } = o;
+
+      let address;
+      try {
+        address = bitcoin.address.fromOutputScript(
+          o.script,
+          NETWORK
+        );
+      } catch (e) {
+        return;
+      }
+
+      if (Object.keys(addresses).includes(address)) {
+        payments.push(hash);
+
+        let user = await db.User.findOne({
+          where: {
+            username: addresses[address]
+          }
+        });
+
+        let invoices = await db.Payment.findAll({
+          limit: 1,
+          where: {
+            address,
+            received: null,
+            amount: {
+              [Op.gt]: 0
+            }
+          },
+          order: [["createdAt", "DESC"]]
+        });
+
+        let tip = null;
+        if (invoices.length) tip = invoices[0].tip;
+
+        let confirmed = false;
+
+        user.address = await bc.getNewAddress("", "bech32");
+        user.pending += value;
+
+        await user.save();
+        emit(user.username, "user", user);
+        
+        addresses[user.address] = user.username;
+
+        const payment = await db.Payment.create({
+          user_id: user.id,
+          hash,
+          amount: value,
+          currency: user.currency,
+          rate: app.get("rates")[user.currency],
+          received: true,
+          tip,
+          confirmed,
+          address,
+          asset: 'BTC',
+        });
+
+        l.info("bitcoin detected", user.username, o.value);
+        emit(user.username, "payment", payment);
+      }
+    })
+  );
+});
+
+let queue = {};
+
+zmqRawBlock.on("message", async (topic, message, sequence) => {
+  const payments = await db.Payment.findAll({
+    where: { confirmed: false }
+  });
+
+  const hashes = payments.map(p => p.hash);
+
+  let block = bitcoin.Block.fromHex(message.toString("hex"));
+  block.transactions.map(tx => {
+    let hash = reverse(tx.getHash()).toString("hex");
+    if (hashes.includes(hash)) queue[hash] = 1;
+  }); 
+});
+
+setInterval(async () => {
+  let arr = Object.keys(queue);
+  for (let i = 0; i < arr.length; i++) {
+    let hash = arr[i];
+
+    let p = await db.Payment.findOne({
+      include: [{ model: db.User, as: "user" }],
+      where: { hash, confirmed: 0 }
+    })
+
+    p.confirmed = 1;
+
+    let user = await p.getUser();
+    user.balance += p.amount;
+    user.pending -= Math.min(user.pending, p.amount);
+    emit(user.username, "user", user);
+
+    await user.save();
+    await p.save();
+
+    let payments = await db.Payment.findAll({
+      where: { 
+        user_id: user.id,
+        received: { 
+          [Op.ne]: null
+        },
+      },
+      order: [['id', 'DESC']],
+      limit: 12
+    });
+
+    l.info("bitcoin confirmed", user.username, p.amount);
+    emit(user.username, "payments", payments);
+    delete queue[hash];
+  }
+}, 1000);
diff --git a/routes/bitcoin/send.js b/routes/bitcoin/send.js
new file mode 100644
index 00000000..3c545178
--- /dev/null
+++ b/routes/bitcoin/send.js
@@ -0,0 +1,93 @@
+const bitcoin = require("bitcoinjs-lib");
+const reverse = require("buffer-reverse");
+
+module.exports = async (req, res) => {
+  let { user } = req;
+  let { address, tx } = req.body;
+  let { hex } = tx;
+
+  let fee = toSats(tx.fee);
+
+  const isChange = async address => 
+    !((await bc.getAddressInfo(address)).ismine) ||
+    !Object.keys(addresses).includes(address) || 
+    address === user.address;
+
+  tx = await bc.decodeRawTransaction(hex);
+
+  let total = 0;
+  let change = 0;
+
+  for (let i = 0; i < tx.vout.length; i++) {
+    let o = tx.vout[i];
+    total += toSats(o.value);
+   
+    if (o.scriptPubKey.addresses) {
+      if ((await isChange(o.scriptPubKey.addresses[0]))) {
+        change += toSats(o.value);
+      }
+    }
+  } 
+
+  total = (total - change) + fee;
+  let amount = total - fee;
+
+  try {
+    await db.transaction(async transaction => {
+      let { balance } = await db.User.findOne({
+        where: {
+          username: user.username
+        },
+        lock: transaction.LOCK.UPDATE,
+        transaction
+      });
+
+      if (amount !== balance && total > balance) {
+        l.error("amount exceeds balance", amount, fee, balance);
+        throw new Error("insufficient funds");
+      }
+
+      user.balance -= total;
+      await user.save({ transaction });
+    });
+  } catch (e) {
+    l.warn("insufficient funds for bitcoin payment", e);
+    return res.status(500).send("Not enough satoshis");
+  }
+
+  try {
+    if (config.bitcoin.walletpass)
+      await bc.walletPassphrase(config.bitcoin.walletpass, 300);
+
+    hex = (await bc.signRawTransactionWithWallet(hex)).hex;
+    let txid = await bc.sendRawTransaction(hex);
+
+    await db.transaction(async transaction => {
+      await user.save({ transaction });
+      emit(user.username, "user", user);
+
+      const payment = await db.Payment.create(
+        {
+          amount: -total,
+          fee,
+          user_id: user.id,
+          hash: txid,
+          rate: app.get("rates")[user.currency],
+          currency: user.currency,
+          address,
+          confirmed: true,
+          received: false,
+          asset: "BTC"
+        },
+        { transaction }
+      );
+
+      l.info("sent bitcoin", user.username, total);
+      emit(user.username, "payment", payment);
+      res.send(payment);
+    });
+  } catch (e) {
+    l.error("error sending bitcoin", e);
+    return res.status(500).send(e.message);
+  }
+};