130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package admin
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/cloudreve/Cloudreve/v4/application/dependency"
|
|
"github.com/cloudreve/Cloudreve/v4/inventory"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type (
|
|
LoginAuditListService struct {
|
|
Page int `json:"page"`
|
|
PageSize int `json:"page_size"`
|
|
UserID *int `json:"user_id,omitempty"`
|
|
IP string `json:"ip"`
|
|
Status string `json:"status"` // success / fail / 空 全部
|
|
Username string `json:"username"`
|
|
StartTime string `json:"start_time"` // RFC3339
|
|
EndTime string `json:"end_time"` // RFC3339
|
|
}
|
|
LoginAuditListParamCtx struct{}
|
|
|
|
LoginAuditDailyStatsService struct {
|
|
Start string `json:"start" form:"start"` // RFC3339 或 YYYY-MM-DD
|
|
End string `json:"end" form:"end"`
|
|
Days int `json:"days" form:"days"` // 优先用此参数，默认7天
|
|
}
|
|
LoginAuditDailyStatsParamCtx struct{}
|
|
)
|
|
|
|
func parseTimeRange(start, end string, defaultDays int) (time.Time, time.Time) {
|
|
now := time.Now().Truncate(24 * time.Hour)
|
|
endTime := now
|
|
startTime := endTime.AddDate(0, 0, -(defaultDays - 1))
|
|
|
|
if start != "" {
|
|
if t, err := time.ParseInLocation(time.RFC3339, start, time.Local); err == nil {
|
|
startTime = t.Truncate(24 * time.Hour)
|
|
} else if t, err := time.ParseInLocation("2006-01-02", start, time.Local); err == nil {
|
|
startTime = t
|
|
}
|
|
}
|
|
if end != "" {
|
|
if t, err := time.ParseInLocation(time.RFC3339, end, time.Local); err == nil {
|
|
endTime = t.Truncate(24 * time.Hour)
|
|
} else if t, err := time.ParseInLocation("2006-01-02", end, time.Local); err == nil {
|
|
endTime = t
|
|
}
|
|
}
|
|
|
|
// 如果 days > 0 优先
|
|
if defaultDays > 0 && start == "" && end == "" {
|
|
endTime = now
|
|
startTime = endTime.AddDate(0, 0, -(defaultDays - 1))
|
|
}
|
|
|
|
return startTime, endTime
|
|
}
|
|
|
|
// List 分页查询登录审计
|
|
func (s *LoginAuditListService) List(c *gin.Context) (*inventory.LoginAuditListResult, error) {
|
|
dep := dependency.FromContext(c)
|
|
client := dep.LoginAuditClient()
|
|
|
|
page := s.Page
|
|
if page < 1 {
|
|
page = 1
|
|
}
|
|
pageSize := s.PageSize
|
|
if pageSize < 1 {
|
|
pageSize = 20
|
|
}
|
|
if pageSize > 500 {
|
|
pageSize = 500
|
|
}
|
|
|
|
args := &inventory.LoginAuditListArgs{
|
|
Page: page,
|
|
PageSize: pageSize,
|
|
UserID: s.UserID,
|
|
IP: s.IP,
|
|
Username: s.Username,
|
|
}
|
|
switch s.Status {
|
|
case "success":
|
|
st := inventory.LoginAuditStatusSuccess
|
|
args.Status = &st
|
|
case "fail":
|
|
st := inventory.LoginAuditStatusFail
|
|
args.Status = &st
|
|
}
|
|
|
|
if s.StartTime != "" {
|
|
if t, err := time.ParseInLocation(time.RFC3339, s.StartTime, time.Local); err == nil {
|
|
args.StartTime = &t
|
|
} else if t, err := time.ParseInLocation("2006-01-02", s.StartTime, time.Local); err == nil {
|
|
args.StartTime = &t
|
|
}
|
|
}
|
|
if s.EndTime != "" {
|
|
if t, err := time.ParseInLocation(time.RFC3339, s.EndTime, time.Local); err == nil {
|
|
args.EndTime = &t
|
|
} else if t, err := time.ParseInLocation("2006-01-02", s.EndTime, time.Local); err == nil {
|
|
// end_time 是该日的结束（+1 day，左闭右开）
|
|
t = t.AddDate(0, 0, 1)
|
|
args.EndTime = &t
|
|
}
|
|
}
|
|
|
|
return client.List(c.Request.Context(), args)
|
|
}
|
|
|
|
// DailyStats 每日登录统计
|
|
func (s *LoginAuditDailyStatsService) DailyStats(c *gin.Context) ([]inventory.LoginAuditDailyItem, error) {
|
|
dep := dependency.FromContext(c)
|
|
client := dep.LoginAuditClient()
|
|
|
|
days := s.Days
|
|
if days < 1 {
|
|
days = 7
|
|
}
|
|
if days > 365 {
|
|
days = 365
|
|
}
|
|
start, end := parseTimeRange(s.Start, s.End, days)
|
|
|
|
return client.DailyStats(c.Request.Context(), start, end)
|
|
}
|