package service

import (
	"dianshang/internal/model"
	"time"

	"gorm.io/gorm"
)

// LogService 日志服务
type LogService struct {
	db *gorm.DB
}

// NewLogService 创建日志服务
func NewLogService(db *gorm.DB) *LogService {
	return &LogService{
		db: db,
	}
}

// CreateLoginLog 创建登录日志
func (s *LogService) CreateLoginLog(userID uint, ip, userAgent string, status int, remark string) error {
	log := &model.UserLoginLog{
		UserID:    userID,
		LoginIP:   ip,
		UserAgent: userAgent,
		LoginTime: time.Now(),
		Status:    status,
		Remark:    remark,
	}
	return s.db.Create(log).Error
}

// GetUserLoginLogs 获取用户登录日志
func (s *LogService) GetUserLoginLogs(userID uint, page, pageSize int) ([]model.UserLoginLog, map[string]interface{}, error) {
	var logs []model.UserLoginLog
	var total int64
	
	query := s.db.Model(&model.UserLoginLog{}).Where("user_id = ?", userID)
	
	// 获取总数
	err := query.Count(&total).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 分页查询
	offset := (page - 1) * pageSize
	err = query.Offset(offset).Limit(pageSize).Order("login_time DESC").Find(&logs).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 构建分页信息
	pagination := map[string]interface{}{
		"total":       total,
		"page":        page,
		"page_size":   pageSize,
		"total_pages": (total + int64(pageSize) - 1) / int64(pageSize),
	}
	
	return logs, pagination, nil
}

// GetLoginLogList 获取登录日志列表（管理后台）
func (s *LogService) GetLoginLogList(page, pageSize int, conditions map[string]interface{}) ([]model.UserLoginLog, map[string]interface{}, error) {
	var logs []model.UserLoginLog
	var total int64
	
	query := s.db.Model(&model.UserLoginLog{}).Preload("User")
	
	// 应用查询条件
	if userID, ok := conditions["user_id"]; ok && userID != "" {
		query = query.Where("user_id = ?", userID)
	}
	
	if ip, ok := conditions["ip"]; ok && ip != "" {
		query = query.Where("login_ip LIKE ?", "%"+ip.(string)+"%")
	}
	
	if status, ok := conditions["status"]; ok && status != "" {
		query = query.Where("status = ?", status)
	}
	
	if startDate, ok := conditions["start_date"]; ok && startDate != "" {
		query = query.Where("login_time >= ?", startDate)
	}
	
	if endDate, ok := conditions["end_date"]; ok && endDate != "" {
		query = query.Where("login_time <= ?", endDate)
	}
	
	// 获取总数
	err := query.Count(&total).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 分页查询
	offset := (page - 1) * pageSize
	err = query.Offset(offset).Limit(pageSize).Order("login_time DESC").Find(&logs).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 构建分页信息
	pagination := map[string]interface{}{
		"total":       total,
		"page":        page,
		"page_size":   pageSize,
		"total_pages": (total + int64(pageSize) - 1) / int64(pageSize),
	}
	
	return logs, pagination, nil
}

// CreateOperationLog 创建操作日志
func (s *LogService) CreateOperationLog(userID uint, module, action, description, ip, userAgent, requestData string) error {
	log := &model.UserOperationLog{
		UserID:      userID,
		Module:      module,
		Action:      action,
		Description: description,
		IP:          ip,
		UserAgent:   userAgent,
		RequestData: requestData,
		CreatedAt:   time.Now(),
	}
	return s.db.Create(log).Error
}

// GetUserOperationLogs 获取用户操作日志
func (s *LogService) GetUserOperationLogs(userID uint, page, pageSize int) ([]model.UserOperationLog, map[string]interface{}, error) {
	var logs []model.UserOperationLog
	var total int64
	
	query := s.db.Model(&model.UserOperationLog{}).Where("user_id = ?", userID)
	
	// 获取总数
	err := query.Count(&total).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 分页查询
	offset := (page - 1) * pageSize
	err = query.Offset(offset).Limit(pageSize).Order("created_at DESC").Find(&logs).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 构建分页信息
	pagination := map[string]interface{}{
		"total":       total,
		"page":        page,
		"page_size":   pageSize,
		"total_pages": (total + int64(pageSize) - 1) / int64(pageSize),
	}
	
	return logs, pagination, nil
}

// GetOperationLogList 获取操作日志列表（管理后台）
func (s *LogService) GetOperationLogList(page, pageSize int, conditions map[string]interface{}) ([]model.UserOperationLog, map[string]interface{}, error) {
	var logs []model.UserOperationLog
	var total int64
	
	query := s.db.Model(&model.UserOperationLog{}).Preload("User")
	
	// 应用查询条件
	if userID, ok := conditions["user_id"]; ok && userID != "" {
		query = query.Where("user_id = ?", userID)
	}
	
	if module, ok := conditions["module"]; ok && module != "" {
		query = query.Where("module = ?", module)
	}
	
	if action, ok := conditions["action"]; ok && action != "" {
		query = query.Where("action = ?", action)
	}
	
	if ip, ok := conditions["ip"]; ok && ip != "" {
		query = query.Where("ip LIKE ?", "%"+ip.(string)+"%")
	}
	
	if startDate, ok := conditions["start_date"]; ok && startDate != "" {
		query = query.Where("created_at >= ?", startDate)
	}
	
	if endDate, ok := conditions["end_date"]; ok && endDate != "" {
		query = query.Where("created_at <= ?", endDate)
	}
	
	// 获取总数
	err := query.Count(&total).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 分页查询
	offset := (page - 1) * pageSize
	err = query.Offset(offset).Limit(pageSize).Order("created_at DESC").Find(&logs).Error
	if err != nil {
		return nil, nil, err
	}
	
	// 构建分页信息
	pagination := map[string]interface{}{
		"total":       total,
		"page":        page,
		"page_size":   pageSize,
		"total_pages": (total + int64(pageSize) - 1) / int64(pageSize),
	}
	
	return logs, pagination, nil
}

// GetLoginStatistics 获取登录统计
func (s *LogService) GetLoginStatistics(startDate, endDate string) (map[string]interface{}, error) {
	result := make(map[string]interface{})
	
	// 总登录次数
	var totalLogins int64
	query := s.db.Model(&model.UserLoginLog{})
	if startDate != "" && endDate != "" {
		query = query.Where("DATE(login_time) BETWEEN ? AND ?", startDate, endDate)
	}
	query.Count(&totalLogins)
	result["total_logins"] = totalLogins
	
	// 成功登录次数
	var successLogins int64
	query = s.db.Model(&model.UserLoginLog{}).Where("status = 1")
	if startDate != "" && endDate != "" {
		query = query.Where("DATE(login_time) BETWEEN ? AND ?", startDate, endDate)
	}
	query.Count(&successLogins)
	result["success_logins"] = successLogins
	
	// 失败登录次数
	var failedLogins int64
	query = s.db.Model(&model.UserLoginLog{}).Where("status = 0")
	if startDate != "" && endDate != "" {
		query = query.Where("DATE(login_time) BETWEEN ? AND ?", startDate, endDate)
	}
	query.Count(&failedLogins)
	result["failed_logins"] = failedLogins
	
	// 独立用户数
	var uniqueUsers int64
	query = s.db.Model(&model.UserLoginLog{}).Distinct("user_id")
	if startDate != "" && endDate != "" {
		query = query.Where("DATE(login_time) BETWEEN ? AND ?", startDate, endDate)
	}
	query.Count(&uniqueUsers)
	result["unique_users"] = uniqueUsers
	
	return result, nil
}

// GetOperationStatistics 获取操作统计
func (s *LogService) GetOperationStatistics(startDate, endDate string) (map[string]interface{}, error) {
	result := make(map[string]interface{})
	
	// 总操作次数
	var totalOperations int64
	query := s.db.Model(&model.UserOperationLog{})
	if startDate != "" && endDate != "" {
		query = query.Where("DATE(created_at) BETWEEN ? AND ?", startDate, endDate)
	}
	query.Count(&totalOperations)
	result["total_operations"] = totalOperations
	
	// 按模块统计
	var moduleStats []struct {
		Module string `json:"module"`
		Count  int64  `json:"count"`
	}
	query = s.db.Model(&model.UserOperationLog{}).Select("module, COUNT(*) as count").Group("module")
	if startDate != "" && endDate != "" {
		query = query.Where("DATE(created_at) BETWEEN ? AND ?", startDate, endDate)
	}
	query.Scan(&moduleStats)
	result["module_stats"] = moduleStats
	
	// 按操作统计
	var actionStats []struct {
		Action string `json:"action"`
		Count  int64  `json:"count"`
	}
	query = s.db.Model(&model.UserOperationLog{}).Select("action, COUNT(*) as count").Group("action")
	if startDate != "" && endDate != "" {
		query = query.Where("DATE(created_at) BETWEEN ? AND ?", startDate, endDate)
	}
	query.Scan(&actionStats)
	result["action_stats"] = actionStats
	
	return result, nil
}

// CleanOldLogs 清理旧日志
func (s *LogService) CleanOldLogs(days int) error {
	cutoffDate := time.Now().AddDate(0, 0, -days)
	
	// 清理登录日志
	if err := s.db.Where("login_time < ?", cutoffDate).Delete(&model.UserLoginLog{}).Error; err != nil {
		return err
	}
	
	// 清理操作日志
	if err := s.db.Where("created_at < ?", cutoffDate).Delete(&model.UserOperationLog{}).Error; err != nil {
		return err
	}
	
	return nil
}

// GetLoginLogByID 根据ID获取登录日志
func (s *LogService) GetLoginLogByID(id uint) (*model.UserLoginLog, error) {
	var log model.UserLoginLog
	err := s.db.Preload("User").First(&log, id).Error
	if err != nil {
		return nil, err
	}
	return &log, nil
}

// GetTodayLoginCount 获取今日登录次数
func (s *LogService) GetTodayLoginCount() (int64, error) {
	var count int64
	today := time.Now().Format("2006-01-02")
	err := s.db.Model(&model.UserLoginLog{}).Where("DATE(login_time) = ?", today).Count(&count).Error
	return count, err
}

// GetTodayOperationCount 获取今日操作次数
func (s *LogService) GetTodayOperationCount() (int64, error) {
	var count int64
	today := time.Now().Format("2006-01-02")
	err := s.db.Model(&model.UserOperationLog{}).Where("DATE(created_at) = ?", today).Count(&count).Error
	return count, err
}

// GetOnlineUserCount 获取在线用户数（最近30分钟有登录记录的用户）
func (s *LogService) GetOnlineUserCount() (int64, error) {
	var count int64
	thirtyMinutesAgo := time.Now().Add(-30 * time.Minute)
	err := s.db.Model(&model.UserLoginLog{}).
		Where("login_time >= ? AND status = 1", thirtyMinutesAgo).
		Distinct("user_id").
		Count(&count).Error
	return count, err
}

// GetLoginTrend 获取登录趋势
func (s *LogService) GetLoginTrend(days int) ([]map[string]interface{}, error) {
	var results []map[string]interface{}
	
	for i := days - 1; i >= 0; i-- {
		date := time.Now().AddDate(0, 0, -i).Format("2006-01-02")
		var count int64
		s.db.Model(&model.UserLoginLog{}).
			Where("DATE(login_time) = ?", date).
			Count(&count)
		
		results = append(results, map[string]interface{}{
			"date":  date,
			"count": count,
		})
	}
	
	return results, nil
}

// GetOperationTrend 获取操作趋势
func (s *LogService) GetOperationTrend(days int) ([]map[string]interface{}, error) {
	var results []map[string]interface{}
	
	for i := days - 1; i >= 0; i-- {
		date := time.Now().AddDate(0, 0, -i).Format("2006-01-02")
		var count int64
		s.db.Model(&model.UserOperationLog{}).
			Where("DATE(created_at) = ?", date).
			Count(&count)
		
		results = append(results, map[string]interface{}{
			"date":  date,
			"count": count,
		})
	}
	
	return results, nil
}

// GetOperationLogByID 根据ID获取操作日志
func (s *LogService) GetOperationLogByID(id uint) (*model.UserOperationLog, error) {
	var log model.UserOperationLog
	err := s.db.Preload("User").First(&log, id).Error
	if err != nil {
		return nil, err
	}
	return &log, nil
}