getAccountUtxo.ts

import { throwError } from '@trezor/utils';
import { discovery } from '@trezor/utxo-lib';
import type { GetAccountUtxo as Req } from '@trezor/blockchain-link-types/src/messages';
import type { GetAccountUtxo as Res } from '@trezor/blockchain-link-types/src/responses';
import type { Utxo } from '@trezor/blockchain-link-types/src/electrum';

import { Api, tryGetScripthash, discoverAddress } from '../utils';

const transformUtxo =
    (currentHeight: number, addressInfo: { address?: string; path?: string } = {}) =>
    ({ height, tx_hash, tx_pos, value }: Utxo): Res['payload'][number] => ({
        txid: tx_hash,
        vout: tx_pos,
        amount: value.toString(),
        address: '',
        path: '',
        ...addressInfo,
        ...(height
            ? {
                  blockHeight: height,
                  confirmations: currentHeight - height + 1,
              }
            : {
                  blockHeight: -1,
                  confirmations: 0,
              }),
    });

const getAccountUtxo: Api<Req, Res> = async (client, descriptor) => {
    const {
        block: { height },
        network,
    } = client.getInfo() || throwError('Client not initialized');

    const parsed = tryGetScripthash(descriptor, network);

    if (parsed.valid) {
        const utxos = await client.request('blockchain.scripthash.listunspent', parsed.scripthash);

        return utxos.map(transformUtxo(height));
    }

    const discover = discoverAddress(client);
    const receive = await discovery(discover, descriptor, 'receive', network);
    const change = await discovery(discover, descriptor, 'change', network);
    const result = await Promise.all(
        receive
            .concat(change)
            .filter(a => a.history.length)
            .map(({ address, path, scripthash }) =>
                client
                    .request('blockchain.scripthash.listunspent', scripthash)
                    .then(utxos => utxos.map(transformUtxo(height, { address, path }))),
            ),
    ).then(res => res.flat());

    return result;
};

export default getAccountUtxo;