llvy.ltd/casdoor
1
0
Fork 0
mirror of https://github.com/casdoor/casdoor.git synced 2025-05-23 02:35:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor GenerateCasToken()

Browse Source
This commit is contained in:
Yang Luo 2023-09-06 18:35:13 +08:00
parent a9de7d3aef
commit 3c2fd574a6
1 changed files with 42 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
object/token_cas.go
View File

@ -185,9 +185,14 @@ func StoreCasTokenForProxyTicket(token *CasAuthenticationSuccess, targetService,
} }
func GenerateCasToken(userId string, service string) (string, error) { func GenerateCasToken(userId string, service string) (string, error) {
if user, err := GetUser(userId); err != nil { user, err := GetUser(userId)
if err != nil {
return "", err return "", err
} else if user != nil { }
if user == nil {
return "", fmt.Errorf("The user: %s doesn't exist", userId)
}
authenticationSuccess := CasAuthenticationSuccess{ authenticationSuccess := CasAuthenticationSuccess{
User: user.Name, User: user.Name,
Attributes: &CasAttributes{ Attributes: &CasAttributes{
@ -196,9 +201,18 @@ func GenerateCasToken(userId string, service string) (string, error) {
}, },
ProxyGrantingTicket: fmt.Sprintf("PGTIOU-%s", util.GenerateId()), ProxyGrantingTicket: fmt.Sprintf("PGTIOU-%s", util.GenerateId()),
} }
data, _ := json.Marshal(user)
data, err := json.Marshal(user)
if err != nil {
return "", err
}
tmp := map[string]string{} tmp := map[string]string{}
json.Unmarshal(data, &tmp) err = json.Unmarshal(data, &tmp)
if err != nil {
return "", err
}
for k, v := range tmp { for k, v := range tmp {
if v != "" { if v != "" {
authenticationSuccess.Attributes.UserAttributes.Attributes = append(authenticationSuccess.Attributes.UserAttributes.Attributes, &CasNamedAttribute{ authenticationSuccess.Attributes.UserAttributes.Attributes = append(authenticationSuccess.Attributes.UserAttributes.Attributes, &CasNamedAttribute{
@ -207,6 +221,7 @@ func GenerateCasToken(userId string, service string) (string, error) {
}) })
} }
} }
st := fmt.Sprintf("ST-%d", rand.Int()) st := fmt.Sprintf("ST-%d", rand.Int())
stToServiceResponse.Store(st, &CasAuthenticationSuccessWrapper{ stToServiceResponse.Store(st, &CasAuthenticationSuccessWrapper{
AuthenticationSuccess: &authenticationSuccess, AuthenticationSuccess: &authenticationSuccess,
@ -214,9 +229,6 @@ func GenerateCasToken(userId string, service string) (string, error) {
UserId: userId, UserId: userId,
}) })
return st, nil return st, nil
} else {
return "", fmt.Errorf("invalid user Id")
}
} }
// GetValidationBySaml // GetValidationBySaml