webhooks.go•5.53 kB
package inspection
import (
"fmt"
"time"
"github.com/duke-git/lancet/v2/slice"
"github.com/gin-gonic/gin"
"github.com/weibaohui/k8m/internal/dao"
"github.com/weibaohui/k8m/pkg/comm/utils"
"github.com/weibaohui/k8m/pkg/comm/utils/amis"
"github.com/weibaohui/k8m/pkg/models"
"github.com/weibaohui/k8m/pkg/webhook"
"gorm.io/gorm"
)
type Controller struct {
}
func RegisterAdminWebhookRoutes(admin *gin.RouterGroup) {
ctrl := &Controller{}
admin.GET("/inspection/webhook/list", ctrl.WebhookList)
admin.POST("/inspection/webhook/delete/:ids", ctrl.WebhookDelete)
admin.POST("/inspection/webhook/save", ctrl.WebhookSave)
admin.POST("/inspection/webhook/id/:id/test", ctrl.WebhookTest)
admin.GET("/inspection/webhook/option_list", ctrl.WebhookOptionList)
// Webhook记录相关接口
admin.GET("/inspection/webhook/records", ctrl.WebhookRecordList)
admin.GET("/inspection/webhook/records/:id", ctrl.WebhookRecordDetail)
admin.GET("/inspection/webhook/records/statistics", ctrl.WebhookRecordStatistics)
}
// @Summary 获取Webhook接收器选项列表
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/inspection/webhook/option_list [get]
func (s *Controller) WebhookOptionList(c *gin.Context) {
m := models.WebhookReceiver{}
params := dao.BuildParams(c)
params.PerPage = 100000
list, _, err := m.List(params)
if err != nil {
amis.WriteJsonData(c, gin.H{
"options": make([]map[string]string, 0),
})
return
}
var hooks []map[string]string
for _, n := range list {
hooks = append(hooks, map[string]string{
"label": n.Name,
"value": fmt.Sprintf("%d", n.ID),
})
}
slice.SortBy(hooks, func(a, b map[string]string) bool {
return a["label"] < b["label"]
})
amis.WriteJsonData(c, gin.H{
"options": hooks,
})
}
func (s *Controller) WebhookTest(c *gin.Context) {
id := c.Param("id")
params := dao.BuildParams(c)
m := &models.WebhookReceiver{
ID: utils.ToUInt(id),
}
m, err := m.GetOne(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
ret := webhook.PushMsgToSingleTarget("test", "", m)
if ret != nil {
amis.WriteJsonOKMsg(c, ret.RespBody)
return
}
amis.WriteJsonError(c, fmt.Errorf("unsupported platform: %s", m.Platform))
}
// @Summary 获取Webhook接收器列表
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/inspection/webhook/list [get]
func (s *Controller) WebhookList(c *gin.Context) {
params := dao.BuildParams(c)
m := &models.WebhookReceiver{}
items, total, err := m.List(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonListWithTotal(c, total, items)
}
// @Summary 创建或更新Webhook接收器
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/inspection/webhook/save [post]
func (s *Controller) WebhookSave(c *gin.Context) {
params := dao.BuildParams(c)
m := models.WebhookReceiver{}
err := c.ShouldBindJSON(&m)
if err != nil {
amis.WriteJsonError(c, err)
return
}
err = m.Save(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonOK(c)
}
// @Summary 删除Webhook接收器
// @Security BearerAuth
// @Param ids path string true "Webhook接收器ID，多个用逗号分隔"
// @Success 200 {object} string
// @Router /admin/inspection/webhook/delete/{ids} [post]
func (s *Controller) WebhookDelete(c *gin.Context) {
ids := c.Param("ids")
params := dao.BuildParams(c)
m := &models.WebhookReceiver{}
err := m.Delete(params, ids)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonOK(c)
}
// @Summary 获取Webhook记录列表
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/inspection/webhook/records [get]
func (s *Controller) WebhookRecordList(c *gin.Context) {
params := dao.BuildParams(c)
m := &models.WebhookLogRecord{}
items, total, err := m.List(params, func(db *gorm.DB) *gorm.DB {
return db.Order("created_at DESC")
})
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonListWithTotal(c, total, items)
}
// @Summary 获取Webhook记录详情
// @Security BearerAuth
// @Param id path string true "记录ID"
// @Success 200 {object} string
// @Router /admin/inspection/webhook/records/{id} [get]
func (s *Controller) WebhookRecordDetail(c *gin.Context) {
id := c.Param("id")
params := dao.BuildParams(c)
m := &models.WebhookLogRecord{
ID: utils.ToUInt(id),
}
record, err := m.GetOne(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonData(c, record)
}
// @Summary 获取Webhook记录统计信息
// @Security BearerAuth
// @Success 200 {object} string
// @Router /admin/inspection/webhook/records/statistics [get]
func (s *Controller) WebhookRecordStatistics(c *gin.Context) {
webhookID := utils.ToUInt(c.Query("webhook_id"))
startTimeStr := c.Query("start_time")
endTimeStr := c.Query("end_time")
var startTime, endTime time.Time
var err error
if startTimeStr != "" {
startTime, err = time.Parse("2006-01-02 15:04:05", startTimeStr)
if err != nil {
amis.WriteJsonError(c, fmt.Errorf("invalid start_time format: %v", err))
return
}
}
if endTimeStr != "" {
endTime, err = time.Parse("2006-01-02 15:04:05", endTimeStr)
if err != nil {
amis.WriteJsonError(c, fmt.Errorf("invalid end_time format: %v", err))
return
}
}
m := &models.WebhookLogRecord{}
statistics, err := m.GetStatistics(webhookID, startTime, endTime)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonData(c, statistics)
}