databricks-cli/libs/auth/service_principal.go

package auth

import (
	"context"

	"github.com/databricks/databricks-sdk-go"
	"github.com/databricks/databricks-sdk-go/apierr"
)

// Determines whether a given user id is a service principal.
// This function uses a heuristic: if no user exists with this id, we assume
// it's a service principal. Unfortunately, the standard service principal API is too
// slow for our purposes.
func IsServicePrincipal(ctx context.Context, ws *databricks.WorkspaceClient, userId string) (bool, error) {
	_, err := ws.Users.GetById(ctx, userId)
	if apierr.IsMissing(err) {
		return true, nil
	}
	return false, err
}