From b4775777479db5d1cd4858f4a84143bd414835df Mon Sep 17 00:00:00 2001 From: Gordon <46924906+FGadvancer@users.noreply.github.com> Date: Wed, 10 Apr 2024 11:44:59 +0800 Subject: [PATCH] refactor: all module update. --- internal/rpc/user/user.go | 6 +++--- pkg/authverify/token.go | 11 ----------- 2 files changed, 3 insertions(+), 14 deletions(-) diff --git a/internal/rpc/user/user.go b/internal/rpc/user/user.go index ba39f24e7..a50bcd25f 100644 --- a/internal/rpc/user/user.go +++ b/internal/rpc/user/user.go @@ -534,7 +534,7 @@ func (s *userServer) ProcessUserCommandGetAll(ctx context.Context, req *pbuser.P } func (s *userServer) AddNotificationAccount(ctx context.Context, req *pbuser.AddNotificationAccountReq) (*pbuser.AddNotificationAccountResp, error) { - if err := authverify.CheckIMAdmin(ctx, s.config); err != nil { + if err := authverify.CheckAdmin(ctx, &s.config.Share.IMAdmin); err != nil { return nil, err } @@ -577,7 +577,7 @@ func (s *userServer) AddNotificationAccount(ctx context.Context, req *pbuser.Add } func (s *userServer) UpdateNotificationAccountInfo(ctx context.Context, req *pbuser.UpdateNotificationAccountInfoReq) (*pbuser.UpdateNotificationAccountInfoResp, error) { - if err := authverify.CheckIMAdmin(ctx, s.config); err != nil { + if err := authverify.CheckAdmin(ctx, &s.config.Share.IMAdmin); err != nil { return nil, err } @@ -604,7 +604,7 @@ func (s *userServer) UpdateNotificationAccountInfo(ctx context.Context, req *pbu func (s *userServer) SearchNotificationAccount(ctx context.Context, req *pbuser.SearchNotificationAccountReq) (*pbuser.SearchNotificationAccountResp, error) { // Check if user is an admin - if err := authverify.CheckIMAdmin(ctx, s.config); err != nil { + if err := authverify.CheckAdmin(ctx, &s.config.Share.IMAdmin); err != nil { return nil, err } diff --git a/pkg/authverify/token.go b/pkg/authverify/token.go index eb9428fb7..95e7588ec 100644 --- a/pkg/authverify/token.go +++ b/pkg/authverify/token.go @@ -55,17 +55,6 @@ func CheckAdmin(ctx context.Context, imAdmin *config.IMAdmin) error { return servererrs.ErrNoPermission.WrapMsg(fmt.Sprintf("user %s is not admin userID", mcontext.GetOpUserID(ctx))) } -func CheckIMAdmin(ctx context.Context, config *config.GlobalConfig) error { - if datautil.Contain(mcontext.GetOpUserID(ctx), config.IMAdmin.UserID...) { - return nil - } - return servererrs.ErrNoPermission.WrapMsg(fmt.Sprintf("user %s is not CheckIMAdmin userID", mcontext.GetOpUserID(ctx))) -} - -func ParseRedisInterfaceToken(redisToken any, secret string) (*tokenverify.Claims, error) { - return tokenverify.GetClaimFromToken(string(redisToken.([]uint8)), Secret(secret)) -} - func IsManagerUserID(opUserID string, imAdmin *config.IMAdmin) bool { return datautil.Contain(opUserID, imAdmin.UserID...) }