feat(queries): login policy idp links (#2767)

* fix(idp): set type in projection * correct table * user idp links * refactor: user idp link query * add not null constraint * refactor: idp user links * rename file * fix(idp): correct resource owner * refactor: rename test * fix(query): implement idp login policy links * unify naming of idp links * test prepare * fix(api): convert idp type * rename migration
2025-08-12 01:37:31 +00:00 · 2021-12-08 14:49:19 +01:00
parent 7bf7379a05
commit c9face4ea4
15 changed files with 336 additions and 89 deletions
--- a/internal/query/idp_user_link.go
+++ b/internal/query/idp_user_link.go
@@ -0,0 +1,158 @@
+package query
+
+import (
+	"context"
+	"database/sql"
+
+	sq "github.com/Masterminds/squirrel"
+	"github.com/caos/zitadel/internal/domain"
+	"github.com/caos/zitadel/internal/errors"
+	"github.com/caos/zitadel/internal/query/projection"
+)
+
+type IDPUserLink struct {
+	IDPID            string
+	UserID           string
+	IDPName          string
+	ProvidedUserID   string
+	ProvidedUsername string
+	IDPType          domain.IDPConfigType
+}
+
+type IDPUserLinks struct {
+	SearchResponse
+	Links []*IDPUserLink
+}
+
+type IDPUserLinksSearchQuery struct {
+	SearchRequest
+	Queries []SearchQuery
+}
+
+func (q *IDPUserLinksSearchQuery) toQuery(query sq.SelectBuilder) sq.SelectBuilder {
+	query = q.SearchRequest.toQuery(query)
+	for _, q := range q.Queries {
+		query = q.toQuery(query)
+	}
+	return query
+}
+
+var (
+	idpUserLinkTable = table{
+		name: projection.IDPUserLinkTable,
+	}
+	IDPUserLinkIDPIDCol = Column{
+		name:  projection.IDPUserLinkIDPIDCol,
+		table: idpUserLinkTable,
+	}
+	IDPUserLinkUserIDCol = Column{
+		name:  projection.IDPUserLinkUserIDCol,
+		table: idpUserLinkTable,
+	}
+	IDPUserLinkExternalUserIDCol = Column{
+		name:  projection.IDPUserLinkExternalUserIDCol,
+		table: idpUserLinkTable,
+	}
+	IDPUserLinkCreationDateCol = Column{
+		name:  projection.IDPUserLinkCreationDateCol,
+		table: idpUserLinkTable,
+	}
+	IDPUserLinkChangeDateCol = Column{
+		name:  projection.IDPUserLinkChangeDateCol,
+		table: idpUserLinkTable,
+	}
+	IDPUserLinkSequenceCol = Column{
+		name:  projection.IDPUserLinkSequenceCol,
+		table: idpUserLinkTable,
+	}
+	IDPUserLinkResourceOwnerCol = Column{
+		name:  projection.IDPUserLinkResourceOwnerCol,
+		table: idpUserLinkTable,
+	}
+	IDPUserLinkDisplayNameCol = Column{
+		name:  projection.IDPUserLinkDisplayNameCol,
+		table: idpUserLinkTable,
+	}
+)
+
+func (q *Queries) IDPUserLinks(ctx context.Context, queries *IDPUserLinksSearchQuery) (idps *IDPUserLinks, err error) {
+	query, scan := prepareIDPUserLinksQuery()
+	stmt, args, err := queries.toQuery(query).ToSql()
+	if err != nil {
+		return nil, errors.ThrowInvalidArgument(err, "QUERY-4zzFK", "Errors.Query.InvalidRequest")
+	}
+
+	rows, err := q.client.QueryContext(ctx, stmt, args...)
+	if err != nil {
+		return nil, errors.ThrowInternal(err, "QUERY-C1E4D", "Errors.Internal")
+	}
+	idps, err = scan(rows)
+	if err != nil {
+		return nil, err
+	}
+	idps.LatestSequence, err = q.latestSequence(ctx, idpUserLinkTable)
+	return idps, err
+}
+
+func NewIDPUserLinksUserIDSearchQuery(value string) (SearchQuery, error) {
+	return NewTextQuery(IDPUserLinkUserIDCol, value, TextEquals)
+}
+
+func NewIDPUserLinksResourceOwnerSearchQuery(value string) (SearchQuery, error) {
+	return NewTextQuery(IDPUserLinkResourceOwnerCol, value, TextEquals)
+}
+
+func prepareIDPUserLinksQuery() (sq.SelectBuilder, func(*sql.Rows) (*IDPUserLinks, error)) {
+	return sq.Select(
+			IDPUserLinkIDPIDCol.identifier(),
+			IDPUserLinkUserIDCol.identifier(),
+			IDPNameCol.identifier(),
+			IDPUserLinkExternalUserIDCol.identifier(),
+			IDPUserLinkDisplayNameCol.identifier(),
+			IDPTypeCol.identifier(),
+			countColumn.identifier()).
+			From(idpUserLinkTable.identifier()).
+			LeftJoin(join(IDPIDCol, IDPUserLinkIDPIDCol)).PlaceholderFormat(sq.Dollar),
+		func(rows *sql.Rows) (*IDPUserLinks, error) {
+			idps := make([]*IDPUserLink, 0)
+			var count uint64
+			for rows.Next() {
+				var (
+					idpName = sql.NullString{}
+					idpType = sql.NullInt16{}
+					idp     = new(IDPUserLink)
+				)
+				err := rows.Scan(
+					&idp.IDPID,
+					&idp.UserID,
+					&idpName,
+					&idp.ProvidedUserID,
+					&idp.ProvidedUsername,
+					&idpType,
+					&count,
+				)
+				if err != nil {
+					return nil, err
+				}
+				idp.IDPName = idpName.String
+				//IDPType 0 is oidc so we have to set unspecified manually
+				if idpType.Valid {
+					idp.IDPType = domain.IDPConfigType(idpType.Int16)
+				} else {
+					idp.IDPType = domain.IDPConfigTypeUnspecified
+				}
+				idps = append(idps, idp)
+			}
+
+			if err := rows.Close(); err != nil {
+				return nil, errors.ThrowInternal(err, "QUERY-nwx6U", "Errors.Query.CloseRows")
+			}
+
+			return &IDPUserLinks{
+				Links: idps,
+				SearchResponse: SearchResponse{
+					Count: count,
+				},
+			}, nil
+		}
+}