2020-04-14 16:20:20 +00:00
|
|
|
package view
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
caos_errs "github.com/caos/zitadel/internal/errors"
|
|
|
|
"github.com/caos/zitadel/internal/model"
|
|
|
|
"github.com/jinzhu/gorm"
|
2020-06-19 13:32:03 +00:00
|
|
|
"github.com/lib/pq"
|
2020-04-14 16:20:20 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type SearchRequest interface {
|
|
|
|
GetLimit() uint64
|
|
|
|
GetOffset() uint64
|
|
|
|
GetSortingColumn() ColumnKey
|
|
|
|
GetAsc() bool
|
|
|
|
GetQueries() []SearchQuery
|
|
|
|
}
|
|
|
|
|
|
|
|
type SearchQuery interface {
|
|
|
|
GetKey() ColumnKey
|
|
|
|
GetMethod() model.SearchMethod
|
|
|
|
GetValue() interface{}
|
|
|
|
}
|
|
|
|
|
|
|
|
type ColumnKey interface {
|
|
|
|
ToColumnName() string
|
|
|
|
}
|
|
|
|
|
|
|
|
func PrepareSearchQuery(table string, request SearchRequest) func(db *gorm.DB, res interface{}) (int, error) {
|
|
|
|
return func(db *gorm.DB, res interface{}) (int, error) {
|
|
|
|
count := 0
|
|
|
|
query := db.Table(table)
|
|
|
|
if column := request.GetSortingColumn(); column != nil {
|
|
|
|
order := "DESC"
|
|
|
|
if request.GetAsc() {
|
|
|
|
order = "ASC"
|
|
|
|
}
|
|
|
|
query = query.Order(fmt.Sprintf("%s %s", column.ToColumnName(), order))
|
|
|
|
}
|
|
|
|
for _, q := range request.GetQueries() {
|
|
|
|
var err error
|
|
|
|
query, err = SetQuery(query, q.GetKey(), q.GetValue(), q.GetMethod())
|
|
|
|
if err != nil {
|
|
|
|
return count, caos_errs.ThrowInvalidArgument(err, "VIEW-KaGue", "query is invalid")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
query = query.Count(&count)
|
|
|
|
if request.GetLimit() != 0 {
|
|
|
|
query = query.Limit(request.GetLimit())
|
|
|
|
}
|
|
|
|
query = query.Offset(request.GetOffset())
|
|
|
|
err := query.Find(res).Error
|
|
|
|
if err != nil {
|
|
|
|
return count, caos_errs.ThrowInternal(err, "VIEW-muSDK", "unable to find result")
|
|
|
|
}
|
|
|
|
return count, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func SetQuery(query *gorm.DB, key ColumnKey, value interface{}, method model.SearchMethod) (*gorm.DB, error) {
|
|
|
|
column := key.ToColumnName()
|
|
|
|
if column == "" {
|
|
|
|
return nil, caos_errs.ThrowInvalidArgument(nil, "VIEW-7dz3w", "Column name missing")
|
|
|
|
}
|
|
|
|
|
|
|
|
switch method {
|
|
|
|
case model.SEARCHMETHOD_EQUALS:
|
|
|
|
query = query.Where(""+column+" = ?", value)
|
|
|
|
case model.SEARCHMETHOD_EQUALS_IGNORE_CASE:
|
|
|
|
valueText, ok := value.(string)
|
|
|
|
if !ok {
|
2020-05-11 10:16:29 +00:00
|
|
|
return nil, caos_errs.ThrowInvalidArgument(nil, "VIEW-idu8e", "Equal ignore case only possible for strings")
|
2020-04-14 16:20:20 +00:00
|
|
|
}
|
|
|
|
query = query.Where("LOWER("+column+") = LOWER(?)", valueText)
|
|
|
|
case model.SEARCHMETHOD_STARTS_WITH:
|
|
|
|
valueText, ok := value.(string)
|
|
|
|
if !ok {
|
|
|
|
return nil, caos_errs.ThrowInvalidArgument(nil, "VIEW-idu8e", "Starts with only possible for strings")
|
|
|
|
}
|
|
|
|
query = query.Where(column+" LIKE ?", valueText+"%")
|
|
|
|
case model.SEARCHMETHOD_STARTS_WITH_IGNORE_CASE:
|
|
|
|
valueText, ok := value.(string)
|
|
|
|
if !ok {
|
2020-05-11 10:16:29 +00:00
|
|
|
return nil, caos_errs.ThrowInvalidArgument(nil, "VIEW-eidus", "Starts with ignore case only possible for strings")
|
2020-04-14 16:20:20 +00:00
|
|
|
}
|
|
|
|
query = query.Where("LOWER("+column+") LIKE LOWER(?)", valueText+"%")
|
|
|
|
case model.SEARCHMETHOD_CONTAINS:
|
|
|
|
valueText, ok := value.(string)
|
|
|
|
if !ok {
|
|
|
|
return nil, caos_errs.ThrowInvalidArgument(nil, "VIEW-3ids", "Contains with only possible for strings")
|
|
|
|
}
|
|
|
|
query = query.Where(column+" LIKE ?", "%"+valueText+"%")
|
|
|
|
case model.SEARCHMETHOD_CONTAINS_IGNORE_CASE:
|
|
|
|
valueText, ok := value.(string)
|
|
|
|
if !ok {
|
2020-05-11 10:16:29 +00:00
|
|
|
return nil, caos_errs.ThrowInvalidArgument(nil, "VIEW-eid73", "Contains with ignore case only possible for strings")
|
2020-04-14 16:20:20 +00:00
|
|
|
}
|
|
|
|
query = query.Where("LOWER("+column+") LIKE LOWER(?)", "%"+valueText+"%")
|
2020-05-11 10:16:29 +00:00
|
|
|
case model.SEARCHMETHOD_NOT_EQUALS:
|
|
|
|
query = query.Where(""+column+" <> ?", value)
|
2020-06-05 05:50:04 +00:00
|
|
|
case model.SEARCHMETHOD_GREATER_THAN:
|
|
|
|
query = query.Where(column+" > ?", value)
|
|
|
|
case model.SEARCHMETHOD_LESS_THAN:
|
|
|
|
query = query.Where(column+" < ?", value)
|
2020-06-19 13:32:03 +00:00
|
|
|
case model.SEARCHMETHOD_IS_ONE_OF:
|
2020-06-05 05:50:04 +00:00
|
|
|
query = query.Where(column+" IN (?)", value)
|
2020-06-19 13:32:03 +00:00
|
|
|
case model.SEARCHMETHOD_LIST_CONTAINS:
|
|
|
|
valueText, ok := value.(string)
|
|
|
|
if !ok {
|
|
|
|
return nil, caos_errs.ThrowInvalidArgument(nil, "VIEW-Psois", "list contains only possible for strings")
|
|
|
|
}
|
|
|
|
query = query.Where("? <@ "+column, pq.Array([]string{valueText}))
|
2020-04-14 16:20:20 +00:00
|
|
|
default:
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
return query, nil
|
|
|
|
}
|