bugfix: support target parameter for prometheus (#1)

Fixes #1
jop-software · Aug 2, 2022 · 27b5046 · 27b5046
1 parent e0b1fa0
commit 27b5046
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -15,7 +15,7 @@ IMAP_PASSWORD=""
 ### Probe
 
 ```txt
-http://127.0.0.1:9101/probe?mailbox=INBOX
+http://127.0.0.1:9101/probe?target=INBOX
 ```
 
 ### Provided metrics

diff --git a/main.go b/main.go
@@ -81,12 +81,18 @@ func main() {
 	})
 
 	http.HandleFunc("/probe", func(w http.ResponseWriter, r *http.Request) {
-		mailbox := r.URL.Query().Get("mailbox")
-		if mailbox == "" {
-			http.Error(w, "Mailbox parameter is missing", http.StatusBadRequest)
-			return
+		// Maily use the target parameter, but support mailbox as a fallback.
+		target := r.URL.Query().Get("target")
+		if target == "" {
+			target = r.URL.Query().Get("mailbox")
+			if target == "" {
+				http.Error(w, "Mailbox parameter is missing", http.StatusBadRequest)
+				return
+			}
 		}
 
+		mailbox := target
+
 		probeCountGauge := prometheus.NewGauge(prometheus.GaugeOpts{
 			Name: "probe_mailbox_count",
 			Help: "Displays the count of mails found in the mailbox",