improve: 飞书事件鉴权与日志打印

ncuhome · Nov 2, 2023 · c048e27 · c048e27
1 parent abf22bb
commit c048e27
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/internal/api/controllers/feishu/webhook.go b/internal/api/controllers/feishu/webhook.go
@@ -3,7 +3,9 @@ package controllers
 import (
 	"github.com/gin-gonic/gin"
 	"github.com/ncuhome/GeniusAuthoritarian/internal/api/callback"
+	"github.com/ncuhome/GeniusAuthoritarian/internal/global"
 	"github.com/ncuhome/GeniusAuthoritarian/internal/pkg/feishu"
+	log "github.com/sirupsen/logrus"
 )
 
 func Webhook(c *gin.Context) {
@@ -14,4 +16,11 @@ func Webhook(c *gin.Context) {
 		return
 	}
 
+	if event.Header.Token != global.Config.Feishu.WebhookVerificationToken {
+		callback.Error(c, callback.ErrUnauthorized, err)
+		return
+	}
+
+	log.Infof("[飞书事件] %s:%s", event.Header.EventType, event.Header.EventID)
+
 }
diff --git a/internal/global/models/config.go b/internal/global/models/config.go
@@ -18,8 +18,9 @@ type Config struct {
 }
 
 type Feishu struct {
-	ClientID string `yaml:"clientID"`
-	Secret   string `yaml:"secret"`
+	ClientID                 string `yaml:"clientID"`
+	Secret                   string `yaml:"secret"`
+	WebhookVerificationToken string `json:"webhook_verification_token"`
 }
 
 type DingTalk struct {