aboutsummaryrefslogtreecommitdiff
path: root/weed/notification/webhook/filter.go
blob: f346d6c93665f1cc9fc651612ae98fb2d65e233c (plain)
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
package webhook

import (
	"strings"

	"github.com/seaweedfs/seaweedfs/weed/glog"
	"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
)

type filter struct {
	eventTypes   map[eventType]bool
	pathPrefixes []string
}

func newFilter(cfg *config) *filter {
	f := &filter{
		eventTypes:   make(map[eventType]bool),
		pathPrefixes: cfg.pathPrefixes,
	}

	if len(cfg.eventTypes) == 0 {
		f.eventTypes[eventTypeCreate] = true
		f.eventTypes[eventTypeDelete] = true
		f.eventTypes[eventTypeUpdate] = true
		f.eventTypes[eventTypeRename] = true
	} else {
		for _, et := range cfg.eventTypes {
			t := eventType(et)
			if !t.valid() {
				glog.Warningf("invalid event type: %v", t)

				continue
			}

			f.eventTypes[t] = true
		}
	}

	return f
}

func (f *filter) shouldPublish(key string, notification *filer_pb.EventNotification) bool {
	if !f.matchesPath(key) {
		return false
	}

	eventType := detectEventType(notification)

	return f.eventTypes[eventType]
}

func (f *filter) matchesPath(key string) bool {
	if len(f.pathPrefixes) == 0 {
		return true
	}

	for _, prefix := range f.pathPrefixes {
		if strings.HasPrefix(key, prefix) {
			return true
		}
	}

	return false
}