2020-08-26 07:56:23 +00:00
|
|
|
package view
|
|
|
|
|
|
|
|
import (
|
2022-04-19 06:26:12 +00:00
|
|
|
"github.com/jinzhu/gorm"
|
|
|
|
|
2022-04-26 23:01:45 +00:00
|
|
|
"github.com/zitadel/zitadel/internal/domain"
|
|
|
|
iam_model "github.com/zitadel/zitadel/internal/iam/model"
|
|
|
|
"github.com/zitadel/zitadel/internal/iam/repository/view/model"
|
|
|
|
"github.com/zitadel/zitadel/internal/view/repository"
|
2023-12-08 14:30:55 +00:00
|
|
|
"github.com/zitadel/zitadel/internal/zerrors"
|
2020-08-26 07:56:23 +00:00
|
|
|
)
|
|
|
|
|
2022-04-19 06:26:12 +00:00
|
|
|
func IDPByID(db *gorm.DB, table, idpID, instanceID string) (*model.IDPConfigView, error) {
|
2020-08-26 07:56:23 +00:00
|
|
|
idp := new(model.IDPConfigView)
|
2021-03-01 07:48:50 +00:00
|
|
|
idpIDQuery := &model.IDPConfigSearchQuery{Key: iam_model.IDPConfigSearchKeyIdpConfigID, Value: idpID, Method: domain.SearchMethodEquals}
|
2022-04-19 06:26:12 +00:00
|
|
|
instanceIDQuery := &model.IDPConfigSearchQuery{Key: iam_model.IDPConfigSearchKeyInstanceID, Value: instanceID, Method: domain.SearchMethodEquals}
|
2022-11-30 16:01:17 +00:00
|
|
|
ownerRemovedQuery := &model.IDPConfigSearchQuery{Key: iam_model.IDPConfigSearchKeyOwnerRemoved, Value: false, Method: domain.SearchMethodEquals}
|
|
|
|
query := repository.PrepareGetByQuery(table, idpIDQuery, instanceIDQuery, ownerRemovedQuery)
|
2020-08-26 07:56:23 +00:00
|
|
|
err := query(db, idp)
|
2023-12-08 14:30:55 +00:00
|
|
|
if zerrors.IsNotFound(err) {
|
|
|
|
return nil, zerrors.ThrowNotFound(nil, "VIEW-Ahq2s", "Errors.IDP.NotExisting")
|
2020-08-26 07:56:23 +00:00
|
|
|
}
|
|
|
|
return idp, err
|
|
|
|
}
|
|
|
|
|
2022-04-19 06:26:12 +00:00
|
|
|
func GetIDPConfigsByAggregateID(db *gorm.DB, table string, aggregateID, instanceID string) ([]*model.IDPConfigView, error) {
|
2020-09-18 11:26:28 +00:00
|
|
|
idps := make([]*model.IDPConfigView, 0)
|
|
|
|
queries := []*iam_model.IDPConfigSearchQuery{
|
|
|
|
{
|
|
|
|
Key: iam_model.IDPConfigSearchKeyAggregateID,
|
|
|
|
Value: aggregateID,
|
2021-03-01 07:48:50 +00:00
|
|
|
Method: domain.SearchMethodEquals,
|
2022-04-19 06:26:12 +00:00
|
|
|
}, {
|
|
|
|
Key: iam_model.IDPConfigSearchKeyInstanceID,
|
|
|
|
Value: instanceID,
|
|
|
|
Method: domain.SearchMethodEquals,
|
2020-09-18 11:26:28 +00:00
|
|
|
},
|
2022-11-30 16:01:17 +00:00
|
|
|
{
|
|
|
|
Key: iam_model.IDPConfigSearchKeyOwnerRemoved,
|
|
|
|
Value: false,
|
|
|
|
Method: domain.SearchMethodEquals,
|
|
|
|
},
|
2020-09-18 11:26:28 +00:00
|
|
|
}
|
|
|
|
query := repository.PrepareSearchQuery(table, model.IDPConfigSearchRequest{Queries: queries})
|
|
|
|
_, err := query(db, &idps)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return idps, nil
|
|
|
|
}
|
|
|
|
|
2020-08-26 07:56:23 +00:00
|
|
|
func SearchIDPs(db *gorm.DB, table string, req *iam_model.IDPConfigSearchRequest) ([]*model.IDPConfigView, uint64, error) {
|
|
|
|
idps := make([]*model.IDPConfigView, 0)
|
|
|
|
query := repository.PrepareSearchQuery(table, model.IDPConfigSearchRequest{Limit: req.Limit, Offset: req.Offset, Queries: req.Queries})
|
|
|
|
count, err := query(db, &idps)
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, err
|
|
|
|
}
|
|
|
|
return idps, count, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func PutIDP(db *gorm.DB, table string, idp *model.IDPConfigView) error {
|
|
|
|
save := repository.PrepareSave(table)
|
|
|
|
return save(db, idp)
|
|
|
|
}
|
|
|
|
|
2022-04-19 06:26:12 +00:00
|
|
|
func DeleteIDP(db *gorm.DB, table, idpID, instanceID string) error {
|
|
|
|
delete := repository.PrepareDeleteByKeys(table,
|
|
|
|
repository.Key{model.IDPConfigSearchKey(iam_model.IDPConfigSearchKeyIdpConfigID), idpID},
|
|
|
|
repository.Key{model.IDPConfigSearchKey(iam_model.IDPConfigSearchKeyInstanceID), instanceID},
|
|
|
|
)
|
2022-10-26 13:06:48 +00:00
|
|
|
return delete(db)
|
|
|
|
}
|
2020-08-26 07:56:23 +00:00
|
|
|
|
2022-11-30 16:01:17 +00:00
|
|
|
func UpdateOrgOwnerRemovedIDPs(db *gorm.DB, table, instanceID, aggID string) error {
|
|
|
|
update := repository.PrepareUpdateByKeys(table,
|
|
|
|
model.IDPConfigSearchKey(iam_model.IDPConfigSearchKeyOwnerRemoved),
|
|
|
|
true,
|
|
|
|
repository.Key{Key: model.IDPConfigSearchKey(iam_model.IDPConfigSearchKeyInstanceID), Value: instanceID},
|
|
|
|
repository.Key{Key: model.IDPConfigSearchKey(iam_model.IDPConfigSearchKeyAggregateID), Value: aggID},
|
|
|
|
)
|
|
|
|
return update(db)
|
|
|
|
}
|
|
|
|
|
2022-10-26 13:06:48 +00:00
|
|
|
func DeleteInstanceIDPs(db *gorm.DB, table, instanceID string) error {
|
|
|
|
delete := repository.PrepareDeleteByKey(table, model.IDPConfigSearchKey(iam_model.IDPConfigSearchKeyInstanceID), instanceID)
|
2020-08-26 07:56:23 +00:00
|
|
|
return delete(db)
|
|
|
|
}
|