pistachio/filter.go at main · winebarrel/pistachio · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package pistachio

import (
	"github.com/winebarrel/orderedmap"
	"github.com/winebarrel/pistachio/model"
)

func (f *FilterOptions) filterTables(tables *orderedmap.Map[string, *model.Table]) *orderedmap.Map[string, *model.Table] {
	if !f.IsTypeEnabled("table") {
		return orderedmap.New[string, *model.Table]()
	}
	if len(f.Include) == 0 && len(f.Exclude) == 0 {
		return tables
	}

	filtered := orderedmap.New[string, *model.Table]()
	for k, t := range tables.All() {
		if f.MatchName(t.Name) {
			filtered.Set(k, t)
		}
	}
	return filtered
}

func (f *FilterOptions) filterViews(views *orderedmap.Map[string, *model.View]) *orderedmap.Map[string, *model.View] {
	if !f.IsTypeEnabled("view") {
		return orderedmap.New[string, *model.View]()
	}
	if len(f.Include) == 0 && len(f.Exclude) == 0 {
		return views
	}

	filtered := orderedmap.New[string, *model.View]()
	for k, v := range views.All() {
		if f.MatchName(v.Name) {
			filtered.Set(k, v)
		}
	}
	return filtered
}

func (f *FilterOptions) filterEnums(enums *orderedmap.Map[string, *model.Enum]) *orderedmap.Map[string, *model.Enum] {
	if !f.IsTypeEnabled("enum") {
		return orderedmap.New[string, *model.Enum]()
	}
	if len(f.Include) == 0 && len(f.Exclude) == 0 {
		return enums
	}

	filtered := orderedmap.New[string, *model.Enum]()
	for k, e := range enums.All() {
		if f.MatchName(e.Name) {
			filtered.Set(k, e)
		}
	}
	return filtered
}

func (f *FilterOptions) filterDomains(domains *orderedmap.Map[string, *model.Domain]) *orderedmap.Map[string, *model.Domain] {
	if !f.IsTypeEnabled("domain") {
		return orderedmap.New[string, *model.Domain]()
	}
	if len(f.Include) == 0 && len(f.Exclude) == 0 {
		return domains
	}

	filtered := orderedmap.New[string, *model.Domain]()
	for k, d := range domains.All() {
		if f.MatchName(d.Name) {
			filtered.Set(k, d)
		}
	}
	return filtered
}