feat: New compose strategies for ES256

ory · Jun 25, 2020 · 3280e77 · 3280e77
1 parent 901e206
commit 3280e77
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/compose/compose_strategy.go b/compose/compose_strategy.go
@@ -22,6 +22,7 @@
 package compose
 
 import (
+	"crypto/ecdsa"
 	"crypto/rsa"
 
 	"github.com/ory/fosite/handler/oauth2"
@@ -58,6 +59,15 @@ func NewOAuth2JWTStrategy(key *rsa.PrivateKey, strategy *oauth2.HMACSHAStrategy)
 	}
 }
 
+func NewOAuth2JWTECDSAStrategy(key *ecdsa.PrivateKey, strategy *oauth2.HMACSHAStrategy) *oauth2.DefaultJWTStrategy {
+	return &oauth2.DefaultJWTStrategy{
+		JWTStrategy: &jwt.ES256JWTStrategy{
+			PrivateKey: key,
+		},
+		HMACSHAStrategy: strategy,
+	}
+}
+
 func NewOAuth2JWTStrategyWithIssuer(key *rsa.PrivateKey, strategy *oauth2.HMACSHAStrategy, issuer string) *oauth2.DefaultJWTStrategy {
 	return &oauth2.DefaultJWTStrategy{
 		JWTStrategy: &jwt.RS256JWTStrategy{
@@ -68,6 +78,16 @@ func NewOAuth2JWTStrategyWithIssuer(key *rsa.PrivateKey, strategy *oauth2.HMACSH
 	}
 }
 
+func NewOAuth2JWTECDSAStrategyWithIssuer(key *ecdsa.PrivateKey, strategy *oauth2.HMACSHAStrategy, issuer string) *oauth2.DefaultJWTStrategy {
+	return &oauth2.DefaultJWTStrategy{
+		JWTStrategy: &jwt.ES256JWTStrategy{
+			PrivateKey: key,
+		},
+		HMACSHAStrategy: strategy,
+		Issuer: issuer,
+	}
+}
+
 func NewOpenIDConnectStrategy(config *Config, key *rsa.PrivateKey) *openid.DefaultStrategy {
 	return &openid.DefaultStrategy{
 		JWTStrategy: &jwt.RS256JWTStrategy{
@@ -77,3 +97,13 @@ func NewOpenIDConnectStrategy(config *Config, key *rsa.PrivateKey) *openid.Defau
 		Issuer: config.IDTokenIssuer,
 	}
 }
+
+func NewOpenIDConnectECDSAStrategy(config *Config, key *ecdsa.PrivateKey) *openid.DefaultStrategy {
+	return &openid.DefaultStrategy{
+		JWTStrategy: &jwt.ES256JWTStrategy{
+			PrivateKey: key,
+		},
+		Expiry: config.GetIDTokenLifespan(),
+		Issuer: config.IDTokenIssuer,
+	}
+}