package middleware

import (
	"context"
	"errors"
	"net/http"
	"time"

	systemmodel "management/internal/erpserver/model/system"
	v1 "management/internal/erpserver/service/v1"
	"management/internal/pkg/know"
	"management/internal/pkg/session"

	"github.com/drhin/logger"
	"go.uber.org/zap"
)

func Audit(sess session.Manager, auditLogService v1.AuditLogService, log *logger.Logger) func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			start := time.Now()

			// 提前获取用户信息（同步操作）
			user, err := sess.GetUser(r.Context(), know.StoreName)
			if err != nil {
				log.Error("获取用户会话失败", err)
				next.ServeHTTP(w, r) // 继续处理请求
				return
			}

			defer func() {
				go func() {
					if user.ID == 0 {
						log.Error("用户信息为空", errors.New("scs get user is empty"))
						return
					}

					ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
					defer cancel()

					al := systemmodel.NewAuditLog(r, user.Email, user.OS, user.Browser, start, time.Now())
					if err := auditLogService.Create(ctx, al); err != nil {
						log.Error(err.Error(), err,
							zap.Int32("user_id", user.ID),
							zap.String("user", user.Email),
							zap.String("ip", al.Ip),
							zap.String("os", al.Os),
							zap.String("method", al.Method),
							zap.String("path", al.Url),
						)
					}
				}()
			}()

			next.ServeHTTP(w, r)
		})
	}
}