github.com/polarismesh/polaris@v1.17.8/common/api/v1/auth_response.go

/**
 * Tencent is pleased to support the open source community by making Polaris available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * 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 v1

import (
	"github.com/golang/protobuf/ptypes/wrappers"
	apimodel "github.com/polarismesh/specification/source/go/api/v1/model"
	apisecurity "github.com/polarismesh/specification/source/go/api/v1/security"
	apiservice "github.com/polarismesh/specification/source/go/api/v1/service_manage"
)

// NewAuthResponse 创建回复消息
func NewAuthResponse(code apimodel.Code) *apiservice.Response {
	return &apiservice.Response{
		Code: &wrappers.UInt32Value{Value: uint32(code)},
		Info: &wrappers.StringValue{Value: code2info[uint32(code)]},
	}
}

// NewAuthResponseWithMsg 创建回复消息
func NewAuthResponseWithMsg(code apimodel.Code, msg string) *apiservice.Response {
	return &apiservice.Response{
		Code: &wrappers.UInt32Value{Value: uint32(code)},
		Info: &wrappers.StringValue{Value: code2info[uint32(code)] + ":" + msg},
	}
}

// NewAuthBatchWriteResponse 创建批量回复
func NewAuthBatchWriteResponse(code apimodel.Code) *apiservice.BatchWriteResponse {
	return &apiservice.BatchWriteResponse{
		Code: &wrappers.UInt32Value{Value: uint32(code)},
		Info: &wrappers.StringValue{Value: code2info[uint32(code)]},
		Size: &wrappers.UInt32Value{Value: 0},
	}
}

// NewAuthBatchQueryResponse 创建批量查询回复
func NewAuthBatchQueryResponse(code apimodel.Code) *apiservice.BatchQueryResponse {
	return &apiservice.BatchQueryResponse{
		Code:   &wrappers.UInt32Value{Value: uint32(code)},
		Info:   &wrappers.StringValue{Value: code2info[uint32(code)]},
		Amount: &wrappers.UInt32Value{Value: 0},
		Size:   &wrappers.UInt32Value{Value: 0},
	}
}

// NewAuthBatchQueryResponseWithMsg 创建带详细信息的批量查询回复
func NewAuthBatchQueryResponseWithMsg(code apimodel.Code, msg string) *apiservice.BatchQueryResponse {
	resp := NewAuthBatchQueryResponse(code)
	resp.Info.Value += ": " + msg
	return resp
}

// NewUserResponse 创建回复带用户信息
func NewUserResponse(code apimodel.Code, user *apisecurity.User) *apiservice.Response {
	return &apiservice.Response{
		Code: &wrappers.UInt32Value{Value: uint32(code)},
		Info: &wrappers.StringValue{Value: code2info[uint32(code)]},
		User: user,
	}
}

// NewUserResponse 创建回复带用户信息
func NewUserResponseWithMsg(code apimodel.Code, info string, user *apisecurity.User) *apiservice.Response {
	return &apiservice.Response{
		Code: &wrappers.UInt32Value{Value: uint32(code)},
		Info: &wrappers.StringValue{Value: info},
		User: user,
	}
}

// NewGroupResponse 创建回复带用户组信息
func NewGroupResponse(code apimodel.Code, user *apisecurity.UserGroup) *apiservice.Response {
	return &apiservice.Response{
		Code:      &wrappers.UInt32Value{Value: uint32(code)},
		Info:      &wrappers.StringValue{Value: code2info[uint32(code)]},
		UserGroup: user,
	}
}

// NewModifyGroupResponse 创建修改用户组的响应信息
func NewModifyGroupResponse(code apimodel.Code, group *apisecurity.ModifyUserGroup) *apiservice.Response {
	return &apiservice.Response{
		Code:            &wrappers.UInt32Value{Value: uint32(code)},
		Info:            &wrappers.StringValue{Value: code2info[uint32(code)]},
		ModifyUserGroup: group,
	}
}

// NewGroupRelationResponse 创建用户组关联关系的响应体
func NewGroupRelationResponse(code apimodel.Code, relation *apisecurity.UserGroupRelation) *apiservice.Response {
	return &apiservice.Response{
		Code:     &wrappers.UInt32Value{Value: uint32(code)},
		Info:     &wrappers.StringValue{Value: code2info[uint32(code)]},
		Relation: relation,
	}
}

// NewAuthStrategyResponse 创建鉴权策略响应体
func NewAuthStrategyResponse(code apimodel.Code, req *apisecurity.AuthStrategy) *apiservice.Response {
	return &apiservice.Response{
		Code:         &wrappers.UInt32Value{Value: uint32(code)},
		Info:         &wrappers.StringValue{Value: code2info[uint32(code)]},
		AuthStrategy: req,
	}
}

// NewAuthStrategyResponseWithMsg 创建鉴权策略响应体并自定义Info
func NewAuthStrategyResponseWithMsg(
	code apimodel.Code, msg string, req *apisecurity.AuthStrategy) *apiservice.Response {
	return &apiservice.Response{
		Code:         &wrappers.UInt32Value{Value: uint32(code)},
		Info:         &wrappers.StringValue{Value: msg},
		AuthStrategy: req,
	}
}

// NewModifyAuthStrategyResponse 创建修改鉴权策略响应体
func NewModifyAuthStrategyResponse(code apimodel.Code, req *apisecurity.ModifyAuthStrategy) *apiservice.Response {
	return &apiservice.Response{
		Code:               &wrappers.UInt32Value{Value: uint32(code)},
		Info:               &wrappers.StringValue{Value: code2info[uint32(code)]},
		ModifyAuthStrategy: req,
	}
}

// NewStrategyResourcesResponse 创建修改鉴权策略响应体
func NewStrategyResourcesResponse(code apimodel.Code, ret *apisecurity.StrategyResources) *apiservice.Response {
	return &apiservice.Response{
		Code:      &wrappers.UInt32Value{Value: uint32(code)},
		Info:      &wrappers.StringValue{Value: code2info[uint32(code)]},
		Resources: ret,
	}
}

// NewLoginResponse 创建登录响应体
func NewLoginResponse(code apimodel.Code, loginResponse *apisecurity.LoginResponse) *apiservice.Response {
	return &apiservice.Response{
		Code:          &wrappers.UInt32Value{Value: uint32(code)},
		Info:          &wrappers.StringValue{Value: code2info[uint32(code)]},
		LoginResponse: loginResponse,
	}
}