mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.2 KiB
80 lines
2.2 KiB
// 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/bitbus/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/auto/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: ccBuild(db, cc.BuildAuthorizationManage),
|
|
}
|
|
}
|