paopao-ce/internal/dao/sakila/authority.go

// Copyright 2023 ROC. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

package sakila

import (
	"github.com/jmoiron/sqlx"
	"github.com/rocboss/paopao-ce/internal/core"
	"github.com/rocboss/paopao-ce/internal/core/cs"
	"github.com/rocboss/paopao-ce/internal/core/ms"
	"github.com/rocboss/paopao-ce/internal/dao/sakila/yesql/cc"
	"github.com/rocboss/paopao-ce/pkg/types"
	"github.com/sirupsen/logrus"
)

var (
	_ core.AuthorizationManageService = (*authorizationManageSrv)(nil)
)

type authorizationManageSrv struct {
	*sqlxSrv
	q *cc.AuthorizationManage
}

func (s *authorizationManageSrv) IsAllow(user *ms.User, action *ms.Action) bool {
	// user is activation if had bind phone
	isActivation := (len(user.Phone) != 0)
	isFriend := s.isFriend(user.ID, action.UserId)
	// TODO: just use defaut act authorization chek rule now
	return action.Act.IsAllow(user, action.UserId, isFriend, isActivation)
}

func (s *authorizationManageSrv) MyFriendSet(userId int64) ms.FriendSet {
	var ids []string
	if err := s.q.MyFriendSet.Get(&ids, userId); err != nil {
		logrus.Warnf("get my FriendSet error: %s", err)
		return ms.FriendSet{}
	}
	resp := make(ms.FriendSet, len(ids))
	for _, id := range ids {
		resp[id] = types.Empty{}
	}
	return resp
}

func (s *authorizationManageSrv) BeFriendFilter(userId int64) ms.FriendFilter {
	var ids []int64
	if err := s.q.BeFriendIds.Get(&ids, userId); err != nil {
		logrus.Warnf("get my BeFriendFilter error: %s", err)
		return ms.FriendFilter{}
	}
	resp := make(ms.FriendFilter, len(ids))
	for _, id := range ids {
		resp[id] = types.Empty{}
	}
	return resp
}

func (s *authorizationManageSrv) BeFriendIds(userId int64) (res []int64, err error) {
	err = s.q.BeFriendIds.Get(&res, userId)
	return
}

func (s *authorizationManageSrv) isFriend(userId int64, friendId int64) bool {
	var status int8
	err := s.q.IsFriend.Get(&status, userId, friendId)
	if err == nil || status == cs.ContactStatusAgree {
		return true
	}
	return false
}

func newAuthorizationManageService(db *sqlx.DB) core.AuthorizationManageService {
	return &authorizationManageSrv{
		sqlxSrv: newSqlxSrv(db),
		q:       mustBuild(db, cc.BuildAuthorizationManage),
	}
}