siyuan/kernel/av/filter.go

248 lines
10 KiB
Go

// SiYuan - Refactor your thinking
// Copyright (c) 2020-present, b3log.org
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package av
import (
"github.com/siyuan-note/siyuan/kernel/util"
"strings"
)
type Filterable interface {
FilterRows()
}
type ViewFilter struct {
Column string `json:"column"`
Operator FilterOperator `json:"operator"`
Value *Value `json:"value"`
}
type FilterOperator string
const (
FilterOperatorIsEqual FilterOperator = "="
FilterOperatorIsNotEqual FilterOperator = "!="
FilterOperatorIsGreater FilterOperator = ">"
FilterOperatorIsGreaterOrEqual FilterOperator = ">="
FilterOperatorIsLess FilterOperator = "<"
FilterOperatorIsLessOrEqual FilterOperator = "<="
FilterOperatorContains FilterOperator = "Contains"
FilterOperatorDoesNotContain FilterOperator = "Does not contains"
FilterOperatorIsEmpty FilterOperator = "Is empty"
FilterOperatorIsNotEmpty FilterOperator = "Is not empty"
FilterOperatorStartsWith FilterOperator = "Starts with"
FilterOperatorEndsWith FilterOperator = "Ends with"
FilterOperatorIsBetween FilterOperator = "Is between"
FilterOperatorIsRelativeToToday FilterOperator = "Is relative to today"
FilterOperatorIsTrue FilterOperator = "Is true"
FilterOperatorIsFalse FilterOperator = "Is false"
)
func (filter *ViewFilter) GetAffectValue(key *Key) (ret *Value) {
// Improve adding rows of the filtered database table view https://github.com/siyuan-note/siyuan/issues/10025
ret = filter.Value.Clone()
switch filter.Value.Type {
case KeyTypeBlock:
switch filter.Operator {
case FilterOperatorIsEqual:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: filter.Value.Block.Content}
case FilterOperatorIsNotEqual:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: strings.TrimSpace(filter.Value.Block.Content + " Untitled")}
case FilterOperatorContains:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: filter.Value.Block.Content}
case FilterOperatorDoesNotContain:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: strings.ReplaceAll("Untitled", filter.Value.Block.Content, "")}
case FilterOperatorStartsWith:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: filter.Value.Block.Content}
case FilterOperatorEndsWith:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: filter.Value.Block.Content}
case FilterOperatorIsEmpty:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: ""}
case FilterOperatorIsNotEmpty:
ret.Block = &ValueBlock{ID: filter.Value.Block.ID, Content: "Untitled"}
}
case KeyTypeText:
switch filter.Operator {
case FilterOperatorIsEqual:
ret.Text = &ValueText{Content: filter.Value.Text.Content}
case FilterOperatorIsNotEqual:
ret.Text = &ValueText{Content: strings.TrimSpace(filter.Value.Text.Content + " Untitled")}
case FilterOperatorContains:
ret.Text = &ValueText{Content: filter.Value.Text.Content}
case FilterOperatorDoesNotContain:
ret.Text = &ValueText{Content: strings.ReplaceAll("Untitled", filter.Value.Text.Content, "")}
case FilterOperatorStartsWith:
ret.Text = &ValueText{Content: filter.Value.Text.Content}
case FilterOperatorEndsWith:
ret.Text = &ValueText{Content: filter.Value.Text.Content}
case FilterOperatorIsEmpty:
ret.Text = &ValueText{Content: ""}
case FilterOperatorIsNotEmpty:
ret.Text = &ValueText{Content: "Untitled"}
}
case KeyTypeNumber:
switch filter.Operator {
case FilterOperatorIsEqual:
ret.Number = &ValueNumber{Content: filter.Value.Number.Content, IsNotEmpty: false}
case FilterOperatorIsNotEqual:
if 0 == filter.Value.Number.Content {
ret.Number = &ValueNumber{Content: 1, IsNotEmpty: true}
} else {
ret.Number = &ValueNumber{Content: 0, IsNotEmpty: true}
}
case FilterOperatorIsGreater:
ret.Number = &ValueNumber{Content: filter.Value.Number.Content + 1, IsNotEmpty: true}
case FilterOperatorIsGreaterOrEqual:
ret.Number = &ValueNumber{Content: filter.Value.Number.Content, IsNotEmpty: true}
case FilterOperatorIsLess:
ret.Number = &ValueNumber{Content: filter.Value.Number.Content - 1, IsNotEmpty: true}
case FilterOperatorIsLessOrEqual:
ret.Number = &ValueNumber{Content: filter.Value.Number.Content, IsNotEmpty: true}
case FilterOperatorIsEmpty:
ret.Number = &ValueNumber{Content: 0, IsNotEmpty: false}
case FilterOperatorIsNotEmpty:
ret.Number = &ValueNumber{Content: 0, IsNotEmpty: true}
}
case KeyTypeDate:
switch filter.Operator {
case FilterOperatorIsEqual:
ret.Date = &ValueDate{Content: filter.Value.Date.Content, IsNotEmpty: true}
case FilterOperatorIsNotEqual:
ret.Date = &ValueDate{Content: util.CurrentTimeMillis(), IsNotEmpty: true}
case FilterOperatorIsGreater:
ret.Date = &ValueDate{Content: filter.Value.Date.Content + 1000*60, IsNotEmpty: true}
case FilterOperatorIsGreaterOrEqual:
ret.Date = &ValueDate{Content: filter.Value.Date.Content, IsNotEmpty: true}
case FilterOperatorIsLess:
ret.Date = &ValueDate{Content: filter.Value.Date.Content - 1000*60, IsNotEmpty: true}
case FilterOperatorIsLessOrEqual:
ret.Date = &ValueDate{Content: filter.Value.Date.Content, IsNotEmpty: true}
case FilterOperatorIsBetween:
ret.Date = &ValueDate{Content: filter.Value.Date.Content - 1000*60, IsNotEmpty: true}
case FilterOperatorIsEmpty:
ret.Date = &ValueDate{Content: 0, IsNotEmpty: false}
case FilterOperatorIsNotEmpty:
ret.Date = &ValueDate{Content: util.CurrentTimeMillis(), IsNotEmpty: true}
}
case KeyTypeMSelect:
switch filter.Operator {
case FilterOperatorIsEqual, FilterOperatorContains:
if 0 < len(filter.Value.MSelect) {
ret.MSelect = []*ValueSelect{{Content: filter.Value.MSelect[0].Content, Color: filter.Value.MSelect[0].Color}}
}
case FilterOperatorIsNotEqual, FilterOperatorDoesNotContain:
if 0 < len(filter.Value.MSelect) {
ret.MSelect = []*ValueSelect{{Content: filter.Value.MSelect[0].Content + " Untitled", Color: "1"}}
}
case FilterOperatorIsEmpty:
ret.MSelect = []*ValueSelect{}
case FilterOperatorIsNotEmpty:
if 0 < len(key.Options) {
ret.MSelect = []*ValueSelect{{Content: key.Options[0].Name, Color: key.Options[0].Color}}
}
}
case KeyTypeURL:
switch filter.Operator {
case FilterOperatorIsEqual:
ret.URL = &ValueURL{Content: filter.Value.URL.Content}
case FilterOperatorIsNotEqual:
ret.URL = &ValueURL{Content: filter.Value.URL.Content + " Untitled"}
case FilterOperatorContains:
ret.URL = &ValueURL{Content: filter.Value.URL.Content}
case FilterOperatorDoesNotContain:
ret.URL = &ValueURL{Content: strings.ReplaceAll("Untitled", filter.Value.URL.Content, "")}
case FilterOperatorStartsWith:
ret.URL = &ValueURL{Content: filter.Value.URL.Content}
case FilterOperatorEndsWith:
ret.URL = &ValueURL{Content: filter.Value.URL.Content}
case FilterOperatorIsEmpty:
ret.URL = &ValueURL{Content: ""}
case FilterOperatorIsNotEmpty:
ret.URL = &ValueURL{Content: "Untitled"}
}
case KeyTypeEmail:
switch filter.Operator {
case FilterOperatorIsEqual:
ret.Email = &ValueEmail{Content: filter.Value.Email.Content}
case FilterOperatorIsNotEqual:
ret.Email = &ValueEmail{Content: filter.Value.Email.Content + " Untitled"}
case FilterOperatorContains:
ret.Email = &ValueEmail{Content: filter.Value.Email.Content}
case FilterOperatorDoesNotContain:
ret.Email = &ValueEmail{Content: strings.ReplaceAll("Untitled", filter.Value.Email.Content, "")}
case FilterOperatorStartsWith:
ret.Email = &ValueEmail{Content: filter.Value.Email.Content}
case FilterOperatorEndsWith:
ret.Email = &ValueEmail{Content: filter.Value.Email.Content}
case FilterOperatorIsEmpty:
ret.Email = &ValueEmail{Content: ""}
case FilterOperatorIsNotEmpty:
ret.Email = &ValueEmail{Content: "Untitled"}
}
case KeyTypePhone:
switch filter.Operator {
case FilterOperatorIsEqual:
ret.Phone = &ValuePhone{Content: filter.Value.Phone.Content}
case FilterOperatorIsNotEqual:
ret.Phone = &ValuePhone{Content: filter.Value.Phone.Content + " Untitled"}
case FilterOperatorContains:
ret.Phone = &ValuePhone{Content: filter.Value.Phone.Content}
case FilterOperatorDoesNotContain:
ret.Phone = &ValuePhone{Content: strings.ReplaceAll("Untitled", filter.Value.Phone.Content, "")}
case FilterOperatorStartsWith:
ret.Phone = &ValuePhone{Content: filter.Value.Phone.Content}
case FilterOperatorEndsWith:
ret.Phone = &ValuePhone{Content: filter.Value.Phone.Content}
case FilterOperatorIsEmpty:
ret.Phone = &ValuePhone{Content: ""}
case FilterOperatorIsNotEmpty:
ret.Phone = &ValuePhone{Content: "Untitled"}
}
case KeyTypeMAsset:
switch filter.Operator {
case FilterOperatorIsEqual, FilterOperatorContains:
if 0 < len(filter.Value.MAsset) {
ret.MAsset = []*ValueAsset{{Type: filter.Value.MAsset[0].Type, Name: filter.Value.MAsset[0].Name, Content: filter.Value.MAsset[0].Content}}
}
case FilterOperatorIsNotEqual, FilterOperatorDoesNotContain:
case FilterOperatorIsEmpty:
ret.MAsset = []*ValueAsset{}
case FilterOperatorIsNotEmpty:
}
case KeyTypeCheckbox:
switch filter.Operator {
case FilterOperatorIsTrue:
ret.Checkbox = &ValueCheckbox{Checked: true}
case FilterOperatorIsFalse:
ret.Checkbox = &ValueCheckbox{Checked: false}
}
case KeyTypeRelation:
switch filter.Operator {
case FilterOperatorContains:
if 0 < len(filter.Value.Relation.Contents) {
ret.Relation = &ValueRelation{Contents: filter.Value.Relation.Contents}
}
case FilterOperatorDoesNotContain:
case FilterOperatorIsEmpty:
ret.Relation = &ValueRelation{Contents: []string{}}
case FilterOperatorIsNotEmpty:
}
}
return
}