address+universe: allow syncing of specific group key

We'll want to be able to query the asset leaves for a group key to map
them to one or more asset IDs.

Author: guggero

address/book.go

@@ -82,8 +82,8 @@ type QueryParams struct {
 // known to universe servers in our federation.
 type AssetSyncer interface {
 	// SyncAssetInfo queries the universes in our federation for genesis
-	// and asset group information about the given asset ID.
-	SyncAssetInfo(ctx context.Context, assetID *asset.ID) error
+	// and asset group information about the given asset.
+	SyncAssetInfo(ctx context.Context, specifier asset.Specifier) error
 
 	// EnableAssetSync updates the sync config for the given asset so that
 	// we sync future issuance proofs.
@@ -232,7 +232,7 @@ func (b *Book) QueryAssetInfo(ctx context.Context,
 	log.Debugf("Asset %v is unknown, attempting to bootstrap", id.String())
 
 	// Use the AssetSyncer to query our universe federation for the asset.
-	err = b.cfg.Syncer.SyncAssetInfo(ctx, &id)
+	err = b.cfg.Syncer.SyncAssetInfo(ctx, asset.NewSpecifierFromId(id))
 	if err != nil {
 		return nil, err
 	}
@@ -264,6 +264,27 @@ func (b *Book) QueryAssetInfo(ctx context.Context,
 	return assetGroup, nil
 }
 
+// SyncAssetGroup attempts to enable asset sync for the given asset group, then
+// perform an initial sync with the federation for that group.
+func (b *Book) SyncAssetGroup(ctx context.Context,
+	groupKey btcec.PublicKey) error {
+
+	groupInfo := &asset.AssetGroup{
+		GroupKey: &asset.GroupKey{
+			GroupPubKey: groupKey,
+		},
+	}
+	err := b.cfg.Syncer.EnableAssetSync(ctx, groupInfo)
+	if err != nil {
+		return fmt.Errorf("unable to enable asset sync: %w", err)
+	}
+
+	// Use the AssetSyncer to query our universe federation for the asset.
+	return b.cfg.Syncer.SyncAssetInfo(
+		ctx, asset.NewSpecifierFromGroupKey(groupKey),
+	)
+}
+
 // FetchAssetMetaByHash attempts to fetch an asset meta based on an asset hash.
 func (b *Book) FetchAssetMetaByHash(ctx context.Context,
 	metaHash [asset.MetaHashLen]byte) (*proof.MetaReveal, error) {
@@ -296,7 +317,7 @@ func (b *Book) FetchAssetMetaForAsset(ctx context.Context,
 		assetID.String())
 
 	// Use the AssetSyncer to query our universe federation for the asset.
-	err = b.cfg.Syncer.SyncAssetInfo(ctx, &assetID)
+	err = b.cfg.Syncer.SyncAssetInfo(ctx, asset.NewSpecifierFromId(assetID))
 	if err != nil {
 		return nil, err
 	}

tapgarden/mock.go

@@ -823,8 +823,8 @@ func (m *MockAssetSyncer) FetchAsset(id asset.ID) (*asset.AssetGroup, error) {
 	}
 }
 
-func (m *MockAssetSyncer) SyncAssetInfo(_ context.Context,
-	id *asset.ID) error {
+func (m *MockAssetSyncer) SyncAssetInfo(_ context.Context, id *asset.ID,
+	_ *btcec.PublicKey) error {
 
 	if id == nil {
 		return fmt.Errorf("no asset ID provided")

universe/auto_syncer.go

@@ -7,6 +7,7 @@ import (
 	"sync"
 	"time"
 
+	"github.com/btcsuite/btcd/btcec/v2"
 	"github.com/davecgh/go-spew/spew"
 	"github.com/lightninglabs/taproot-assets/address"
 	"github.com/lightninglabs/taproot-assets/asset"
@@ -807,29 +808,43 @@ func (f *FederationEnvoy) tryFetchServers() ([]ServerAddr, error) {
 // SyncAssetInfo queries the universes in our federation for genesis and asset
 // group information about the given asset ID.
 func (f *FederationEnvoy) SyncAssetInfo(ctx context.Context,
-	assetID *asset.ID) error {
+	specifier asset.Specifier) error {
 
-	if assetID == nil {
-		return fmt.Errorf("no asset ID provided")
+	uniID := Identifier{
+		ProofType: ProofTypeIssuance,
 	}
 
+	switch {
+	case specifier.HasId() && specifier.HasGroupPubKey():
+		return fmt.Errorf("must set either asset ID or group key for " +
+			"asset sync")
+
+	case !specifier.HasId() && !specifier.HasGroupPubKey():
+		return fmt.Errorf("no asset ID or group key provided")
+	}
+
+	specifier.WhenId(func(id asset.ID) {
+		uniID.AssetID = id
+	})
+	specifier.WhenGroupPubKey(func(groupKey btcec.PublicKey) {
+		uniID.GroupKey = &groupKey
+	})
+
 	// Fetch the set of universe servers in our federation.
 	fedServers, err := f.tryFetchServers()
 	if err != nil {
 		return err
 	}
 
 	assetConfig := FedUniSyncConfig{
-		UniverseID: Identifier{
-			AssetID:   *assetID,
-			ProofType: ProofTypeIssuance,
-		},
+		UniverseID:      uniID,
 		AllowSyncInsert: true,
 		AllowSyncExport: false,
 	}
 	fullConfig := SyncConfigs{
 		UniSyncConfigs: []*FedUniSyncConfig{&assetConfig},
 	}
+
 	// We'll sync with Universe servers in parallel and collect the diffs
 	// from any successful syncs. There can only be one diff per server, as
 	// we're only syncing one universe root.
@@ -846,8 +861,8 @@ func (f *FederationEnvoy) SyncAssetInfo(ctx context.Context,
 		// Sync failures are expected from Universe servers that do not
 		// have a relevant universe root.
 		if err != nil {
-			log.Warnf("Asset lookup failed: asset_id=%v, "+
-				"remote_server=%v: %v", assetID.String(),
+			log.Warnf("Asset lookup failed: id=%v, "+
+				"remote_server=%v: %v", uniID.String(),
 				addr.HostStr(), err)
 
 			// We don't want to abort syncing here, as this might
@@ -863,8 +878,8 @@ func (f *FederationEnvoy) SyncAssetInfo(ctx context.Context,
 			if len(syncDiff) != 1 {
 				log.Warnf("Unexpected number of sync diffs "+
 					"when looking up asset: num_diffs=%d, "+
-					"asset_id=%v, remote_server=%v",
-					len(syncDiff), assetID.String(),
+					"id=%v, remote_server=%v",
+					len(syncDiff), uniID.String(),
 					addr.HostStr())
 
 				// We don't want to abort syncing here, as this
@@ -891,12 +906,11 @@ func (f *FederationEnvoy) SyncAssetInfo(ctx context.Context,
 
 	syncDiffs := fn.Collect(returnedSyncDiffs)
 	log.Infof("Synced new Universe leaves for asset %v, diff_size=%v",
-		assetID.String(), len(syncDiffs))
+		uniID.String(), len(syncDiffs))
 
-	// TODO(jhb): Log successful syncs?
 	if len(syncDiffs) == 0 {
 		return fmt.Errorf("asset lookup failed for asset: %v",
-			assetID.String())
+			uniID.String())
 	}
 
 	return nil