54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
|
package user
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"forge.cadoles.com/Cadoles/emissary/internal/jwk"
|
||
|
|
"github.com/lestrrat-go/jwx/v2/jwa"
|
||
|
|
"github.com/lestrrat-go/jwx/v2/jws"
|
||
|
|
"github.com/lestrrat-go/jwx/v2/jwt"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
const keyRole = "role"
|
||
|
|
|
||
|
|
func parseToken(ctx context.Context, keys jwk.Set, issuer string, rawToken string) (jwt.Token, error) {
|
||
|
|
token, err := jwt.Parse(
|
||
|
|
[]byte(rawToken),
|
||
|
|
jwt.WithKeySet(keys, jws.WithRequireKid(false)),
|
||
|
|
jwt.WithIssuer(issuer),
|
||
|
|
jwt.WithValidate(true),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return token, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func GenerateToken(ctx context.Context, key jwk.Key, role Role) (string, error) {
|
||
|
|
token := jwt.New()
|
||
|
|
|
||
|
|
if err := token.Set(keyRole, role); err != nil {
|
||
|
|
return "", errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
now := time.Now()
|
||
|
|
|
||
|
|
if err := token.Set(jwt.NotBeforeKey, now); err != nil {
|
||
|
|
return "", errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := token.Set(jwt.IssuedAtKey, now); err != nil {
|
||
|
|
return "", errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
rawToken, err := jwt.Sign(token, jwt.WithKey(jwa.RS256, key))
|
||
|
|
if err != nil {
|
||
|
|
return "", errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(rawToken), nil
|
||
|
|
}
|