import * as bitcoinJs from 'bitcoinjs-lib';

import { BitcoinNetworkModes } from '@leather.io/models';

// TODO - this PR was merged so we could update this
// https://github.com/paulmillr/scure-btc-signer/blob/main/src/utils.ts
// See this PR https://github.com/paulmillr/@scure/btc-signer/pull/15
// Atttempting to add these directly to the library
export interface BtcSignerNetwork {
  bech32: string;
  pubKeyHash: number;
  scriptHash: number;
  wif: number;
}

const bitcoinMainnet: BtcSignerNetwork = {
  bech32: 'bc',
  pubKeyHash: 0x00,
  scriptHash: 0x05,
  wif: 0x80,
};

const bitcoinTestnet: BtcSignerNetwork = {
  bech32: 'tb',
  pubKeyHash: 0x6f,
  scriptHash: 0xc4,
  wif: 0xef,
};

const bitcoinRegtest: BtcSignerNetwork = {
  bech32: 'bcrt',
  pubKeyHash: 0x6f,
  scriptHash: 0xc4,
  wif: 0xef,
};

const btcSignerLibNetworks: Record<BitcoinNetworkModes, BtcSignerNetwork> = {
  mainnet: bitcoinMainnet,
  testnet: bitcoinTestnet,
  regtest: bitcoinRegtest,
  // Signet originally was going to have its own prefix but authors decided to
  // copy testnet
  signet: bitcoinTestnet,
};

export function getBtcSignerLibNetworkConfigByMode(network: BitcoinNetworkModes) {
  return btcSignerLibNetworks[network];
}

const bitcoinJsLibNetworks: Record<BitcoinNetworkModes, bitcoinJs.Network> = {
  mainnet: bitcoinJs.networks.bitcoin,
  testnet: bitcoinJs.networks.testnet,
  regtest: bitcoinJs.networks.regtest,
  signet: bitcoinJs.networks.testnet,
};

export function getBitcoinJsLibNetworkConfigByMode(network: BitcoinNetworkModes) {
  return bitcoinJsLibNetworks[network];
}