2

internal/pkg/middleware/audit.go

@@ -3,13 +3,9 @@ package middleware
 import (
 	"context"
 	"net/http"
-	"strconv"
-	"strings"
 	"time"
 
-	db "management/internal/db/sqlc"
-
-	"github.com/zhang2092/browser"
+	systemmodel "management/internal/erpserver/model/system"
 )
 
 func (m *middleware) Audit(next http.Handler) http.Handler {
@@ -27,47 +23,11 @@ func (m *middleware) Audit(next http.Handler) http.Handler {
 
 func (m *middleware) writeLog(req *http.Request, start time.Time) {
 	end := time.Now()
-	duration := end.Sub(start)
-	var params string
-	method := req.Method
-	if method == "GET" {
-		params = req.URL.Query().Encode()
-	} else if method == "POST" {
-		contentType := req.Header.Get("Content-Type")
-		if strings.Contains(contentType, "application/json") {
-			body := make([]byte, req.ContentLength)
-			req.Body.Read(body)
-			params = string(body)
-		} else if strings.Contains(contentType, "application/x-www-form-urlencoded") {
-			params = req.Form.Encode()
-		}
-	}
-
-	ctx := req.Context()
-	au := m.AuthUser(ctx)
-	arg := &db.CreateSysAuditLogParams{
-		CreatedAt:  time.Now(),
-		Email:      au.Email,
-		Username:   au.Username,
-		UserUuid:   au.Uuid,
-		StartAt:    start,
-		EndAt:      end,
-		Duration:   strconv.FormatInt(duration.Milliseconds(), 10),
-		Url:        req.URL.RequestURI(),
-		Method:     method,
-		Parameters: params,
-		RefererUrl: req.Header.Get("Referer"),
-		Ip:         req.RemoteAddr,
-		Remark:     "",
-	}
-	br, err := browser.NewBrowser(req.Header.Get("User-Agent"))
-	if err == nil {
-		arg.Os = br.Platform().Name()
-		arg.Browser = br.Name()
-	}
+	user := m.AuthUser(req.Context())
+	al := systemmodel.NewAuditLog(req, user.Email, user.OS, user.Browser, start, end)
 
 	c, cancel := context.WithTimeout(context.Background(), time.Second*3)
 	defer cancel()
 
-	_ = m.biz.AuditBiz().Create(c, arg)
+	_ = m.auditLogsvc.Create(c, al)
 }

internal/pkg/middleware/authorize.go

@@ -42,7 +42,7 @@ func (m *middleware) Authorize(next http.Handler) http.Handler {
 			return
 		}
 
-		menus, err := m.biz.MenuBiz().MapOwnerMenuByRoleID(ctx, user.RoleID)
+		menus, err := m.menusvc.ListByRoleIDToMap(ctx, user.RoleID)
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusInternalServerError)
 			return
@@ -80,3 +80,23 @@ func (m *middleware) AuthUser(ctx context.Context) dto.AuthorizeUser {
 	}
 	return user
 }
+
+func (m *middleware) IsAuth(ctx context.Context) bool {
+	var user dto.AuthorizeUser
+	b := m.session.GetBytes(ctx, know.StoreName)
+	if err := json.Unmarshal(b, &user); err == nil {
+		return true
+	}
+	return false
+}
+
+func (m *middleware) RefreshToken(ctx context.Context) bool {
+	if err := m.session.RenewToken(ctx); err == nil {
+		return true
+	}
+	return false
+}
+
+func (m *middleware) Destroy(ctx context.Context) error {
+	return m.session.Destroy(ctx)
+}

internal/pkg/middleware/middleware.go

@@ -5,28 +5,33 @@ import (
 	"net/http"
 
 	"management/internal/db/model/dto"
-	systemv1 "management/internal/erpserver/biz/v1/system"
+	v1 "management/internal/erpserver/service/v1"
 	"management/internal/pkg/session"
 )
 
-type IMiddleware interface {
+type Middleware interface {
 	Audit(next http.Handler) http.Handler
 	NoSurf(next http.Handler) http.Handler
 	LoadSession(next http.Handler) http.Handler
 	Authorize(next http.Handler) http.Handler
 	AuthUser(ctx context.Context) dto.AuthorizeUser
+	IsAuth(ctx context.Context) bool
+	RefreshToken(ctx context.Context) bool
+	Destroy(ctx context.Context) error
 }
 
 type middleware struct {
-	biz     systemv1.SystemBiz
-	session session.ISession
+	session     session.Session
+	menusvc     v1.MenuService
+	auditLogsvc v1.AuditLogService
 }
 
-var _ IMiddleware = (*middleware)(nil)
+var _ Middleware = (*middleware)(nil)
 
-func New(biz systemv1.SystemBiz, session session.ISession) IMiddleware {
+func New(session session.Session, menusvc v1.MenuService, auditLogsvc v1.AuditLogService) Middleware {
 	return &middleware{
-		biz:     biz,
-		session: session,
+		session:     session,
+		menusvc:     menusvc,
+		auditLogsvc: auditLogsvc,
 	}
 }