wallet-storage/src/services/chaintracker/ChaintracksChainTracker.ts

BRC100 conforming wallet, wallet storage and wallet signer components

import { sdk, wait } from '../../index.client'
import { ChainTracker } from "@bsv/sdk";
import { BlockHeader } from "./chaintracks";
import { ChaintracksServiceClient } from "./chaintracks/ChaintracksServiceClient";

export interface ChaintracksChainTrackerOptions {
    maxRetries?: number
}

export class ChaintracksChainTracker implements ChainTracker {
    chaintracks: ChaintracksServiceClient
    cache: Record<number, string>
    options: ChaintracksChainTrackerOptions

    constructor(chain?: sdk.Chain, chaintracks?: ChaintracksServiceClient, options?: ChaintracksChainTrackerOptions) {

        chain ||= 'main'
        this.chaintracks = chaintracks ?? new ChaintracksServiceClient(chain, `https://npm-registry.babbage.systems:808${chain === 'main' ? '4' : '3'}`)
        this.cache = {}
        this.options = options || {}
    }

    async currentHeight() : Promise<number> {
        return await this.chaintracks.getPresentHeight()
    }

    async isValidRootForHeight(root: string, height: number) : Promise<boolean> {
        const cachedRoot = this.cache[height]
        if (cachedRoot) {
            return cachedRoot === root
        }

        let header: BlockHeader | undefined

        const retries = this.options.maxRetries || 3

        let error: sdk.WalletError | undefined = undefined

        for (let tryCount = 1; tryCount <= retries; tryCount++) {

            try {
                header = await this.chaintracks.findHeaderForHeight(height)

                if (!header) {
                    return false
                }

                break
            } catch (eu: unknown) {
                error = sdk.WalletError.fromUnknown(eu)
                if (tryCount > retries) {
                    throw error
                }
                await wait(1000)
            }
        }

        if (!header)
            throw new sdk.WERR_INTERNAL('no header should have returned false or thrown an error.')

        this.cache[height] = header.merkleRoot

        if (header.merkleRoot !== root) {
            return false
        }

        return true
    }
}