Adjust monitoring to the latest tBTC SDK

monitoring/src/block-explorer.ts

@@ -1,6 +1,6 @@
 import { context, Environment } from "./context"
 
-import type { BitcoinTransactionHash, Hex } from "@keep-network/tbtc-v2.ts"
+import type { BitcoinTxHash, Hex } from "@keep-network/tbtc-v2.ts"
 
 const ethTxUrlPrefixMapping = {
   [Environment.Mainnet]: "https://etherscan.io/tx",
@@ -18,6 +18,6 @@ const btcTxUrlPrefixMapping = {
   [Environment.Testnet]: "https://mempool.space/testnet/tx",
 }
 
-export function createBtcTxUrl(txHash: BitcoinTransactionHash) {
+export function createBtcTxUrl(txHash: BitcoinTxHash) {
   return `${btcTxUrlPrefixMapping[context.environment]}/${txHash.toString()}`
 }

monitoring/src/blocks.ts

@@ -0,0 +1,21 @@
+import { providers } from "ethers"
+
+import { context } from "./context"
+
+const resolve = () => {
+  const provider = new providers.JsonRpcProvider(context.ethereumUrl)
+
+  const latestBlock = async () => {
+    const block = await provider.getBlock("latest")
+    return block.number
+  }
+
+  const blockTimestamp = async (blockNumber: number): Promise<number> => {
+    const block = await provider.getBlock(blockNumber)
+    return block.timestamp
+  }
+
+  return { latestBlock, blockTimestamp }
+}
+
+export const blocks = resolve()

monitoring/src/context.ts

@@ -1,3 +1,5 @@
+import { BitcoinNetwork, Chains } from "@keep-network/tbtc-v2.ts"
+
 // List of environment variables used by the monitoring package.
 const {
   ENVIRONMENT,
@@ -15,6 +17,16 @@ export enum Environment {
   Testnet = "testnet",
 }
 
+const ethereumEnvironmentMapping = {
+  [Environment.Mainnet]: Chains.Ethereum.Mainnet,
+  [Environment.Testnet]: Chains.Ethereum.Sepolia,
+}
+
+const bitcoinEnvironmentMapping = {
+  [Environment.Mainnet]: BitcoinNetwork.Mainnet,
+  [Environment.Testnet]: BitcoinNetwork.Testnet,
+}
+
 const resolveEnvironment = () => {
   switch (ENVIRONMENT) {
     case Environment.Mainnet: {
@@ -57,4 +69,6 @@ export const context = {
   dataDirPath: DATA_DIR_PATH ?? "./data",
   sentryDsn: SENTRY_DSN,
   discordWebhookUrl: DISCORD_WEBHOOK_URL,
+  ethereumEnvironmentMapping,
+  bitcoinEnvironmentMapping,
 }

monitoring/src/deposit-monitor.ts

@@ -5,8 +5,10 @@ import { context } from "./context"
 import { createBtcTxUrl, createEthTxUrl } from "./block-explorer"
 
 import type { SystemEvent, Monitor as SystemEventMonitor } from "./system-event"
-import type { DepositRevealedEvent as DepositRevealedChainEvent } from "@keep-network/tbtc-v2.ts/dist/src/deposit"
-import type { Bridge } from "@keep-network/tbtc-v2.ts/dist/src/chain"
+import type {
+  Bridge,
+  DepositRevealedEvent as DepositRevealedChainEvent,
+} from "@keep-network/tbtc-v2.ts"
 
 export const satsToRoundedBTC = (sats: BigNumber): string =>
   (sats.div(BigNumber.from(1e6)).toNumber() / 100).toFixed(2)

monitoring/src/index.ts

@@ -1,8 +1,12 @@
-import { ElectrumClient } from "@keep-network/tbtc-v2.ts"
+import {
+  ElectrumClient,
+  loadEthereumCoreContracts,
+  TBTC,
+} from "@keep-network/tbtc-v2.ts"
+import { providers } from "ethers"
 
 import { Manager as SystemEventManager } from "./system-event"
 import { DepositMonitor } from "./deposit-monitor"
-import { contracts } from "./contracts"
 import { DiscordReceiver } from "./discord-receiver"
 import { SentryReceiver } from "./sentry-receiver"
 import {
@@ -15,53 +19,87 @@ import { WalletMonitor } from "./wallet-monitor"
 import { SupplyMonitor } from "./supply-monitor"
 import { RedemptionMonitor } from "./redemption-monitor"
 
-import type { Client as BitcoinClient } from "@keep-network/tbtc-v2.ts/dist/src/bitcoin"
 import type {
   Monitor as SystemEventMonitor,
   Receiver as SystemEventReceiver,
 } from "./system-event"
 
-const btcClient: BitcoinClient = ElectrumClient.fromUrl(context.electrumUrl)
+async function setupSDK(): Promise<TBTC> {
+  const provider = new providers.JsonRpcProvider(context.ethereumUrl)
+  const chainId = context.ethereumEnvironmentMapping[context.environment]
+  const tbtcContracts = await loadEthereumCoreContracts(provider, chainId)
 
-const monitors: SystemEventMonitor[] = [
-  new DepositMonitor(contracts.bridge),
-  new MintingMonitor(contracts.bridge, contracts.tbtcVault, btcClient),
-  new SupplyMonitor(contracts.tbtcToken, new SupplyMonitorFilePersistence()),
-  new WalletMonitor(contracts.bridge),
-  new RedemptionMonitor(contracts.bridge),
-]
+  const bitcoinNetwork = context.bitcoinEnvironmentMapping[context.environment]
+  const btcClient = ElectrumClient.fromUrl(context.electrumUrl)
+  if ((await btcClient.getNetwork()) !== bitcoinNetwork) {
+    throw new Error("Bitcoin network mismatch")
+  }
 
-const receivers: SystemEventReceiver[] = ((): SystemEventReceiver[] => {
-  const registered: SystemEventReceiver[] = []
+  return TBTC.initializeCustom(tbtcContracts, btcClient)
+}
 
-  if (context.discordWebhookUrl) {
-    console.log("registered Discord receiver")
-    registered.push(new DiscordReceiver(context.discordWebhookUrl))
-  }
+async function setupMonitoring(sdk: TBTC): Promise<SystemEventManager> {
+  const { tbtcContracts, bitcoinClient } = sdk
 
-  if (context.sentryDsn) {
-    console.log("registered Sentry receiver")
-    registered.push(new SentryReceiver(context.sentryDsn))
-  }
+  const monitors: SystemEventMonitor[] = [
+    new DepositMonitor(tbtcContracts.bridge),
+    new MintingMonitor(
+      tbtcContracts.bridge,
+      tbtcContracts.tbtcVault,
+      bitcoinClient
+    ),
+    new SupplyMonitor(
+      tbtcContracts.tbtcToken,
+      new SupplyMonitorFilePersistence()
+    ),
+    new WalletMonitor(tbtcContracts.bridge),
+    new RedemptionMonitor(tbtcContracts.bridge),
+  ]
 
-  return registered
-})()
+  const receivers: SystemEventReceiver[] = ((): SystemEventReceiver[] => {
+    const registered: SystemEventReceiver[] = []
 
-const manager = new SystemEventManager(
-  monitors,
-  receivers,
-  new SystemEventFilePersistence()
-)
+    if (context.discordWebhookUrl) {
+      // eslint-disable-next-line no-console
+      console.log("registered Discord receiver")
+      registered.push(new DiscordReceiver(context.discordWebhookUrl))
+    }
 
-manager.trigger().then((report) => {
-  switch (report.status) {
-    case "success": {
-      console.log(report)
-      break
+    if (context.sentryDsn) {
+      // eslint-disable-next-line no-console
+      console.log("registered Sentry receiver")
+      registered.push(new SentryReceiver(context.sentryDsn))
     }
-    case "failure": {
-      console.error(report)
-      break
+
+    return registered
+  })()
+
+  return new SystemEventManager(
+    monitors,
+    receivers,
+    new SystemEventFilePersistence()
+  )
+}
+
+async function setup(): Promise<SystemEventManager> {
+  const sdk = await setupSDK()
+  return setupMonitoring(sdk)
+}
+
+setup().then((manager) => {
+  manager.trigger().then((report) => {
+    // eslint-disable-next-line default-case
+    switch (report.status) {
+      case "success": {
+        // eslint-disable-next-line no-console
+        console.log(report)
+        break
+      }
+      case "failure": {
+        // eslint-disable-next-line no-console
+        console.error(report)
+        break
+      }
     }
-  }
+  })
 })