feat: add stdin support and retry logic for all search commandsHEADv1.1.1v0.1.14master

1 files changed, 21 insertions, 15 deletions

diff --git a/cmd/files.go b/cmd/files.go
index aee7d14..6b608ca 100644
--- a/cmd/files.go
+++ b/cmd/files.go
@@ -11,7 +11,7 @@ var filesCmd = &cobra.Command{
 	Aliases: []string{"f"},
 	Short:   "Search open directory files",
 	Long:    `Search for files in open directories using various filters (free OSINT endpoint)`,
-	Args:    cobra.ExactArgs(1),
+	Args:    argsOrStdin(1),
 	RunE:    runFilesSearch,
 }
 
@@ -30,19 +30,25 @@ func runFilesSearch(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
-	params := &models.OpenDirectorySearchParams{
-		URL: args[0],
-	}
-	params.Filename, _ = cmd.Flags().GetString("filename")
-	params.Extension, _ = cmd.Flags().GetString("extension")
-	params.Exclude, _ = cmd.Flags().GetString("exclude")
-	params.Size, _ = cmd.Flags().GetInt("size")
-	params.From, _ = cmd.Flags().GetInt("from")
-
-	response, err := c.SearchOpenDirectoryFiles(params)
-	if err != nil {
-		return err
-	}
+	filename, _ := cmd.Flags().GetString("filename")
+	extension, _ := cmd.Flags().GetString("extension")
+	exclude, _ := cmd.Flags().GetString("exclude")
+	size, _ := cmd.Flags().GetInt("size")
+	from, _ := cmd.Flags().GetInt("from")
 
-	return formatter.FormatFilesResults(*response, IsJSONOutput())
+	return forEachQuery(args, func(url string) error {
+		params := &models.OpenDirectorySearchParams{
+			URL:       url,
+			Filename:  filename,
+			Extension: extension,
+			Exclude:   exclude,
+			Size:      size,
+			From:      from,
+		}
+		response, err := c.SearchOpenDirectoryFiles(params)
+		if err != nil {
+			return err
+		}
+		return formatter.FormatFilesResults(*response, IsJSONOutput())
+	})
 }