1Panel/backend/app/service/operation_log.go

package service

import (
	"encoding/json"

	"github.com/1Panel-dev/1Panel/backend/app/dto"
	"github.com/1Panel-dev/1Panel/backend/app/model"
	"github.com/1Panel-dev/1Panel/backend/constant"
	"github.com/1Panel-dev/1Panel/backend/global"
	"github.com/jinzhu/copier"
	"github.com/pkg/errors"
)

type OperationService struct{}

type IOperationService interface {
	Page(search dto.PageInfo) (int64, interface{}, error)
	Create(operation model.OperationLog) error
	BatchDelete(ids []uint) error
}

func NewIOperationService() IOperationService {
	return &OperationService{}
}

func (u *OperationService) Create(operation model.OperationLog) error {
	return operationRepo.Create(&operation)
}

func (u *OperationService) Page(search dto.PageInfo) (int64, interface{}, error) {
	total, ops, err := operationRepo.Page(search.Page, search.PageSize, commonRepo.WithOrderBy("created_at desc"))
	var dtoOps []dto.OperationLogBack
	for _, op := range ops {
		var item dto.OperationLogBack
		if err := copier.Copy(&item, &op); err != nil {
			return 0, nil, errors.WithMessage(constant.ErrStructTransform, err.Error())
		}
		item.Body = filterSensitive(item.Body)
		var res dto.Response
		if err := json.Unmarshal([]byte(item.Resp), &res); err != nil {
			global.LOG.Errorf("unmarshal failed, err: %+v", err)
			dtoOps = append(dtoOps, item)
			continue
		}
		item.Status = res.Code
		if item.Status != 200 {
			item.ErrorMessage = res.Msg
		}
		dtoOps = append(dtoOps, item)
	}
	return total, dtoOps, err
}

func (u *OperationService) BatchDelete(ids []uint) error {
	return operationRepo.Delete(commonRepo.WithIdsIn(ids))
}

func filterSensitive(vars string) string {
	var Sensitives = []string{"password", "Password", "credential", "privateKey"}
	ops := make(map[string]interface{})
	if err := json.Unmarshal([]byte(vars), &ops); err != nil {
		return vars
	}
	for k := range ops {
		for _, sen := range Sensitives {
			if k == sen {
				delete(ops, k)
				continue
			}
		}
	}
	backStr, err := json.Marshal(ops)
	if err != nil {
		return ""
	}
	return string(backStr)
}