Compare commits

..

1 Commits

Author SHA1 Message Date
UUBulb
163231bbb1
Merge 2c8ab28efe into a9c2abe71e 2024-12-16 23:46:04 +08:00
9 changed files with 26 additions and 194 deletions

View File

@ -50,9 +50,6 @@ func createAlertRule(c *gin.Context) (uint64, error) {
return 0, err return 0, err
} }
uid := getUid(c)
r.UserID = uid
r.Name = arf.Name r.Name = arf.Name
r.Rules = arf.Rules r.Rules = arf.Rules
r.FailTriggerTasks = arf.FailTriggerTasks r.FailTriggerTasks = arf.FailTriggerTasks
@ -103,10 +100,6 @@ func updateAlertRule(c *gin.Context) (any, error) {
return nil, singleton.Localizer.ErrorT("alert id %d does not exist", id) return nil, singleton.Localizer.ErrorT("alert id %d does not exist", id)
} }
if !r.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
}
r.Name = arf.Name r.Name = arf.Name
r.Rules = arf.Rules r.Rules = arf.Rules
r.FailTriggerTasks = arf.FailTriggerTasks r.FailTriggerTasks = arf.FailTriggerTasks
@ -140,22 +133,10 @@ func updateAlertRule(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/alert-rule [post] // @Router /batch-delete/alert-rule [post]
func batchDeleteAlertRule(c *gin.Context) (any, error) { func batchDeleteAlertRule(c *gin.Context) (any, error) {
var arr []uint64
if err := c.ShouldBindJSON(&arr); err != nil {
return nil, err
}
var ars []model.AlertRule
if err := singleton.DB.Where("id in (?)", arr).Find(&ars).Error; err != nil {
return nil, err
}
var ar []uint64 var ar []uint64
for _, a := range ars {
if !a.HasPermission(c) { if err := c.ShouldBindJSON(&ar); err != nil {
return nil, singleton.Localizer.ErrorT("permission denied") return nil, err
}
ar = append(ar, a.ID)
} }
if err := singleton.DB.Unscoped().Delete(&model.AlertRule{}, "id in (?)", ar).Error; err != nil { if err := singleton.DB.Unscoped().Delete(&model.AlertRule{}, "id in (?)", ar).Error; err != nil {

View File

@ -1,6 +1,7 @@
package controller package controller
import ( import (
"fmt"
"strconv" "strconv"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
@ -49,7 +50,6 @@ func createCron(c *gin.Context) (uint64, error) {
return 0, err return 0, err
} }
cr.UserID = getUid(c)
cr.TaskType = cf.TaskType cr.TaskType = cf.TaskType
cr.Name = cf.Name cr.Name = cf.Name
cr.Scheduler = cf.Scheduler cr.Scheduler = cf.Scheduler
@ -106,11 +106,7 @@ func updateCron(c *gin.Context) (any, error) {
var cr model.Cron var cr model.Cron
if err := singleton.DB.First(&cr, id).Error; err != nil { if err := singleton.DB.First(&cr, id).Error; err != nil {
return nil, singleton.Localizer.ErrorT("task id %d does not exist", id) return nil, fmt.Errorf("task id %d does not exist", id)
}
if !cr.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
} }
cr.TaskType = cf.TaskType cr.TaskType = cf.TaskType
@ -160,15 +156,12 @@ func manualTriggerCron(c *gin.Context) (any, error) {
return nil, err return nil, err
} }
singleton.CronLock.RLock() var cr model.Cron
cr, ok := singleton.Crons[id] if err := singleton.DB.First(&cr, id).Error; err != nil {
if !ok {
singleton.CronLock.RUnlock()
return nil, singleton.Localizer.ErrorT("task id %d does not exist", id) return nil, singleton.Localizer.ErrorT("task id %d does not exist", id)
} }
singleton.CronLock.RUnlock()
singleton.ManualTrigger(cr) singleton.ManualTrigger(&cr)
return nil, nil return nil, nil
} }
@ -184,24 +177,12 @@ func manualTriggerCron(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/cron [post] // @Router /batch-delete/cron [post]
func batchDeleteCron(c *gin.Context) (any, error) { func batchDeleteCron(c *gin.Context) (any, error) {
var crr []uint64 var cr []uint64
if err := c.ShouldBindJSON(&crr); err != nil {
if err := c.ShouldBindJSON(&cr); err != nil {
return nil, err return nil, err
} }
var cr []uint64
singleton.CronLock.RLock()
for _, crID := range crr {
if crn, ok := singleton.Crons[crID]; ok {
if !crn.HasPermission(c) {
singleton.CronLock.RUnlock()
return nil, singleton.Localizer.ErrorT("permission denied")
}
cr = append(cr, crn.ID)
}
}
singleton.CronLock.RUnlock()
if err := singleton.DB.Unscoped().Delete(&model.Cron{}, "id in (?)", cr).Error; err != nil { if err := singleton.DB.Unscoped().Delete(&model.Cron{}, "id in (?)", cr).Error; err != nil {
return nil, newGormError("%v", err) return nil, newGormError("%v", err)
} }

View File

@ -56,7 +56,6 @@ func createDDNS(c *gin.Context) (uint64, error) {
return 0, singleton.Localizer.ErrorT("the retry count must be an integer between 1 and 10") return 0, singleton.Localizer.ErrorT("the retry count must be an integer between 1 and 10")
} }
p.UserID = getUid(c)
p.Name = df.Name p.Name = df.Name
enableIPv4 := df.EnableIPv4 enableIPv4 := df.EnableIPv4
enableIPv6 := df.EnableIPv6 enableIPv6 := df.EnableIPv6
@ -126,10 +125,6 @@ func updateDDNS(c *gin.Context) (any, error) {
return nil, singleton.Localizer.ErrorT("profile id %d does not exist", id) return nil, singleton.Localizer.ErrorT("profile id %d does not exist", id)
} }
if !p.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
}
p.Name = df.Name p.Name = df.Name
enableIPv4 := df.EnableIPv4 enableIPv4 := df.EnableIPv4
enableIPv6 := df.EnableIPv6 enableIPv6 := df.EnableIPv6
@ -177,25 +172,12 @@ func updateDDNS(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/ddns [post] // @Router /batch-delete/ddns [post]
func batchDeleteDDNS(c *gin.Context) (any, error) { func batchDeleteDDNS(c *gin.Context) (any, error) {
var ddnsConfigsr []uint64 var ddnsConfigs []uint64
if err := c.ShouldBindJSON(&ddnsConfigsr); err != nil { if err := c.ShouldBindJSON(&ddnsConfigs); err != nil {
return nil, err return nil, err
} }
var ddnsConfigs []uint64
singleton.DDNSCacheLock.RLock()
for _, pid := range ddnsConfigsr {
if p, ok := singleton.DDNSCache[pid]; ok {
if !p.HasPermission(c) {
singleton.DDNSCacheLock.RUnlock()
return nil, singleton.Localizer.ErrorT("permission denied")
}
ddnsConfigs = append(ddnsConfigs, p.ID)
}
}
singleton.DDNSCacheLock.RUnlock()
if err := singleton.DB.Unscoped().Delete(&model.DDNSProfile{}, "id in (?)", ddnsConfigs).Error; err != nil { if err := singleton.DB.Unscoped().Delete(&model.DDNSProfile{}, "id in (?)", ddnsConfigs).Error; err != nil {
return nil, newGormError("%v", err) return nil, newGormError("%v", err)
} }

View File

@ -51,9 +51,6 @@ func createNAT(c *gin.Context) (uint64, error) {
return 0, err return 0, err
} }
uid := getUid(c)
n.UserID = uid
n.Name = nf.Name n.Name = nf.Name
n.Domain = nf.Domain n.Domain = nf.Domain
n.Host = nf.Host n.Host = nf.Host
@ -98,10 +95,6 @@ func updateNAT(c *gin.Context) (any, error) {
return nil, singleton.Localizer.ErrorT("profile id %d does not exist", id) return nil, singleton.Localizer.ErrorT("profile id %d does not exist", id)
} }
if !n.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
}
n.Name = nf.Name n.Name = nf.Name
n.Domain = nf.Domain n.Domain = nf.Domain
n.Host = nf.Host n.Host = nf.Host
@ -128,24 +121,12 @@ func updateNAT(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/nat [post] // @Router /batch-delete/nat [post]
func batchDeleteNAT(c *gin.Context) (any, error) { func batchDeleteNAT(c *gin.Context) (any, error) {
var nr []uint64 var n []uint64
if err := c.ShouldBindJSON(&nr); err != nil {
if err := c.ShouldBindJSON(&n); err != nil {
return nil, err return nil, err
} }
var n []uint64
singleton.NATCacheRwLock.RLock()
for _, id := range nr {
if p, ok := singleton.NATCache[singleton.NATIDToDomain[id]]; ok {
if !p.HasPermission(c) {
singleton.NATCacheRwLock.RUnlock()
return nil, singleton.Localizer.ErrorT("permission denied")
}
n = append(n, p.ID)
}
}
singleton.NATCacheRwLock.RUnlock()
if err := singleton.DB.Unscoped().Delete(&model.NAT{}, "id in (?)", n).Error; err != nil { if err := singleton.DB.Unscoped().Delete(&model.NAT{}, "id in (?)", n).Error; err != nil {
return nil, newGormError("%v", err) return nil, newGormError("%v", err)
} }

View File

@ -48,7 +48,6 @@ func createNotification(c *gin.Context) (uint64, error) {
} }
var n model.Notification var n model.Notification
n.UserID = getUid(c)
n.Name = nf.Name n.Name = nf.Name
n.RequestMethod = nf.RequestMethod n.RequestMethod = nf.RequestMethod
n.RequestType = nf.RequestType n.RequestType = nf.RequestType
@ -107,10 +106,6 @@ func updateNotification(c *gin.Context) (any, error) {
return nil, singleton.Localizer.ErrorT("notification id %d does not exist", id) return nil, singleton.Localizer.ErrorT("notification id %d does not exist", id)
} }
if !n.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
}
n.Name = nf.Name n.Name = nf.Name
n.RequestMethod = nf.RequestMethod n.RequestMethod = nf.RequestMethod
n.RequestType = nf.RequestType n.RequestType = nf.RequestType
@ -153,23 +148,12 @@ func updateNotification(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/notification [post] // @Router /batch-delete/notification [post]
func batchDeleteNotification(c *gin.Context) (any, error) { func batchDeleteNotification(c *gin.Context) (any, error) {
var nr []uint64 var n []uint64
if err := c.ShouldBindJSON(&nr); err != nil {
if err := c.ShouldBindJSON(&n); err != nil {
return nil, err return nil, err
} }
var n []uint64
singleton.NotificationsLock.RLock()
for _, nid := range nr {
if ns, ok := singleton.NotificationMap[nid]; ok {
if !ns.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
}
n = append(n, ns.ID)
}
}
singleton.NotificationsLock.RUnlock()
err := singleton.DB.Transaction(func(tx *gorm.DB) error { err := singleton.DB.Transaction(func(tx *gorm.DB) error {
if err := tx.Unscoped().Delete(&model.Notification{}, "id in (?)", n).Error; err != nil { if err := tx.Unscoped().Delete(&model.Notification{}, "id in (?)", n).Error; err != nil {
return err return err

View File

@ -68,11 +68,8 @@ func createNotificationGroup(c *gin.Context) (uint64, error) {
} }
ngf.Notifications = slices.Compact(ngf.Notifications) ngf.Notifications = slices.Compact(ngf.Notifications)
uid := getUid(c)
var ng model.NotificationGroup var ng model.NotificationGroup
ng.Name = ngf.Name ng.Name = ngf.Name
ng.UserID = uid
var count int64 var count int64
if err := singleton.DB.Model(&model.Notification{}).Where("id in (?)", ngf.Notifications).Count(&count).Error; err != nil { if err := singleton.DB.Model(&model.Notification{}).Where("id in (?)", ngf.Notifications).Count(&count).Error; err != nil {
@ -89,9 +86,6 @@ func createNotificationGroup(c *gin.Context) (uint64, error) {
} }
for _, n := range ngf.Notifications { for _, n := range ngf.Notifications {
if err := tx.Create(&model.NotificationGroupNotification{ if err := tx.Create(&model.NotificationGroupNotification{
Common: model.Common{
UserID: uid,
},
NotificationGroupID: ng.ID, NotificationGroupID: ng.ID,
NotificationID: n, NotificationID: n,
}).Error; err != nil { }).Error; err != nil {
@ -137,10 +131,6 @@ func updateNotificationGroup(c *gin.Context) (any, error) {
return nil, singleton.Localizer.ErrorT("group id %d does not exist", id) return nil, singleton.Localizer.ErrorT("group id %d does not exist", id)
} }
if !ngDB.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
}
ngDB.Name = ngf.Name ngDB.Name = ngf.Name
ngf.Notifications = slices.Compact(ngf.Notifications) ngf.Notifications = slices.Compact(ngf.Notifications)
@ -152,8 +142,6 @@ func updateNotificationGroup(c *gin.Context) (any, error) {
return nil, singleton.Localizer.ErrorT("have invalid notification id") return nil, singleton.Localizer.ErrorT("have invalid notification id")
} }
uid := getUid(c)
err = singleton.DB.Transaction(func(tx *gorm.DB) error { err = singleton.DB.Transaction(func(tx *gorm.DB) error {
if err := tx.Save(&ngDB).Error; err != nil { if err := tx.Save(&ngDB).Error; err != nil {
return err return err
@ -164,9 +152,6 @@ func updateNotificationGroup(c *gin.Context) (any, error) {
for _, n := range ngf.Notifications { for _, n := range ngf.Notifications {
if err := tx.Create(&model.NotificationGroupNotification{ if err := tx.Create(&model.NotificationGroupNotification{
Common: model.Common{
UserID: uid,
},
NotificationGroupID: ngDB.ID, NotificationGroupID: ngDB.ID,
NotificationID: n, NotificationID: n,
}).Error; err != nil { }).Error; err != nil {
@ -195,22 +180,9 @@ func updateNotificationGroup(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/notification-group [post] // @Router /batch-delete/notification-group [post]
func batchDeleteNotificationGroup(c *gin.Context) (any, error) { func batchDeleteNotificationGroup(c *gin.Context) (any, error) {
var ngnr []uint64
if err := c.ShouldBindJSON(&ngnr); err != nil {
return nil, err
}
var ng []model.NotificationGroup
if err := singleton.DB.Where("id in (?)", ng).Find(&ng).Error; err != nil {
return nil, err
}
var ngn []uint64 var ngn []uint64
for _, n := range ng { if err := c.ShouldBindJSON(&ngn); err != nil {
if !n.HasPermission(c) { return nil, err
return nil, singleton.Localizer.ErrorT("permission denied")
}
ngn = append(ngn, n.ID)
} }
err := singleton.DB.Transaction(func(tx *gorm.DB) error { err := singleton.DB.Transaction(func(tx *gorm.DB) error {

View File

@ -62,7 +62,7 @@ func updateServer(c *gin.Context) (any, error) {
} }
if !s.HasPermission(c) { if !s.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied") return nil, singleton.Localizer.ErrorT("unauthorized")
} }
s.Name = sf.Name s.Name = sf.Name
@ -113,7 +113,6 @@ func batchDeleteServer(c *gin.Context) (any, error) {
for _, sid := range serversRaw { for _, sid := range serversRaw {
if s, ok := singleton.ServerList[sid]; ok { if s, ok := singleton.ServerList[sid]; ok {
if !s.HasPermission(c) { if !s.HasPermission(c) {
singleton.ServerLock.RUnlock()
return nil, singleton.Localizer.ErrorT("permission denied") return nil, singleton.Localizer.ErrorT("permission denied")
} }
servers = append(servers, s.ID) servers = append(servers, s.ID)

View File

@ -67,11 +67,8 @@ func createServerGroup(c *gin.Context) (uint64, error) {
} }
sgf.Servers = slices.Compact(sgf.Servers) sgf.Servers = slices.Compact(sgf.Servers)
uid := getUid(c)
var sg model.ServerGroup var sg model.ServerGroup
sg.Name = sgf.Name sg.Name = sgf.Name
sg.UserID = uid
var count int64 var count int64
if err := singleton.DB.Model(&model.Server{}).Where("id in (?)", sgf.Servers).Count(&count).Error; err != nil { if err := singleton.DB.Model(&model.Server{}).Where("id in (?)", sgf.Servers).Count(&count).Error; err != nil {
@ -87,9 +84,6 @@ func createServerGroup(c *gin.Context) (uint64, error) {
} }
for _, s := range sgf.Servers { for _, s := range sgf.Servers {
if err := tx.Create(&model.ServerGroupServer{ if err := tx.Create(&model.ServerGroupServer{
Common: model.Common{
UserID: uid,
},
ServerGroupId: sg.ID, ServerGroupId: sg.ID,
ServerId: s, ServerId: s,
}).Error; err != nil { }).Error; err != nil {
@ -135,11 +129,6 @@ func updateServerGroup(c *gin.Context) (any, error) {
if err := singleton.DB.First(&sgDB, id).Error; err != nil { if err := singleton.DB.First(&sgDB, id).Error; err != nil {
return nil, singleton.Localizer.ErrorT("group id %d does not exist", id) return nil, singleton.Localizer.ErrorT("group id %d does not exist", id)
} }
if !sgDB.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("unauthorized")
}
sgDB.Name = sg.Name sgDB.Name = sg.Name
var count int64 var count int64
@ -150,8 +139,6 @@ func updateServerGroup(c *gin.Context) (any, error) {
return nil, singleton.Localizer.ErrorT("have invalid server id") return nil, singleton.Localizer.ErrorT("have invalid server id")
} }
uid := getUid(c)
err = singleton.DB.Transaction(func(tx *gorm.DB) error { err = singleton.DB.Transaction(func(tx *gorm.DB) error {
if err := tx.Save(&sgDB).Error; err != nil { if err := tx.Save(&sgDB).Error; err != nil {
return err return err
@ -162,9 +149,6 @@ func updateServerGroup(c *gin.Context) (any, error) {
for _, s := range sg.Servers { for _, s := range sg.Servers {
if err := tx.Create(&model.ServerGroupServer{ if err := tx.Create(&model.ServerGroupServer{
Common: model.Common{
UserID: uid,
},
ServerGroupId: sgDB.ID, ServerGroupId: sgDB.ID,
ServerId: s, ServerId: s,
}).Error; err != nil { }).Error; err != nil {
@ -192,22 +176,9 @@ func updateServerGroup(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/server-group [post] // @Router /batch-delete/server-group [post]
func batchDeleteServerGroup(c *gin.Context) (any, error) { func batchDeleteServerGroup(c *gin.Context) (any, error) {
var sgsr []uint64
if err := c.ShouldBindJSON(&sgsr); err != nil {
return nil, err
}
var sg []model.ServerGroup
if err := singleton.DB.Where("id in (?)", sgsr).Find(&sg).Error; err != nil {
return nil, err
}
var sgs []uint64 var sgs []uint64
for _, s := range sg { if err := c.ShouldBindJSON(&sgs); err != nil {
if !s.HasPermission(c) { return nil, err
return nil, singleton.Localizer.ErrorT("permission denied")
}
sgs = append(sgs, s.ID)
} }
err := singleton.DB.Transaction(func(tx *gorm.DB) error { err := singleton.DB.Transaction(func(tx *gorm.DB) error {

View File

@ -263,11 +263,6 @@ func updateService(c *gin.Context) (any, error) {
if err := singleton.DB.First(&m, id).Error; err != nil { if err := singleton.DB.First(&m, id).Error; err != nil {
return nil, singleton.Localizer.ErrorT("service id %d does not exist", id) return nil, singleton.Localizer.ErrorT("service id %d does not exist", id)
} }
if !m.HasPermission(c) {
return nil, singleton.Localizer.ErrorT("permission denied")
}
m.Name = mf.Name m.Name = mf.Name
m.Target = strings.TrimSpace(mf.Target) m.Target = strings.TrimSpace(mf.Target)
m.Type = mf.Type m.Type = mf.Type
@ -322,24 +317,10 @@ func updateService(c *gin.Context) (any, error) {
// @Success 200 {object} model.CommonResponse[any] // @Success 200 {object} model.CommonResponse[any]
// @Router /batch-delete/service [post] // @Router /batch-delete/service [post]
func batchDeleteService(c *gin.Context) (any, error) { func batchDeleteService(c *gin.Context) (any, error) {
var idsr []uint64 var ids []uint64
if err := c.ShouldBindJSON(&idsr); err != nil { if err := c.ShouldBindJSON(&ids); err != nil {
return nil, err return nil, err
} }
var ids []uint64
singleton.ServiceSentinelShared.ServicesLock.RLock()
for _, id := range idsr {
if ss, ok := singleton.ServiceSentinelShared.Services[id]; ok {
if !ss.HasPermission(c) {
singleton.ServiceSentinelShared.ServicesLock.RUnlock()
return nil, singleton.Localizer.ErrorT("permission denied")
}
ids = append(ids, ss.ID)
}
}
singleton.ServiceSentinelShared.ServicesLock.RUnlock()
err := singleton.DB.Transaction(func(tx *gorm.DB) error { err := singleton.DB.Transaction(func(tx *gorm.DB) error {
if err := tx.Unscoped().Delete(&model.Service{}, "id in (?)", ids).Error; err != nil { if err := tx.Unscoped().Delete(&model.Service{}, "id in (?)", ids).Error; err != nil {
return err return err