gim/internal/business/domain/user/service/auth.go

package service

import (
	"context"
	"errors"
	"fmt"
	"gim/internal/business/comm/db"
	"gim/internal/business/comm/utils"
	"gim/internal/business/domain/user/model"
	"gim/internal/business/domain/user/repo"
	"gim/pkg/gerrors"
	"gim/pkg/pb"
	"gim/pkg/rpc"
	"gim/pkg/util"
	"strconv"
	"time"
)

type authService struct{}

var AuthService = new(authService)

// SignIn 登录
func (*authService) SignIn(ctx context.Context, phoneNumber, code string, masterId, deviceId int64, pushAlia string) (bool, int64, string, int64, error) {
	if !Verify(phoneNumber, code) {
		return false, 0, "", 0, gerrors.ErrBadCode
	}

	user, err := repo.UserRepo.GetByPhoneNumber(phoneNumber, masterId)
	if err != nil {
		return false, 0, "", 0, err
	}

	var isNew = false
	if user == nil {
		user = &model.User{
			PhoneNumber: phoneNumber,
			MasterId:    masterId,
			CreateTime:  time.Now(),
			UpdateTime:  time.Now(),
		}
		err := repo.UserRepo.Save(user)
		if err != nil {
			return false, 0, "", 0, err
		}
		isNew = true
	}

	resp, err := rpc.GetLogicIntClient().GetDevice(ctx, &pb.GetDeviceReq{DeviceId: deviceId})
	if err != nil {
		return false, 0, "", 0, err
	}

	// 方便测试
	//token := "0"
	token := util.RandString(40)
	err = repo.AuthRepo.Set(user.Id, resp.Device.DeviceId, model.Device{
		Type:   resp.Device.Type,
		Token:  token,
		Expire: time.Now().AddDate(0, 3, 0).Unix(),
	})
	if err != nil {
		return false, 0, "", 0, err
	}

	if pushAlia != "" {
		userPushForJg, err := db.DbUserPushForJg.UserPushForJgGetOne(strconv.FormatInt(user.Id, 10), masterId)
		if err != nil {
			return false, 0, "", 0, err
		}
		if userPushForJg == nil {
			save := db.DbUserPushForJg.UserPushForJgInsert(user.Id, masterId, pushAlia)
			if !save {
				return false, 0, "", 0, errors.New("插入user_push_for_jg记录失败")
			}
		} else {
			update := db.DbUserPushForJg.UserPushForJgUpdate(user.Id, masterId, pushAlia)
			if !update {
				return false, 0, "", 0, errors.New("修改user_push_for_jg记录失败")
			}
		}
	}
	return isNew, user.Id, token, masterId, nil
}

func Verify(phoneNumber, code string) bool {
	// 假装他成功了
	return true
}

// Auth 验证用户是否登录
func (*authService) Auth(ctx context.Context, userId, deviceId int64, token string) error {
	device, err := repo.AuthRepo.Get(userId, deviceId)
	if err != nil {
		return err
	}

	if device == nil {
		return gerrors.ErrUnauthorized
	}

	if device.Expire < time.Now().Unix() {
		return gerrors.ErrUnauthorized
	}

	if device.Token != token {
		return gerrors.ErrUnauthorized
	}
	return nil
}

// AuthMaster 验证站长权限
func (*authService) AuthMaster(ctx context.Context, masterId string) error {
	master, err := repo.MasterRepo.Get(utils.StrToInt64(masterId))
	if err != nil {
		return err
	}
	if master == nil {
		return gerrors.ErrMasterUnauthorized
	}
	packageExpireTime, _ := utils.StrToTime(master.PackageExpireTime)
	fmt.Println(">>>>>>>>>>>>>>packageExpireTime>>>>>>>>>>>>>>>", packageExpireTime, time.Now().Unix())
	if packageExpireTime < time.Now().Unix() {
		return gerrors.ErrMasterUnauthorized
	}

	return nil
}