fix: apply context to GenerateRSA

cmd/vela-server/token.go

@@ -28,7 +28,7 @@ func setupTokenManager(c *cli.Context, db database.Interface) (*token.Manager, e
 	}
 
 	// generate a new RSA key pair
-	err := tm.GenerateRSA(db)
+	err := tm.GenerateRSA(c.Context, db)
 	if err != nil {
 		return nil, err
 	}

internal/token/generate_rsa.go

@@ -14,7 +14,7 @@ import (
 )
 
 // GenerateRSA creates an RSA key pair and sets it in the token manager and saves the JWK in the database.
-func (tm *Manager) GenerateRSA(db database.Interface) error {
+func (tm *Manager) GenerateRSA(ctx context.Context, db database.Interface) error {
 	// generate key pair
 	privateRSAKey, err := rsa.GenerateKey(rand.Reader, 2048)
 	if err != nil {

internal/token/mint.go

@@ -181,14 +181,14 @@ func (tm *Manager) MintIDToken(ctx context.Context, mto *MintTokenOpts, db datab
 	tk := jwt.NewWithClaims(jwt.SigningMethodRS256, claims)
 
 	// verify key is active in the database before signing
-	_, err = db.GetActiveJWK(context.TODO(), tm.RSAKeySet.KID)
+	_, err = db.GetActiveJWK(ctx, tm.RSAKeySet.KID)
 	if err != nil {
 		if !errors.Is(err, gorm.ErrRecordNotFound) {
 			return "", fmt.Errorf("unable to get active public key: %w", err)
 		}
 
 		// generate a new RSA key pair if previous key is inactive (rotated)
-		err = tm.GenerateRSA(db)
+		err = tm.GenerateRSA(ctx, db)
 		if err != nil {
 			return "", fmt.Errorf("unable to generate RSA key pair: %w", err)
 		}