复制项目

pkg/common/mctx/get.go

@@ -0,0 +1,136 @@
+// Copyright © 2023 OpenIM open source community. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package mctx
+
+import (
+	"context"
+	"strconv"
+
+	"github.com/openimsdk/tools/utils/datautil"
+
+	constantpb "git.imall.cloud/openim/protocol/constant"
+	"github.com/openimsdk/tools/errs"
+
+	"git.imall.cloud/openim/chat/pkg/common/constant"
+	"git.imall.cloud/openim/chat/pkg/common/tokenverify"
+)
+
+func HaveOpUser(ctx context.Context) bool {
+	return ctx.Value(constant.RpcOpUserID) != nil
+}
+
+func Check(ctx context.Context) (string, int32, error) {
+	opUserIDVal := ctx.Value(constant.RpcOpUserID)
+	opUserID, ok := opUserIDVal.(string)
+	if !ok {
+		return "", 0, errs.ErrNoPermission.WrapMsg("no opUserID")
+	}
+	if opUserID == "" {
+		return "", 0, errs.ErrNoPermission.WrapMsg("opUserID empty")
+	}
+	opUserTypeArr, ok := ctx.Value(constant.RpcOpUserType).([]string)
+	if !ok {
+		return "", 0, errs.ErrNoPermission.WrapMsg("missing user type")
+	}
+	if len(opUserTypeArr) == 0 {
+		return "", 0, errs.ErrNoPermission.WrapMsg("user type empty")
+	}
+	userType, err := strconv.Atoi(opUserTypeArr[0])
+	if err != nil {
+		return "", 0, errs.ErrNoPermission.WrapMsg("user type invalid " + err.Error())
+	}
+	if !(userType == constant.AdminUser || userType == constant.NormalUser) {
+		return "", 0, errs.ErrNoPermission.WrapMsg("user type invalid")
+	}
+	return opUserID, int32(userType), nil
+}
+
+func CheckAdmin(ctx context.Context) (string, error) {
+	userID, userType, err := Check(ctx)
+	if err != nil {
+		return "", err
+	}
+	if userType != constant.AdminUser {
+		return "", errs.ErrNoPermission.WrapMsg("not admin")
+	}
+	return userID, nil
+}
+
+func CheckUser(ctx context.Context) (string, error) {
+	userID, userType, err := Check(ctx)
+	if err != nil {
+		return "", err
+	}
+	if userType != constant.NormalUser {
+		return "", errs.ErrNoPermission.WrapMsg("not user")
+	}
+	return userID, nil
+}
+
+func CheckAdminOrUser(ctx context.Context) (string, int32, error) {
+	userID, userType, err := Check(ctx)
+	if err != nil {
+		return "", 0, err
+	}
+	return userID, userType, nil
+}
+
+func CheckAdminOr(ctx context.Context, userIDs ...string) error {
+	userID, userType, err := Check(ctx)
+	if err != nil {
+		return err
+	}
+	if userType == tokenverify.TokenAdmin {
+		return nil
+	}
+	for _, id := range userIDs {
+		if userID == id {
+			return nil
+		}
+	}
+	return errs.ErrNoPermission.WrapMsg("not admin or not in userIDs")
+}
+
+func GetOpUserID(ctx context.Context) string {
+	userID, _ := ctx.Value(constantpb.OpUserID).(string)
+	return userID
+}
+
+func GetUserType(ctx context.Context) (int, error) {
+	userTypeArr, _ := ctx.Value(constant.RpcOpUserType).([]string)
+	userType, err := strconv.Atoi(userTypeArr[0])
+	if err != nil {
+		return 0, errs.ErrNoPermission.WrapMsg("user type invalid " + err.Error())
+	}
+	return userType, nil
+}
+
+func WithOpUserID(ctx context.Context, opUserID string, userType int) context.Context {
+	headers, _ := ctx.Value(constant.RpcCustomHeader).([]string)
+	ctx = context.WithValue(ctx, constant.RpcOpUserID, opUserID)
+	ctx = context.WithValue(ctx, constant.RpcOpUserType, []string{strconv.Itoa(userType)})
+	if datautil.IndexOf(constant.RpcOpUserType, headers...) < 0 {
+		ctx = context.WithValue(ctx, constant.RpcCustomHeader, append(headers, constant.RpcOpUserType))
+	}
+	return ctx
+}
+
+func WithAdminUser(ctx context.Context, userID string) context.Context {
+	return WithOpUserID(ctx, userID, constant.AdminUser)
+}
+
+func WithApiToken(ctx context.Context, token string) context.Context {
+	return context.WithValue(ctx, constant.CtxApiToken, token)
+}