1 files changed, 33 insertions, 0 deletions

diff --git a/web/middleware/dns_manage.go b/web/middleware/dns_manage.go
new file mode 100644
index 0000000..d3b420b
--- /dev/null
+++ b/web/middleware/dns_manage.go
@@ -0,0 +1,33 @@
+package middleware
+
+import (
+        "net/http"
+
+        "github.com/gin-gonic/gin"
+)
+
+const dnsUserId = "DNSManageUserID"
+
+func DnsManageAuthMiddleware(c *gin.Context) {
+        cfg := GetServerConfig(c)
+
+        user, pwd, ok := c.Request.BasicAuth()
+        if !ok {
+                c.Request.Header.Set("WWW-Authenticate", `Basic realm="closed site"`)
+                c.AbortWithStatus(http.StatusUnauthorized)
+                return
+        }
+
+        if !cfg.DNSUserAuth(user, pwd) {
+                c.AbortWithStatus(http.StatusForbidden)
+                return
+        } else {
+                c.Set(dnsUserId, user)
+        }
+
+        c.Next()
+}
+
+func GetDnsAuthContext(c *gin.Context) string {
+        return c.GetString(dnsUserId)
+}