ea4a6301ea
Define uuid_generate_v7() in the first migration and switch all table primary key defaults to it. Remove the uuid-ossp extension dependency. Update refresh token and request ID generation in Go code to use uuid.NewV7() from the existing google/uuid v1.6.0 library. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type JWTManager struct {
|
|
secret []byte
|
|
accessDuration time.Duration
|
|
refreshDuration time.Duration
|
|
}
|
|
|
|
type Claims struct {
|
|
UserID string `json:"user_id"`
|
|
Plan string `json:"plan"`
|
|
jwt.RegisteredClaims
|
|
}
|
|
|
|
func NewJWTManager(secret string, accessDuration, refreshDuration time.Duration) *JWTManager {
|
|
return &JWTManager{
|
|
secret: []byte(secret),
|
|
accessDuration: accessDuration,
|
|
refreshDuration: refreshDuration,
|
|
}
|
|
}
|
|
|
|
func (j *JWTManager) GenerateAccessToken(userID, plan string) (string, error) {
|
|
claims := Claims{
|
|
UserID: userID,
|
|
Plan: plan,
|
|
RegisteredClaims: jwt.RegisteredClaims{
|
|
ExpiresAt: jwt.NewNumericDate(time.Now().Add(j.accessDuration)),
|
|
IssuedAt: jwt.NewNumericDate(time.Now()),
|
|
},
|
|
}
|
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
|
return token.SignedString(j.secret)
|
|
}
|
|
|
|
func (j *JWTManager) GenerateRefreshToken() (string, time.Time) {
|
|
token := uuid.Must(uuid.NewV7()).String()
|
|
expiresAt := time.Now().Add(j.refreshDuration)
|
|
return token, expiresAt
|
|
}
|
|
|
|
func (j *JWTManager) ValidateAccessToken(tokenStr string) (*Claims, error) {
|
|
token, err := jwt.ParseWithClaims(tokenStr, &Claims{}, func(t *jwt.Token) (interface{}, error) {
|
|
if _, ok := t.Method.(*jwt.SigningMethodHMAC); !ok {
|
|
return nil, fmt.Errorf("unexpected signing method: %v", t.Header["alg"])
|
|
}
|
|
return j.secret, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
claims, ok := token.Claims.(*Claims)
|
|
if !ok || !token.Valid {
|
|
return nil, fmt.Errorf("invalid token")
|
|
}
|
|
return claims, nil
|
|
}
|
|
|
|
func (j *JWTManager) AccessDuration() time.Duration {
|
|
return j.accessDuration
|
|
}
|