Adapt to iterator API changes.

VirusTotal · Jun 6, 2019 · ecdc493 · ecdc493
1 parent f5e10f6
commit ecdc493
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 14 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -229,9 +229,9 @@ func (p *ObjectPrinter) Print(objType string, args []string, argRe *regexp.Regex
 // specified by the collection URL.
 func (p *ObjectPrinter) PrintCollection(collection *url.URL) error {
 	it, err := p.client.Iterator(collection,
-		vt.WithLimit(viper.GetInt("limit")),
-		vt.WithCursor(viper.GetString("cursor")),
-		vt.WithFilter(viper.GetString("filter")))
+		vt.IteratorLimit(viper.GetInt("limit")),
+		vt.IteratorCursor(viper.GetString("cursor")),
+		vt.IteratorFilter(viper.GetString("filter")))
 	if err != nil {
 		return err
 	}

diff --git a/cmd/retrohunt.go b/cmd/retrohunt.go
@@ -52,8 +52,8 @@ func retrohuntListTable(cmd *cobra.Command) error {
 
 	it, err := client.Iterator(
 		vt.URL("intelligence/retrohunt_jobs"),
-		vt.WithLimit(limit),
-		vt.WithFilter(viper.GetString("filter")))
+		vt.IteratorLimit(limit),
+		vt.IteratorFilter(viper.GetString("filter")))
 
 	if err != nil {
 		return err

diff --git a/cmd/search.go b/cmd/search.go
@@ -56,10 +56,10 @@ func runSearchCmd(cmd *cobra.Command, args []string) error {
 	}
 
 	it, err := client.Search(args[0],
-		vt.WithLimit(viper.GetInt("limit")),
-		vt.WithCursor(viper.GetString("cursor")),
-		vt.WithBatchSize(batchSize),
-		vt.WithDescriptorsOnly(
+		vt.IteratorLimit(viper.GetInt("limit")),
+		vt.IteratorCursor(viper.GetString("cursor")),
+		vt.IteratorBatchSize(batchSize),
+		vt.IteratorDescriptorsOnly(
 			viper.GetBool("identifiers-only") || viper.GetBool("download")))
 
 	if err != nil {
@@ -180,10 +180,10 @@ func runContentSearchCmd(cmd *cobra.Command, args []string) error {
 	}
 
 	it, err := client.Search(strings.Join(terms, " "),
-		vt.WithLimit(viper.GetInt("limit")),
-		vt.WithCursor(viper.GetString("cursor")),
-		vt.WithBatchSize(batchSize),
-		vt.WithDescriptorsOnly(
+		vt.IteratorLimit(viper.GetInt("limit")),
+		vt.IteratorCursor(viper.GetString("cursor")),
+		vt.IteratorBatchSize(batchSize),
+		vt.IteratorDescriptorsOnly(
 			viper.GetBool("identifiers-only") || viper.GetBool("download")))
 
 	if err != nil {