feat: add bootstrap mode to go peer

go-peer/main.go

@@ -123,6 +123,10 @@ func main() {
 	keyPath := flag.String("tls-key-path", "", "path to the tls key file (for websockets")
 	useLogger := flag.Bool("logger", false, "write logs to file")
 	headless := flag.Bool("headless", false, "run without chat UI")
+	bootstrapper := flag.Bool("bootstrapper", false, "run as a bootstrap peer")
+
+	var directPeers stringSlice
+	flag.Var(&directPeers, "directpeer", "reciprocal gossipsub bootstrap peers (can be used multiple times)")
 
 	var addrsToConnectTo stringSlice
 	flag.Var(&addrsToConnectTo, "connect", "address to connect to (can be used multiple times)")
@@ -194,8 +198,28 @@ func main() {
 		panic(err)
 	}
 
+	gossipSubOpts := []pubsub.Option{}
+
+	if *bootstrapper {
+		gossipSubOpts = append(gossipSubOpts, pubsub.WithPeerExchange(true))
+
+		if len(directPeers) > 0 {
+			dp := peerStrSliceToAddrInfoSlice(directPeers)
+			gossipSubOpts = append(gossipSubOpts, pubsub.WithDirectPeers(dp))
+			pubsub.WithDirectConnectTicks(60) // attempt to reconnect to direct peers every 60 ticks (seconds)
+		}
+
+		pubsub.WithFloodPublish(true)
+
+		// See https://github.com/libp2p/specs/blob/master/pubsub/gossipsub/gossipsub-v1.1.md#recommendations-for-network-operators
+		pubsub.GossipSubD = 0
+		pubsub.GossipSubDlo = 0
+		pubsub.GossipSubDhi = 0
+		pubsub.GossipSubDout = 0
+	}
+
 	// create a new PubSub service using the GossipSub router
-	ps, err := pubsub.NewGossipSub(ctx, h)
+	ps, err := pubsub.NewGossipSub(ctx, h, gossipSubOpts...)
 	if err != nil {
 		panic(err)
 	}
@@ -266,6 +290,24 @@ func main() {
 	}
 }
 
+func peerStrSliceToAddrInfoSlice(peerStrs []string) []peer.AddrInfo {
+	var addrInfos []peer.AddrInfo
+
+	if len(peerStrs) > 0 {
+		for _, addr := range peerStrs {
+			peerInfo, err := peer.AddrInfoFromString(addr)
+			if err != nil {
+				LogMsgf("Failed to parse multiaddr: %s", err.Error())
+				continue
+			}
+
+			addrInfos = append(addrInfos, *peerInfo)
+		}
+	}
+
+	return addrInfos
+}
+
 // printErr is like fmt.Printf, but writes to stderr.
 func printErr(m string, args ...interface{}) {
 	fmt.Fprintf(os.Stderr, m, args...)