module JSON::JOSE

def with_jwk_support(key)
  case key
  when JSON::JWK
    key.to_key
  when JSON::JWK::Set
    key[kid]&.to_key or raise JWK::Set::KidNotFound
  else
    key
  end
end