Add OAuth client branding for trust centers

Serve CIMD logos from dedicated endpoints, expose client metadata
branding through GraphQL, and resolve branding from OAuth clients.

Signed-off-by: Bryan Frimin <bryan@probo.com>
This commit is contained in:
Bryan Frimin
2026-07-15 17:33:07 +02:00
parent 01acda1d84
commit d447fa295f
12 changed files with 441 additions and 24 deletions

View File

@@ -526,3 +526,29 @@ func (s *Service) validateCIMDScopes(scopes coredata.OAuth2Scopes) error {
return nil
}
func (s *Service) CIMDClientMetadata(ctx context.Context, clientIDRaw string) (*ClientMetadataDocument, error) {
if !IsCIMDClientID(clientIDRaw) {
return nil, nil
}
doc, err := s.cimd.fetch(ctx, clientIDRaw)
if err != nil {
return nil, fmt.Errorf("cannot fetch cimd client metadata: %w", err)
}
return doc, nil
}
func (s *Service) CIMDClientDisplayName(ctx context.Context, clientIDRaw string) (*string, error) {
doc, err := s.CIMDClientMetadata(ctx, clientIDRaw)
if err != nil {
return nil, err
}
if doc == nil || doc.ClientName == "" {
return nil, nil
}
return &doc.ClientName, nil
}

View File

@@ -0,0 +1,67 @@
// Copyright (c) 2026 Probo Inc <hello@probo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package oauth2
import (
"context"
"errors"
"go.probo.inc/probo/pkg/coredata"
)
type ClientBranding struct {
Name string
ClientURL *string
LogoURL *string
}
func (s *Service) ClientBranding(ctx context.Context, clientIDRaw string) (*ClientBranding, error) {
if clientIDRaw == "" {
return nil, nil
}
client, err := s.resolveClient(ctx, nil, clientIDRaw)
if err != nil {
if _, ok := errors.AsType[*OAuth2Error](err); ok {
return nil, nil
}
return nil, err
}
return ClientBrandingFromClient(client), nil
}
func ClientBrandingFromClient(client *coredata.OAuth2Client) *ClientBranding {
if client == nil || client.ClientName == "" {
return nil
}
branding := &ClientBranding{
Name: client.ClientName,
}
if client.ClientURI != nil {
clientURL := client.ClientURI.String()
branding.ClientURL = &clientURL
}
if client.LogoURI != nil {
logoURL := client.LogoURI.String()
branding.LogoURL = &logoURL
}
return branding
}

View File

@@ -0,0 +1,66 @@
// Copyright (c) 2026 Probo Inc <hello@probo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package oauth2
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"go.gearno.de/kit/log"
"go.probo.inc/probo/pkg/coredata"
"go.probo.inc/probo/pkg/uri"
)
func TestClientBrandingFromClient(t *testing.T) {
t.Parallel()
clientURL := uri.URI("https://example.com")
logoURL := uri.URI("https://example.com/logo.png")
branding := ClientBrandingFromClient(
&coredata.OAuth2Client{
ClientName: "Acme",
ClientURI: &clientURL,
LogoURI: &logoURL,
},
)
require.NotNil(t, branding)
assert.Equal(t, "Acme", branding.Name)
assert.Equal(t, "https://example.com", *branding.ClientURL)
assert.Equal(t, "https://example.com/logo.png", *branding.LogoURL)
}
func TestClientBranding_EmptyClientID(t *testing.T) {
t.Parallel()
svc := NewService(nil, nil, "", log.NewLogger())
branding, err := svc.ClientBranding(context.Background(), "")
require.NoError(t, err)
assert.Nil(t, branding)
}
func TestClientBranding_InvalidClientID(t *testing.T) {
t.Parallel()
svc := NewService(nil, nil, "", log.NewLogger())
branding, err := svc.ClientBranding(context.Background(), "not-a-cimd-client")
require.NoError(t, err)
assert.Nil(t, branding)
}