bunker: repeat connection info every now and then.

bunker.go

@@ -1,10 +1,13 @@
 package main
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	"net/url"
 	"os"
+	"sync"
+	"time"
 
 	"github.com/fatih/color"
 	"github.com/nbd-wtf/go-nostr"
@@ -65,13 +68,18 @@ var bunker = &cli.Command{
 			return err
 		}
 		npub, _ := nip19.EncodePublicKey(pubkey)
+		bunkerURI := fmt.Sprintf("bunker://%s?%s", pubkey, qs.Encode())
 		bold := color.New(color.Bold).Sprint
-		log("listening at %v:\n  pubkey: %s \n  npub: %s\n  bunker: %s\n\n",
-			bold(relayURLs),
-			bold(pubkey),
-			bold(npub),
-			bold(fmt.Sprintf("bunker://%s?%s", pubkey, qs.Encode())),
-		)
+
+		printBunkerInfo := func() {
+			log("listening at %v:\n  pubkey: %s \n  npub: %s\n  bunker: %s\n\n",
+				bold(relayURLs),
+				bold(pubkey),
+				bold(npub),
+				bold(bunkerURI),
+			)
+		}
+		printBunkerInfo()
 
 		alwaysYes := c.Bool("yes")
 
@@ -85,10 +93,21 @@ var bunker = &cli.Command{
 		})
 
 		signer := nip46.NewStaticKeySigner(sec)
+		handlerWg := sync.WaitGroup{}
+		printLock := sync.Mutex{}
+
+		// just a gimmick
+		var cancelPreviousBunkerInfoPrint context.CancelFunc
+		_, cancel := context.WithCancel(c.Context)
+		cancelPreviousBunkerInfoPrint = cancel
+
 		for ie := range events {
+			cancelPreviousBunkerInfoPrint() // this prevents us from printing a million bunker info blocks
+
+			// handle the NIP-46 request event
 			req, resp, eventResponse, harmless, err := signer.HandleRequest(ie.Event)
 			if err != nil {
-				log("< failed to handle request from %s: %s", ie.Event.PubKey, err.Error())
+				log("< failed to handle request from %s: %s\n", ie.Event.PubKey, err.Error())
 				continue
 			}
 
@@ -98,16 +117,39 @@ var bunker = &cli.Command{
 			log("~ responding with %s\n", string(jresp))
 
 			if alwaysYes || harmless || askProceed(ie.Event.PubKey) {
+				handlerWg.Add(len(relayURLs))
 				for _, relayURL := range relayURLs {
-					if relay, _ := pool.EnsureRelay(relayURL); relay != nil {
-						if err := relay.Publish(c.Context, eventResponse); err == nil {
-							log("* sent response through %s\n", relay.URL)
-						} else {
-							log("* failed to send response: %s\n", err)
+					go func(relayURL string) {
+						if relay, _ := pool.EnsureRelay(relayURL); relay != nil {
+							err := relay.Publish(c.Context, eventResponse)
+							printLock.Lock()
+							if err == nil {
+								log("* sent response through %s\n", relay.URL)
+							} else {
+								log("* failed to send response: %s\n", err)
+							}
+							printLock.Unlock()
+							handlerWg.Done()
 						}
-					}
+					}(relayURL)
 				}
+				handlerWg.Wait()
 			}
+
+			// just after handling one request we trigger this
+			go func() {
+				ctx, cancel := context.WithCancel(c.Context)
+				defer cancel()
+				cancelPreviousBunkerInfoPrint = cancel
+				// the idea is that we will print the bunker URL again so it is easier to copy-paste by users
+				// but we will only do if the bunker is inactive for more than 5 minutes
+				select {
+				case <-ctx.Done():
+				case <-time.After(time.Minute * 5):
+					fmt.Fprintf(os.Stderr, "\n")
+					printBunkerInfo()
+				}
+			}()
 		}
 
 		return nil