WIP: supplycommit: push supply commitment to remote uni during CommitFinalizeState

Author: ffranr

tapcfg/server.go

@@ -626,6 +626,7 @@ func genServerConfig(cfg *Config, cfgLogger btclog.Logger,
 			AssetLookup:    tapdbAddrBook,
 			KeyRing:        keyRing,
 			Chain:          chainBridge,
+			SupplySyncer:   &supplySyncer,
 			DaemonAdapters: lndFsmDaemonAdapters,
 			StateLog:       supplyCommitStore,
 			ChainParams:    *tapChainParams.Params,

universe/supplycommit/env.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"crypto/sha256"
 	"fmt"
+	"net/url"
 
 	"github.com/btcsuite/btcd/btcec/v2"
 	"github.com/btcsuite/btcd/btcutil"
@@ -100,6 +101,39 @@ type SupplyLeaves struct {
 	IgnoreLeafEntries []NewIgnoreEvent
 }
 
+// NewSupplyLeavesFromEvents creates a SupplyLeaves instance from a slice of
+// SupplyUpdateEvent instances.
+func NewSupplyLeavesFromEvents(events []SupplyUpdateEvent) (SupplyLeaves,
+	error) {
+
+	var leaves SupplyLeaves
+	for idx := range events {
+		event := events[idx]
+
+		switch e := event.(type) {
+		case *NewMintEvent:
+			leaves.IssuanceLeafEntries = append(
+				leaves.IssuanceLeafEntries, *e,
+			)
+
+		case *NewBurnEvent:
+			leaves.BurnLeafEntries = append(
+				leaves.BurnLeafEntries, *e,
+			)
+
+		case *NewIgnoreEvent:
+			leaves.IgnoreLeafEntries = append(
+				leaves.IgnoreLeafEntries, *e,
+			)
+
+		default:
+			return leaves, fmt.Errorf("unknown event type: %T", e)
+		}
+	}
+
+	return leaves, nil
+}
+
 // AssetLookup is an interface that allows us to query for asset
 // information, such as asset groups and asset metadata.
 type AssetLookup interface {
@@ -498,6 +532,17 @@ type StateMachineStore interface {
 		asset.Specifier) ([]SupplyUpdateEvent, error)
 }
 
+// SupplySyncer is an interface that allows the state machine to insert
+// supply commitments into the remote universe server.
+type SupplySyncer interface {
+	// PushSupplyCommitment pushes a supply commitment to the remote
+	// universe server. This function should block until the sync insertion
+	// is complete.
+	PushSupplyCommitment(ctx context.Context, assetSpec asset.Specifier,
+		commitment RootCommitment, updateLeaves SupplyLeaves,
+		chainProof ChainProof, canonicalUniverses []url.URL) error
+}
+
 // Environment is a set of dependencies that a state machine may need to carry
 // out the logic for a given state transition. All fields are to be considered
 // immutable, and will be fixed for the lifetime of the state machine.
@@ -518,6 +563,10 @@ type Environment struct {
 	// Wallet is the main wallet interface used to managed PSBT packets.
 	Wallet Wallet
 
+	// AssetLookup is used to look up asset information such as asset groups
+	// and asset metadata.
+	AssetLookup AssetLookup
+
 	// KeyRing is the main key ring interface used to manage keys.
 	KeyRing KeyRing
 
@@ -526,6 +575,10 @@ type Environment struct {
 	// TODO(roasbeef): can make a slimmer version of
 	Chain tapgarden.ChainBridge
 
+	// SupplySyncer is used to insert supply commitments into the remote
+	// universe server.
+	SupplySyncer SupplySyncer
+
 	// StateLog is the main state log that is used to track the state of the
 	// state machine. This is used to persist the state of the state machine
 	// across restarts.

universe/supplycommit/manager.go

@@ -64,6 +64,10 @@ type ManagerCfg struct {
 	// TODO(roasbeef): can make a slimmer version of
 	Chain tapgarden.ChainBridge
 
+	// SupplySyncer is used to insert supply commitments into the remote
+	// universe server.
+	SupplySyncer SupplySyncer
+
 	// DaemonAdapters is a set of adapters that allow the state machine to
 	// interact with external daemons whilst processing internal events.
 	DaemonAdapters DaemonAdapters
@@ -227,8 +231,10 @@ func (m *Manager) fetchStateMachine(
 		TreeView:         m.cfg.TreeView,
 		Commitments:      m.cfg.Commitments,
 		Wallet:           m.cfg.Wallet,
+		AssetLookup:      m.cfg.AssetLookup,
 		KeyRing:          m.cfg.KeyRing,
 		Chain:            m.cfg.Chain,
+		SupplySyncer:     m.cfg.SupplySyncer,
 		StateLog:         m.cfg.StateLog,
 		CommitConfTarget: DefaultCommitConfTarget,
 		ChainParams:      m.cfg.ChainParams,

universe/supplycommit/mock.go

@@ -2,6 +2,7 @@ package supplycommit
 
 import (
 	"context"
+	"net/url"
 	"sync"
 
 	"github.com/btcsuite/btcd/btcec/v2"
@@ -451,3 +452,18 @@ func (m *mockAssetLookup) FetchInternalKeyLocator(ctx context.Context,
 	args := m.Called(ctx, rawKey)
 	return args.Get(0).(keychain.KeyLocator), args.Error(1)
 }
+
+// mockSupplySyncer is a mock implementation of the SupplySyncer interface.
+type mockSupplySyncer struct {
+	mock.Mock
+}
+
+func (m *mockSupplySyncer) PushSupplyCommitment(ctx context.Context,
+	assetSpec asset.Specifier, commitment RootCommitment,
+	updateLeaves SupplyLeaves, chainProof ChainProof,
+	canonicalUniverses []url.URL) error {
+
+	args := m.Called(ctx, assetSpec, commitment, updateLeaves, chainProof,
+		canonicalUniverses)
+	return args.Error(0)
+}