refactor(p2p): use map for enabled transports instead of separate flags

Author: gacevicljubisa

pkg/p2p/libp2p/export_test.go

@@ -7,6 +7,7 @@ package libp2p
 import (
 	"context"
 
+	"github.com/ethersphere/bee/v2/pkg/bzz"
 	handshake "github.com/ethersphere/bee/v2/pkg/p2p/libp2p/internal/handshake"
 	libp2pm "github.com/libp2p/go-libp2p"
 	"github.com/libp2p/go-libp2p/core/host"
@@ -63,7 +64,9 @@ func (s *Service) FilterSupportedAddresses(addrs []ma.Multiaddr) []ma.Multiaddr
 }
 
 func (s *Service) SetTransportFlags(hasTCP, hasWS, hasWSS bool) {
-	s.hasTCPTransport = hasTCP
-	s.hasWSTransport = hasWS
-	s.hasWSSTransport = hasWSS
+	s.enabledTransports = map[bzz.TransportType]bool{
+		bzz.TransportTCP: hasTCP,
+		bzz.TransportWS:  hasWS,
+		bzz.TransportWSS: hasWSS,
+	}
 }

pkg/p2p/libp2p/libp2p.go

@@ -124,9 +124,7 @@ type Service struct {
 	enableWS           bool
 	autoTLSCertManager autoTLSCertManager
 	zapLogger          *zap.Logger
-	hasTCPTransport    bool
-	hasWSTransport     bool
-	hasWSSTransport    bool
+	enabledTransports  map[bzz.TransportType]bool
 }
 
 type lightnodes interface {
@@ -549,9 +547,11 @@ func New(ctx context.Context, signer beecrypto.Signer, networkID uint64, overlay
 		enableWS:           o.EnableWS,
 		autoTLSCertManager: certManager,
 		zapLogger:          zapLogger,
-		hasTCPTransport:    true, // TCP transport is always included
-		hasWSTransport:     o.EnableWS,
-		hasWSSTransport:    o.EnableWSS,
+		enabledTransports: map[bzz.TransportType]bool{
+			bzz.TransportTCP: true, // TCP transport is always included
+			bzz.TransportWS:  o.EnableWS,
+			bzz.TransportWSS: o.EnableWSS,
+		},
 	}
 
 	peerRegistry.setDisconnecter(s)
@@ -806,16 +806,7 @@ func (s *Service) handleIncoming(stream network.Stream) {
 
 // isTransportSupported checks if the given transport type is supported by this service.
 func (s *Service) isTransportSupported(t bzz.TransportType) bool {
-	switch t {
-	case bzz.TransportTCP:
-		return s.hasTCPTransport
-	case bzz.TransportWS:
-		return s.hasWSTransport
-	case bzz.TransportWSS:
-		return s.hasWSSTransport
-	default:
-		return false
-	}
+	return s.enabledTransports[t]
 }
 
 // filterSupportedAddresses filters multiaddresses to only include those