Add covenant-based congestion tree (#62)

* covenant based tx builder

* remove relative time delta

* txbuilder/covenant add leaf boolean in node

* txbuilder/covenant final version

* support covenantType

* add GetLeafOutputScript

* remove printLn

* fix linting

* Update asp/internal/app-config/config.go

Co-authored-by: João Bordalo <bordalix@users.noreply.github.com>
Signed-off-by: Louis Singer <41042567+louisinger@users.noreply.github.com>

---------

Signed-off-by: Louis Singer <41042567+louisinger@users.noreply.github.com>
Co-authored-by: João Bordalo <bordalix@users.noreply.github.com>

asp/internal/core/application/service.go

@@ -12,9 +12,7 @@ import (
 	"github.com/ark-network/ark/internal/core/ports"
 	"github.com/decred/dcrd/dcrec/secp256k1/v4"
 	log "github.com/sirupsen/logrus"
-	"github.com/vulpemventures/go-elements/address"
 	"github.com/vulpemventures/go-elements/network"
-	"github.com/vulpemventures/go-elements/payment"
 	"github.com/vulpemventures/go-elements/psetv2"
 )
 
@@ -267,26 +265,20 @@ func (s *service) startFinalization() {
 		return
 	}
 
-	signedPoolTx, err := s.builder.BuildPoolTx(s.pubkey, s.wallet, payments)
+	signedPoolTx, tree, err := s.builder.BuildPoolTx(s.pubkey, s.wallet, payments)
 	if err != nil {
 		changes = round.Fail(fmt.Errorf("failed to create pool tx: %s", err))
 		log.WithError(err).Warn("failed to create pool tx")
 		return
 	}
 
-	tree, err := s.builder.BuildCongestionTree(s.pubkey, signedPoolTx, payments)
-	if err != nil {
-		changes = round.Fail(fmt.Errorf("failed to create congestion tree: %s", err))
-		log.WithError(err).Warn("failed to create congestion tree")
-		return
-	}
-
 	connectors, forfeitTxs, err := s.builder.BuildForfeitTxs(s.pubkey, signedPoolTx, payments)
 	if err != nil {
 		changes = round.Fail(fmt.Errorf("failed to create connectors and forfeit txs: %s", err))
 		log.WithError(err).Warn("failed to create connectors and forfeit txs")
 		return
 	}
+
 	events, _ := round.StartFinalization(connectors, tree, signedPoolTx)
 	changes = append(changes, events...)
 
@@ -354,7 +346,7 @@ func (s *service) updateProjectionStore(round *domain.Round) {
 			}
 		}
 
-		newVtxos := getNewVtxos(s.onchainNework, round)
+		newVtxos := s.getNewVtxos(round)
 		for {
 			if err := repo.AddVtxos(ctx, newVtxos); err != nil {
 				log.WithError(err).Warn("failed to add new vtxos, retrying soon")
@@ -393,7 +385,7 @@ func (s *service) propagateEvents(round *domain.Round) {
 	}
 }
 
-func getNewVtxos(net network.Network, round *domain.Round) []domain.Vtxo {
+func (s *service) getNewVtxos(round *domain.Round) []domain.Vtxo {
 	leaves := round.CongestionTree.Leaves()
 	vtxos := make([]domain.Vtxo, 0)
 	for _, node := range leaves {
@@ -405,9 +397,7 @@ func getNewVtxos(net network.Network, round *domain.Round) []domain.Vtxo {
 				for _, r := range p.Receivers {
 					buf, _ := hex.DecodeString(r.Pubkey)
 					pk, _ := secp256k1.ParsePubKey(buf)
-					p2wpkh := payment.FromPublicKey(pk, &net, nil)
-					addr, _ := p2wpkh.WitnessPubKeyHash()
-					script, _ := address.ToOutputScript(addr)
+					script, _ := s.builder.GetLeafOutputScript(pk, s.pubkey)
 					if bytes.Equal(script, out.Script) {
 						found = true
 						pubkey = r.Pubkey

asp/internal/core/domain/congestion_tree.go

@@ -22,6 +22,19 @@ func (c CongestionTree) Leaves() []Node {
 	return leaves
 }
 
+func (c CongestionTree) Children(nodeTxid string) []Node {
+	var children []Node
+	for _, level := range c {
+		for _, node := range level {
+			if node.ParentTxid == nodeTxid {
+				children = append(children, node)
+			}
+		}
+	}
+
+	return children
+}
+
 func (c CongestionTree) NumberOfNodes() int {
 	var count int
 	for _, level := range c {

asp/internal/core/ports/tx_builder.go

@@ -8,11 +8,9 @@ import (
 type TxBuilder interface {
 	BuildPoolTx(
 		aspPubkey *secp256k1.PublicKey, wallet WalletService, payments []domain.Payment,
-	) (poolTx string, err error)
-	BuildCongestionTree(
-		aspPubkey *secp256k1.PublicKey, poolTx string, payments []domain.Payment,
-	) (congestionTree domain.CongestionTree, err error)
+	) (poolTx string, congestionTree domain.CongestionTree, err error)
 	BuildForfeitTxs(
 		aspPubkey *secp256k1.PublicKey, poolTx string, payments []domain.Payment,
 	) (connectors []string, forfeitTxs []string, err error)
+	GetLeafOutputScript(userPubkey, aspPubkey *secp256k1.PublicKey) ([]byte, error)
 }