130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
package api
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"next-terminal/server/global/security"
|
|
"next-terminal/server/model"
|
|
"next-terminal/server/utils"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
func SecurityCreateEndpoint(c echo.Context) error {
|
|
var item model.AccessSecurity
|
|
if err := c.Bind(&item); err != nil {
|
|
return err
|
|
}
|
|
|
|
item.ID = utils.UUID()
|
|
item.Source = "管理员添加"
|
|
|
|
if err := accessSecurityRepository.Create(&item); err != nil {
|
|
return err
|
|
}
|
|
// 更新内存中的安全规则
|
|
rule := &security.Security{
|
|
ID: item.ID,
|
|
IP: item.IP,
|
|
Rule: item.Rule,
|
|
Priority: item.Priority,
|
|
}
|
|
security.GlobalSecurityManager.Add <- rule
|
|
|
|
return Success(c, "")
|
|
}
|
|
|
|
func ReloadAccessSecurity() error {
|
|
rules, err := accessSecurityRepository.FindAll()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(rules) > 0 {
|
|
// 先清空
|
|
security.GlobalSecurityManager.Clear()
|
|
// 再添加到全局的安全管理器中
|
|
for i := 0; i < len(rules); i++ {
|
|
rule := &security.Security{
|
|
ID: rules[i].ID,
|
|
IP: rules[i].IP,
|
|
Rule: rules[i].Rule,
|
|
Priority: rules[i].Priority,
|
|
}
|
|
security.GlobalSecurityManager.Add <- rule
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func SecurityPagingEndpoint(c echo.Context) error {
|
|
pageIndex, _ := strconv.Atoi(c.QueryParam("pageIndex"))
|
|
pageSize, _ := strconv.Atoi(c.QueryParam("pageSize"))
|
|
ip := c.QueryParam("ip")
|
|
rule := c.QueryParam("rule")
|
|
|
|
order := c.QueryParam("order")
|
|
field := c.QueryParam("field")
|
|
|
|
items, total, err := accessSecurityRepository.Find(pageIndex, pageSize, ip, rule, order, field)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return Success(c, H{
|
|
"total": total,
|
|
"items": items,
|
|
})
|
|
}
|
|
|
|
func SecurityUpdateEndpoint(c echo.Context) error {
|
|
id := c.Param("id")
|
|
|
|
var item model.AccessSecurity
|
|
if err := c.Bind(&item); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := accessSecurityRepository.UpdateById(&item, id); err != nil {
|
|
return err
|
|
}
|
|
// 更新内存中的安全规则
|
|
security.GlobalSecurityManager.Del <- id
|
|
rule := &security.Security{
|
|
ID: item.ID,
|
|
IP: item.IP,
|
|
Rule: item.Rule,
|
|
Priority: item.Priority,
|
|
}
|
|
security.GlobalSecurityManager.Add <- rule
|
|
|
|
return Success(c, nil)
|
|
}
|
|
|
|
func SecurityDeleteEndpoint(c echo.Context) error {
|
|
ids := c.Param("id")
|
|
|
|
split := strings.Split(ids, ",")
|
|
for i := range split {
|
|
id := split[i]
|
|
if err := accessSecurityRepository.DeleteById(id); err != nil {
|
|
return err
|
|
}
|
|
// 更新内存中的安全规则
|
|
security.GlobalSecurityManager.Del <- id
|
|
}
|
|
|
|
return Success(c, nil)
|
|
}
|
|
|
|
func SecurityGetEndpoint(c echo.Context) error {
|
|
id := c.Param("id")
|
|
|
|
item, err := accessSecurityRepository.FindById(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return Success(c, item)
|
|
}
|