From 08df88213f66140d8412a3dc421a5bccbb18189b Mon Sep 17 00:00:00 2001
From: Juan P Lopez <dolcalmi@gmail.com>
Date: Thu, 17 Oct 2024 20:19:39 -0500
Subject: [PATCH] feat(core): update exporter metrics

---
 core/api/src/servers/exporter.ts | 115 +++++++++++++++++++++++++++++--
 1 file changed, 110 insertions(+), 5 deletions(-)

diff --git a/core/api/src/servers/exporter.ts b/core/api/src/servers/exporter.ts
index 51d26a8222..7b725c33d9 100644
--- a/core/api/src/servers/exporter.ts
+++ b/core/api/src/servers/exporter.ts
@@ -9,7 +9,7 @@ import {
   SECS_PER_5_MINS,
 } from "@/config"
 
-import { Lightning, OnChain } from "@/app"
+import { Lightning, OnChain, Prices } from "@/app"
 
 import { toSeconds } from "@/domain/primitives"
 
@@ -32,6 +32,7 @@ import { activateLndHealthCheck } from "@/services/lnd/health"
 import { ledgerAdmin, setupMongoConnection } from "@/services/mongodb"
 
 import { timeoutWithCancel } from "@/utils"
+import { displayAmountFromNumber, UsdDisplayCurrency } from "@/domain/fiat"
 
 const TIMEOUT_WALLET_BALANCE = 30000
 
@@ -40,12 +41,35 @@ const logger = baseLogger.child({ module: "exporter" })
 const prefix = "galoy"
 
 const main = async () => {
-  const { getLiabilitiesBalance, getLndBalance, getBitcoindBalance, getOnChainBalance } =
-    ledgerAdmin
+  const { getLndBalance, getBitcoindBalance, getOnChainBalance } = ledgerAdmin
+
+  createGauge({
+    name: "assets",
+    description: "how much money (BTC) is on books",
+    collect: async () => {
+      const getDealerBalance = async (getId: () => Promise<WalletId>) => {
+        const walletId = await getId()
+        return getWalletBalance(walletId)
+      }
+
+      const btcAssets = await ledgerAdmin.getAssetsBalance()
+      const dealerBtcLiabilities = await getDealerBalance(getDealerBtcWalletId)
+
+      // Dealer BTC liabilities must be deducted from assets because
+      // Stablesats deposits and withdrawals are processed directly through Bria.
+      return Math.abs(btcAssets) - dealerBtcLiabilities
+    },
+  })
+
   createGauge({
     name: "liabilities",
-    description: "how much money customers has",
-    collect: getLiabilitiesBalance,
+    description: "how much money (BTC) customers has",
+    collect: async () => {
+      const liabilities = await getUserLiabilities()
+      if (liabilities instanceof Error) return 0
+
+      return liabilities
+    },
   })
 
   createGauge({
@@ -109,6 +133,12 @@ const main = async () => {
     },
   })
 
+  createGauge({
+    name: "realAssetsVsLiabilities",
+    description: "do we have enough Bitcoin to cover users' liabilities",
+    collect: getRealAssetsVersusLiabilities,
+  })
+
   createGauge({
     name: "assetsEqLiabilities",
     description: "do we have a balanced book",
@@ -311,6 +341,81 @@ const getAssetsLiabilitiesDifference = async () => {
   return assets + liabilities
 }
 
+const getUserLiabilities = async () => {
+  const getDealerBalance = async (getId: () => Promise<WalletId>) => {
+    const walletId = await getId()
+    return getWalletBalance(walletId)
+  }
+
+  const btcLiabilities = await ledgerAdmin.getLiabilitiesBalance()
+  const dealerBtcLiabilities = await getDealerBalance(getDealerBtcWalletId)
+
+  // Dealer BTC liabilities must be deducted from liabilities because
+  // Stablesats deposits and withdrawals are processed directly through Bria.
+  const customerBtcLiabilities = btcLiabilities - dealerBtcLiabilities
+
+  const dealerUsdLiabilities = await getDealerBalance(getDealerUsdWalletId)
+  logger.info(
+    {
+      btcLiabilities,
+      dealerBtcLiabilities,
+      customerBtcLiabilities,
+      dealerUsdLiabilities,
+    },
+    "getUserLiabilities balances",
+  )
+  const dealerUsdLiabilitiesDisplay = displayAmountFromNumber({
+    amount: Math.abs(dealerUsdLiabilities),
+    currency: UsdDisplayCurrency,
+  })
+  if (dealerUsdLiabilitiesDisplay instanceof Error) return dealerUsdLiabilitiesDisplay
+
+  const dealerUsdLiabilitiesInSatsAmount = await Prices.estimateWalletsAmounts({
+    amount: Number(dealerUsdLiabilitiesDisplay.displayInMajor),
+    currency: UsdDisplayCurrency,
+  })
+  logger.info(
+    {
+      mayor: dealerUsdLiabilitiesDisplay.displayInMajor,
+      currency: UsdDisplayCurrency,
+      dealerUsdLiabilitiesInSatsAmount,
+    },
+    "getUserLiabilities usd balances",
+  )
+  if (dealerUsdLiabilitiesInSatsAmount instanceof Error)
+    return dealerUsdLiabilitiesInSatsAmount
+
+  return (
+    customerBtcLiabilities + Number(dealerUsdLiabilitiesInSatsAmount.btcSatAmount.amount)
+  )
+}
+
+const getRealAssetsVersusLiabilities = async () => {
+  const [liabilitiesBalance, lndBalance, coldStorage, hotBalance] = await Promise.all([
+    getUserLiabilities(),
+    Lightning.getTotalBalance(),
+    OnChain.getColdBalance(),
+    OnChain.getHotBalance(),
+  ])
+
+  const liabilities = liabilitiesBalance instanceof Error ? 0 : liabilitiesBalance
+  const lnd = lndBalance instanceof Error ? 0 : lndBalance
+  const briaHot = hotBalance instanceof Error ? 0 : Number(hotBalance.amount)
+  const briaCold = coldStorage instanceof Error ? 0 : Number(coldStorage.amount)
+
+  logger.error(
+    { liabilities, lnd, briaHot, briaCold },
+    "getRealAssetsVersusLiabilities balances",
+  )
+
+  return (
+    lnd + // physical assets
+    briaCold + // physical assets
+    briaHot + // physical assets
+    liabilities // value in accounting
+  )
+}
+
 export const getBookingVersusRealWorldAssets = async () => {
   const [lightning, bitcoin, onChain, lndBalance, coldStorage, hotBalance] =
     await Promise.all([