diff options
Diffstat (limited to 'cmd/sl.go')
| -rw-r--r-- | cmd/sl.go | 13 |
1 files changed, 10 insertions, 3 deletions
@@ -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") |