1 files changed, 40 insertions, 0 deletions

diff --git a/web/middleware/ddns.go b/web/middleware/ddns.go
new file mode 100644
index 0000000..b213926
--- /dev/null
+++ b/web/middleware/ddns.go
@@ -0,0 +1,40 @@
+package middleware
+
+import (
+        "log"
+        "net/http"
+
+        "github.com/gin-gonic/gin"
+)
+
+const DDNS_AUTH_KEY = "DDNSAuthZone"
+
+type DDNSUpdateRequest struct {
+        Key string `form:"key" binding:"required"`
+}
+
+func DDNSAuthMiddleware(c *gin.Context) {
+        cfg := GetServerConfig(c)
+
+        var req DDNSUpdateRequest
+        if err := c.ShouldBind(&req); err != nil {
+                log.Println("ddnsAuthMiddleware: No key in request")
+                c.AbortWithStatus(http.StatusNotFound)
+                return
+        }
+
+        res := cfg.GetDDNSZoneName(req.Key)
+        if res == "" {
+                log.Println("ddnsAuthMiddleware: Unknown secret")
+                c.AbortWithStatus(http.StatusNotFound)
+                return
+        } else {
+                c.Set(DDNS_AUTH_KEY, res)
+        }
+
+        c.Next()
+}
+
+func GetDDNSAuthKey(c *gin.Context) string {
+        return c.GetString(DDNS_AUTH_KEY)
+}