package logic

import (
	"context"
	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	admin "lone-services/rpc/admin/pb"
	"lone-services/services/admin/internal/dao"
	"lone-services/services/admin/internal/model"
	"lone-services/services/admin/internal/svc"
	"lone-services/services/admin/validator"
	"strconv"

	"github.com/zeromicro/go-zero/core/logx"
)

type AuthoritySortLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
	BaseLogic
}

func NewAuthoritySortLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AuthoritySortLogic {
	return &AuthoritySortLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *AuthoritySortLogic) AuthoritySort(in *admin.AuthoritySortRequest) (*admin.Response, error) {
	var v validator.AuthoritySortValidator
	if fail := l.checkParams(in, &v); fail != nil {
		return fail, nil
	}
	adminInfo := utils.GetUserFromCtx(l.ctx)
	if adminInfo.ID < utils.NumberOne {
		return l.fail(utils.ErrorNoLoginInfo)
	}
	info := dao.AuthoritySort{}
	modelObj := model.AuthorityModel{}.Init()
	w := modelbase.Params{Eq: map[string]string{"id": strconv.FormatInt(in.Id, utils.NumberTen)}}
	err := modelObj.GetOne(w, &info)
	if err != nil {
		l.Logger.Error(err)
		return l.fail(utils.Fail)
	}
	if info.Id < utils.NumberOne {
		return l.fail(utils.ErrorNotFund)
	}

	w = modelbase.Params{
		Eq:    map[string]string{"parent_id": strconv.FormatInt(info.ParentId, utils.NumberTen)},
		Order: "sort ASC",
	}
	sortKey := "sort > ?"
	if in.Type == dao.SortTypeUp {
		sortKey = "sort < ?"
		w.Order = "sort DESC"
	}
	w.Other = map[string]string{sortKey: strconv.Itoa(int(info.Sort))}
	other := dao.AuthoritySort{}
	err = modelObj.GetOne(w, &other)
	if err != nil {
		l.Logger.Error(err)
		return l.fail(utils.Fail)
	}
	action := utils.ActionAdd{}
	action.Type = utils.LogActionTypeSort
	action.ModuleName = utils.LogActionModuleAuthority
	action.OldContent = info
	if info.Id < utils.NumberOne {
		return l.fail(utils.ErrorNotFund)
	}
	w = modelbase.Params{Eq: map[string]string{"id": strconv.FormatInt(info.Id, utils.NumberTen)}}

	infoSort := info.Sort
	info.Sort = other.Sort
	info.AdminId = adminInfo.ID
	info.AdminName = adminInfo.Name
	num, err := modelObj.Edit(w, &info)
	if err != nil || num < utils.NumberOne {
		l.Logger.Error(err)
		return l.fail(utils.Fail)
	}
	action.NewContent = info
	utils.SetActionLog(adminInfo, action)
	action.OldContent = other
	w = modelbase.Params{Eq: map[string]string{"id": strconv.FormatInt(other.Id, utils.NumberTen)}}
	other.Sort = infoSort
	other.AdminId = info.AdminId
	other.AdminName = info.AdminName
	num, err = modelObj.Edit(w, &other)
	if err != nil || num < utils.NumberOne {
		l.Logger.Error(err)
		return l.fail(utils.Fail)
	}
	action.NewContent = other
	utils.SetActionLog(adminInfo, action)

	return l.ok(utils.NumberOne)
}