mirror of
https://github.com/skyle1995/NetworkAuth.git
synced 2026-05-25 02:24:05 +08:00
288 lines
8.4 KiB
Go
288 lines
8.4 KiB
Go
|
package admin
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"NetworkAuth/controllers"
|
|||
|
|
"NetworkAuth/models"
|
|||
|
|
"NetworkAuth/services"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/sirupsen/logrus"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ============================================================================
|
|||
|
|
// 全局变量
|
|||
|
|
// ============================================================================
|
|||
|
|
|
|||
|
|
var logBaseController = controllers.NewBaseController()
|
|||
|
|
|
|||
|
|
// ============================================================================
|
|||
|
|
// 登录日志 API处理器
|
|||
|
|
// ============================================================================
|
|||
|
|
|
|||
|
|
// DashboardLoginLogsHandler 获取管理员最近登录日志
|
|||
|
|
func DashboardLoginLogsHandler(c *gin.Context) {
|
|||
|
|
db, ok := logBaseController.GetDB(c)
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取分页参数
|
|||
|
|
page, limit := logBaseController.GetPaginationParams(c)
|
|||
|
|
|
|||
|
|
// 获取当前管理员信息
|
|||
|
|
uuid := c.GetString("admin_uuid")
|
|||
|
|
username := c.GetString("admin_username")
|
|||
|
|
|
|||
|
|
var total int64
|
|||
|
|
query := db.Model(&models.LoginLog{})
|
|||
|
|
|
|||
|
|
// 如果有用户名，则仅过滤该用户的日志
|
|||
|
|
if uuid != "" {
|
|||
|
|
query = query.Where("uuid = ? OR (uuid = '' AND username = ?)", uuid, username)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
logs, total, err := services.Paginate[models.LoginLog](query, page, limit, "created_at desc")
|
|||
|
|
if err != nil {
|
|||
|
|
logBaseController.HandleInternalError(c, "获取登录日志失败", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
data := gin.H{
|
|||
|
|
"total": total,
|
|||
|
|
"list": logs,
|
|||
|
|
}
|
|||
|
|
logBaseController.HandleSuccess(c, "获取登录日志成功", data)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// LoginLogsListHandler 登录日志列表API处理器
|
|||
|
|
func LoginLogsListHandler(c *gin.Context) {
|
|||
|
|
// 获取分页参数
|
|||
|
|
page, limit := logBaseController.GetPaginationParams(c)
|
|||
|
|
|
|||
|
|
// 获取数据库连接
|
|||
|
|
db, ok := logBaseController.GetDB(c)
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
query := db.Model(&models.LoginLog{})
|
|||
|
|
|
|||
|
|
// 筛选条件：账号或UUID合并搜索
|
|||
|
|
if username := strings.TrimSpace(c.Query("username")); username != "" {
|
|||
|
|
query = query.Where("username = ? OR uuid = ?", username, username)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 筛选条件：IP
|
|||
|
|
if ip := strings.TrimSpace(c.Query("ip")); ip != "" {
|
|||
|
|
query = query.Where("ip = ?", ip)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 筛选条件：状态
|
|||
|
|
if statusStr := strings.TrimSpace(c.Query("status")); statusStr != "" {
|
|||
|
|
if status, err := strconv.Atoi(statusStr); err == nil {
|
|||
|
|
query = query.Where("status = ?", status)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 筛选条件：时间范围
|
|||
|
|
query = logBaseController.ApplyTimeRangeQuery(c, query, "created_at")
|
|||
|
|
|
|||
|
|
// 泛型分页查询
|
|||
|
|
logs, total, err := services.Paginate[models.LoginLog](query, page, limit, "created_at DESC")
|
|||
|
|
if err != nil {
|
|||
|
|
logBaseController.HandleInternalError(c, "获取日志列表失败", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 转换数据格式
|
|||
|
|
var list []map[string]interface{}
|
|||
|
|
for _, log := range logs {
|
|||
|
|
list = append(list, map[string]interface{}{
|
|||
|
|
"id": log.ID,
|
|||
|
|
"uuid": log.UUID,
|
|||
|
|
"username": log.Username,
|
|||
|
|
"ip": log.IP,
|
|||
|
|
"status": log.Status,
|
|||
|
|
"message": log.Message,
|
|||
|
|
"user_agent": log.UserAgent,
|
|||
|
|
"created_at": log.CreatedAt,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
logBaseController.HandleSuccess(c, "ok", gin.H{
|
|||
|
|
"list": list,
|
|||
|
|
"total": total,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// LoginLogsClearHandler 清空登录日志API处理器
|
|||
|
|
func LoginLogsClearHandler(c *gin.Context) {
|
|||
|
|
// 鉴权拦截：仅超级管理员 (role=0) 允许清空日志
|
|||
|
|
if role, exists := c.Get("admin_role"); !exists || role.(int) != 0 {
|
|||
|
|
logBaseController.HandleValidationError(c, "权限不足，仅超级管理员可清空日志")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
db, ok := logBaseController.GetDB(c)
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查数据库类型
|
|||
|
|
dbType := db.Dialector.Name()
|
|||
|
|
|
|||
|
|
if dbType == "sqlite" {
|
|||
|
|
// SQLite 不支持 TRUNCATE，直接使用 DELETE 和重置自增序列
|
|||
|
|
if err := db.Session(&gorm.Session{AllowGlobalUpdate: true}).Unscoped().Where("1 = 1").Delete(&models.LoginLog{}).Error; err != nil {
|
|||
|
|
logrus.WithError(err).Error("Failed to clear login logs")
|
|||
|
|
logBaseController.HandleInternalError(c, "清空登录日志失败", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 重置 sqlite 的自增序列
|
|||
|
|
db.Exec("UPDATE sqlite_sequence SET seq = 0 WHERE name = 'login_logs'")
|
|||
|
|
// 释放空间
|
|||
|
|
db.Exec("VACUUM")
|
|||
|
|
} else {
|
|||
|
|
// 其他数据库（如 MySQL/PostgreSQL）尝试使用 TRUNCATE
|
|||
|
|
if err := db.Exec("TRUNCATE TABLE login_logs").Error; err != nil {
|
|||
|
|
// 如果 TRUNCATE 失败，回退到 DELETE
|
|||
|
|
if err := db.Session(&gorm.Session{AllowGlobalUpdate: true}).Unscoped().Where("1 = 1").Delete(&models.LoginLog{}).Error; err != nil {
|
|||
|
|
logrus.WithError(err).Error("Failed to clear login logs")
|
|||
|
|
logBaseController.HandleInternalError(c, "清空登录日志失败", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 记录操作日志
|
|||
|
|
var operator, operatorUUID string
|
|||
|
|
operator = c.GetString("admin_username")
|
|||
|
|
operatorUUID = c.GetString("admin_uuid")
|
|||
|
|
if operator == "" {
|
|||
|
|
operator = "system"
|
|||
|
|
operatorUUID = "system"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log := models.OperationLog{
|
|||
|
|
OperationType: "清空登录日志",
|
|||
|
|
Operator: operator,
|
|||
|
|
OperatorUUID: operatorUUID,
|
|||
|
|
Details: "管理员清空了所有登录日志",
|
|||
|
|
CreatedAt: time.Now(),
|
|||
|
|
}
|
|||
|
|
db.Create(&log)
|
|||
|
|
|
|||
|
|
logBaseController.HandleSuccess(c, "登录日志已清空", nil)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ============================================================================
|
|||
|
|
// 操作日志 API处理器
|
|||
|
|
// ============================================================================
|
|||
|
|
|
|||
|
|
// LogsListHandler 日志列表API处理器
|
|||
|
|
func LogsListHandler(c *gin.Context) {
|
|||
|
|
// 获取分页参数
|
|||
|
|
page, limit := logBaseController.GetPaginationParams(c)
|
|||
|
|
|
|||
|
|
// 获取搜索参数
|
|||
|
|
operationType := strings.TrimSpace(c.Query("operation_type"))
|
|||
|
|
operator := strings.TrimSpace(c.Query("operator"))
|
|||
|
|
|
|||
|
|
// 获取数据库连接
|
|||
|
|
db, ok := logBaseController.GetDB(c)
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
query := db.Model(&models.OperationLog{})
|
|||
|
|
|
|||
|
|
// 筛选条件
|
|||
|
|
if operationType != "" {
|
|||
|
|
query = query.Where("operation_type = ?", operationType)
|
|||
|
|
}
|
|||
|
|
if operator != "" {
|
|||
|
|
// 支持按 UUID 或 用户名 筛选
|
|||
|
|
query = query.Where("operator_uuid = ? OR operator = ?", operator, operator)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 筛选条件：时间范围
|
|||
|
|
query = logBaseController.ApplyTimeRangeQuery(c, query, "created_at")
|
|||
|
|
|
|||
|
|
// 泛型分页查询
|
|||
|
|
logs, total, err := services.Paginate[models.OperationLog](query, page, limit, "created_at DESC")
|
|||
|
|
if err != nil {
|
|||
|
|
logrus.WithError(err).Error("查询日志列表失败")
|
|||
|
|
logBaseController.HandleInternalError(c, "查询日志列表失败", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
logBaseController.HandleSuccess(c, "获取日志列表成功", gin.H{
|
|||
|
|
"list": logs,
|
|||
|
|
"total": total,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// LogsClearHandler 清空日志API处理器
|
|||
|
|
func LogsClearHandler(c *gin.Context) {
|
|||
|
|
// 鉴权拦截：仅超级管理员 (role=0) 允许清空日志
|
|||
|
|
if role, exists := c.Get("admin_role"); !exists || role.(int) != 0 {
|
|||
|
|
logBaseController.HandleValidationError(c, "权限不足，仅超级管理员可清空日志")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
db, ok := logBaseController.GetDB(c)
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查数据库类型
|
|||
|
|
dbType := db.Dialector.Name()
|
|||
|
|
|
|||
|
|
if dbType == "sqlite" {
|
|||
|
|
// SQLite 不支持 TRUNCATE，直接使用 DELETE 和重置自增序列
|
|||
|
|
if err := db.Session(&gorm.Session{AllowGlobalUpdate: true}).Unscoped().Where("1 = 1").Delete(&models.OperationLog{}).Error; err != nil {
|
|||
|
|
logrus.WithError(err).Error("清空操作日志失败")
|
|||
|
|
logBaseController.HandleInternalError(c, "清空操作日志失败", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 重置 sqlite 的自增序列
|
|||
|
|
db.Exec("UPDATE sqlite_sequence SET seq = 0 WHERE name = 'operation_logs'")
|
|||
|
|
// 释放空间
|
|||
|
|
db.Exec("VACUUM")
|
|||
|
|
} else {
|
|||
|
|
// 其他数据库（如 MySQL/PostgreSQL）尝试使用 TRUNCATE
|
|||
|
|
if err := db.Exec("TRUNCATE TABLE operation_logs").Error; err != nil {
|
|||
|
|
// 如果 TRUNCATE 失败，回退到 DELETE
|
|||
|
|
if err := db.Session(&gorm.Session{AllowGlobalUpdate: true}).Unscoped().Where("1 = 1").Delete(&models.OperationLog{}).Error; err != nil {
|
|||
|
|
logrus.WithError(err).Error("清空操作日志失败")
|
|||
|
|
logBaseController.HandleInternalError(c, "清空操作日志失败", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 记录操作日志 (因为刚刚清空了，这条将是第一条)
|
|||
|
|
var operator, operatorUUID string
|
|||
|
|
operator = c.GetString("admin_username")
|
|||
|
|
operatorUUID = c.GetString("admin_uuid")
|
|||
|
|
if operator == "" {
|
|||
|
|
operator = "system"
|
|||
|
|
operatorUUID = "system"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log := models.OperationLog{
|
|||
|
|
OperationType: "清空日志",
|
|||
|
|
Operator: operator,
|
|||
|
|
OperatorUUID: operatorUUID,
|
|||
|
|
Details: "管理员清空了所有操作日志",
|
|||
|
|
CreatedAt: time.Now(),
|
|||
|
|
}
|
|||
|
|
db.Create(&log)
|
|||
|
|
|
|||
|
|
logBaseController.HandleSuccess(c, "日志已清空", nil)
|
|||
|
|
}
|