From 94b972630643b6f4b472a4b03f58fa757d3ca6a5 Mon Sep 17 00:00:00 2001
From: Jakub Guzik <jakubmguzik@gmail.com>
Date: Sat, 10 Apr 2021 00:01:05 +0200
Subject: Add network prune filters support to bindings

Signed-off-by: Jakub Guzik <jakubmguzik@gmail.com>
---
 pkg/bindings/network/network.go             | 10 ++++--
 pkg/bindings/network/types.go               |  3 ++
 pkg/bindings/network/types_prune_options.go | 16 ++++++++++
 pkg/bindings/test/networks_test.go          | 47 +++++++++++++++++++++++++++++
 pkg/domain/infra/tunnel/network.go          |  3 +-
 5 files changed, 76 insertions(+), 3 deletions(-)

(limited to 'pkg')

diff --git a/pkg/bindings/network/network.go b/pkg/bindings/network/network.go
index 6f3aa8594..17451c273 100644
--- a/pkg/bindings/network/network.go
+++ b/pkg/bindings/network/network.go
@@ -184,7 +184,13 @@ func Exists(ctx context.Context, nameOrID string, options *ExistsOptions) (bool,
 
 // Prune removes unused CNI networks
 func Prune(ctx context.Context, options *PruneOptions) ([]*entities.NetworkPruneReport, error) {
-	// TODO Filters is not implemented
+	if options == nil {
+		options = new(PruneOptions)
+	}
+	params, err := options.ToParams()
+	if err != nil {
+		return nil, err
+	}
 	var (
 		prunedNetworks []*entities.NetworkPruneReport
 	)
@@ -193,7 +199,7 @@ func Prune(ctx context.Context, options *PruneOptions) ([]*entities.NetworkPrune
 		return nil, err
 	}
 
-	response, err := conn.DoRequest(nil, http.MethodPost, "/networks/prune", nil, nil)
+	response, err := conn.DoRequest(nil, http.MethodPost, "/networks/prune", params, nil)
 	if err != nil {
 		return nil, err
 	}
diff --git a/pkg/bindings/network/types.go b/pkg/bindings/network/types.go
index 47dce67c7..e62ae8f52 100644
--- a/pkg/bindings/network/types.go
+++ b/pkg/bindings/network/types.go
@@ -79,4 +79,7 @@ type ExistsOptions struct {
 // PruneOptions are optional options for removing unused
 // CNI networks
 type PruneOptions struct {
+	// Filters are applied to the prune of networks to be more
+	// specific on choosing
+	Filters map[string][]string
 }
diff --git a/pkg/bindings/network/types_prune_options.go b/pkg/bindings/network/types_prune_options.go
index 84e1a8b32..f17e09d69 100644
--- a/pkg/bindings/network/types_prune_options.go
+++ b/pkg/bindings/network/types_prune_options.go
@@ -19,3 +19,19 @@ func (o *PruneOptions) Changed(fieldName string) bool {
 func (o *PruneOptions) ToParams() (url.Values, error) {
 	return util.ToParams(o)
 }
+
+// WithFilters
+func (o *PruneOptions) WithFilters(value map[string][]string) *PruneOptions {
+	v := value
+	o.Filters = v
+	return o
+}
+
+// GetFilters
+func (o *PruneOptions) GetFilters() map[string][]string {
+	var filters map[string][]string
+	if o.Filters == nil {
+		return filters
+	}
+	return o.Filters
+}
diff --git a/pkg/bindings/test/networks_test.go b/pkg/bindings/test/networks_test.go
index ef20235ae..df7d7cd1c 100644
--- a/pkg/bindings/test/networks_test.go
+++ b/pkg/bindings/test/networks_test.go
@@ -37,6 +37,53 @@ var _ = Describe("Podman networks", func() {
 		bt.cleanup()
 	})
 
+	It("podman prune unused networks with filters", func() {
+		name := "foobar"
+		opts := network.CreateOptions{
+			Name: &name,
+		}
+		_, err = network.Create(connText, &opts)
+		Expect(err).To(BeNil())
+
+		// Invalid filters should return error
+		filtersIncorrect := map[string][]string{
+			"status": {"dummy"},
+		}
+		_, err = network.Prune(connText, new(network.PruneOptions).WithFilters(filtersIncorrect))
+		Expect(err).ToNot(BeNil())
+
+		// List filter params should not work with prune.
+		filtersIncorrect = map[string][]string{
+			"name": {name},
+		}
+		_, err = network.Prune(connText, new(network.PruneOptions).WithFilters(filtersIncorrect))
+		Expect(err).ToNot(BeNil())
+
+		// Mismatched label, correct filter params => no network should be pruned.
+		filtersIncorrect = map[string][]string{
+			"label": {"xyz"},
+		}
+		pruneResponse, err := network.Prune(connText, new(network.PruneOptions).WithFilters(filtersIncorrect))
+		Expect(err).To(BeNil())
+		Expect(len(pruneResponse)).To(Equal(0))
+
+		// Mismatched until, correct filter params => no network should be pruned.
+		filters := map[string][]string{
+			"until": {"50"}, // January 1, 1970
+		}
+		pruneResponse, err = network.Prune(connText, new(network.PruneOptions).WithFilters(filters))
+		Expect(err).To(BeNil())
+		Expect(len(pruneResponse)).To(Equal(0))
+
+		// Valid filter params => network should be pruned now.
+		filters = map[string][]string{
+			"until": {"5000000000"}, //June 11, 2128
+		}
+		pruneResponse, err = network.Prune(connText, new(network.PruneOptions).WithFilters(filters))
+		Expect(err).To(BeNil())
+		Expect(len(pruneResponse)).To(Equal(1))
+	})
+
 	It("create network", func() {
 		// create a network with blank config should work
 		_, err = network.Create(connText, &network.CreateOptions{})
diff --git a/pkg/domain/infra/tunnel/network.go b/pkg/domain/infra/tunnel/network.go
index adf34460c..7e59e44c2 100644
--- a/pkg/domain/infra/tunnel/network.go
+++ b/pkg/domain/infra/tunnel/network.go
@@ -92,5 +92,6 @@ func (ic *ContainerEngine) NetworkExists(ctx context.Context, networkname string
 
 // Network prune removes unused cni networks
 func (ic *ContainerEngine) NetworkPrune(ctx context.Context, options entities.NetworkPruneOptions) ([]*entities.NetworkPruneReport, error) {
-	return network.Prune(ic.ClientCtx, nil)
+	opts := new(network.PruneOptions).WithFilters(options.Filters)
+	return network.Prune(ic.ClientCtx, opts)
 }
-- 
cgit v1.2.3-54-g00ecf