otterscan/cmd/otter/server.go

package main

import (
	"encoding/json"
	"net/http"
	"os"

	"gfx.cafe/open/4bytes/sigs"
	"gfx.cafe/open/4bytes/topics"
	"gfx.cafe/open/4bytes/triemap"
	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
	"github.com/wmitsuda/otterscan/cmd/otter/cli/httpcfg"
)

func RouteServer(r chi.Router, cfg httpcfg.HttpCfg) {
	r.Use(middleware.Logger)
	r.Use(middleware.Recoverer)
	r.HandleFunc("/signatures/{hash}", triemap.HttpHandler(sigs.Both))
	r.HandleFunc("/topic0/{hash}", triemap.HttpHandler(topics.Both))
	r.HandleFunc("/config.json", func(w http.ResponseWriter, r *http.Request) {
		json.NewEncoder(w).Encode(map[string]any{
			"erigonURL":       cfg.OtsRpcDaemonUrl,
			"beaconAPI":       cfg.OtsBeaconApiUrl,
			"assetsURLPrefix": cfg.OtsExternalAssetUrl,
		})
	})

	filesDir := http.Dir(cfg.OtsStaticDir)
	fileServer := http.FileServer(filesDir)
	r.Handle("/*", http.StripPrefix("/", fileServer))

	r.Get("/*", func(w http.ResponseWriter, r *http.Request) {
		if _, err := os.Stat(cfg.OtsStaticDir + r.RequestURI); os.IsNotExist(err) {
			http.StripPrefix(r.RequestURI, fileServer).ServeHTTP(w, r)
		} else {
			fileServer.ServeHTTP(w, r)
		}
	})
}