muyu-apiserver/gateway/internal/logic/auth/loginLogic.go

package auth

import (
	"context"
	"errors"

	"muyu-apiserver/gateway/internal/svc"
	"muyu-apiserver/gateway/internal/types"
	jwtpkg "muyu-apiserver/pkg/jwt"
	"muyu-apiserver/rpc/system/pb"

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

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

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

func (l *LoginLogic) Login(req *types.LoginReq) (resp *types.LoginResp, err error) {
	rpcResp, err := l.svcCtx.SystemRpc.Login(l.ctx, &pb.LoginReq{
		Username: req.Username,
		Password: req.Password,
	})
	if err != nil {
		return nil, err
	}

	if req.TenantId == "" {
		return nil, errors.New("请选择要登录的租户")
	}

	verified, verr := l.svcCtx.CrmRepo.VerifyUserTenant(l.ctx, rpcResp.UserId, req.TenantId)
	if verr != nil || !verified {
		return nil, errors.New("用户不属于该租户，禁止登录")
	}
	tenantId := req.TenantId

	token, err := jwtpkg.GenerateToken(
		l.svcCtx.Config.Auth.AccessSecret,
		l.svcCtx.Config.Auth.AccessExpire,
		rpcResp.UserId,
		rpcResp.RoleKey,
		tenantId,
	)
	if err != nil {
		return nil, err
	}

	return &types.LoginResp{
		Token: token,
		UserInfo: types.UserInfo{
			UserId:   rpcResp.UserId,
			Username: rpcResp.Username,
			RealName: rpcResp.RealName,
			Avatar:   rpcResp.Avatar,
			RoleName: rpcResp.RoleName,
			TenantId: tenantId,
		},
	}, nil
}