Add membership source

Signed-off-by: Bryan Frimin <bryan@getprobo.com>
2026-01-04 16:25:44 +01:00
parent 3564bd5495
commit b6799cbb01
14 changed files with 225 additions and 36 deletions
--- a/pkg/server/api/connect/v1/schema.graphql
+++ b/pkg/server/api/connect/v1/schema.graphql
@@ -234,6 +234,13 @@ enum MembershipRole
    @goEnum(value: "go.probo.inc/probo/pkg/coredata.MembershipRoleAuditor")
 }

+enum MembershipSource
+  @goModel(model: "go.probo.inc/probo/pkg/coredata.MembershipSource") {
+  MANUAL
+    @goEnum(value: "go.probo.inc/probo/pkg/coredata.MembershipSourceManual")
+  SAML @goEnum(value: "go.probo.inc/probo/pkg/coredata.MembershipSourceSAML")
+}
+
 type Membership implements Node {
  id: ID!
  createdAt: Datetime!
@@ -241,6 +248,7 @@ type Membership implements Node {
  profile: MembershipProfile @goField(forceResolver: true)
  organization: Organization @goField(forceResolver: true)
  role: MembershipRole!
+  source: MembershipSource!

  lastSession: Session @goField(forceResolver: true) @isViewer

--- a/pkg/server/api/connect/v1/schema/schema.go
+++ b/pkg/server/api/connect/v1/schema/schema.go
@@ -170,6 +170,7 @@ type ComplexityRoot struct {
 		Permission   func(childComplexity int, action string) int
 		Profile      func(childComplexity int) int
 		Role         func(childComplexity int) int
+		Source       func(childComplexity int) int
 	}

 	MembershipConnection struct {
@@ -862,6 +863,12 @@ func (e *executableSchema) Complexity(ctx context.Context, typeName, field strin
 		}

 		return e.complexity.Membership.Role(childComplexity), true
+	case "Membership.source":
+		if e.complexity.Membership.Source == nil {
+			break
+		}
+
+		return e.complexity.Membership.Source(childComplexity), true

 	case "MembershipConnection.edges":
 		if e.complexity.MembershipConnection.Edges == nil {
@@ -2198,6 +2205,13 @@ enum MembershipRole
    @goEnum(value: "go.probo.inc/probo/pkg/coredata.MembershipRoleAuditor")
 }

+enum MembershipSource
+  @goModel(model: "go.probo.inc/probo/pkg/coredata.MembershipSource") {
+  MANUAL
+    @goEnum(value: "go.probo.inc/probo/pkg/coredata.MembershipSourceManual")
+  SAML @goEnum(value: "go.probo.inc/probo/pkg/coredata.MembershipSourceSAML")
+}
+
 type Membership implements Node {
  id: ID!
  createdAt: Datetime!
@@ -2205,6 +2219,7 @@ type Membership implements Node {
  profile: MembershipProfile @goField(forceResolver: true)
  organization: Organization @goField(forceResolver: true)
  role: MembershipRole!
+  source: MembershipSource!

  lastSession: Session @goField(forceResolver: true) @isViewer

@@ -5163,6 +5178,35 @@ func (ec *executionContext) fieldContext_Membership_role(_ context.Context, fiel
 	return fc, nil
 }

+func (ec *executionContext) _Membership_source(ctx context.Context, field graphql.CollectedField, obj *types.Membership) (ret graphql.Marshaler) {
+	return graphql.ResolveField(
+		ctx,
+		ec.OperationContext,
+		field,
+		ec.fieldContext_Membership_source,
+		func(ctx context.Context) (any, error) {
+			return obj.Source, nil
+		},
+		nil,
+		ec.marshalNMembershipSource2goᚗproboᚗincᚋproboᚋpkgᚋcoredataᚐMembershipSource,
+		true,
+		true,
+	)
+}
+
+func (ec *executionContext) fieldContext_Membership_source(_ context.Context, field graphql.CollectedField) (fc *graphql.FieldContext, err error) {
+	fc = &graphql.FieldContext{
+		Object:     "Membership",
+		Field:      field,
+		IsMethod:   false,
+		IsResolver: false,
+		Child: func(ctx context.Context, field graphql.CollectedField) (*graphql.FieldContext, error) {
+			return nil, errors.New("field of type MembershipSource does not have child fields")
+		},
+	}
+	return fc, nil
+}
+
 func (ec *executionContext) _Membership_lastSession(ctx context.Context, field graphql.CollectedField, obj *types.Membership) (ret graphql.Marshaler) {
 	return graphql.ResolveField(
 		ctx,
@@ -5421,6 +5465,8 @@ func (ec *executionContext) fieldContext_MembershipEdge_node(_ context.Context,
 				return ec.fieldContext_Membership_organization(ctx, field)
 			case "role":
 				return ec.fieldContext_Membership_role(ctx, field)
+			case "source":
+				return ec.fieldContext_Membership_source(ctx, field)
 			case "lastSession":
 				return ec.fieldContext_Membership_lastSession(ctx, field)
 			case "permission":
@@ -7711,6 +7757,8 @@ func (ec *executionContext) fieldContext_Organization_viewerMembership(_ context
 				return ec.fieldContext_Membership_organization(ctx, field)
 			case "role":
 				return ec.fieldContext_Membership_role(ctx, field)
+			case "source":
+				return ec.fieldContext_Membership_source(ctx, field)
 			case "lastSession":
 				return ec.fieldContext_Membership_lastSession(ctx, field)
 			case "permission":
@@ -7864,6 +7912,8 @@ func (ec *executionContext) fieldContext_OrganizationSessionCreated_membership(_
 				return ec.fieldContext_Membership_organization(ctx, field)
 			case "role":
 				return ec.fieldContext_Membership_role(ctx, field)
+			case "source":
+				return ec.fieldContext_Membership_source(ctx, field)
 			case "lastSession":
 				return ec.fieldContext_Membership_lastSession(ctx, field)
 			case "permission":
@@ -10496,6 +10546,8 @@ func (ec *executionContext) fieldContext_UpdateMembershipPayload_membership(_ co
 				return ec.fieldContext_Membership_organization(ctx, field)
 			case "role":
 				return ec.fieldContext_Membership_role(ctx, field)
+			case "source":
+				return ec.fieldContext_Membership_source(ctx, field)
 			case "lastSession":
 				return ec.fieldContext_Membership_lastSession(ctx, field)
 			case "permission":
@@ -14408,6 +14460,11 @@ func (ec *executionContext) _Membership(ctx context.Context, sel ast.SelectionSe
 			if out.Values[i] == graphql.Null {
 				atomic.AddUint32(&out.Invalids, 1)
 			}
+		case "source":
+			out.Values[i] = ec._Membership_source(ctx, field, obj)
+			if out.Values[i] == graphql.Null {
+				atomic.AddUint32(&out.Invalids, 1)
+			}
 		case "lastSession":
 			field := field

@@ -17574,6 +17631,34 @@ var (
 	}
 )

+func (ec *executionContext) unmarshalNMembershipSource2goᚗproboᚗincᚋproboᚋpkgᚋcoredataᚐMembershipSource(ctx context.Context, v any) (coredata.MembershipSource, error) {
+	tmp, err := graphql.UnmarshalString(v)
+	res := unmarshalNMembershipSource2goᚗproboᚗincᚋproboᚋpkgᚋcoredataᚐMembershipSource[tmp]
+	return res, graphql.ErrorOnPath(ctx, err)
+}
+
+func (ec *executionContext) marshalNMembershipSource2goᚗproboᚗincᚋproboᚋpkgᚋcoredataᚐMembershipSource(ctx context.Context, sel ast.SelectionSet, v coredata.MembershipSource) graphql.Marshaler {
+	_ = sel
+	res := graphql.MarshalString(marshalNMembershipSource2goᚗproboᚗincᚋproboᚋpkgᚋcoredataᚐMembershipSource[v])
+	if res == graphql.Null {
+		if !graphql.HasFieldError(ctx, graphql.GetFieldContext(ctx)) {
+			graphql.AddErrorf(ctx, "the requested element is null which the schema does not allow")
+		}
+	}
+	return res
+}
+
+var (
+	unmarshalNMembershipSource2goᚗproboᚗincᚋproboᚋpkgᚋcoredataᚐMembershipSource = map[string]coredata.MembershipSource{
+		"MANUAL": coredata.MembershipSourceManual,
+		"SAML":   coredata.MembershipSourceSAML,
+	}
+	marshalNMembershipSource2goᚗproboᚗincᚋproboᚋpkgᚋcoredataᚐMembershipSource = map[coredata.MembershipSource]string{
+		coredata.MembershipSourceManual: "MANUAL",
+		coredata.MembershipSourceSAML:   "SAML",
+	}
+)
+
 func (ec *executionContext) unmarshalNOrderDirection2goᚗproboᚗincᚋproboᚋpkgᚋpageᚐOrderDirection(ctx context.Context, v any) (page.OrderDirection, error) {
 	tmp, err := graphql.UnmarshalString(v)
 	res := unmarshalNOrderDirection2goᚗproboᚗincᚋproboᚋpkgᚋpageᚐOrderDirection[tmp]
--- a/pkg/server/api/connect/v1/types/membership.go
+++ b/pkg/server/api/connect/v1/types/membership.go
@@ -72,10 +72,7 @@ func NewMembership(membership *coredata.Membership) *Membership {
 		Profile: &MembershipProfile{
 			ID: membership.ID,
 		},
-		Role: membership.Role,
-		// Permissions:   membership.Permissions,
-		// ProvisionedBy: membership.ProvisionedBy,
-		// Active:        membership.Active,
-		// LastSyncedAt:  membership.LastSyncedAt,
+		Role:   membership.Role,
+		Source: membership.Source,
 	}
 }
--- a/pkg/server/api/connect/v1/types/types.go
+++ b/pkg/server/api/connect/v1/types/types.go
@@ -186,14 +186,15 @@ type InviteMemberPayload struct {
 }

 type Membership struct {
-	ID           gid.GID                 `json:"id"`
-	CreatedAt    time.Time               `json:"createdAt"`
-	Identity     *Identity               `json:"identity,omitempty"`
-	Profile      *MembershipProfile      `json:"profile,omitempty"`
-	Organization *Organization           `json:"organization,omitempty"`
-	Role         coredata.MembershipRole `json:"role"`
-	LastSession  *Session                `json:"lastSession,omitempty"`
-	Permission   bool                    `json:"permission"`
+	ID           gid.GID                   `json:"id"`
+	CreatedAt    time.Time                 `json:"createdAt"`
+	Identity     *Identity                 `json:"identity,omitempty"`
+	Profile      *MembershipProfile        `json:"profile,omitempty"`
+	Organization *Organization             `json:"organization,omitempty"`
+	Role         coredata.MembershipRole   `json:"role"`
+	Source       coredata.MembershipSource `json:"source"`
+	LastSession  *Session                  `json:"lastSession,omitempty"`
+	Permission   bool                      `json:"permission"`
 }

 func (Membership) IsNode()             {}