package admin_agent

import (
	"context"
	"database/sql"
	"regexp"
	"time"

	"bd-server/app/main/api/internal/svc"
	"bd-server/app/main/api/internal/types"
	"bd-server/common/xerr"
	"bd-server/pkg/lzkit/crypto"

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

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

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

func (l *AdminUpdateAgentLogic) AdminUpdateAgent(req *types.AdminUpdateAgentReq) (resp *types.AdminUpdateAgentResp, err error) {
	agent, err := l.svcCtx.AgentModel.FindOne(l.ctx, req.Id)
	if err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理失败 err: %v", err)
	}

	if ok, _ := regexp.MatchString(`^1[3-9]\d{9}$`, req.Mobile); !ok {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), "手机号格式错误: %s", req.Mobile)
	}

	encryptedMobile, err := crypto.EncryptMobile(req.Mobile, l.svcCtx.Config.Encrypt.SecretKey)
	if err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err)
	}

	expiryTime, err := time.ParseInLocation("2006-01-02 15:04:05", req.MembershipExpiryTime, time.Local)
	if err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), "会员到期时间格式错误: %v", err)
	}

	agent.Mobile = encryptedMobile
	agent.LevelName = req.LevelName
	agent.Region = req.Region
	agent.MembershipExpiryTime = sql.NullTime{
		Time:  expiryTime,
		Valid: true,
	}

	if _, err = l.svcCtx.AgentModel.Update(l.ctx, nil, agent); err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新代理失败 err: %v", err)
	}

	return &types.AdminUpdateAgentResp{Success: true}, nil
}