package admin
import (
"fmt"
"time"
"github.com/duke-git/lancet/v2/slice"
"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/plugins/modules/webhook/core"
"github.com/weibaohui/k8m/pkg/plugins/modules/webhook/models"
"github.com/weibaohui/k8m/pkg/response"
"gorm.io/gorm"
)
type Controller struct{}
func (s *Controller) WebhookOptionList(c *response.Context) {
m := models.WebhookReceiver{}
params := dao.BuildParams(c)
params.PerPage = 100000
list, _, err := m.List(params)
if err != nil {
amis.WriteJsonData(c, response.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, response.H{
"options": hooks,
})
}
func (s *Controller) WebhookTest(c *response.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 := core.PushMsgToSingleTarget("test", "", m)
if ret != nil {
amis.WriteJsonOKMsg(c, ret.RespBody)
return
}
amis.WriteJsonError(c, fmt.Errorf("unsupported platform: %s", m.Platform))
}
func (s *Controller) WebhookList(c *response.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)
}
func (s *Controller) WebhookSave(c *response.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)
}
func (s *Controller) WebhookDelete(c *response.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)
}
func (s *Controller) WebhookRecordList(c *response.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)
}
func (s *Controller) WebhookRecordDetail(c *response.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)
}
func (s *Controller) WebhookRecordStatistics(c *response.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)
}
