POST endpoint to subscribe to messages

bruno/POST Privmsg Subscription.bru

@@ -0,0 +1,19 @@
+meta {
+  name: POST Privmsg Subscription
+  type: http
+  seq: 3
+}
+
+post {
+  url: http://localhost:8080/subscription
+  body: json
+  auth: none
+}
+
+body:json {
+  {
+    "Target": "Test",
+    "URL": "http://localhost/404",
+    "Password": "{{WMB_PASSWORD}}"
+  }
+}

main.go

@@ -27,6 +27,7 @@ func main() {
 		router.POST("/message", requesthandlers.PostMessage)
 		router.GET("/message", requesthandlers.QueryMessage)
 		router.POST("/directedMessage", requesthandlers.PostDirectedMessage)
+		router.POST("/subscription", requesthandlers.PostSubscribePrivmsg)
 
 		err := router.Run(listenAddress)
 

model/model.go

@@ -12,6 +12,12 @@ type DirectedIncomingMessage struct {
 }
 
 type DirectedOutgoingMessage struct {
-	Target  string `json:"target"`
-	Message string `json:"message"`
+	Target  string
+	Message string
+}
+
+type PrivmsgSubscription struct {
+	Target   string
+	URL      string
+	Password string
 }

requesthandlers/http.go

@@ -7,6 +7,7 @@ import (
 	"github.com/cfindlayisme/wmb/env"
 	"github.com/cfindlayisme/wmb/ircclient"
 	"github.com/cfindlayisme/wmb/model"
+	"github.com/cfindlayisme/wmb/webhook"
 	"github.com/gin-gonic/gin"
 )
 
@@ -15,13 +16,20 @@ func validateMessage(msg model.IncomingMessage, c *gin.Context) bool {
 		c.String(http.StatusBadRequest, "Message cannot contain newline characters")
 		return false
 	}
-	if env.GetWebhookPassword() != msg.Password {
-		c.String(http.StatusUnauthorized, "Invalid password")
-		return false
-	}
+	validatePassword(msg.Password, c)
+
 	return true
 }
 
+func validatePassword(password string, c *gin.Context) bool {
+	if env.GetWebhookPassword() == password {
+		return true
+	}
+	c.String(http.StatusUnauthorized, "Invalid password")
+
+	return false
+}
+
 func PostMessage(c *gin.Context) {
 	var msg model.IncomingMessage
 
@@ -71,7 +79,6 @@ func QueryMessage(c *gin.Context) {
 	var msg model.IncomingMessage
 
 	if err := c.ShouldBindQuery(&msg); err != nil {
-
 		c.String(http.StatusBadRequest, "Invalid query parameters")
 		return
 	}
@@ -88,3 +95,30 @@ func QueryMessage(c *gin.Context) {
 	c.String(http.StatusOK, "Message sent")
 
 }
+
+func PostSubscribePrivmsg(c *gin.Context) {
+	var subscription model.PrivmsgSubscription
+
+	if err := c.BindJSON(&subscription); err != nil {
+		c.String(http.StatusBadRequest, "Invalid query parameters")
+		return
+	}
+
+	if !validatePassword(subscription.Password, c) {
+		return
+	}
+
+	success := webhook.SubscribePrivmsg(subscription.Target, subscription.URL)
+
+	if success {
+		c.JSON(http.StatusOK, gin.H{
+			"status":  "success",
+			"message": "Subscription successful",
+		})
+	} else {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"status":  "failure",
+			"message": "Subscription failed",
+		})
+	}
+}