tapcfg+tapfreighter: configure sweeping with wallet flag

darioAnongba · darioAnongba · commit 7b10b2e3be52 · 2025-10-30T16:56:33.000+01:00
diff --git a/config.go b/config.go
@@ -190,6 +190,10 @@ type Config struct {
 
 	ChainPorter tapfreighter.Porter
 
+	// SweepOrphanUtxos toggles sweeping orphaned UTXOs into anchor
+	// transactions for sends and burns.
+	SweepOrphanUtxos bool
+
 	// FsmDaemonAdapters is a set of adapters that allow a state machine to
 	// interact with external daemons.
 	FsmDaemonAdapters *lndservices.LndFsmDaemonAdapters
diff --git a/itest/assertions.go b/itest/assertions.go
@@ -1276,10 +1276,6 @@ func AssertAssetOutboundTransferWithOutputs(t *testing.T,
 		scripts[string(o.ScriptKey)] = struct{}{}
 	}
 
-	sendRespJSON, err := formatProtoJSON(transfer)
-	require.NoError(t, err)
-	t.Logf("Got response from sending assets: %v", sendRespJSON)
-
 	// Mine a block to force the send event to complete (confirm on-chain).
 	var newBlock *wire.MsgBlock
 	if confirm {
@@ -1325,16 +1321,6 @@ func AssertAssetOutboundTransferWithOutputs(t *testing.T,
 				return slices.Contains(actualInputAssetIDs, id)
 			})
 	}, defaultTimeout, wait.PollInterval)
-	require.NoError(t, err)
-
-	transferResp, err := sender.ListTransfers(
-		ctxb, &taprpc.ListTransfersRequest{},
-	)
-	require.NoError(t, err)
-
-	transferRespJSON, err := formatProtoJSON(transferResp)
-	require.NoError(t, err)
-	t.Logf("Got response from list transfers: %v", transferRespJSON)
 
 	return newBlock
 }
@@ -2410,7 +2396,7 @@ func AssertBalances(t *testing.T, client taprpc.TaprootAssetsClient,
 			actualBalance := balanceSum(resp, false)
 			if balance != actualBalance {
 				return fmt.Errorf("asset balance, wanted %d, "+
-					"got: %v", balance, toJSON(t, resp))
+					"got: %v", balance, actualBalance)
 			}
 
 			// If we query for grouped asset balances too, it means
@@ -2528,7 +2514,7 @@ func AssertBalances(t *testing.T, client taprpc.TaprootAssetsClient,
 		}
 		if balance != totalBalance {
 			return fmt.Errorf("ListAssets balance, wanted %d, "+
-				"got: %v", balance, toJSON(t, resp))
+				"got: %v", balance, totalBalance)
 		}
 
 		// The number of UTXOs means asset outputs in this case. We
@@ -2605,7 +2591,7 @@ func AssertBalances(t *testing.T, client taprpc.TaprootAssetsClient,
 
 		if balance != totalBalance {
 			return fmt.Errorf("ListUtxos balance, wanted %d, "+
-				"got: %v", balance, toJSON(t, resp))
+				"got: %v", balance, totalBalance)
 		}
 
 		if config.numAnchorUtxos > 0 {
diff --git a/itest/burn_test.go b/itest/burn_test.go
@@ -246,11 +246,6 @@ func testBurnAssets(t *harnessTest) {
 		WithScriptKeyType(asset.ScriptKeyBurn),
 	)
 
-	_, err = t.tapd.ListAssets(ctxt, &taprpc.ListAssetRequest{
-		IncludeSpent: true,
-	})
-	require.NoError(t.t, err)
-
 	// Test case 4: Burn some units of a grouped asset. We start by making
 	// sure we still have the full balance before burning.
 	AssertBalanceByID(
diff --git a/itest/integration_test.go b/itest/integration_test.go
@@ -120,6 +120,7 @@ func TestTaprootAssetsDaemon(t *testing.T) {
 			tapdHarness, uniHarness, proofCourier := setupHarnesses(
 				t1, ht, lndHarness, uniServerLndHarness,
 				testCase.proofCourierType,
+				testCase.tapdOptions...,
 			)
 
 			ht := ht.newHarnessTest(
diff --git a/itest/tapd_harness.go b/itest/tapd_harness.go
@@ -145,6 +145,10 @@ type harnessOpts struct {
 	// verification should only occur for commitments submitted by peers,
 	// not via on-chain spend detection.
 	disableSupplyVerifierChainWatch bool
+
+	// sweepOrphanUtxos indicates whether orphaned anchor UTXOs should be
+	// swept into anchor transactions.
+	sweepOrphanUtxos bool
 }
 
 type harnessOption func(*harnessOpts)
@@ -256,6 +260,10 @@ func newTapdHarness(t *testing.T, ht *harnessTest, cfg tapdConfig,
 	// nolint: lll
 	tapCfg.Universe.DisableSupplyVerifierChainWatch = opts.disableSupplyVerifierChainWatch
 
+	// Pass through the sweep orphan UTXOs flag. If the option was not set,
+	// this will be false, which is the default.
+	tapCfg.Wallet.SweepOrphanUtxos = opts.sweepOrphanUtxos
+
 	switch {
 	case len(opts.oracleServerAddress) > 0:
 		tapCfg.Experimental.Rfq.PriceOracleAddress =
diff --git a/itest/test_harness.go b/itest/test_harness.go
@@ -70,6 +70,7 @@ type testCase struct {
 	name             string
 	test             func(t *harnessTest)
 	proofCourierType proof.CourierType
+	tapdOptions      []Option
 }
 
 // harnessTest wraps a regular testing.T providing enhanced error detection
@@ -270,7 +271,7 @@ func (h *harnessTest) addFederationServer(host string, target *tapdHarness) {
 // to each other through an in-memory gRPC connection.
 func setupHarnesses(t *testing.T, ht *harnessTest,
 	lndHarness *lntest.HarnessTest, uniServerLndHarness *node.HarnessNode,
-	proofCourierType proof.CourierType) (*tapdHarness,
+	proofCourierType proof.CourierType, tapdOpts ...Option) (*tapdHarness,
 	*universeServerHarness, proof.CourierHarness) {
 
 	// Create a new universe server harness and start it.
@@ -306,10 +307,11 @@ func setupHarnesses(t *testing.T, ht *harnessTest,
 	alice := lndHarness.NewNodeWithCoins("Alice", nil)
 
 	// Create a tapd that uses Alice and connect it to the universe server.
+	tapdOptions := append(tapdOpts, func(params *tapdHarnessParams) {
+		params.proofCourier = proofCourier
+	})
 	tapdHarness := setupTapdHarness(
-		t, ht, alice, universeServer, func(params *tapdHarnessParams) {
-			params.proofCourier = proofCourier
-		},
+		t, ht, alice, universeServer, tapdOptions...,
 	)
 	return tapdHarness, universeServer, proofCourier
 }
@@ -372,6 +374,10 @@ type tapdHarnessParams struct {
 	// sendPriceHint indicates whether the tapd should send price hints from
 	// the local oracle to the counterparty when requesting a quote.
 	sendPriceHint bool
+
+	// sweepOrphanUtxos indicates whether orphaned UTXOs should be swept
+	// into anchor transactions.
+	sweepOrphanUtxos bool
 }
 
 // Option is a tapd harness option.
@@ -400,6 +406,14 @@ func WithSendPriceHint() Option {
 	}
 }
 
+// WithSweepOrphanUtxos enables sweeping zero-value anchor UTXOs for
+// the tapd harness created with this option.
+func WithSweepOrphanUtxos() Option {
+	return func(th *tapdHarnessParams) {
+		th.sweepOrphanUtxos = true
+	}
+}
+
 // setupTapdHarness creates a new tapd that connects to the given lnd node
 // and to the given universe server.
 func setupTapdHarness(t *testing.T, ht *harnessTest,
@@ -434,6 +448,7 @@ func setupTapdHarness(t *testing.T, ht *harnessTest,
 		ho.disableSyncCache = params.disableSyncCache
 		ho.oracleServerAddress = params.oracleServerAddress
 		ho.sendPriceHint = params.sendPriceHint
+		ho.sweepOrphanUtxos = params.sweepOrphanUtxos
 	}
 
 	tapdCfg := tapdConfig{
diff --git a/itest/test_list_on_test.go b/itest/test_list_on_test.go
@@ -104,6 +104,13 @@ var allTestCases = []*testCase{
 	{
 		name: "zero value anchor sweep",
 		test: testZeroValueAnchorSweep,
+		tapdOptions: []Option{
+			WithSweepOrphanUtxos(),
+		},
+	},
+	{
+		name: "zero value anchor accumulation",
+		test: testZeroValueAnchorAccumulation,
 	},
 	{
 		name: "restart receiver check balance",
diff --git a/itest/zero_value_anchor_test.go b/itest/zero_value_anchor_test.go
@@ -113,7 +113,7 @@ func testZeroValueAnchorSweep(t *harnessTest) {
 			AssetId: genInfo3.AssetId,
 		},
 		AmountToBurn:     assetAmount,
-		ConfirmationText: "assets will be destroyed",
+		ConfirmationText: taprootassets.AssetBurnConfirmationText,
 	})
 	require.NoError(t.t, err)
 
@@ -290,7 +290,7 @@ func testZeroValueAnchorAccumulation(t *harnessTest) {
 	require.NoError(t.t, t.tapd.stop(false))
 
 	// Enable sweeping in the config.
-	t.tapd.clientCfg.Wallet.SweepZeroValueAnchorUtxos = true
+	t.tapd.clientCfg.Wallet.SweepOrphanUtxos = true
 
 	// Restart with the modified config.
 	require.NoError(t.t, t.tapd.start(false))
diff --git a/rpcserver.go b/rpcserver.go
@@ -2671,11 +2671,18 @@ func (r *rpcServer) AnchorVirtualPsbts(ctx context.Context,
 			prevID.OutPoint.String())
 	}
 
-	// Fetch zero-value UTXOs that should be swept as additional inputs.
-	zeroValueInputs, err := r.cfg.AssetStore.FetchZeroValueAnchorUTXOs(ctx)
-	if err != nil {
-		return nil, fmt.Errorf("unable to fetch zero-value "+
-			"UTXOs: %w", err)
+	// Fetch zero-value UTXOs that should be swept as additional inputs if
+	// the feature is enabled.
+	var (
+		zeroValueInputs []*tapfreighter.ZeroValueInput
+		err             error
+	)
+	if r.cfg.SweepOrphanUtxos {
+		zeroValueInputs, err = r.cfg.AssetStore.FetchOrphanUTXOs(ctx)
+		if err != nil {
+			return nil, fmt.Errorf("unable to fetch zero-value "+
+				"UTXOs: %w", err)
+		}
 	}
 
 	resp, err := r.cfg.ChainPorter.RequestShipment(
diff --git a/sample-tapd.conf b/sample-tapd.conf
@@ -412,6 +412,10 @@
 ; Value must be a valid float ranging from 0.00 to 1.00.
 ; wallet.psbt-max-fee-ratio=0.75
 
+; Sweep orphaned UTXOs into send and burn anchor transactions.
+; Disabled by default.
+; wallet.sweep-orphan-utxos=false
+
 [address]
 
 ; If true, tapd will not try to sync issuance proofs for unknown assets when
diff --git a/tapcfg/config.go b/tapcfg/config.go
@@ -294,6 +294,10 @@ type WalletConfig struct {
 	// high fee environments the total fees paid may outweigh the anchor
 	// amount. The allowed values for this argument range from 0.00 to 1.00.
 	PsbtMaxFeeRatio float64 `long:"psbt-max-fee-ratio" description:"The maximum fees to total output amount ratio to use when funding PSBTs for asset transfers. Value must be between 0.00 and 1.00"`
+
+	// SweepOrphanUtxos, when true, sweeps orphaned UTXOs into anchor
+	// transactions created during sends and burns.
+	SweepOrphanUtxos bool `long:"sweep-orphan-utxos" description:"Sweep orphaned UTXOs into anchor transactions created during sends and burns. Disabled by default."`
 }
 
 // UniverseConfig is the config that houses any Universe related config
@@ -492,7 +496,8 @@ func DefaultConfig() Config {
 			DisableSupplyVerifierChainWatch: false,
 		},
 		Wallet: &WalletConfig{
-			PsbtMaxFeeRatio: DefaultPsbtMaxFeeRatio,
+			PsbtMaxFeeRatio:  DefaultPsbtMaxFeeRatio,
+			SweepOrphanUtxos: false,
 		},
 		AddrBook: &AddrBookConfig{
 			DisableSyncer: false,
diff --git a/tapcfg/server.go b/tapcfg/server.go
@@ -443,6 +443,7 @@ func genServerConfig(cfg *Config, cfgLogger btclog.Logger,
 		IgnoreChecker:    ignoreCheckerOpt,
 		Wallet:           walletAnchor,
 		ChainParams:      &tapChainParams,
+		SweepOrphanUtxos: cfg.Wallet.SweepOrphanUtxos,
 	})
 
 	// Addresses can have different proof couriers configured, but both
@@ -750,6 +751,7 @@ func genServerConfig(cfg *Config, cfgLogger btclog.Logger,
 		AssetWallet:              assetWallet,
 		CoinSelect:               coinSelect,
 		ChainPorter:              chainPorter,
+		SweepOrphanUtxos:         cfg.Wallet.SweepOrphanUtxos,
 		FsmDaemonAdapters:        lndFsmDaemonAdapters,
 		SupplyCommitManager:      supplyCommitManager,
 		IgnoreChecker:            ignoreChecker,
diff --git a/tapdb/assets_store.go b/tapdb/assets_store.go
@@ -1349,9 +1349,9 @@ func (a *AssetStore) MarkManagedUTXOAsSwept(ctx context.Context,
 	})
 }
 
-// FetchZeroValueAnchorUTXOs fetches all managed UTXOs that contain only
+// FetchOrphanUTXOs fetches all managed UTXOs that contain only
 // zero-value assets (tombstones and burns).
-func (a *AssetStore) FetchZeroValueAnchorUTXOs(ctx context.Context) (
+func (a *AssetStore) FetchOrphanUTXOs(ctx context.Context) (
 	[]*tapfreighter.ZeroValueInput, error) {
 
 	// Strategy: fetch all managed UTXOs and filter in-memory.
diff --git a/tapfreighter/coin_select.go b/tapfreighter/coin_select.go
@@ -185,17 +185,17 @@ func (s *CoinSelect) selectForAmount(minTotalAmount uint64,
 	return selectedCommitments, nil
 }
 
-// SelectZeroValueCoins fetches all managed UTXOs that contain only
+// SelectOrphanCoins fetches all managed UTXOs that contain only
 // zero-value assets (tombstones and burns). The selected UTXOs are
 // leased for the default lease duration.
-func (s *CoinSelect) SelectZeroValueCoins(ctx context.Context) (
+func (s *CoinSelect) SelectOrphanCoins(ctx context.Context) (
 	[]*ZeroValueInput, error) {
 
 	s.coinLock.Lock()
 	defer s.coinLock.Unlock()
 
 	// Fetch all zero-value UTXOs that are eligible for sweeping.
-	zeroValueInputs, err := s.coinLister.FetchZeroValueAnchorUTXOs(ctx)
+	zeroValueInputs, err := s.coinLister.FetchOrphanUTXOs(ctx)
 	if err != nil {
 		return nil, fmt.Errorf("unable to fetch zero-value UTXOs: %w",
 			err)
diff --git a/tapfreighter/coin_select_test.go b/tapfreighter/coin_select_test.go
@@ -60,7 +60,7 @@ func (m *mockCoinLister) DeleteExpiredLeases(ctx context.Context) error {
 	return nil
 }
 
-func (m *mockCoinLister) FetchZeroValueAnchorUTXOs(
+func (m *mockCoinLister) FetchOrphanUTXOs(
 	context.Context) ([]*ZeroValueInput, error) {
 
 	return nil, nil
diff --git a/tapfreighter/interface.go b/tapfreighter/interface.go
@@ -169,10 +169,9 @@ type CoinLister interface {
 	// DeleteExpiredLeases deletes all expired leases from the database.
 	DeleteExpiredLeases(ctx context.Context) error
 
-	// FetchZeroValueAnchorUTXOs fetches all managed UTXOs that contain only
+	// FetchOrphanUTXOs fetches all managed UTXOs that contain only
 	// zero-value assets (tombstones and burns).
-	FetchZeroValueAnchorUTXOs(ctx context.Context) ([]*ZeroValueInput,
-		error)
+	FetchOrphanUTXOs(ctx context.Context) ([]*ZeroValueInput, error)
 }
 
 // MultiCommitmentSelectStrategy is an enum that describes the strategy that
@@ -201,10 +200,10 @@ type CoinSelector interface {
 	// makes them available for coin selection again.
 	ReleaseCoins(ctx context.Context, utxoOutpoints ...wire.OutPoint) error
 
-	// SelectZeroValueCoins selects all managed UTXOs that contain only
+	// SelectOrphanCoins selects all managed UTXOs that contain only
 	// zero-value assets (tombstones and burns). The selected UTXOs are
 	// leased for the default lease duration.
-	SelectZeroValueCoins(ctx context.Context) ([]*ZeroValueInput, error)
+	SelectOrphanCoins(ctx context.Context) ([]*ZeroValueInput, error)
 }
 
 // ZeroValueInput represents a zero-value UTXO that should be swept.
diff --git a/tapfreighter/wallet.go b/tapfreighter/wallet.go
@@ -225,6 +225,10 @@ type WalletConfig struct {
 
 	// ChainParams is the chain params of the chain we operate on.
 	ChainParams *address.ChainParams
+
+	// SweepOrphanUtxos specifies whether orphaned UTXOs should be swept
+	// into send and burn anchor transactions.
+	SweepOrphanUtxos bool
 }
 
 // AssetWallet is an implementation of the Wallet interface that can create
@@ -663,10 +667,13 @@ func (f *AssetWallet) FundPacket(ctx context.Context,
 		return nil, err
 	}
 
-	zeroValueInputs, err := f.cfg.CoinSelector.SelectZeroValueCoins(ctx)
-	if err != nil {
-		return nil, fmt.Errorf("unable to select zero-value "+
-			"UTXOs: %w", err)
+	var zeroValueInputs []*ZeroValueInput
+	if f.cfg.SweepOrphanUtxos {
+		zeroValueInputs, err = f.cfg.CoinSelector.SelectOrphanCoins(ctx)
+		if err != nil {
+			return nil, fmt.Errorf("unable to select zero-value "+
+				"UTXOs: %w", err)
+		}
 	}
 
 	// If we return with an error, we want to release all the coins we've
@@ -740,10 +747,15 @@ func (f *AssetWallet) FundBurn(ctx context.Context,
 		return nil, err
 	}
 
-	zeroValueInputs, err := f.cfg.CoinSelector.SelectZeroValueCoins(ctx)
-	if err != nil {
-		return nil, fmt.Errorf("unable to select zero-value "+
-			"UTXOs: %w", err)
+	var zeroValueInputs []*ZeroValueInput
+	if f.cfg.SweepOrphanUtxos {
+		zeroValueInputs, err = f.cfg.CoinSelector.SelectOrphanCoins(
+			ctx,
+		)
+		if err != nil {
+			return nil, fmt.Errorf("unable to select zero-value "+
+				"UTXOs: %w", err)
+		}
 	}
 
 	// If we return with an error, we want to release all the coins we've

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,7 @@ func TestTaprootAssetsDaemon(t *testing.T) {`
`120`	`120`	`tapdHarness, uniHarness, proofCourier := setupHarnesses(`
`121`	`121`	`t1, ht, lndHarness, uniServerLndHarness,`
`122`	`122`	`testCase.proofCourierType,`
	`123`	`+ testCase.tapdOptions...,`
`123`	`124`	`)`
`124`	`125`
`125`	`126`	`ht := ht.newHarnessTest(`
Original file line number	Diff line number	Diff line change
`@@ -104,6 +104,13 @@ var allTestCases = []*testCase{`
`104`	`104`	`{`
`105`	`105`	`name: "zero value anchor sweep",`
`106`	`106`	`test: testZeroValueAnchorSweep,`
	`107`	`+ tapdOptions: []Option{`
	`108`	`+ WithSweepOrphanUtxos(),`
	`109`	`+ },`
	`110`	`+ },`
	`111`	`+ {`
	`112`	`+ name: "zero value anchor accumulation",`
	`113`	`+ test: testZeroValueAnchorAccumulation,`
`107`	`114`	`},`
`108`	`115`	`{`
`109`	`116`	`name: "restart receiver check balance",`