zitadel/internal/query/password_complexity_policy.go
Silvan 28760ab4b3
fix(query): add tracing for each method (#4777)
* fix(query): add tracing for each method
2022-12-01 09:18:53 +01:00

192 lines
5.8 KiB
Go

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"
"github.com/zitadel/zitadel/internal/telemetry/tracing"
)
type PasswordComplexityPolicy struct {
ID string
Sequence uint64
CreationDate time.Time
ChangeDate time.Time
ResourceOwner string
State domain.PolicyState
MinLength uint64
HasLowercase bool
HasUppercase bool
HasNumber bool
HasSymbol bool
IsDefault bool
}
func (q *Queries) PasswordComplexityPolicyByOrg(ctx context.Context, shouldTriggerBulk bool, orgID string, withOwnerRemoved bool) (_ *PasswordComplexityPolicy, err error) {
ctx, span := tracing.NewSpan(ctx)
defer func() { span.EndWithError(err) }()
if shouldTriggerBulk {
projection.PasswordComplexityProjection.Trigger(ctx)
}
eq := sq.Eq{PasswordComplexityColInstanceID.identifier(): authz.GetInstance(ctx).InstanceID()}
if !withOwnerRemoved {
eq[PasswordComplexityColOwnerRemoved.identifier()] = false
}
stmt, scan := preparePasswordComplexityPolicyQuery()
query, args, err := stmt.Where(
sq.And{
eq,
sq.Or{
sq.Eq{PasswordComplexityColID.identifier(): orgID},
sq.Eq{PasswordComplexityColID.identifier(): authz.GetInstance(ctx).InstanceID()},
},
}).
OrderBy(PasswordComplexityColIsDefault.identifier()).
Limit(1).ToSql()
if err != nil {
return nil, errors.ThrowInternal(err, "QUERY-lDnrk", "Errors.Query.SQLStatement")
}
row := q.client.QueryRowContext(ctx, query, args...)
return scan(row)
}
func (q *Queries) DefaultPasswordComplexityPolicy(ctx context.Context, shouldTriggerBulk bool) (_ *PasswordComplexityPolicy, err error) {
ctx, span := tracing.NewSpan(ctx)
defer func() { span.EndWithError(err) }()
if shouldTriggerBulk {
projection.PasswordComplexityProjection.Trigger(ctx)
}
stmt, scan := preparePasswordComplexityPolicyQuery()
query, args, err := stmt.Where(sq.Eq{
PasswordComplexityColID.identifier(): authz.GetInstance(ctx).InstanceID(),
PasswordComplexityColInstanceID.identifier(): authz.GetInstance(ctx).InstanceID(),
}).
OrderBy(PasswordComplexityColIsDefault.identifier()).
Limit(1).ToSql()
if err != nil {
return nil, errors.ThrowInternal(err, "QUERY-h4Uyr", "Errors.Query.SQLStatement")
}
row := q.client.QueryRowContext(ctx, query, args...)
return scan(row)
}
var (
passwordComplexityTable = table{
name: projection.PasswordComplexityTable,
instanceIDCol: projection.ComplexityPolicyInstanceIDCol,
}
PasswordComplexityColID = Column{
name: projection.ComplexityPolicyIDCol,
table: passwordComplexityTable,
}
PasswordComplexityColSequence = Column{
name: projection.ComplexityPolicySequenceCol,
table: passwordComplexityTable,
}
PasswordComplexityColCreationDate = Column{
name: projection.ComplexityPolicyCreationDateCol,
table: passwordComplexityTable,
}
PasswordComplexityColChangeDate = Column{
name: projection.ComplexityPolicyChangeDateCol,
table: passwordComplexityTable,
}
PasswordComplexityColResourceOwner = Column{
name: projection.ComplexityPolicyResourceOwnerCol,
table: passwordComplexityTable,
}
PasswordComplexityColInstanceID = Column{
name: projection.ComplexityPolicyInstanceIDCol,
table: passwordComplexityTable,
}
PasswordComplexityColMinLength = Column{
name: projection.ComplexityPolicyMinLengthCol,
table: passwordComplexityTable,
}
PasswordComplexityColHasLowercase = Column{
name: projection.ComplexityPolicyHasLowercaseCol,
table: passwordComplexityTable,
}
PasswordComplexityColHasUpperCase = Column{
name: projection.ComplexityPolicyHasUppercaseCol,
table: passwordComplexityTable,
}
PasswordComplexityColHasNumber = Column{
name: projection.ComplexityPolicyHasNumberCol,
table: passwordComplexityTable,
}
PasswordComplexityColHasSymbol = Column{
name: projection.ComplexityPolicyHasSymbolCol,
table: passwordComplexityTable,
}
PasswordComplexityColIsDefault = Column{
name: projection.ComplexityPolicyIsDefaultCol,
table: passwordComplexityTable,
}
PasswordComplexityColState = Column{
name: projection.ComplexityPolicyStateCol,
table: passwordComplexityTable,
}
PasswordComplexityColOwnerRemoved = Column{
name: projection.ComplexityPolicyOwnerRemovedCol,
table: passwordComplexityTable,
}
)
func preparePasswordComplexityPolicyQuery() (sq.SelectBuilder, func(*sql.Row) (*PasswordComplexityPolicy, error)) {
return sq.Select(
PasswordComplexityColID.identifier(),
PasswordComplexityColSequence.identifier(),
PasswordComplexityColCreationDate.identifier(),
PasswordComplexityColChangeDate.identifier(),
PasswordComplexityColResourceOwner.identifier(),
PasswordComplexityColMinLength.identifier(),
PasswordComplexityColHasLowercase.identifier(),
PasswordComplexityColHasUpperCase.identifier(),
PasswordComplexityColHasNumber.identifier(),
PasswordComplexityColHasSymbol.identifier(),
PasswordComplexityColIsDefault.identifier(),
PasswordComplexityColState.identifier(),
).
From(passwordComplexityTable.identifier()).PlaceholderFormat(sq.Dollar),
func(row *sql.Row) (*PasswordComplexityPolicy, error) {
policy := new(PasswordComplexityPolicy)
err := row.Scan(
&policy.ID,
&policy.Sequence,
&policy.CreationDate,
&policy.ChangeDate,
&policy.ResourceOwner,
&policy.MinLength,
&policy.HasLowercase,
&policy.HasUppercase,
&policy.HasNumber,
&policy.HasSymbol,
&policy.IsDefault,
&policy.State,
)
if err != nil {
if errs.Is(err, sql.ErrNoRows) {
return nil, errors.ThrowNotFound(err, "QUERY-63mtI", "Errors.PasswordComplexity.NotFound")
}
return nil, errors.ThrowInternal(err, "QUERY-uulCZ", "Errors.Internal")
}
return policy, nil
}
}