mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
95889cf576
* fix(api): use organization instead of organisation * fix test * docs: add deprecation notice * remove validation
80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/zitadel/zitadel/internal/api/authz"
|
|
grpc_util "github.com/zitadel/zitadel/internal/api/grpc"
|
|
"github.com/zitadel/zitadel/internal/api/http"
|
|
"github.com/zitadel/zitadel/internal/telemetry/tracing"
|
|
)
|
|
|
|
func AuthorizationInterceptor(verifier *authz.TokenVerifier, authConfig authz.Config) grpc.UnaryServerInterceptor {
|
|
return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
|
|
return authorize(ctx, req, info, handler, verifier, authConfig)
|
|
}
|
|
}
|
|
|
|
func authorize(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler, verifier *authz.TokenVerifier, authConfig authz.Config) (_ interface{}, err error) {
|
|
authOpt, needsToken := verifier.CheckAuthMethod(info.FullMethod)
|
|
if !needsToken {
|
|
return handler(ctx, req)
|
|
}
|
|
|
|
authCtx, span := tracing.NewServerInterceptorSpan(ctx)
|
|
defer func() { span.EndWithError(err) }()
|
|
|
|
authToken := grpc_util.GetAuthorizationHeader(authCtx)
|
|
if authToken == "" {
|
|
return nil, status.Error(codes.Unauthenticated, "auth header missing")
|
|
}
|
|
|
|
orgID, orgDomain := orgIDAndDomainFromRequest(authCtx, req)
|
|
ctxSetter, err := authz.CheckUserAuthorization(authCtx, req, authToken, orgID, orgDomain, verifier, authConfig, authOpt, info.FullMethod)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
span.End()
|
|
return handler(ctxSetter(ctx), req)
|
|
}
|
|
|
|
func orgIDAndDomainFromRequest(ctx context.Context, req interface{}) (id, domain string) {
|
|
orgID := grpc_util.GetHeader(ctx, http.ZitadelOrgID)
|
|
o, ok := req.(OrganizationFromRequest)
|
|
if !ok {
|
|
return orgID, ""
|
|
}
|
|
id = o.OrganizationFromRequest().ID
|
|
domain = o.OrganizationFromRequest().Domain
|
|
if id != "" || domain != "" {
|
|
return id, domain
|
|
}
|
|
// check if the deprecated organisation is used.
|
|
// to be removed before going GA (https://github.com/zitadel/zitadel/issues/6718)
|
|
id = o.OrganisationFromRequest().ID
|
|
domain = o.OrganisationFromRequest().Domain
|
|
if id != "" || domain != "" {
|
|
return id, domain
|
|
}
|
|
return orgID, domain
|
|
}
|
|
|
|
// Deprecated: will be removed in favor of OrganizationFromRequest (https://github.com/zitadel/zitadel/issues/6718)
|
|
type OrganisationFromRequest interface {
|
|
OrganisationFromRequest() *Organization
|
|
}
|
|
|
|
type Organization struct {
|
|
ID string
|
|
Domain string
|
|
}
|
|
|
|
type OrganizationFromRequest interface {
|
|
OrganizationFromRequest() *Organization
|
|
OrganisationFromRequest
|
|
}
|