clean: add --filter flag to allow override of configured client filter

l3uddz · Apr 20, 2021 · 7951a61 · 7951a61
1 parent 1dd9fa3
commit 7951a61
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/cmd/clean.go b/cmd/clean.go
@@ -54,6 +54,13 @@ var cleanCmd = &cobra.Command{
 			log.WithError(err).Fatal("Failed retrieving client filter")
 		}
 
+		if flagFilterName != "" {
+			clientFilter, err = getFilter(flagFilterName)
+			if err != nil {
+				log.WithError(err).Fatal("Failed retrieving specified filter")
+			}
+		}
+
 		// compile client filters
 		exp, err := expression.Compile(clientFilter)
 		if err != nil {
@@ -115,4 +122,6 @@ var cleanCmd = &cobra.Command{
 
 func init() {
 	rootCmd.AddCommand(cleanCmd)
+
+	cleanCmd.Flags().StringVar(&flagFilterName, "filter", "", "Filter to use instead of client")
 }
diff --git a/cmd/relabel.go b/cmd/relabel.go
@@ -11,10 +11,6 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var (
-	flagFilterName string
-)
-
 var relabelCmd = &cobra.Command{
 	Use:   "relabel [CLIENT]",
 	Short: "Check torrent client for torrents to relabel",

diff --git a/cmd/root.go b/cmd/root.go
@@ -21,6 +21,7 @@ var (
 	flagConfigFolder = config.GetDefaultConfigDirectory("tqm", flagConfigFile)
 	flagLogFile      = "activity.log"
 
+	flagFilterName string
 	flagDryRun bool
 
 	// Global vars