diff --git a/CHANGELOG.md b/CHANGELOG.md index 3e23430..37e097f 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,7 @@ +# 5.1.5 +__added__ +- `Psbt` now has `getFee(): number` for use when all inputs are finalized. It returns the satoshi fee of the transaction. Calling getFee, getFeeRate, or extractTransaction will cache these values so if you call one after the other, the second call will return immediately. + # 5.1.4 __changed__ - `Psbt` inputs using segwit scripts can now work with nonWitnessUtxo as well as the original witnessUtxo. The reasoning for this is that nonWitnessUtxo has all the information contained in the witnessUtxo, so rejecting signing even though we have all the info we need is unnecessary. Trying to sign a non-segwit script with a witnessUtxo will still throw an Error as it should. diff --git a/package-lock.json b/package-lock.json index bb292ce..c2966f8 100644 --- a/package-lock.json +++ b/package-lock.json @@ -1,6 +1,6 @@ { "name": "bitcoinjs-lib", - "version": "5.1.4", + "version": "5.1.5", "lockfileVersion": 1, "requires": true, "dependencies": { diff --git a/package.json b/package.json index 07bb419..d701877 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "bitcoinjs-lib", - "version": "5.1.4", + "version": "5.1.5", "description": "Client-side Bitcoin JavaScript library", "main": "./src/index.js", "types": "./types/index.d.ts", diff --git a/src/psbt.js b/src/psbt.js index e134774..57a159c 100644 --- a/src/psbt.js +++ b/src/psbt.js @@ -153,6 +153,7 @@ class Psbt { if (input.nonWitnessUtxo) { addNonWitnessTxCache(this.__CACHE, input, inputIndex); } + c.__FEE = undefined; c.__FEE_RATE = undefined; c.__EXTRACTED_TX = undefined; return this; @@ -171,6 +172,7 @@ class Psbt { } const c = this.__CACHE; this.data.addOutput(outputData); + c.__FEE = undefined; c.__FEE_RATE = undefined; c.__EXTRACTED_TX = undefined; return this; @@ -187,20 +189,15 @@ class Psbt { return tx; } getFeeRate() { - if (!this.data.inputs.every(isFinalized)) - throw new Error('PSBT must be finalized to calculate fee rate'); - const c = this.__CACHE; - if (c.__FEE_RATE) return c.__FEE_RATE; - let tx; - let mustFinalize = true; - if (c.__EXTRACTED_TX) { - tx = c.__EXTRACTED_TX; - mustFinalize = false; - } else { - tx = c.__TX.clone(); - } - inputFinalizeGetAmts(this.data.inputs, tx, c, mustFinalize); - return c.__FEE_RATE; + return getTxCacheValue( + '__FEE_RATE', + 'fee rate', + this.data.inputs, + this.__CACHE, + ); + } + getFee() { + return getTxCacheValue('__FEE', 'fee', this.data.inputs, this.__CACHE); } finalizeAllInputs() { utils_1.checkForInput(this.data.inputs, 0); // making sure we have at least one @@ -724,6 +721,23 @@ const checkWitnessScript = scriptCheckerFactory( payments.p2wsh, 'Witness script', ); +function getTxCacheValue(key, name, inputs, c) { + if (!inputs.every(isFinalized)) + throw new Error(`PSBT must be finalized to calculate ${name}`); + if (key === '__FEE_RATE' && c.__FEE_RATE) return c.__FEE_RATE; + if (key === '__FEE' && c.__FEE) return c.__FEE; + let tx; + let mustFinalize = true; + if (c.__EXTRACTED_TX) { + tx = c.__EXTRACTED_TX; + mustFinalize = false; + } else { + tx = c.__TX.clone(); + } + inputFinalizeGetAmts(inputs, tx, c, mustFinalize); + if (key === '__FEE_RATE') return c.__FEE_RATE; + else if (key === '__FEE') return c.__FEE; +} function getFinalScripts( script, scriptType, @@ -1124,6 +1138,7 @@ function inputFinalizeGetAmts(inputs, tx, cache, mustFinalize) { throw new Error('Outputs are spending more than Inputs'); } const bytes = tx.virtualSize(); + cache.__FEE = fee; cache.__EXTRACTED_TX = tx; cache.__FEE_RATE = Math.floor(fee / bytes); } diff --git a/test/psbt.js b/test/psbt.js index 467e426..54eb5e4 100644 --- a/test/psbt.js +++ b/test/psbt.js @@ -149,6 +149,11 @@ describe(`Psbt`, () => { const fr1 = psbt5.getFeeRate() const fr2 = psbt5.getFeeRate() assert.strictEqual(fr1, fr2) + + const psbt6 = Psbt.fromBase64(f.psbt) + const f1 = psbt6.getFee() + const f2 = psbt6.getFee() + assert.strictEqual(f1, f2) }) }) }) diff --git a/ts_src/psbt.ts b/ts_src/psbt.ts index 0431056..73f7cfc 100644 --- a/ts_src/psbt.ts +++ b/ts_src/psbt.ts @@ -194,6 +194,7 @@ export class Psbt { if (input.nonWitnessUtxo) { addNonWitnessTxCache(this.__CACHE, input, inputIndex); } + c.__FEE = undefined; c.__FEE_RATE = undefined; c.__EXTRACTED_TX = undefined; return this; @@ -214,6 +215,7 @@ export class Psbt { } const c = this.__CACHE; this.data.addOutput(outputData); + c.__FEE = undefined; c.__FEE_RATE = undefined; c.__EXTRACTED_TX = undefined; return this; @@ -232,20 +234,16 @@ export class Psbt { } getFeeRate(): number { - if (!this.data.inputs.every(isFinalized)) - throw new Error('PSBT must be finalized to calculate fee rate'); - const c = this.__CACHE; - if (c.__FEE_RATE) return c.__FEE_RATE; - let tx: Transaction; - let mustFinalize = true; - if (c.__EXTRACTED_TX) { - tx = c.__EXTRACTED_TX; - mustFinalize = false; - } else { - tx = c.__TX.clone(); - } - inputFinalizeGetAmts(this.data.inputs, tx, c, mustFinalize); - return c.__FEE_RATE!; + return getTxCacheValue( + '__FEE_RATE', + 'fee rate', + this.data.inputs, + this.__CACHE, + )!; + } + + getFee(): number { + return getTxCacheValue('__FEE', 'fee', this.data.inputs, this.__CACHE)!; } finalizeAllInputs(): this { @@ -610,6 +608,7 @@ interface PsbtCache { __TX_IN_CACHE: { [index: string]: number }; __TX: Transaction; __FEE_RATE?: number; + __FEE?: number; __EXTRACTED_TX?: Transaction; } @@ -920,6 +919,30 @@ const checkWitnessScript = scriptCheckerFactory( 'Witness script', ); +type TxCacheNumberKey = '__FEE_RATE' | '__FEE'; +function getTxCacheValue( + key: TxCacheNumberKey, + name: string, + inputs: PsbtInput[], + c: PsbtCache, +): number | undefined { + if (!inputs.every(isFinalized)) + throw new Error(`PSBT must be finalized to calculate ${name}`); + if (key === '__FEE_RATE' && c.__FEE_RATE) return c.__FEE_RATE; + if (key === '__FEE' && c.__FEE) return c.__FEE; + let tx: Transaction; + let mustFinalize = true; + if (c.__EXTRACTED_TX) { + tx = c.__EXTRACTED_TX; + mustFinalize = false; + } else { + tx = c.__TX.clone(); + } + inputFinalizeGetAmts(inputs, tx, c, mustFinalize); + if (key === '__FEE_RATE') return c.__FEE_RATE!; + else if (key === '__FEE') return c.__FEE!; +} + function getFinalScripts( script: Buffer, scriptType: string, @@ -1398,6 +1421,7 @@ function inputFinalizeGetAmts( throw new Error('Outputs are spending more than Inputs'); } const bytes = tx.virtualSize(); + cache.__FEE = fee; cache.__EXTRACTED_TX = tx; cache.__FEE_RATE = Math.floor(fee / bytes); } diff --git a/types/psbt.d.ts b/types/psbt.d.ts index 6a56636..a0c669d 100644 --- a/types/psbt.d.ts +++ b/types/psbt.d.ts @@ -57,6 +57,7 @@ export declare class Psbt { addOutput(outputData: PsbtOutputExtended): this; extractTransaction(disableFeeCheck?: boolean): Transaction; getFeeRate(): number; + getFee(): number; finalizeAllInputs(): this; finalizeInput(inputIndex: number): this; validateSignaturesOfAllInputs(): boolean;