chore: move the go code into a subfolder

2025-08-12 03:37:34 +00:00 · 2025-08-05 15:20:32 -07:00
parent 4ad22ba456
commit cd2921de26
2978 changed files with 373 additions and 300 deletions
--- a/apps/api/internal/query/iam_member.go
+++ b/apps/api/internal/query/iam_member.go
@@ -0,0 +1,190 @@
+package query
+
+import (
+	"context"
+	"database/sql"
+
+	sq "github.com/Masterminds/squirrel"
+
+	"github.com/zitadel/zitadel/internal/api/authz"
+	"github.com/zitadel/zitadel/internal/domain"
+	"github.com/zitadel/zitadel/internal/query/projection"
+	"github.com/zitadel/zitadel/internal/telemetry/tracing"
+	"github.com/zitadel/zitadel/internal/zerrors"
+)
+
+var (
+	instanceMemberTable = table{
+		name:          projection.InstanceMemberProjectionTable,
+		alias:         "members",
+		instanceIDCol: projection.MemberInstanceID,
+	}
+	InstanceMemberUserID = Column{
+		name:  projection.MemberUserIDCol,
+		table: instanceMemberTable,
+	}
+	InstanceMemberRoles = Column{
+		name:  projection.MemberRolesCol,
+		table: instanceMemberTable,
+	}
+	InstanceMemberCreationDate = Column{
+		name:  projection.MemberCreationDate,
+		table: instanceMemberTable,
+	}
+	InstanceMemberChangeDate = Column{
+		name:  projection.MemberChangeDate,
+		table: instanceMemberTable,
+	}
+	InstanceMemberSequence = Column{
+		name:  projection.MemberSequence,
+		table: instanceMemberTable,
+	}
+	InstanceMemberResourceOwner = Column{
+		name:  projection.MemberResourceOwner,
+		table: instanceMemberTable,
+	}
+	InstanceMemberUserResourceOwner = Column{
+		name:  projection.MemberUserResourceOwner,
+		table: instanceMemberTable,
+	}
+	InstanceMemberInstanceID = Column{
+		name:  projection.MemberInstanceID,
+		table: instanceMemberTable,
+	}
+	InstanceMemberIAMID = Column{
+		name:  projection.InstanceMemberIAMIDCol,
+		table: instanceMemberTable,
+	}
+)
+
+type IAMMembersQuery struct {
+	MembersQuery
+}
+
+func (q *IAMMembersQuery) toQuery(query sq.SelectBuilder) sq.SelectBuilder {
+	return q.MembersQuery.
+		toQuery(query)
+}
+
+func (q *Queries) IAMMembers(ctx context.Context, queries *IAMMembersQuery) (members *Members, err error) {
+	ctx, span := tracing.NewSpan(ctx)
+	defer func() { span.EndWithError(err) }()
+
+	query, scan := prepareInstanceMembersQuery()
+	eq := sq.Eq{InstanceMemberInstanceID.identifier(): authz.GetInstance(ctx).InstanceID()}
+	stmt, args, err := queries.toQuery(query).Where(eq).ToSql()
+	if err != nil {
+		return nil, zerrors.ThrowInvalidArgument(err, "QUERY-USNwM", "Errors.Query.InvalidRequest")
+	}
+
+	currentSequence, err := q.latestState(ctx, instanceMemberTable)
+	if err != nil {
+		return nil, err
+	}
+
+	err = q.client.QueryContext(ctx, func(rows *sql.Rows) error {
+		members, err = scan(rows)
+		return err
+	}, stmt, args...)
+	if err != nil {
+		return nil, zerrors.ThrowInternal(err, "QUERY-Pdg1I", "Errors.Internal")
+	}
+	members.State = currentSequence
+	return members, err
+}
+
+func prepareInstanceMembersQuery() (sq.SelectBuilder, func(*sql.Rows) (*Members, error)) {
+	return sq.Select(
+			InstanceMemberCreationDate.identifier(),
+			InstanceMemberChangeDate.identifier(),
+			InstanceMemberSequence.identifier(),
+			InstanceMemberResourceOwner.identifier(),
+			InstanceMemberUserResourceOwner.identifier(),
+			InstanceMemberUserID.identifier(),
+			InstanceMemberRoles.identifier(),
+			LoginNameNameCol.identifier(),
+			HumanEmailCol.identifier(),
+			HumanFirstNameCol.identifier(),
+			HumanLastNameCol.identifier(),
+			HumanDisplayNameCol.identifier(),
+			MachineNameCol.identifier(),
+			HumanAvatarURLCol.identifier(),
+			UserTypeCol.identifier(),
+			countColumn.identifier(),
+		).From(instanceMemberTable.identifier()).
+			LeftJoin(join(HumanUserIDCol, InstanceMemberUserID)).
+			LeftJoin(join(MachineUserIDCol, InstanceMemberUserID)).
+			LeftJoin(join(UserIDCol, InstanceMemberUserID)).
+			LeftJoin(join(LoginNameUserIDCol, InstanceMemberUserID)).
+			Where(
+				sq.Eq{LoginNameIsPrimaryCol.identifier(): true},
+			).PlaceholderFormat(sq.Dollar),
+		func(rows *sql.Rows) (*Members, error) {
+			members := make([]*Member, 0)
+			var count uint64
+
+			for rows.Next() {
+				member := new(Member)
+
+				var (
+					preferredLoginName = sql.NullString{}
+					email              = sql.NullString{}
+					firstName          = sql.NullString{}
+					lastName           = sql.NullString{}
+					displayName        = sql.NullString{}
+					machineName        = sql.NullString{}
+					avatarURL          = sql.NullString{}
+					userType           = sql.NullInt32{}
+				)
+
+				err := rows.Scan(
+					&member.CreationDate,
+					&member.ChangeDate,
+					&member.Sequence,
+					&member.ResourceOwner,
+					&member.UserResourceOwner,
+					&member.UserID,
+					&member.Roles,
+					&preferredLoginName,
+					&email,
+					&firstName,
+					&lastName,
+					&displayName,
+					&machineName,
+					&avatarURL,
+					&userType,
+
+					&count,
+				)
+
+				if err != nil {
+					return nil, err
+				}
+
+				member.PreferredLoginName = preferredLoginName.String
+				member.Email = email.String
+				member.FirstName = firstName.String
+				member.LastName = lastName.String
+				member.AvatarURL = avatarURL.String
+				if displayName.Valid {
+					member.DisplayName = displayName.String
+				} else {
+					member.DisplayName = machineName.String
+				}
+				member.UserType = domain.UserType(userType.Int32)
+
+				members = append(members, member)
+			}
+
+			if err := rows.Close(); err != nil {
+				return nil, zerrors.ThrowInternal(err, "QUERY-EqJFc", "Errors.Query.CloseRows")
+			}
+
+			return &Members{
+				Members: members,
+				SearchResponse: SearchResponse{
+					Count: count,
+				},
+			}, nil
+		}
+}