provide addrbook.json

Author: 0xbcdev

client/html/index.tmpl

@@ -61,6 +61,20 @@
                     </div>
                 </div>
             </div>
+            <div class="accordion-item">
+                <h3 class="accordion-header" id="headingAddrbook">
+                    <span class="accordion-button cursor-pointer" data-bs-toggle="collapse" data-bs-target="#collapseAddrbook" aria-expanded="true" aria-controls="collapseAddrbook">
+                        Addrbook
+                    </span>
+                </h3>
+                <div id="collapseAddrbook" class="accordion-collapse collapse show" aria-labelledby="headingAddrbook" data-bs-parent="#accordionMain">
+                    <div class="card card-body">
+                        <p>Download: <a href="/download/addrbook.json">addrbook.json</a></p>
+                        <pre class="border p-2"><code>wget -O addrbook.json https://{[{ .host }]}/download/addrbook.json --inet4-only
+mv addrbook.json $HOME/{[{ .generalNodeHomeName }]}/config</code></pre>
+                    </div>
+                </div>
+            </div>
             <div class="accordion-item">
                 <h3 class="accordion-header" id="headingSnapshot">
                     <span class="accordion-button cursor-pointer" data-bs-toggle="collapse" data-bs-target="#collapseSnapshot" aria-expanded="true" aria-controls="collapseSnapshot">

services/web_server/download-addrbook.go

@@ -0,0 +1,73 @@
+package web_server
+
+import (
+	webtypes "github.com/bcdevtools/node-management/services/web_server/types"
+	"github.com/bcdevtools/node-management/types"
+	"github.com/bcdevtools/node-management/utils"
+	"github.com/gin-gonic/gin"
+	"github.com/pkg/errors"
+	"net/http"
+	"time"
+)
+
+var cacheAddrBook *types.TimeBasedCache
+
+func HandleDownloadAddrBook(c *gin.Context) {
+	w := wrapGin(c)
+
+	addrBook, err := getAddrbook(w.Config())
+	if err != nil {
+		utils.PrintlnStdErr("ERR: failed to get addrbook.json:", err)
+		w.PrepareDefaultErrorResponse().
+			WithResult("failed to get addrbook.json").
+			SendResponse()
+		return
+	}
+	if addrBook == nil || len(addrBook.Addrs) == 0 {
+		w.PrepareDefaultErrorResponse().
+			WithHttpStatusCode(http.StatusServiceUnavailable).
+			WithResult("failed to get addrbook.json").
+			SendResponse()
+		return
+	}
+
+	c.Header("Content-Disposition", "attachment; filename=addrbook.json")
+	c.JSON(http.StatusOK, addrBook)
+}
+
+func getAddrbook(cfg webtypes.Config) (*types.AddrBook, error) {
+	if addrBook := cacheAddrBook.GetRL(); addrBook != nil {
+		return addrBook.(*types.AddrBook), nil
+	}
+
+	addrBook, err := cacheAddrBook.UpdateWL(func() (any, error) {
+		addrBook := &types.AddrBook{}
+		if err := addrBook.ReadAddrBook(cfg.GetAddrBookFilePath()); err != nil {
+			return nil, errors.Wrap(err, "failed to read addrbook")
+		}
+
+		livePeers := addrBook.GetLivePeers(48 * time.Hour)
+
+		if len(livePeers) == 0 && cfg.Debug {
+			// load random, include dead peers, on debug mode
+			livePeers = addrBook.Addrs
+			if len(livePeers) > 10 {
+				livePeers = livePeers[:10]
+			}
+		}
+
+		addrBook.Addrs = livePeers
+
+		return addrBook, nil
+	}, true)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return addrBook.(*types.AddrBook), nil
+}
+
+func init() {
+	cacheAddrBook = types.NewTimeBasedCache(60 * time.Second)
+}

services/web_server/handle_node_peers.go

@@ -20,6 +20,7 @@ func HandleApiNodeLivePeers(c *gin.Context) {
 	if err != nil {
 		utils.PrintlnStdErr("ERR: failed to get live peers:", err)
 		w.PrepareDefaultErrorResponse().WithResult("failed to get live peers").SendResponse()
+		return
 	}
 
 	w.PrepareDefaultSuccessResponse(peers).SendResponse()

services/web_server/handle_web_page.go

@@ -46,6 +46,7 @@ func HandleWebIndex(c *gin.Context) {
 	fmt.Println(chainDescriptionLines)
 
 	c.HTML(http.StatusOK, "index.tmpl", gin.H{
+		"host":                c.Request.Host,
 		"title":               fmt.Sprintf("%s snapshot by %s", cfg.ChainName, cfg.Brand),
 		"description":         fmt.Sprintf("Snapshot data, live-peers for %s (%s) by %s", cfg.ChainName, cfg.ChainID, cfg.Brand),
 		"chainName":           cfg.ChainName,

services/web_server/web_server.go

@@ -60,6 +60,7 @@ func StartWebServer(cfg webtypes.Config) {
 
 	// Web
 	r.GET("/", HandleWebIndex)
+	r.GET("/download/addrbook.json", HandleDownloadAddrBook)
 
 	fmt.Println("INF: starting Web service at", binding)