1 files changed, 33 insertions, 0 deletions

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