feat(wasm-utxo): add supportsScriptType function

OttoAllmendinger · llm-git · OttoAllmendinger · commit 09db15f9f5ec · 2026-01-09T14:05:46.000+01:00
Add `supportsScriptType` function to check if a given network supports a particular wallet script type (p2sh, p2wsh, p2tr, etc). This provides a simple API for frontends to determine script compatibility before attempting to create wallets. Major improvements: - Refactored script type handling to use a dedicated OutputScriptType enum - Separated script type from derivation chain/scope concerns - Added network compatibility checks for script types - Exposed JS API with proper TypeScript definitions Issue: BTC-2916 Co-authored-by: llm-git <llm-git@ttll.de>
diff --git a/packages/wasm-utxo/js/fixedScriptWallet/index.ts b/packages/wasm-utxo/js/fixedScriptWallet/index.ts
@@ -1,3 +1,6 @@
+import { FixedScriptWalletNamespace } from "../wasm/wasm_utxo.js";
+import type { CoinName } from "../coinName.js";
+
 export { RootWalletKeys, type WalletKeysArg, type IWalletKeys } from "./RootWalletKeys.js";
 export { ReplayProtection, type ReplayProtectionArg } from "./ReplayProtection.js";
 export { outputScript, address } from "./address.js";
@@ -26,3 +29,36 @@ export {
   type ZcashNetworkName,
   type CreateEmptyZcashOptions,
 } from "./ZcashBitGoPsbt.js";
+
+/**
+ * Fixed-script wallet script types (2-of-3 multisig)
+ *
+ * This type represents the abstract script type, independent of chain (external/internal).
+ * Use this for checking network support or when you need the script type without derivation info.
+ */
+export type ScriptType = "p2sh" | "p2shP2wsh" | "p2wsh" | "p2tr" | "p2trMusig2";
+
+/**
+ * Check if a network supports a given fixed-script wallet script type
+ *
+ * @param coin - Coin name (e.g., "btc", "ltc", "doge")
+ * @param scriptType - Script type to check
+ * @returns `true` if the network supports the script type, `false` otherwise
+ *
+ * @example
+ * ```typescript
+ * // Bitcoin supports all script types
+ * supportsScriptType("btc", "p2tr"); // true
+ *
+ * // Litecoin supports segwit but not taproot
+ * supportsScriptType("ltc", "p2wsh"); // true
+ * supportsScriptType("ltc", "p2tr"); // false
+ *
+ * // Dogecoin only supports legacy scripts
+ * supportsScriptType("doge", "p2sh"); // true
+ * supportsScriptType("doge", "p2wsh"); // false
+ * ```
+ */
+export function supportsScriptType(coin: CoinName, scriptType: ScriptType): boolean {
+  return FixedScriptWalletNamespace.supports_script_type(coin, scriptType);
+}
diff --git a/packages/wasm-utxo/src/address/networks.rs b/packages/wasm-utxo/src/address/networks.rs
@@ -13,6 +13,7 @@ use super::{
     ZCASH_TEST,
 };
 use crate::bitcoin::Script;
+use crate::fixed_script_wallet::wallet_scripts::OutputScriptType;
 use crate::networks::Network;
 use miniscript::bitcoin::WitnessVersion;
 
@@ -120,6 +121,15 @@ impl OutputScriptSupport {
         }
         Ok(())
     }
+
+    /// Check if the network supports a given fixed-script wallet script type
+    pub fn supports_script_type(&self, script_type: OutputScriptType) -> bool {
+        match script_type {
+            OutputScriptType::P2sh => true, // all networks support legacy scripts
+            OutputScriptType::P2shP2wsh | OutputScriptType::P2wsh => self.segwit,
+            OutputScriptType::P2trLegacy | OutputScriptType::P2trMusig2 => self.taproot,
+        }
+    }
 }
 
 impl Network {
diff --git a/packages/wasm-utxo/src/fixed_script_wallet/bitgo_psbt/mod.rs b/packages/wasm-utxo/src/fixed_script_wallet/bitgo_psbt/mod.rs
@@ -184,16 +184,13 @@ fn get_default_sighash_type(
     network: Network,
     chain: crate::fixed_script_wallet::wallet_scripts::Chain,
 ) -> miniscript::bitcoin::psbt::PsbtSighashType {
-    use crate::fixed_script_wallet::wallet_scripts::Chain;
+    use crate::fixed_script_wallet::wallet_scripts::OutputScriptType;
     use miniscript::bitcoin::sighash::{EcdsaSighashType, TapSighashType};
 
     // For taproot, always use Default
     if matches!(
-        chain,
-        Chain::P2trInternal
-            | Chain::P2trExternal
-            | Chain::P2trMusig2Internal
-            | Chain::P2trMusig2External
+        chain.script_type,
+        OutputScriptType::P2trLegacy | OutputScriptType::P2trMusig2
     ) {
         return TapSighashType::Default.into();
     }
@@ -758,7 +755,7 @@ impl BitGoPsbt {
         options: WalletInputOptions,
     ) -> Result<usize, String> {
         use crate::fixed_script_wallet::to_pub_triple;
-        use crate::fixed_script_wallet::wallet_scripts::{Chain, WalletScripts};
+        use crate::fixed_script_wallet::wallet_scripts::{Chain, OutputScriptType, WalletScripts};
         use miniscript::bitcoin::psbt::Input;
         use miniscript::bitcoin::taproot::{LeafVersion, TapLeafHash};
         use miniscript::bitcoin::{transaction::Sequence, Amount, OutPoint, TxIn, TxOut};
@@ -799,18 +796,8 @@ impl BitGoPsbt {
         // Create the PSBT input
         let mut psbt_input = Input::default();
 
-        // Determine if segwit based on chain type
-        let is_segwit = matches!(
-            chain_enum,
-            Chain::P2shP2wshExternal
-                | Chain::P2shP2wshInternal
-                | Chain::P2wshExternal
-                | Chain::P2wshInternal
-                | Chain::P2trInternal
-                | Chain::P2trExternal
-                | Chain::P2trMusig2Internal
-                | Chain::P2trMusig2External
-        );
+        // Determine if segwit based on chain type (all types except P2sh are segwit)
+        let is_segwit = chain_enum.script_type != OutputScriptType::P2sh;
 
         if let (false, Some(tx_bytes)) = (is_segwit, options.prev_tx) {
             // Non-segwit with prev_tx: use non_witness_utxo
diff --git a/packages/wasm-utxo/src/fixed_script_wallet/bitgo_psbt/psbt_wallet_input.rs b/packages/wasm-utxo/src/fixed_script_wallet/bitgo_psbt/psbt_wallet_input.rs
@@ -4,7 +4,9 @@ use miniscript::bitcoin::secp256k1::{self, PublicKey};
 use miniscript::bitcoin::{OutPoint, ScriptBuf, TapLeafHash, XOnlyPublicKey};
 
 use crate::bitcoin::bip32::KeySource;
-use crate::fixed_script_wallet::{Chain, ReplayProtection, RootWalletKeys, WalletScripts};
+use crate::fixed_script_wallet::{
+    Chain, OutputScriptType, ReplayProtection, RootWalletKeys, WalletScripts,
+};
 use crate::Network;
 
 pub type Bip32DerivationMap = std::collections::BTreeMap<PublicKey, KeySource>;
@@ -655,12 +657,12 @@ pub enum InputScriptType {
 impl InputScriptType {
     pub fn from_script_id(script_id: ScriptId, psbt_input: &Input) -> Result<Self, String> {
         let chain = Chain::try_from(script_id.chain).map_err(|e| e.to_string())?;
-        match chain {
-            Chain::P2shExternal | Chain::P2shInternal => Ok(InputScriptType::P2sh),
-            Chain::P2shP2wshExternal | Chain::P2shP2wshInternal => Ok(InputScriptType::P2shP2wsh),
-            Chain::P2wshExternal | Chain::P2wshInternal => Ok(InputScriptType::P2wsh),
-            Chain::P2trInternal | Chain::P2trExternal => Ok(InputScriptType::P2trLegacy),
-            Chain::P2trMusig2Internal | Chain::P2trMusig2External => {
+        match chain.script_type {
+            OutputScriptType::P2sh => Ok(InputScriptType::P2sh),
+            OutputScriptType::P2shP2wsh => Ok(InputScriptType::P2shP2wsh),
+            OutputScriptType::P2wsh => Ok(InputScriptType::P2wsh),
+            OutputScriptType::P2trLegacy => Ok(InputScriptType::P2trLegacy),
+            OutputScriptType::P2trMusig2 => {
                 // check if tap_script_sigs or tap_scripts are set
                 if !psbt_input.tap_script_sigs.is_empty() || !psbt_input.tap_scripts.is_empty() {
                     Ok(InputScriptType::P2trMusig2ScriptPath)
diff --git a/packages/wasm-utxo/src/fixed_script_wallet/test_utils/mod.rs b/packages/wasm-utxo/src/fixed_script_wallet/test_utils/mod.rs
@@ -4,7 +4,7 @@ pub mod fixtures;
 pub mod psbt_compare;
 
 use super::wallet_keys::XpubTriple;
-use super::wallet_scripts::{Chain, WalletScripts};
+use super::wallet_scripts::{Chain, OutputScriptType, Scope, WalletScripts};
 use crate::bitcoin::bip32::{DerivationPath, Fingerprint, Xpriv, Xpub};
 use crate::bitcoin::psbt::{Input as PsbtInput, Output as PsbtOutput, Psbt};
 use crate::bitcoin::{Transaction, TxIn, TxOut};
@@ -38,7 +38,7 @@ pub fn create_external_output(seed: &str) -> PsbtOutput {
     let xpubs = get_test_wallet_keys(seed);
     let _scripts = WalletScripts::from_wallet_keys(
         &RootWalletKeys::new(xpubs),
-        Chain::P2wshExternal,
+        Chain::new(OutputScriptType::P2wsh, Scope::External),
         0,
         &Network::Bitcoin.output_script_support(),
     )
diff --git a/packages/wasm-utxo/src/fixed_script_wallet/wallet_scripts/checkmultisig.rs b/packages/wasm-utxo/src/fixed_script_wallet/wallet_scripts/checkmultisig.rs
@@ -100,14 +100,13 @@ mod tests {
     use crate::bitcoin::blockdata::script::Builder;
     use crate::fixed_script_wallet::wallet_keys::tests::get_test_wallet_keys;
     use crate::fixed_script_wallet::wallet_keys::to_pub_triple;
-    use crate::fixed_script_wallet::wallet_scripts::Chain;
 
     #[test]
     fn test_parse_multisig_script_2_of_3_valid() {
         // Get test keys
         let wallet_keys = get_test_wallet_keys("test_parse");
         let derived_keys = wallet_keys
-            .derive_for_chain_and_index(Chain::P2shExternal as u32, 0)
+            .derive_for_chain_and_index(0, 0)
             .unwrap();
         let pub_triple = to_pub_triple(&derived_keys);
 
@@ -127,7 +126,7 @@ mod tests {
         for seed in ["seed1", "seed2", "seed3"] {
             let wallet_keys = get_test_wallet_keys(seed);
             let derived_keys = wallet_keys
-                .derive_for_chain_and_index(Chain::P2shExternal as u32, 42)
+                .derive_for_chain_and_index(0, 42)
                 .unwrap();
             let original_keys = to_pub_triple(&derived_keys);
 
@@ -169,7 +168,7 @@ mod tests {
         // Create a valid key for testing
         let wallet_keys = get_test_wallet_keys("test_wrong_quorum");
         let derived_keys = wallet_keys
-            .derive_for_chain_and_index(Chain::P2shExternal as u32, 0)
+            .derive_for_chain_and_index(0, 0)
             .unwrap();
         let pub_triple = to_pub_triple(&derived_keys);
 
@@ -195,7 +194,7 @@ mod tests {
         // Create a valid key for testing
         let wallet_keys = get_test_wallet_keys("test_wrong_total");
         let derived_keys = wallet_keys
-            .derive_for_chain_and_index(Chain::P2shExternal as u32, 0)
+            .derive_for_chain_and_index(0, 0)
             .unwrap();
         let pub_triple = to_pub_triple(&derived_keys);
 
@@ -221,7 +220,7 @@ mod tests {
         // Create a valid key for testing
         let wallet_keys = get_test_wallet_keys("test_missing_checkmultisig");
         let derived_keys = wallet_keys
-            .derive_for_chain_and_index(Chain::P2shExternal as u32, 0)
+            .derive_for_chain_and_index(0, 0)
             .unwrap();
         let pub_triple = to_pub_triple(&derived_keys);
 
diff --git a/packages/wasm-utxo/src/fixed_script_wallet/wallet_scripts/mod.rs b/packages/wasm-utxo/src/fixed_script_wallet/wallet_scripts/mod.rs
diff --git a/packages/wasm-utxo/src/wasm/fixed_script_wallet/dimensions.rs b/packages/wasm-utxo/src/wasm/fixed_script_wallet/dimensions.rs
diff --git a/packages/wasm-utxo/src/wasm/fixed_script_wallet/mod.rs b/packages/wasm-utxo/src/wasm/fixed_script_wallet/mod.rs
