@trezor/utxo-lib/lib/derivation.js

Client-side Bitcoin-like JavaScript library

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.deriveAddresses = exports.getXpubOrDescriptorInfo = void 0;
const tslib_1 = require("tslib");
const bs58_1 = tslib_1.__importDefault(require("bs58"));
const utils_1 = require("@trezor/utils");
const bip32_1 = require("./bip32");
const networks_1 = require("./networks");
const payments_1 = require("./payments");
const BIP32_PAYMENT_TYPES = {
  0x0488b21e: 'p2pkh',
  0x049d7cb2: 'p2sh',
  0x04b24746: 'p2wpkh',
  0x043587cf: 'p2pkh',
  0x044a5262: 'p2sh',
  0x045f1cf6: 'p2wpkh',
  0x019da462: 'p2pkh',
  0x01b26ef6: 'p2sh'
};
const BIP32_COIN_TYPES = {
  0x0488b21e: "0'",
  0x049d7cb2: "0'",
  0x04b24746: "0'",
  0x043587cf: "1'",
  0x044a5262: "1'",
  0x045f1cf6: "1'",
  0x019da462: "2'",
  0x01b26ef6: "2'"
};
const BIP32_PURPOSES = {
  p2pkh: "44'",
  p2sh: "49'",
  p2wpkh: "84'",
  p2tr: "86'"
};
const validateVersion = version => !!BIP32_PAYMENT_TYPES[version];
const getVersion = xpub => {
  const version = Buffer.from(bs58_1.default.decode(xpub)).readUInt32BE();
  if (!validateVersion(version)) throw new Error(`Unknown xpub version: ${xpub}`);
  return version;
};
const getPubkeyToPayment = (type, network) => pubkey => {
  switch (type) {
    case 'p2pkh':
      return (0, payments_1.p2pkh)({
        pubkey,
        network
      });
    case 'p2sh':
      return (0, payments_1.p2sh)({
        redeem: (0, payments_1.p2wpkh)({
          pubkey,
          network
        }),
        network
      });
    case 'p2wpkh':
      return (0, payments_1.p2wpkh)({
        pubkey,
        network
      });
    case 'p2tr':
      return (0, payments_1.p2tr)({
        pubkey,
        network
      });
    default:
      throw new Error(`Unknown payment type '${type}'`);
  }
};
const getBip32Node = (xpub, version, network) => (0, bip32_1.fromBase58)(xpub, {
  ...network,
  bip32: {
    ...network.bip32,
    public: version
  }
});
const getXpubInfo = (xpub, network) => {
  const version = getVersion(xpub);
  const paymentType = BIP32_PAYMENT_TYPES[version];
  const coinType = network.wif === 0xb0 ? "2'" : BIP32_COIN_TYPES[version];
  const purpose = BIP32_PURPOSES[paymentType];
  const node = getBip32Node(xpub, version, network);
  const account = `${node.index << 1 >>> 1}'`;
  const levels = [purpose, coinType, account];
  return {
    levels,
    paymentType,
    node
  };
};
const getDescriptorInfo = (paymentType, descriptor, network) => {
  const [_match, _script, path, xpub, _checksum] = descriptor.match(/^([a-z]+\()+\[([a-z0-9]{8}(?:\/[0-9]+'?){3,})\]([xyztuv]pub[a-zA-Z0-9]*)\/<0;1>\/\*\)+(#[a-z0-9]{8})?$/) || (0, utils_1.throwError)(`Descriptor cannot be parsed: ${descriptor}`);
  const [_fingerprint, ...levels] = path.split('/');
  const version = getVersion(xpub);
  const node = getBip32Node(xpub, version, network);
  return {
    levels,
    paymentType,
    node
  };
};
const getXpubOrDescriptorInfo = (descriptor, network = networks_1.bitcoin) => {
  if (descriptor.startsWith('pkh(')) {
    return getDescriptorInfo('p2pkh', descriptor, network);
  }
  if (descriptor.startsWith('sh(wpkh(')) {
    return getDescriptorInfo('p2sh', descriptor, network);
  }
  if (descriptor.startsWith('wpkh(')) {
    return getDescriptorInfo('p2wpkh', descriptor, network);
  }
  if (descriptor.startsWith('tr(')) {
    return getDescriptorInfo('p2tr', descriptor, network);
  }
  return getXpubInfo(descriptor, network);
};
exports.getXpubOrDescriptorInfo = getXpubOrDescriptorInfo;
const deriveAddresses = (descriptor, type, from, count, network = networks_1.bitcoin) => {
  const {
    levels,
    node,
    paymentType
  } = (0, exports.getXpubOrDescriptorInfo)(descriptor, network);
  const getAddress = getPubkeyToPayment(paymentType, network);
  const change = type === 'receive' ? 0 : 1;
  const changeNode = node.derive(change);
  return Array.from(Array(count).keys()).map(i => changeNode.derive(from + i).publicKey).map(a => getAddress(a).address || (0, utils_1.throwError)('Cannot convert pubkey to address')).map((address, i) => ({
    address,
    path: `m/${levels.join('/')}/${change}/${from + i}`
  }));
};
exports.deriveAddresses = deriveAddresses;
//# sourceMappingURL=derivation.js.map