124 lines
3.0 KiB
Go
124 lines
3.0 KiB
Go
package account
|
|
|
|
import (
|
|
"crypto/ed25519"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"go.sia.tech/jape"
|
|
|
|
"git.lumeweb.com/LumeWeb/portal/api/router"
|
|
|
|
apiRegistry "git.lumeweb.com/LumeWeb/portal/api/registry"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
)
|
|
|
|
const AUTH_COOKIE_NAME = "auth_token"
|
|
|
|
type JWTPurpose string
|
|
type VerifyTokenFunc func(claim *jwt.RegisteredClaims) error
|
|
|
|
var (
|
|
nopVerifyFunc VerifyTokenFunc = func(claim *jwt.RegisteredClaims) error {
|
|
return nil
|
|
}
|
|
|
|
ErrJWTUnexpectedClaimsType = errors.New("unexpected claims type")
|
|
ErrJWTUnexpectedIssuer = errors.New("unexpected issuer")
|
|
ErrJWTInvalid = errors.New("invalid JWT")
|
|
)
|
|
|
|
const (
|
|
JWTPurposeLogin JWTPurpose = "login"
|
|
JWTPurpose2FA JWTPurpose = "2fa"
|
|
JWTPurposeNone JWTPurpose = ""
|
|
)
|
|
|
|
func JWTGenerateToken(domain string, privateKey ed25519.PrivateKey, userID uint, purpose JWTPurpose) (string, error) {
|
|
return JWTGenerateTokenWithDuration(domain, privateKey, userID, time.Hour*24, purpose)
|
|
}
|
|
|
|
func JWTGenerateTokenWithDuration(domain string, privateKey ed25519.PrivateKey, userID uint, duration time.Duration, purpose JWTPurpose) (string, error) {
|
|
|
|
// Define the claims
|
|
claims := jwt.RegisteredClaims{
|
|
Issuer: domain,
|
|
Subject: strconv.Itoa(int(userID)),
|
|
ExpiresAt: jwt.NewNumericDate(time.Now().Add(duration)),
|
|
IssuedAt: jwt.NewNumericDate(time.Now()),
|
|
Audience: []string{string(purpose)},
|
|
}
|
|
|
|
// Create the token
|
|
token := jwt.NewWithClaims(jwt.SigningMethodEdDSA, claims)
|
|
|
|
// Sign the token with the Ed25519 private key
|
|
tokenString, err := token.SignedString(privateKey)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return tokenString, nil
|
|
}
|
|
|
|
func JWTVerifyToken(token string, domain string, privateKey ed25519.PrivateKey, verifyFunc VerifyTokenFunc) (*jwt.RegisteredClaims, error) {
|
|
validatedToken, err := jwt.ParseWithClaims(token, &jwt.RegisteredClaims{}, func(token *jwt.Token) (interface{}, error) {
|
|
if _, ok := token.Method.(*jwt.SigningMethodEd25519); !ok {
|
|
return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"])
|
|
}
|
|
|
|
publicKey := privateKey.Public()
|
|
|
|
return publicKey, nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if verifyFunc == nil {
|
|
verifyFunc = nopVerifyFunc
|
|
}
|
|
|
|
claim, ok := validatedToken.Claims.(*jwt.RegisteredClaims)
|
|
|
|
if !ok {
|
|
return nil, fmt.Errorf("%w: %s", ErrJWTUnexpectedClaimsType, validatedToken.Claims)
|
|
}
|
|
|
|
if domain != claim.Issuer {
|
|
return nil, fmt.Errorf("%w: %s", ErrJWTUnexpectedIssuer, claim.Issuer)
|
|
}
|
|
|
|
err = verifyFunc(claim)
|
|
|
|
return claim, err
|
|
}
|
|
|
|
func SetAuthCookie(jc jape.Context, apiName string) {
|
|
for name, api := range apiRegistry.GetAllAPIs() {
|
|
routeableApi, ok := api.(router.RoutableAPI)
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if len(apiName) > 0 && apiName != name {
|
|
continue
|
|
}
|
|
|
|
http.SetCookie(jc.ResponseWriter, &http.Cookie{
|
|
Name: name,
|
|
Value: routeableApi.AuthTokenName(),
|
|
Expires: time.Now().Add(24 * time.Hour),
|
|
Secure: true,
|
|
HttpOnly: true,
|
|
Path: "/",
|
|
Domain: routeableApi.Domain(),
|
|
})
|
|
}
|
|
}
|