fix(query): realtime data on defined requests (#3726)

* feat: directly specify factors on addCustomLoginPolicy and return on LoginPolicy responses * fix proto * update login policy * feat: directly specify idp on addCustomLoginPolicy and return on LoginPolicy responses * fix: tests * fix(projection): trigger bulk * refactor: clean projection pkg * instance should bulk * fix(query): should trigger bulk on id calls * tests * build prerelease * fix: add shouldTriggerBulk * fix: test Co-authored-by: Livio Amstutz <livio.a@gmail.com> Co-authored-by: Max Peintner <max@caos.ch>
2025-08-12 00:27:31 +00:00 · 2022-06-14 07:51:00 +02:00
parent 5c805c48db
commit dd2f31683c
146 changed files with 1097 additions and 1239 deletions
--- a/internal/query/domain_policy.go
+++ b/internal/query/domain_policy.go
@@ -0,0 +1,165 @@
+package query
+
+import (
+	"context"
+	"database/sql"
+	errs "errors"
+	"time"
+
+	sq "github.com/Masterminds/squirrel"
+
+	"github.com/zitadel/zitadel/internal/api/authz"
+
+	"github.com/zitadel/zitadel/internal/domain"
+	"github.com/zitadel/zitadel/internal/errors"
+	"github.com/zitadel/zitadel/internal/query/projection"
+)
+
+type DomainPolicy struct {
+	ID            string
+	Sequence      uint64
+	CreationDate  time.Time
+	ChangeDate    time.Time
+	ResourceOwner string
+	State         domain.PolicyState
+
+	UserLoginMustBeDomain                  bool
+	ValidateOrgDomains                     bool
+	SMTPSenderAddressMatchesInstanceDomain bool
+
+	IsDefault bool
+}
+
+var (
+	domainPolicyTable = table{
+		name: projection.DomainPolicyTable,
+	}
+	DomainPolicyColID = Column{
+		name:  projection.DomainPolicyIDCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColSequence = Column{
+		name:  projection.DomainPolicySequenceCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColCreationDate = Column{
+		name:  projection.DomainPolicyCreationDateCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColChangeDate = Column{
+		name:  projection.DomainPolicyChangeDateCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColResourceOwner = Column{
+		name:  projection.DomainPolicyResourceOwnerCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColInstanceID = Column{
+		name:  projection.DomainPolicyInstanceIDCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColUserLoginMustBeDomain = Column{
+		name:  projection.DomainPolicyUserLoginMustBeDomainCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColValidateOrgDomains = Column{
+		name:  projection.DomainPolicyValidateOrgDomainsCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColSMTPSenderAddressMatchesInstanceDomain = Column{
+		name:  projection.DomainPolicySMTPSenderAddressMatchesInstanceDomainCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColIsDefault = Column{
+		name:  projection.DomainPolicyIsDefaultCol,
+		table: domainPolicyTable,
+	}
+	DomainPolicyColState = Column{
+		name:  projection.DomainPolicyStateCol,
+		table: domainPolicyTable,
+	}
+)
+
+func (q *Queries) DomainPolicyByOrg(ctx context.Context, shouldTriggerBulk bool, orgID string) (*DomainPolicy, error) {
+	if shouldTriggerBulk {
+		projection.DomainPolicyProjection.TriggerBulk(ctx)
+	}
+
+	stmt, scan := prepareDomainPolicyQuery()
+	query, args, err := stmt.Where(
+		sq.And{
+			sq.Eq{
+				DomainPolicyColInstanceID.identifier(): authz.GetInstance(ctx).InstanceID(),
+			},
+			sq.Or{
+				sq.Eq{
+					DomainPolicyColID.identifier(): orgID,
+				},
+				sq.Eq{
+					DomainPolicyColID.identifier(): authz.GetInstance(ctx).InstanceID(),
+				},
+			},
+		}).
+		OrderBy(DomainPolicyColIsDefault.identifier()).
+		Limit(1).ToSql()
+	if err != nil {
+		return nil, errors.ThrowInternal(err, "QUERY-D3CqT", "Errors.Query.SQLStatement")
+	}
+
+	row := q.client.QueryRowContext(ctx, query, args...)
+	return scan(row)
+}
+
+func (q *Queries) DefaultDomainPolicy(ctx context.Context) (*DomainPolicy, error) {
+	stmt, scan := prepareDomainPolicyQuery()
+	query, args, err := stmt.Where(sq.Eq{
+		DomainPolicyColID.identifier():         authz.GetInstance(ctx).InstanceID(),
+		DomainPolicyColInstanceID.identifier(): authz.GetInstance(ctx).InstanceID(),
+	}).
+		OrderBy(DomainPolicyColIsDefault.identifier()).
+		Limit(1).ToSql()
+	if err != nil {
+		return nil, errors.ThrowInternal(err, "QUERY-pM7lP", "Errors.Query.SQLStatement")
+	}
+
+	row := q.client.QueryRowContext(ctx, query, args...)
+	return scan(row)
+}
+
+func prepareDomainPolicyQuery() (sq.SelectBuilder, func(*sql.Row) (*DomainPolicy, error)) {
+	return sq.Select(
+			DomainPolicyColID.identifier(),
+			DomainPolicyColSequence.identifier(),
+			DomainPolicyColCreationDate.identifier(),
+			DomainPolicyColChangeDate.identifier(),
+			DomainPolicyColResourceOwner.identifier(),
+			DomainPolicyColUserLoginMustBeDomain.identifier(),
+			DomainPolicyColValidateOrgDomains.identifier(),
+			DomainPolicyColSMTPSenderAddressMatchesInstanceDomain.identifier(),
+			DomainPolicyColIsDefault.identifier(),
+			DomainPolicyColState.identifier(),
+		).
+			From(domainPolicyTable.identifier()).PlaceholderFormat(sq.Dollar),
+		func(row *sql.Row) (*DomainPolicy, error) {
+			policy := new(DomainPolicy)
+			err := row.Scan(
+				&policy.ID,
+				&policy.Sequence,
+				&policy.CreationDate,
+				&policy.ChangeDate,
+				&policy.ResourceOwner,
+				&policy.UserLoginMustBeDomain,
+				&policy.ValidateOrgDomains,
+				&policy.SMTPSenderAddressMatchesInstanceDomain,
+				&policy.IsDefault,
+				&policy.State,
+			)
+			if err != nil {
+				if errs.Is(err, sql.ErrNoRows) {
+					return nil, errors.ThrowNotFound(err, "QUERY-K0Jr5", "Errors.DomainPolicy.NotFound")
+				}
+				return nil, errors.ThrowInternal(err, "QUERY-rIy6j", "Errors.Internal")
+			}
+			return policy, nil
+		}
+}