chore: move the go code into a subfolder

2025-08-12 00:47:33 +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/restrictions.go
+++ b/apps/api/internal/query/restrictions.go
@@ -0,0 +1,122 @@
+package query
+
+import (
+	"context"
+	"database/sql"
+	"errors"
+	"time"
+
+	sq "github.com/Masterminds/squirrel"
+	"golang.org/x/text/language"
+
+	"github.com/zitadel/zitadel/internal/api/authz"
+	"github.com/zitadel/zitadel/internal/database"
+	"github.com/zitadel/zitadel/internal/domain"
+	"github.com/zitadel/zitadel/internal/query/projection"
+	"github.com/zitadel/zitadel/internal/telemetry/tracing"
+	zitade_errors "github.com/zitadel/zitadel/internal/zerrors"
+)
+
+var (
+	restrictionsTable = table{
+		name:          projection.RestrictionsProjectionTable,
+		instanceIDCol: projection.RestrictionsColumnInstanceID,
+	}
+	RestrictionsColumnAggregateID = Column{
+		name:  projection.RestrictionsColumnAggregateID,
+		table: restrictionsTable,
+	}
+	RestrictionsColumnCreationDate = Column{
+		name:  projection.RestrictionsColumnCreationDate,
+		table: restrictionsTable,
+	}
+	RestrictionsColumnChangeDate = Column{
+		name:  projection.RestrictionsColumnChangeDate,
+		table: restrictionsTable,
+	}
+	RestrictionsColumnResourceOwner = Column{
+		name:  projection.RestrictionsColumnResourceOwner,
+		table: restrictionsTable,
+	}
+	RestrictionsColumnInstanceID = Column{
+		name:  projection.RestrictionsColumnInstanceID,
+		table: restrictionsTable,
+	}
+	RestrictionsColumnSequence = Column{
+		name:  projection.RestrictionsColumnSequence,
+		table: restrictionsTable,
+	}
+	RestrictionsColumnDisallowPublicOrgRegistration = Column{
+		name:  projection.RestrictionsColumnDisallowPublicOrgRegistration,
+		table: restrictionsTable,
+	}
+	RestrictionsColumnAllowedLanguages = Column{
+		name:  projection.RestrictionsColumnAllowedLanguages,
+		table: restrictionsTable,
+	}
+)
+
+type Restrictions struct {
+	AggregateID   string
+	CreationDate  time.Time
+	ChangeDate    time.Time
+	ResourceOwner string
+	Sequence      uint64
+
+	DisallowPublicOrgRegistration bool
+	AllowedLanguages              []language.Tag
+}
+
+func (q *Queries) GetInstanceRestrictions(ctx context.Context) (restrictions Restrictions, err error) {
+	ctx, span := tracing.NewSpan(ctx)
+	defer func() { span.EndWithError(err) }()
+
+	stmt, scan := prepareRestrictionsQuery()
+	instanceID := authz.GetInstance(ctx).InstanceID()
+	query, args, err := stmt.Where(sq.Eq{
+		RestrictionsColumnInstanceID.identifier():    instanceID,
+		RestrictionsColumnResourceOwner.identifier(): instanceID,
+	}).ToSql()
+	if err != nil {
+		return restrictions, zitade_errors.ThrowInternal(err, "QUERY-XnLMQ", "Errors.Query.SQLStatement")
+	}
+	err = q.client.QueryRowContext(ctx, func(row *sql.Row) error {
+		restrictions, err = scan(row)
+		return err
+	}, query, args...)
+	if errors.Is(err, sql.ErrNoRows) {
+		// not found is not an error
+		err = nil
+	}
+	return restrictions, err
+}
+
+func prepareRestrictionsQuery() (sq.SelectBuilder, func(*sql.Row) (Restrictions, error)) {
+	return sq.Select(
+			RestrictionsColumnAggregateID.identifier(),
+			RestrictionsColumnCreationDate.identifier(),
+			RestrictionsColumnChangeDate.identifier(),
+			RestrictionsColumnResourceOwner.identifier(),
+			RestrictionsColumnSequence.identifier(),
+			RestrictionsColumnDisallowPublicOrgRegistration.identifier(),
+			RestrictionsColumnAllowedLanguages.identifier(),
+		).
+			From(restrictionsTable.identifier()).
+			PlaceholderFormat(sq.Dollar),
+		func(row *sql.Row) (restrictions Restrictions, err error) {
+			allowedLanguages := database.TextArray[string](make([]string, 0))
+			disallowPublicOrgRegistration := sql.NullBool{}
+			err = row.Scan(
+				&restrictions.AggregateID,
+				&restrictions.CreationDate,
+				&restrictions.ChangeDate,
+				&restrictions.ResourceOwner,
+				&restrictions.Sequence,
+				&disallowPublicOrgRegistration,
+				&allowedLanguages,
+			)
+			restrictions.DisallowPublicOrgRegistration = disallowPublicOrgRegistration.Bool
+			restrictions.AllowedLanguages = domain.StringsToLanguages(allowedLanguages)
+			return restrictions, err
+		}
+}