mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-12 11:04:25 +00:00
77b4fc5487
* beginning with postgres statements * try pgx * use pgx * database * init works for postgres * arrays working * init for cockroach * init * start tests * tests * TESTS * ch * ch * chore: use go 1.18 * read stmts * fix typo * tests * connection string * add missing error handler * cleanup * start all apis * go mod tidy * old update * switch back to minute * on conflict * replace string slice with `database.StringArray` in db models * fix tests and start * update go version in dockerfile * setup go * clean up * remove notification migration * update * docs: add deploy guide for postgres * fix: revert sonyflake * use `database.StringArray` for daos * use `database.StringArray` every where * new tables * index naming, metadata primary key, project grant role key type * docs(postgres): change to beta * chore: correct compose * fix(defaults): add empty postgres config * refactor: remove unused code * docs: add postgres to self hosted * fix broken link * so? * change title * add mdx to link * fix stmt * update goreleaser in test-code * docs: improve postgres example * update more projections * fix: add beta log for postgres * revert index name change * prerelease * fix: add sequence to v1 "reduce paniced" * log if nil * add logging * fix: log output * fix(import): check if org exists and user * refactor: imports * fix(user): ignore malformed events * refactor: method naming * fix: test * refactor: correct errors.Is call * ci: don't build dev binaries on main * fix(go releaser): update version to 1.11.0 * fix(user): projection should not break * fix(user): handle error properly * docs: correct config example * Update .releaserc.js * Update .releaserc.js Co-authored-by: Livio Amstutz <livio.a@gmail.com> Co-authored-by: Elio Bischof <eliobischof@gmail.com>
194 lines
5.2 KiB
Go
194 lines
5.2 KiB
Go
package query
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
sq "github.com/Masterminds/squirrel"
|
|
|
|
"github.com/zitadel/zitadel/internal/api/authz"
|
|
|
|
"github.com/zitadel/zitadel/internal/errors"
|
|
"github.com/zitadel/zitadel/internal/query/projection"
|
|
)
|
|
|
|
var (
|
|
projectGrantMemberTable = table{
|
|
name: projection.ProjectGrantMemberProjectionTable,
|
|
alias: "members",
|
|
}
|
|
ProjectGrantMemberUserID = Column{
|
|
name: projection.MemberUserIDCol,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberRoles = Column{
|
|
name: projection.MemberRolesCol,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberCreationDate = Column{
|
|
name: projection.MemberCreationDate,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberChangeDate = Column{
|
|
name: projection.MemberChangeDate,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberSequence = Column{
|
|
name: projection.MemberSequence,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberResourceOwner = Column{
|
|
name: projection.MemberResourceOwner,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberInstanceID = Column{
|
|
name: projection.MemberInstanceID,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberProjectID = Column{
|
|
name: projection.ProjectGrantMemberProjectIDCol,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
ProjectGrantMemberGrantID = Column{
|
|
name: projection.ProjectGrantMemberGrantIDCol,
|
|
table: projectGrantMemberTable,
|
|
}
|
|
)
|
|
|
|
type ProjectGrantMembersQuery struct {
|
|
MembersQuery
|
|
ProjectID, GrantID, OrgID string
|
|
}
|
|
|
|
func (q *ProjectGrantMembersQuery) toQuery(query sq.SelectBuilder) sq.SelectBuilder {
|
|
return q.MembersQuery.
|
|
toQuery(query).
|
|
Where(sq.And{
|
|
sq.Eq{
|
|
ProjectGrantMemberProjectID.identifier(): q.ProjectID,
|
|
ProjectGrantMemberGrantID.identifier(): q.GrantID,
|
|
},
|
|
sq.Or{
|
|
sq.Eq{ProjectGrantColumnResourceOwner.identifier(): q.OrgID},
|
|
sq.Eq{ProjectGrantColumnGrantedOrgID.identifier(): q.OrgID},
|
|
},
|
|
})
|
|
}
|
|
|
|
func (q *Queries) ProjectGrantMembers(ctx context.Context, queries *ProjectGrantMembersQuery) (*Members, error) {
|
|
query, scan := prepareProjectGrantMembersQuery()
|
|
stmt, args, err := queries.toQuery(query).
|
|
Where(sq.Eq{
|
|
ProjectGrantMemberInstanceID.identifier(): authz.GetInstance(ctx).InstanceID(),
|
|
}).ToSql()
|
|
if err != nil {
|
|
return nil, errors.ThrowInvalidArgument(err, "QUERY-USNwM", "Errors.Query.InvalidRequest")
|
|
}
|
|
|
|
currentSequence, err := q.latestSequence(ctx, projectGrantMemberTable)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
rows, err := q.client.QueryContext(ctx, stmt, args...)
|
|
if err != nil {
|
|
return nil, errors.ThrowInternal(err, "QUERY-Pdg1I", "Errors.Internal")
|
|
}
|
|
members, err := scan(rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
members.LatestSequence = currentSequence
|
|
return members, err
|
|
}
|
|
|
|
func prepareProjectGrantMembersQuery() (sq.SelectBuilder, func(*sql.Rows) (*Members, error)) {
|
|
return sq.Select(
|
|
ProjectGrantMemberCreationDate.identifier(),
|
|
ProjectGrantMemberChangeDate.identifier(),
|
|
ProjectGrantMemberSequence.identifier(),
|
|
ProjectGrantMemberResourceOwner.identifier(),
|
|
ProjectGrantMemberUserID.identifier(),
|
|
ProjectGrantMemberRoles.identifier(),
|
|
LoginNameNameCol.identifier(),
|
|
HumanEmailCol.identifier(),
|
|
HumanFirstNameCol.identifier(),
|
|
HumanLastNameCol.identifier(),
|
|
HumanDisplayNameCol.identifier(),
|
|
MachineNameCol.identifier(),
|
|
HumanAvatarURLCol.identifier(),
|
|
countColumn.identifier(),
|
|
).From(projectGrantMemberTable.identifier()).
|
|
LeftJoin(join(HumanUserIDCol, ProjectGrantMemberUserID)).
|
|
LeftJoin(join(MachineUserIDCol, ProjectGrantMemberUserID)).
|
|
LeftJoin(join(LoginNameUserIDCol, ProjectGrantMemberUserID)).
|
|
LeftJoin(join(ProjectGrantColumnGrantID, ProjectGrantMemberGrantID)).
|
|
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{}
|
|
)
|
|
|
|
err := rows.Scan(
|
|
&member.CreationDate,
|
|
&member.ChangeDate,
|
|
&member.Sequence,
|
|
&member.ResourceOwner,
|
|
&member.UserID,
|
|
&member.Roles,
|
|
&preferredLoginName,
|
|
&email,
|
|
&firstName,
|
|
&lastName,
|
|
&displayName,
|
|
&machineName,
|
|
&avatarURL,
|
|
|
|
&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
|
|
}
|
|
|
|
members = append(members, member)
|
|
}
|
|
|
|
if err := rows.Close(); err != nil {
|
|
return nil, errors.ThrowInternal(err, "QUERY-EqJFc", "Errors.Query.CloseRows")
|
|
}
|
|
|
|
return &Members{
|
|
Members: members,
|
|
SearchResponse: SearchResponse{
|
|
Count: count,
|
|
},
|
|
}, nil
|
|
}
|
|
}
|