impl home.

2026-01-31 16:18:25 -07:00
parent 057a0ab341
commit 7d3da593ca
2 changed files with 72 additions and 3 deletions
--- a/server/server.go
+++ b/server/server.go
@@ -13,8 +13,10 @@ import (
 	"net/http"
 	"net/url"
 	"regexp"
+	"slices"
 	"strconv"
 	"strings"
+	"text/template"
 	"time"

 	"gitea.bel.blue/local/rproxy3/config"
@@ -198,9 +200,13 @@ func (s *Server) Pre(foo http.HandlerFunc) http.HandlerFunc {
 			return
 		}

-		if mapKey(r.Host) == "_" {
+		switch mapKey(r.Host) {
+		case "_":
 			s.List(w)
 			return
+		case "home":
+			s.Home(w, r)
+			return
 		}

 		if auth, err := s.lookupAuth(mapKey(r.Host)); err != nil {
@@ -285,6 +291,50 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	s.Pre(s.Proxy)(w, r)
 }

+func (s *Server) Home(w http.ResponseWriter, r *http.Request) {
+	keys := s.db.Keys(nsRouting)
+	routePrefixes := []string{}
+	for _, key := range keys {
+		u, _ := s.lookup(key)
+		if u != nil && u.String() != "" {
+			routePrefixes = append(routePrefixes, strings.Split(key, "/")[0])
+		}
+	}
+	slices.Sort(routePrefixes)
+	t := strings.ReplaceAll(`
+		<DOCTYPE! html>
+		<html>
+			<header>
+			</header>
+			<body>
+				{{- $baseDomain := .baseDomain }}
+				{{- $scheme := .scheme }}
+				{{- range .routePrefixes }}
+					{{- $url := printf "%s://%s.%s" $scheme . $baseDomain }}
+               <a href="{{ $url }}">{{ . }}</a><br>
+				{{- end }}
+			</body>
+			<footer>
+			</footer>
+		</html>
+	`, "\t", "  ")
+	tmpl, err := template.New("home").Parse(t)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	w.Header().Set("Content-Type", "text/html")
+	if err := tmpl.Execute(w, map[string]any{
+		"routePrefixes": routePrefixes,
+		"domain":        r.Host,
+		"baseDomain":    strings.Join(strings.Split(r.Host, ".")[1:], "."),
+		"scheme":        r.URL.Scheme,
+	}); err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+}
+
 func (s *Server) List(w http.ResponseWriter) {
 	keys := s.db.Keys(nsRouting)
 	hostURL := map[string]string{}
--- a/server/server_test.go
+++ b/server/server_test.go
@@ -38,20 +38,39 @@ func TestServerStart(t *testing.T) {
 }

 func mockServer() *Server {
+	s := _mockServer()
 	portServer := httptest.NewServer(nil)
-	port := strings.Split(portServer.URL, ":")[2]
+	s.addr = ":" + strings.Split(portServer.URL, ":")[2]
 	portServer.Close()
+
+	return s
+}
+
+func _mockServer() *Server {
 	s := &Server{
 		db:      storage.NewMap(),
-		addr:    ":" + port,
 		limiter: rate.NewLimiter(rate.Limit(50), 50),
 	}
 	if err := s.Routes(); err != nil {
 		panic(fmt.Sprintf("cannot initiate server routes; %v", err))
 	}
+	if err := s.Route("test", config.Proxy{To: "http://icanhazip.com"}); err != nil {
+		panic(err.Error())
+	}
 	return s
 }

+func TestServerHome(t *testing.T) {
+	s := _mockServer()
+	r := httptest.NewRequest(http.MethodGet, "https://home.int.bel.blue", nil)
+	w := httptest.NewRecorder()
+	s.Home(w, r)
+	if w.Code != http.StatusOK {
+		t.Fatalf("(%d) %s", w.Code, w.Body.Bytes())
+	}
+	t.Logf("%s", w.Body.Bytes())
+}
+
 func TestServerRoute(t *testing.T) {
 	server := mockServer()
 	p := config.Proxy{