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

1 files changed, 10 insertions, 3 deletions

diff --git a/cmd/sl.go b/cmd/sl.go
index 2f28fc7..b55d458 100644
--- a/cmd/sl.go
+++ b/cmd/sl.go
@@ -3,6 +3,7 @@ package cmd
 import (
 	"fmt"
 
+	"git.db.org.ai/dborg/internal/client"
 	"git.db.org.ai/dborg/internal/formatter"
 	"git.db.org.ai/dborg/internal/models"
 	"github.com/spf13/cobra"
@@ -11,8 +12,8 @@ import (
 var slCmd = &cobra.Command{
 	Use:   "sl [query]",
 	Short: "Search stealer logs",
-	Long:  `Search stealer logs with various filters`,
-	Args:  cobra.ExactArgs(1),
+	Long:  `Search stealer logs with various filters. Accepts a query arg or newline-delimited queries on stdin.`,
+	Args:  argsOrStdin(1),
 	RunE:  runSLSearch,
 }
 
@@ -34,8 +35,14 @@ func runSLSearch(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	return forEachQuery(args, func(query string) error {
+		return runSLSearchOne(cmd, c, query)
+	})
+}
+
+func runSLSearchOne(cmd *cobra.Command, c *client.Client, query string) error {
 	params := &models.SLParams{
-		Query: args[0],
+		Query: query,
 	}
 	params.Filename, _ = cmd.Flags().GetString("filename")
 	params.MaxHits, _ = cmd.Flags().GetInt("max_hits")