Files
probo/pkg/accessreview/drivers/probo_memberships.go
Ludovic Vielle 8094e7cfd0 Truncate access review roles with badge list
Long role strings in the access review table broke row layout when
drivers joined many roles into one comma-separated value. Expose
roles as a string array in GraphQL by splitting the stored role at
the API layer, and render the first three roles as badges with a
"+X more" popover for the rest.

Closes ENG-459.

Signed-off-by: Ludovic Vielle <ludovic@probo.com>
2026-06-16 11:25:31 +02:00

101 lines
2.8 KiB
Go

// 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 drivers
import (
"context"
"fmt"
"strings"
"go.gearno.de/kit/pg"
"go.probo.inc/probo/pkg/coredata"
"go.probo.inc/probo/pkg/gid"
)
// ProboMembershipsDriver is a built-in identity source that queries
// iam_memberships + identities for the organization. No external
// connector is needed.
type ProboMembershipsDriver struct {
pg *pg.Client
scope coredata.Scoper
organizationID gid.GID
}
func NewProboMembershipsDriver(
pgClient *pg.Client,
scope coredata.Scoper,
organizationID gid.GID,
) *ProboMembershipsDriver {
return &ProboMembershipsDriver{
pg: pgClient,
scope: scope,
organizationID: organizationID,
}
}
func (d *ProboMembershipsDriver) ListAccounts(ctx context.Context) ([]AccountRecord, error) {
var records []AccountRecord
err := d.pg.WithConn(
ctx,
func(ctx context.Context, conn pg.Querier) error {
accounts, err := coredata.LoadMembershipAccountsByOrganizationID(
ctx,
conn,
d.scope,
d.organizationID,
)
if err != nil {
return fmt.Errorf("cannot load membership accounts: %w", err)
}
for _, account := range accounts {
role := strings.TrimSpace(account.Role)
roles := []string{}
if role != "" {
roles = []string{role}
}
isAdmin := role == string(coredata.MembershipRoleOwner) || role == string(coredata.MembershipRoleAdmin)
createdAt := account.CreatedAt
records = append(
records,
AccountRecord{
Email: account.Email,
FullName: account.FullName,
Roles: roles,
Active: new(account.State == string(coredata.ProfileStateActive)),
IsAdmin: isAdmin,
ExternalID: account.ID.String(),
CreatedAt: &createdAt,
MFAStatus: coredata.MFAStatusUnknown,
AuthMethod: coredata.AccessReviewEntryAuthMethodUnknown,
AccountType: coredata.AccessReviewEntryAccountTypeUser,
},
)
}
return nil
},
)
if err != nil {
return nil, fmt.Errorf("cannot list probo membership accounts: %w", err)
}
return records, nil
}