server+tapcfg: implement AuxChannelNegotiator

Author: GeorgeTsagk

config.go

@@ -15,6 +15,7 @@ import (
 	"github.com/lightninglabs/taproot-assets/rfq"
 	"github.com/lightninglabs/taproot-assets/tapchannel"
 	"github.com/lightninglabs/taproot-assets/tapdb"
+	"github.com/lightninglabs/taproot-assets/tapfeatures"
 	"github.com/lightninglabs/taproot-assets/tapfreighter"
 	"github.com/lightninglabs/taproot-assets/tapgarden"
 	"github.com/lightninglabs/taproot-assets/universe"
@@ -229,6 +230,8 @@ type Config struct {
 
 	AuxTrafficShaper *tapchannel.AuxTrafficShaper
 
+	AuxChanNegotiator *tapfeatures.AuxChannelNegotiator
+
 	AuxInvoiceManager *tapchannel.AuxInvoiceManager
 
 	AuxChanCloser *tapchannel.AuxChanCloser

server.go

@@ -1319,3 +1319,67 @@ func (s *Server) NotifyBroadcast(req *sweep.BumpRequest,
 
 	return s.cfg.AuxSweeper.NotifyBroadcast(req, tx, fee, outpointToTxIndex)
 }
+
+// GetInitRecords is called when sending an init message to a peer. It returns
+// custom records to include in the init message TLVs. The implementation can
+// decide which records to include based on the peer identity.
+func (s *Server) GetInitRecords(
+	peer route.Vertex) (lnwire.CustomRecords, error) {
+
+	srvrLog.Tracef("GetInitRecords called, peer=%s", peer)
+
+	if err := s.waitForReady(); err != nil {
+		return nil, err
+	}
+
+	return s.cfg.AuxChanNegotiator.GetInitRecords(peer)
+}
+
+// ProcessInitRecords handles received init records from a peer. The
+// implementation can store state internally to affect future channel operations
+// with this peer.
+func (s *Server) ProcessInitRecords(peer route.Vertex,
+	customRecords lnwire.CustomRecords) error {
+
+	srvrLog.Tracef("ProcessInitRecords called, peer=%s", peer)
+
+	if err := s.waitForReady(); err != nil {
+		return err
+	}
+
+	return s.cfg.AuxChanNegotiator.ProcessInitRecords(peer, customRecords)
+}
+
+// ProcessReestablishFeatures handles received channel_reestablish feature TLVs.
+// This is a blocking call - the channel link will wait for this method to
+// complete before continuing channel operations. The implementation can modify
+// aux channel behavior based on the negotiated features.
+func (s *Server) ProcessReestablish(cid lnwire.ChannelID, peer route.Vertex) {
+	srvrLog.Tracef("ProcessReestablishFeatures called, cid=%s",
+		cid.String())
+
+	if err := s.waitForReady(); err != nil {
+		srvrLog.Errorf("Failed to handle ProcessReestablish, server " +
+			"not ready")
+	}
+
+	s.cfg.AuxChanNegotiator.ProcessReestablish(
+		cid, peer,
+	)
+}
+
+// ProcessChannelReady handles the event of marking a channel identified by its
+// channel ID as ready to use. We also provide the peer the channel was
+// established with.
+func (s *Server) ProcessChannelReady(cid lnwire.ChannelID, peer route.Vertex) {
+	srvrLog.Tracef("ProcessChannelReady called, cid=%s, peer=%s", cid, peer)
+
+	if err := s.waitForReady(); err != nil {
+		srvrLog.Errorf("ProcessChannelReady got error while waiting " +
+			"for server ready")
+
+		return
+	}
+
+	s.cfg.AuxChanNegotiator.ProcessChannelReady(cid, peer)
+}

tapcfg/server.go

@@ -20,6 +20,7 @@ import (
 	"github.com/lightninglabs/taproot-assets/tapchannel"
 	"github.com/lightninglabs/taproot-assets/tapdb"
 	"github.com/lightninglabs/taproot-assets/tapdb/sqlc"
+	"github.com/lightninglabs/taproot-assets/tapfeatures"
 	"github.com/lightninglabs/taproot-assets/tapfreighter"
 	"github.com/lightninglabs/taproot-assets/tapgarden"
 	"github.com/lightninglabs/taproot-assets/tapscript"
@@ -474,6 +475,9 @@ func genServerConfig(cfg *Config, cfgLogger btclog.Logger,
 		}
 	}
 
+	// Construct the AuxChannelNegotiator.
+	auxChanNegotiator := tapfeatures.NewAuxChannelNegotiator()
+
 	// Construct the RFQ manager.
 	rfqManager, err := rfq.NewManager(rfq.ManagerCfg{
 		PeerMessenger:             msgTransportClient,
@@ -482,6 +486,7 @@ func genServerConfig(cfg *Config, cfgLogger btclog.Logger,
 		PriceOracle:               priceOracle,
 		ChannelLister:             lndServices.Client,
 		GroupLookup:               tapdbAddrBook,
+		AuxChanNegotiator:         auxChanNegotiator,
 		AliasManager:              lndRouterClient,
 		AcceptPriceDeviationPpm:   rfqCfg.AcceptPriceDeviationPpm,
 		SkipAcceptQuotePriceCheck: rfqCfg.SkipAcceptQuotePriceCheck,
@@ -628,9 +633,10 @@ func genServerConfig(cfg *Config, cfgLogger btclog.Logger,
 	)
 	auxTrafficShaper := tapchannel.NewAuxTrafficShaper(
 		&tapchannel.TrafficShaperConfig{
-			ChainParams: &tapChainParams,
-			RfqManager:  rfqManager,
-			NoopHTLCs:   cfg.Channel.NoopHTLCs,
+			ChainParams:       &tapChainParams,
+			RfqManager:        rfqManager,
+			NoopHTLCs:         cfg.Channel.NoopHTLCs,
+			AuxChanNegotiator: auxChanNegotiator,
 		},
 	)
 	auxInvoiceManager := tapchannel.NewAuxInvoiceManager(
@@ -745,6 +751,7 @@ func genServerConfig(cfg *Config, cfgLogger btclog.Logger,
 		AuxFundingController:     auxFundingController,
 		AuxChanCloser:            auxChanCloser,
 		AuxTrafficShaper:         auxTrafficShaper,
+		AuxChanNegotiator:        auxChanNegotiator,
 		AuxInvoiceManager:        auxInvoiceManager,
 		AuxSweeper:               auxSweeper,
 		LogWriter:                cfg.LogWriter,