feat: webhook server part done (#76)

2023-05-10 11:58:55 +08:00
parent f7999d944e
commit 374400479f
8 changed files with 385 additions and 1 deletions
@@ -116,6 +116,10 @@ func pushMessageHelper(c *gin.Context, message *model.Message) {
 			return
 		}
 	}
+	processMessage(c, message, &user)
+}
+
+func processMessage(c *gin.Context, message *model.Message, user *model.User) {
 	if message.Title == "" {
 		message.Title = common.SystemName
 	}
@@ -133,7 +137,7 @@ func pushMessageHelper(c *gin.Context, message *model.Message) {
 		})
 		return
 	}
-	err = saveAndSendMessage(&user, message, channel_)
+	err = saveAndSendMessage(user, message, channel_)
 	if err != nil {
 		c.JSON(http.StatusOK, gin.H{
 			"success": false,
@@ -0,0 +1,256 @@
+package controller
+
+import (
+	"encoding/json"
+	"github.com/gin-gonic/gin"
+	"github.com/tidwall/gjson"
+	"message-pusher/common"
+	"message-pusher/model"
+	"net/http"
+	"strconv"
+	"strings"
+)
+
+func GetAllWebhooks(c *gin.Context) {
+	userId := c.GetInt("id")
+	p, _ := strconv.Atoi(c.Query("p"))
+	if p < 0 {
+		p = 0
+	}
+	webhooks, err := model.GetWebhooksByUserId(userId, p*common.ItemsPerPage, common.ItemsPerPage)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+		"data":    webhooks,
+	})
+	return
+}
+
+func SearchWebhooks(c *gin.Context) {
+	userId := c.GetInt("id")
+	keyword := c.Query("keyword")
+	webhooks, err := model.SearchWebhooks(userId, keyword)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+		"data":    webhooks,
+	})
+	return
+}
+
+func GetWebhook(c *gin.Context) {
+	id, err := strconv.Atoi(c.Param("id"))
+	userId := c.GetInt("id")
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	webhook_, err := model.GetWebhookById(id, userId)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+		"data":    webhook_,
+	})
+	return
+}
+
+func AddWebhook(c *gin.Context) {
+	webhook_ := model.Webhook{}
+	err := c.ShouldBindJSON(&webhook_)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	if len(webhook_.Name) == 0 || len(webhook_.Name) > 20 {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": "通道名称长度必须在1-20之间",
+		})
+		return
+	}
+	cleanWebhook := model.Webhook{
+		UserId:      c.GetInt("id"),
+		Name:        webhook_.Name,
+		Status:      common.WebhookStatusEnabled,
+		Link:        common.GetUUID(),
+		CreatedTime: common.GetTimestamp(),
+		ExtractRule: webhook_.ExtractRule,
+	}
+	err = cleanWebhook.Insert()
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+	})
+	return
+}
+
+func DeleteWebhook(c *gin.Context) {
+	id, _ := strconv.Atoi(c.Param("id"))
+	userId := c.GetInt("id")
+	_, err := model.DeleteWebhookById(id, userId)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+	})
+	return
+}
+
+func UpdateWebhook(c *gin.Context) {
+	userId := c.GetInt("id")
+	statusOnly := c.Query("status_only")
+	webhook_ := model.Webhook{}
+	err := c.ShouldBindJSON(&webhook_)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	oldWebhook, err := model.GetWebhookById(webhook_.Id, userId)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	cleanWebhook := *oldWebhook
+	if statusOnly != "" {
+		cleanWebhook.Status = webhook_.Status
+	} else {
+		// If you add more fields, please also update webhook_.Update()
+		cleanWebhook.Name = webhook_.Name
+		cleanWebhook.ExtractRule = webhook_.ExtractRule
+		cleanWebhook.ConstructRule = webhook_.ConstructRule
+		cleanWebhook.Channel = webhook_.Channel
+	}
+	err = cleanWebhook.Update()
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"message": "",
+		"data":    cleanWebhook,
+	})
+	return
+}
+
+func TriggerWebhook(c *gin.Context) {
+	var reqText string
+	err := c.Bind(&reqText)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		return
+	}
+	link := c.Param("link")
+	webhook, err := model.GetWebhookByLink(link)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{
+			"success": false,
+			"message": "Webhook 不存在",
+		})
+		return
+	}
+	if webhook.Status != common.WebhookStatusEnabled {
+		c.JSON(http.StatusForbidden, gin.H{
+			"success": false,
+			"message": "Webhook 未启用",
+		})
+		return
+	}
+	user, err := model.GetUserById(webhook.UserId, false)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{
+			"success": false,
+			"message": "用户不存在",
+		})
+		return
+	}
+	if user.Status != common.UserStatusEnabled {
+		c.JSON(http.StatusForbidden, gin.H{
+			"success": false,
+			"message": "用户已被封禁",
+		})
+		return
+	}
+	extractRule := make(map[string]string)
+	err = json.Unmarshal([]byte(webhook.ExtractRule), &extractRule)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"success": false,
+			"message": "Webhook 提取规则解析失败",
+		})
+		return
+	}
+	for key, value := range extractRule {
+		variableValue := gjson.Get(reqText, value).String()
+		webhook.ConstructRule = strings.Replace(webhook.ConstructRule, "$"+key, variableValue, -1)
+	}
+	constructRule := model.WebhookConstructRule{}
+	err = json.Unmarshal([]byte(webhook.ConstructRule), &constructRule)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": "Webhook 构建规则解析失败",
+		})
+		return
+	}
+	message := &model.Message{
+		Channel:     webhook.Channel,
+		Title:       constructRule.Title,
+		Description: constructRule.Description,
+		Content:     constructRule.Content,
+		URL:         constructRule.URL,
+	}
+	processMessage(c, message, user)
+}