88 lines
2.8 KiB
Go
88 lines
2.8 KiB
Go
// Package openidentity 解析开放 IM 场景下的外部访客身份(HTTP Header / Query),与 JSON 请求体 DTO 解耦。
|
||
package openidentity
|
||
|
||
import (
|
||
"cs-agent/internal/pkg/enums"
|
||
"cs-agent/internal/pkg/errorsx"
|
||
"net/url"
|
||
"strings"
|
||
|
||
"github.com/kataras/iris/v12"
|
||
"github.com/mlogclub/simple/common/strs"
|
||
"github.com/mlogclub/simple/web/params"
|
||
)
|
||
|
||
// ExternalInfo 外部访客身份(IM 客户),与站内 AuthPrincipal 区分。
|
||
type ExternalInfo struct {
|
||
ExternalSource enums.ExternalSource `json:"externalSource"`
|
||
ExternalType enums.ExternalType `json:"externalType"`
|
||
ExternalID string `json:"externalId"`
|
||
ExternalName string `json:"externalName"`
|
||
}
|
||
|
||
// GetExternalInfo 从 Header(X-External-*)或 query(externalSource、externalId、externalName)解析身份。
|
||
func GetExternalInfo(ctx iris.Context) (*ExternalInfo, error) {
|
||
externalSource, err := parseExternalSource(ctx)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if !enums.IsAllowedOpenImExternalSource(externalSource) {
|
||
return nil, errorsx.InvalidParam("不支持的外部来源")
|
||
}
|
||
externalID, err := parseExternalID(ctx)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &ExternalInfo{
|
||
ExternalSource: externalSource,
|
||
// TODO: 对接业务系统后,根据业务系统用户信息识别 user;未对接时统一按访客处理。
|
||
ExternalType: enums.ExternalTypeGuest,
|
||
ExternalID: externalID,
|
||
ExternalName: parseExternalName(ctx),
|
||
}, nil
|
||
}
|
||
|
||
func parseExternalSource(ctx iris.Context) (enums.ExternalSource, error) {
|
||
externalSource := ctx.GetHeader("X-External-Source")
|
||
if strs.IsBlank(externalSource) {
|
||
externalSource, _ = params.Get(ctx, "externalSource")
|
||
}
|
||
if strs.IsBlank(externalSource) {
|
||
return "", errorsx.Unauthorized("用户来源不能为空")
|
||
}
|
||
return enums.ExternalSource(strings.TrimSpace(externalSource)), nil
|
||
}
|
||
|
||
func parseExternalID(ctx iris.Context) (string, error) {
|
||
externalID := ctx.GetHeader("X-External-Id")
|
||
if strs.IsBlank(externalID) {
|
||
externalID, _ = params.Get(ctx, "externalId")
|
||
}
|
||
if strs.IsBlank(externalID) {
|
||
return "", errorsx.Unauthorized("用户标识不能为空")
|
||
}
|
||
return strings.TrimSpace(externalID), nil
|
||
}
|
||
|
||
func parseExternalName(ctx iris.Context) string {
|
||
externalName := ctx.GetHeader("X-External-Name")
|
||
if strs.IsBlank(externalName) {
|
||
externalName, _ = params.Get(ctx, "externalName")
|
||
}
|
||
return decodeExternalDisplayName(externalName)
|
||
}
|
||
|
||
// decodeExternalDisplayName 将客户端对 X-External-Name / externalName 做的 encodeURIComponent 还原为 UTF-8。
|
||
// 无百分号编码时 QueryUnescape 原样返回,解码失败则保留原串(兼容异常或旧客户端明文)。
|
||
func decodeExternalDisplayName(s string) string {
|
||
s = strings.TrimSpace(s)
|
||
if s == "" {
|
||
return ""
|
||
}
|
||
dec, err := url.QueryUnescape(s)
|
||
if err != nil {
|
||
return s
|
||
}
|
||
return strings.TrimSpace(dec)
|
||
}
|