generate open token using normal token
This commit is contained in:
@@ -106,6 +106,13 @@ func (m *AuthMiddleware) AuthenticateWithHandler(jwtHandler *jwt.JWTHandler, isO
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if !jwtHandler.IsOpenToken && claims.IsOpenToken {
|
||||||
|
logging.Error("Authentication failed: attempting to authenticate with open token")
|
||||||
|
return ctx.Status(fiber.StatusUnauthorized).JSON(fiber.Map{
|
||||||
|
"error": "Wrong token type used",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
// Additional security: validate user ID format
|
// Additional security: validate user ID format
|
||||||
if claims.UserID == "" || len(claims.UserID) < 10 {
|
if claims.UserID == "" || len(claims.UserID) < 10 {
|
||||||
logging.Error("Authentication failed: invalid user ID in token from IP %s", ip)
|
logging.Error("Authentication failed: invalid user ID in token from IP %s", ip)
|
||||||
|
|||||||
@@ -63,16 +63,11 @@ func (s *MembershipService) Login(ctx context.Context, username, password string
|
|||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
return s.jwtHandler.GenerateToken(user)
|
return s.jwtHandler.GenerateToken(user.ID.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *MembershipService) GenerateOpenToken(ctx context.Context, userId string) (string, error) {
|
func (s *MembershipService) GenerateOpenToken(ctx context.Context, userId string) (string, error) {
|
||||||
user, err := s.repo.GetByID(ctx, userId)
|
return s.openJwtHandler.GenerateToken(userId)
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
return s.openJwtHandler.GenerateToken(user)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// CreateUser creates a new user.
|
// CreateUser creates a new user.
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
Version = "0.10.5"
|
Version = "0.10.6"
|
||||||
Prefix = "v1"
|
Prefix = "v1"
|
||||||
Secret string
|
Secret string
|
||||||
SecretCode string
|
SecretCode string
|
||||||
|
|||||||
@@ -13,7 +13,8 @@ import (
|
|||||||
|
|
||||||
// Claims represents the JWT claims.
|
// Claims represents the JWT claims.
|
||||||
type Claims struct {
|
type Claims struct {
|
||||||
UserID string `json:"user_id"`
|
UserID string `json:"user_id"`
|
||||||
|
IsOpenToken bool `json:"is_open_token"`
|
||||||
jwt.RegisteredClaims
|
jwt.RegisteredClaims
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -70,13 +71,14 @@ func (jh *JWTHandler) GenerateSecretKey() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// GenerateToken generates a new JWT for a given user.
|
// GenerateToken generates a new JWT for a given user.
|
||||||
func (jh *JWTHandler) GenerateToken(user *model.User) (string, error) {
|
func (jh *JWTHandler) GenerateToken(userId string) (string, error) {
|
||||||
expirationTime := time.Now().Add(24 * time.Hour)
|
expirationTime := time.Now().Add(24 * time.Hour)
|
||||||
claims := &Claims{
|
claims := &Claims{
|
||||||
UserID: user.ID.String(),
|
UserID: userId,
|
||||||
RegisteredClaims: jwt.RegisteredClaims{
|
RegisteredClaims: jwt.RegisteredClaims{
|
||||||
ExpiresAt: jwt.NewNumericDate(expirationTime),
|
ExpiresAt: jwt.NewNumericDate(expirationTime),
|
||||||
},
|
},
|
||||||
|
IsOpenToken: jh.IsOpenToken,
|
||||||
}
|
}
|
||||||
|
|
||||||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||||||
@@ -90,6 +92,7 @@ func (jh *JWTHandler) GenerateTokenWithExpiry(user *model.User, expiry time.Time
|
|||||||
RegisteredClaims: jwt.RegisteredClaims{
|
RegisteredClaims: jwt.RegisteredClaims{
|
||||||
ExpiresAt: jwt.NewNumericDate(expirationTime),
|
ExpiresAt: jwt.NewNumericDate(expirationTime),
|
||||||
},
|
},
|
||||||
|
IsOpenToken: jh.IsOpenToken,
|
||||||
}
|
}
|
||||||
|
|
||||||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ func GenerateTestToken() (string, error) {
|
|||||||
jwtHandler := jwt.NewJWTHandler(testSecret)
|
jwtHandler := jwt.NewJWTHandler(testSecret)
|
||||||
|
|
||||||
// Generate JWT token
|
// Generate JWT token
|
||||||
token, err := jwtHandler.GenerateToken(user)
|
token, err := jwtHandler.GenerateToken(user.ID.String())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", fmt.Errorf("failed to generate test token: %w", err)
|
return "", fmt.Errorf("failed to generate test token: %w", err)
|
||||||
}
|
}
|
||||||
@@ -55,7 +55,7 @@ func GenerateTestTokenWithExpiry(expiryTime time.Time) (string, error) {
|
|||||||
testSecret = "test-secret-that-is-at-least-32-bytes-long-for-security"
|
testSecret = "test-secret-that-is-at-least-32-bytes-long-for-security"
|
||||||
}
|
}
|
||||||
jwtHandler := jwt.NewJWTHandler(testSecret)
|
jwtHandler := jwt.NewJWTHandler(testSecret)
|
||||||
|
|
||||||
// Create test user
|
// Create test user
|
||||||
user := &model.User{
|
user := &model.User{
|
||||||
ID: uuid.New(),
|
ID: uuid.New(),
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ func TestJWT_GenerateAndValidateToken(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Test JWT generation
|
// Test JWT generation
|
||||||
token, err := jwtHandler.GenerateToken(user)
|
token, err := jwtHandler.GenerateToken(user.ID.String())
|
||||||
tests.AssertNoError(t, err)
|
tests.AssertNoError(t, err)
|
||||||
tests.AssertNotNil(t, token)
|
tests.AssertNotNil(t, token)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user