···
+
"github.com/lestrrat-go/jwx/v2/jwa"
+
"github.com/lestrrat-go/jwx/v2/jwk"
+
"github.com/lestrrat-go/jwx/v2/jwt"
+
"tangled.sh/tangled.sh/core/spindle/models"
+
const JWKSPath = "/.well-known/jwks.json"
+
// OidcKeyPair represents an OIDC key pair with both private and public keys
+
type OidcKeyPair struct {
+
privateKey *ecdsa.PrivateKey
+
publicKey *ecdsa.PublicKey
+
// OidcTokenGenerator handles OIDC token generation and key management with rotation
+
type OidcTokenGenerator struct {
+
currentKeyPair *OidcKeyPair
+
nextKeyPair *OidcKeyPair
+
// NewOidcTokenGenerator creates a new OIDC token generator with in-memory key management
+
func NewOidcTokenGenerator(issuer string) (*OidcTokenGenerator, error) {
+
currentKeyPair, err := NewOidcKeyPair()
+
return nil, fmt.Errorf("failed to generate initial current key pair: %w", err)
+
return &OidcTokenGenerator{
+
currentKeyPair: currentKeyPair,
+
// NewOidcKeyPair generates a new ECDSA key pair for OIDC token signing
+
func NewOidcKeyPair() (*OidcKeyPair, error) {
+
privKey, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
+
return nil, fmt.Errorf("failed to generate ECDSA key: %w", err)
+
keyID := fmt.Sprintf("spindle-%d", time.Now().Unix())
+
// Create JWK from the private key
+
jwkKey, err := jwk.FromRaw(privKey)
+
return nil, fmt.Errorf("failed to create JWK from private key: %w", err)
+
if err := jwkKey.Set(jwk.KeyIDKey, keyID); err != nil {
+
return nil, fmt.Errorf("failed to set key ID: %w", err)
+
if err := jwkKey.Set(jwk.AlgorithmKey, jwa.ES256); err != nil {
+
return nil, fmt.Errorf("failed to set algorithm: %w", err)
+
if err := jwkKey.Set(jwk.KeyUsageKey, "sig"); err != nil {
+
return nil, fmt.Errorf("failed to set key usage: %w", err)
+
publicKey: &privKey.PublicKey,
+
// LoadOidcKeyPair loads an existing key pair from JWK JSON
+
func LoadOidcKeyPair(jwkJSON []byte) (*OidcKeyPair, error) {
+
jwkKey, err := jwk.ParseKey(jwkJSON)
+
return nil, fmt.Errorf("failed to parse JWK: %w", err)
+
var privKey *ecdsa.PrivateKey
+
if err := jwkKey.Raw(&privKey); err != nil {
+
return nil, fmt.Errorf("failed to extract private key: %w", err)
+
keyID, ok := jwkKey.Get(jwk.KeyIDKey)
+
return nil, fmt.Errorf("JWK missing key ID")
+
keyIDStr, ok := keyID.(string)
+
return nil, fmt.Errorf("JWK key ID is not a string")
+
publicKey: &privKey.PublicKey,
+
// GetKeyID returns the key ID
+
func (k *OidcKeyPair) GetKeyID() string {
+
// RotateKeys performs key rotation: generates new next key, moves next to current
+
func (g *OidcTokenGenerator) RotateKeys() error {
+
// Generate a new key pair for the next key
+
newNextKeyPair, err := NewOidcKeyPair()
+
return fmt.Errorf("failed to generate new next key pair: %w", err)
+
// Perform rotation: next becomes current, new key becomes next
+
g.currentKeyPair = g.nextKeyPair
+
g.nextKeyPair = newNextKeyPair
+
// If we don't have a current key (first time setup), use the new key
+
if g.currentKeyPair == nil {
+
g.currentKeyPair = newNextKeyPair
+
// Generate another new key for next
+
g.nextKeyPair, err = NewOidcKeyPair()
+
return fmt.Errorf("failed to generate next key pair for first setup: %w", err)
+
func (g *OidcTokenGenerator) GetCurrentKeyID() string {
+
if g.currentKeyPair == nil {
+
return g.currentKeyPair.GetKeyID()
+
// GetNextKeyID returns the next key's ID
+
func (g *OidcTokenGenerator) GetNextKeyID() string {
+
if g.nextKeyPair == nil {
+
return g.nextKeyPair.GetKeyID()
+
// HasKeys returns true if the generator has at least a current key
+
func (g *OidcTokenGenerator) HasKeys() bool {
+
return g.currentKeyPair != nil
+
// OidcClaims represents the claims in an OIDC token
+
type OidcClaims struct {
+
Issuer string `json:"iss"`
+
Subject string `json:"sub"`
+
Audience string `json:"aud"`
+
ExpiresAt int64 `json:"exp"`
+
NotBefore int64 `json:"nbf"`
+
IssuedAt int64 `json:"iat"`
+
JWTID string `json:"jti"`
+
// CreateToken creates a signed JWT token for the given OidcToken and pipeline context
+
func (g *OidcTokenGenerator) CreateToken(
+
oidcToken models.OidcToken,
+
pipelineId models.PipelineId,
+
repoOwner, repoName string,
+
exp := now.Add(5 * time.Minute)
+
// Determine audience - use the provided audience or default to issuer
+
audience := fmt.Sprintf(g.issuer)
+
if oidcToken.Aud != nil && *oidcToken.Aud != "" {
+
audience = *oidcToken.Aud
+
pipelineUri := pipelineId.AtUri()
+
// Hardcode the did as did:web of the issuer. At some point knots will have their own DIDs which will be used here
+
Subject: pipelineUri.String(),
+
// Repo owner, name, and id should be global unique but we add timestamp to ensure uniqueness
+
JWTID: fmt.Sprintf("%s/%s-%s-%d", repoOwner, repoName, pipelineUri.RecordKey(), now.Unix()),
+
if err := token.Set(jwt.IssuerKey, claims.Issuer); err != nil {
+
return "", fmt.Errorf("failed to set issuer: %w", err)
+
if err := token.Set(jwt.SubjectKey, claims.Subject); err != nil {
+
return "", fmt.Errorf("failed to set subject: %w", err)
+
if err := token.Set(jwt.AudienceKey, claims.Audience); err != nil {
+
return "", fmt.Errorf("failed to set audience: %w", err)
+
if err := token.Set(jwt.ExpirationKey, claims.ExpiresAt); err != nil {
+
return "", fmt.Errorf("failed to set expiration: %w", err)
+
if err := token.Set(jwt.NotBeforeKey, claims.NotBefore); err != nil {
+
return "", fmt.Errorf("failed to set not before: %w", err)
+
if err := token.Set(jwt.IssuedAtKey, claims.IssuedAt); err != nil {
+
return "", fmt.Errorf("failed to set issued at: %w", err)
+
if err := token.Set(jwt.JwtIDKey, claims.JWTID); err != nil {
+
return "", fmt.Errorf("failed to set JWT ID: %w", err)
+
// Sign the token with the current key
+
if g.currentKeyPair == nil {
+
return "", fmt.Errorf("no current key pair available for signing")
+
signedToken, err := jwt.Sign(token, jwt.WithKey(jwa.ES256, g.currentKeyPair.jwkKey))
+
return "", fmt.Errorf("failed to sign token: %w", err)
+
return string(signedToken), nil
+
// JWKSHandler serves the JWKS endpoint as an HTTP handler
+
func (g *OidcTokenGenerator) JWKSHandler(w http.ResponseWriter, r *http.Request) {
+
// Add current key if available
+
if g.currentKeyPair != nil {
+
pubJWK, err := jwk.PublicKeyOf(g.currentKeyPair.jwkKey)
+
http.Error(w, fmt.Sprintf("failed to extract current public key from JWK: %v", err), http.StatusInternalServerError)
+
keys = append(keys, pubJWK)
+
// Add next key if available
+
if g.nextKeyPair != nil {
+
pubJWK, err := jwk.PublicKeyOf(g.nextKeyPair.jwkKey)
+
http.Error(w, fmt.Sprintf("failed to extract next public key from JWK: %v", err), http.StatusInternalServerError)
+
keys = append(keys, pubJWK)
+
http.Error(w, "no keys available for JWKS", http.StatusInternalServerError)
+
jwks := map[string]interface{}{
+
w.Header().Set("Content-Type", "application/json")
+
if err := json.NewEncoder(w).Encode(jwks); err != nil {
+
http.Error(w, fmt.Sprintf("failed to encode JWKS: %v", err), http.StatusInternalServerError)
+
// DiscoveryHandler serves the OIDC discovery endpoint for JWKS
+
func (g *OidcTokenGenerator) DiscoveryHandler(w http.ResponseWriter, r *http.Request) {
+
claimsSupported := []string{
+
responseTypesSupported := []string{
+
subjectTypesSupported := []string{
+
idTokenSigningAlgValuesSupported := []string{
+
scopesSupported := []string{
+
discovery := map[string]interface{}{
+
"jwks_uri": fmt.Sprintf("%s%s", g.issuer, JWKSPath),
+
"claims_supported": claimsSupported,
+
"response_types_supported": responseTypesSupported,
+
"subject_types_supported": subjectTypesSupported,
+
"id_token_signing_alg_values_supported": idTokenSigningAlgValuesSupported,
+
"scopes_supported": scopesSupported,
+
w.Header().Set("Content-Type", "application/json")
+
if err := json.NewEncoder(w).Encode(discovery); err != nil {
+
http.Error(w, fmt.Sprintf("failed to encode discovery document: %v", err), http.StatusInternalServerError)