overrides

internal/api/handlers.go

@@ -1,7 +1,9 @@
 package api
 
 import (
+	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"net/http"
 	"os"
@@ -188,8 +190,9 @@ func (s *Server) handleDNSTest(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 	var in struct {
-		Name  string `json:"name"`
-		QType string `json:"type"`
+		Name     string `json:"name"`
+		QType    string `json:"type"`
+		Upstream string `json:"upstream"`
 	}
 	if err := decodeJSON(r, &in); err != nil {
 		writeError(w, http.StatusBadRequest, "malformed request body")
@@ -200,7 +203,20 @@ func (s *Server) handleDNSTest(w http.ResponseWriter, r *http.Request) {
 		writeError(w, http.StatusBadRequest, "name is required")
 		return
 	}
-	v, err := s.hooks.Resolve(r.Context(), name, in.QType)
+	resolve := s.hooks.Resolve
+	if in.Upstream != "" {
+		resolve = func(ctx context.Context, n, qt string) (any, error) {
+			if s.hooks.ResolveUpstream == nil {
+				return nil, errors.New("direct upstream lookup unavailable")
+			}
+			return s.hooks.ResolveUpstream(ctx, n, qt, in.Upstream)
+		}
+	}
+	if resolve == nil {
+		writeError(w, http.StatusNotImplemented, "dns test not available")
+		return
+	}
+	v, err := resolve(r.Context(), name, in.QType)
 	if err != nil {
 		writeJSON(w, http.StatusOK, map[string]any{"name": name, "error": err.Error()})
 		return
@@ -208,6 +224,50 @@ func (s *Server) handleDNSTest(w http.ResponseWriter, r *http.Request) {
 	writeJSON(w, http.StatusOK, v)
 }
 
+func (s *Server) handleAddressOverrides(w http.ResponseWriter, r *http.Request) {
+	if s.hooks.AddressOverrides == nil {
+		writeError(w, http.StatusNotImplemented, "address overrides unavailable")
+		return
+	}
+	v, err := s.hooks.AddressOverrides(r.Context())
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+	writeJSON(w, http.StatusOK, v)
+}
+func (s *Server) handleAddressOverridePut(w http.ResponseWriter, r *http.Request) {
+	if s.hooks.SaveAddressOverride == nil {
+		writeError(w, http.StatusNotImplemented, "address overrides are read-only")
+		return
+	}
+	var in struct {
+		Domain  string `json:"domain"`
+		Address string `json:"address"`
+	}
+	if decodeJSON(r, &in) != nil {
+		writeError(w, http.StatusBadRequest, "malformed request body")
+		return
+	}
+	if err := s.hooks.SaveAddressOverride(r.Context(), in.Domain, in.Address); err != nil {
+		writeError(w, http.StatusBadRequest, err.Error())
+		return
+	}
+	writeJSON(w, http.StatusOK, map[string]any{"ok": true})
+}
+func (s *Server) handleAddressOverrideDelete(w http.ResponseWriter, r *http.Request) {
+	if s.hooks.DeleteAddressOverride == nil {
+		writeError(w, http.StatusNotImplemented, "address overrides are read-only")
+		return
+	}
+	domain := r.PathValue("domain")
+	if err := s.hooks.DeleteAddressOverride(r.Context(), domain); err != nil {
+		writeError(w, http.StatusBadRequest, err.Error())
+		return
+	}
+	writeJSON(w, http.StatusOK, map[string]any{"ok": true})
+}
+
 func (s *Server) handleLocalZones(w http.ResponseWriter, r *http.Request) {
 	if s.hooks.LocalZones == nil {
 		writeError(w, http.StatusNotImplemented, "local zone editor is unavailable")