Files
huly-platform/foundations/stream/internal/pkg/token/token.go
T
Denis Bykhov 6e77cbb762 Add 'foundations/stream/' from commit 'b7b6930b0992ae48f08d37745d5e6e5e26517337'
git-subtree-dir: foundations/stream
git-subtree-mainline: 328a7dc269
git-subtree-split: b7b6930b09
2025-11-26 23:06:55 +05:00

75 lines
2.1 KiB
Go

// Copyright © 2025 Hardcore Engineering Inc.
//
// Licensed under the Eclipse Public 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 https://www.eclipse.org/legal/epl-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 token provides functions to work with platform tokens
package token
import (
"fmt"
"github.com/golang-jwt/jwt/v5"
"github.com/google/uuid"
)
// Token represents Claims for the platform token
type Token struct {
jwt.MapClaims
Account string `json:"account"`
Workspace string `json:"workspace,omitempty"`
Extra map[string]any `json:"extra,omitempty"`
}
// NewToken creates a new platform token
func NewToken(serverSecret, workspace, service string) (string, error) {
var res = Token{
Account: uuid.NewString(),
Workspace: workspace,
Extra: map[string]any{
"service": service,
},
}
return res.Encode(serverSecret)
}
// Encode signes a token object and converts it to string
func (t *Token) Encode(serverSecret string) (string, error) {
var token = jwt.NewWithClaims(jwt.SigningMethodHS256, t)
var tokenString, err = token.SignedString([]byte(serverSecret))
if err != nil {
return "", err
}
return tokenString, nil
}
// Decode decodes a token by a passed configuration
func Decode(secretKey, tokenString string) (*Token, error) {
token, err := jwt.ParseWithClaims(tokenString, &Token{}, func(token *jwt.Token) (any, error) {
if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"])
}
return []byte(secretKey), nil
})
if err != nil {
return nil, fmt.Errorf("failed to parse token: %w", err)
}
if claims, ok := token.Claims.(*Token); ok && token.Valid {
return claims, nil
}
return nil, fmt.Errorf("invalid token")
}