Skip to content

Commit c14d6df

Browse files
committed
make regex names backwards compatible
1 parent 5999d0b commit c14d6df

File tree

2 files changed

+43
-10
lines changed

2 files changed

+43
-10
lines changed

pkg/filters/filters.go

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
package filters
22

33
import (
4-
t "github.com/containrrr/watchtower/pkg/types"
5-
"strings"
64
"regexp"
5+
"strings"
6+
7+
t "github.com/containrrr/watchtower/pkg/types"
78
)
89

910
// WatchtowerContainersFilter filters only watchtower containers
@@ -20,11 +21,21 @@ func FilterByNames(names []string, baseFilter t.Filter) t.Filter {
2021

2122
return func(c t.FilterableContainer) bool {
2223
for _, name := range names {
23-
match, _ := (regexp.MatchString(name, c.Name()))
24-
match1, _ := (regexp.MatchString(name, c.Name()[1:]))
25-
if (name == c.Name()) || (name == c.Name()[1:]) || match || match1 {
24+
if name == c.Name() || name == c.Name()[1:] {
2625
return baseFilter(c)
2726
}
27+
28+
if re, err := regexp.Compile(name); err == nil {
29+
indices := re.FindStringIndex(c.Name())
30+
if indices == nil {
31+
continue
32+
}
33+
start := indices[0]
34+
end := indices[1]
35+
if start <= 1 && end >= len(c.Name())-1 {
36+
return baseFilter(c)
37+
}
38+
}
2839
}
2940
return false
3041
}
@@ -80,7 +91,7 @@ func BuildFilter(names []string, enableLabel bool, scope string) (t.Filter, stri
8091
filter = FilterByNames(names, filter)
8192

8293
if len(names) > 0 {
83-
sb.WriteString("with name \"")
94+
sb.WriteString("which name matches \"")
8495
for i, n := range names {
8596
sb.WriteString(n)
8697
if i < len(names)-1 {

pkg/filters/filters_test.go

Lines changed: 26 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,28 @@ func TestFilterByNames(t *testing.T) {
4747
container.AssertExpectations(t)
4848
}
4949

50+
func TestFilterByNamesRegex(t *testing.T) {
51+
names := []string{`ba(b|ll)oon`}
52+
53+
filter := FilterByNames(names, NoFilter)
54+
assert.NotNil(t, filter)
55+
56+
container := new(mocks.FilterableContainer)
57+
container.On("Name").Return("balloon")
58+
assert.True(t, filter(container))
59+
container.AssertExpectations(t)
60+
61+
container = new(mocks.FilterableContainer)
62+
container.On("Name").Return("spoon")
63+
assert.False(t, filter(container))
64+
container.AssertExpectations(t)
65+
66+
container = new(mocks.FilterableContainer)
67+
container.On("Name").Return("baboonious")
68+
assert.False(t, filter(container))
69+
container.AssertExpectations(t)
70+
}
71+
5072
func TestFilterByEnableLabel(t *testing.T) {
5173
filter := FilterByEnableLabel(NoFilter)
5274
assert.NotNil(t, filter)
@@ -68,8 +90,7 @@ func TestFilterByEnableLabel(t *testing.T) {
6890
}
6991

7092
func TestFilterByScope(t *testing.T) {
71-
var scope string
72-
scope = "testscope"
93+
scope := "testscope"
7394

7495
filter := FilterByScope(scope, NoFilter)
7596
assert.NotNil(t, filter)
@@ -111,11 +132,12 @@ func TestFilterByDisabledLabel(t *testing.T) {
111132
}
112133

113134
func TestBuildFilter(t *testing.T) {
114-
var names []string
115-
names = append(names, "test")
135+
names := []string{"test", "valid"}
116136

117137
filter, desc := BuildFilter(names, false, "")
118138
assert.Contains(t, desc, "test")
139+
assert.Contains(t, desc, "or")
140+
assert.Contains(t, desc, "valid")
119141

120142
container := new(mocks.FilterableContainer)
121143
container.On("Name").Return("Invalid")

0 commit comments

Comments
 (0)