package filters import ( t "github.com/containrrr/watchtower/pkg/types" "strings" ) // WatchtowerContainersFilter filters only watchtower containers func WatchtowerContainersFilter(c t.FilterableContainer) bool { return c.IsWatchtower() } // NoFilter will not filter out any containers func NoFilter(t.FilterableContainer) bool { return true } // FilterByNames returns all containers that match the specified name func FilterByNames(names []string, baseFilter t.Filter) t.Filter { if len(names) == 0 { return baseFilter } return func(c t.FilterableContainer) bool { for _, name := range names { if (name == c.Name()) || (name == c.Name()[1:]) { return baseFilter(c) } } return false } } // FilterByEnableLabel returns all containers that have the enabled label set func FilterByEnableLabel(baseFilter t.Filter) t.Filter { return func(c t.FilterableContainer) bool { // If label filtering is enabled, containers should only be considered // if the label is specifically set. _, ok := c.Enabled() if !ok { return false } return baseFilter(c) } } // FilterByDisabledLabel returns all containers that have the enabled label set to disable func FilterByDisabledLabel(baseFilter t.Filter) t.Filter { return func(c t.FilterableContainer) bool { enabledLabel, ok := c.Enabled() if ok && !enabledLabel { // If the label has been set and it demands a disable return false } return baseFilter(c) } } // FilterByScope returns all containers that belongs to a specific scope func FilterByScope(scope string, baseFilter t.Filter) t.Filter { if scope == "" { return baseFilter } return func(c t.FilterableContainer) bool { containerScope, ok := c.Scope() if ok && containerScope == scope { return baseFilter(c) } return false } } // BuildFilter creates the needed filter of containers func BuildFilter(names []string, enableLabel bool, scope string) (t.Filter, string) { sb := strings.Builder{} filter := NoFilter filter = FilterByNames(names, filter) if len(names) > 0 { sb.WriteString("with name \"") for i, n := range names { sb.WriteString(n) if i < len(names)-1 { sb.WriteString(`" or "`) } } sb.WriteString(`", `) } if enableLabel { // If label filtering is enabled, containers should only be considered // if the label is specifically set. filter = FilterByEnableLabel(filter) sb.WriteString("using enable label, ") } if scope != "" { // If a scope has been defined, containers should only be considered // if the scope is specifically set. filter = FilterByScope(scope, filter) sb.WriteString(`in scope "`) sb.WriteString(scope) sb.WriteString(`", `) } filter = FilterByDisabledLabel(filter) filterDesc := "Checking all containers (except explicitly disabled with label)" if sb.Len() > 0 { filterDesc = "Only checking containers " + sb.String() // Remove the last ", " filterDesc = filterDesc[:len(filterDesc)-2] } return filter, filterDesc }