diff options
Diffstat (limited to 'cmd/osint.go')
| -rw-r--r-- | cmd/osint.go | 29 |
1 files changed, 19 insertions, 10 deletions
diff --git a/cmd/osint.go b/cmd/osint.go index 3c3d0de..e29ef54 100644 --- a/cmd/osint.go +++ b/cmd/osint.go @@ -5,8 +5,8 @@ import ( "fmt" "git.db.org.ai/dborg/internal/client" "git.db.org.ai/dborg/internal/config" + "git.db.org.ai/dborg/internal/formatter" "git.db.org.ai/dborg/internal/models" - "git.db.org.ai/dborg/internal/utils" "github.com/spf13/cobra" ) @@ -149,8 +149,17 @@ func runUsernameCheck(cmd *cobra.Command, args []string) error { params.MaxTasks, _ = cmd.Flags().GetInt("max_tasks") err = c.CheckUsernameStream(params, func(result json.RawMessage) error { - fmt.Println(string(result)) - return nil + if IsJSONOutput() { + fmt.Println(string(result)) + return nil + } + + var siteResult models.SiteResult + if err := json.Unmarshal(result, &siteResult); err != nil { + return err + } + + return formatter.FormatUsernameSiteResult(&siteResult) }) if err != nil { @@ -180,7 +189,7 @@ func runBSSIDLookup(cmd *cobra.Command, args []string) error { return err } - return utils.PrintJSON(response) + return formatter.FormatBSSIDResults(*response, IsJSONOutput()) } func runBreachForumSearch(cmd *cobra.Command, args []string) error { @@ -201,7 +210,7 @@ func runBreachForumSearch(cmd *cobra.Command, args []string) error { return err } - return utils.PrintJSON(response) + return formatter.FormatBreachForumResults(response, IsJSONOutput()) } func runFilesSearch(cmd *cobra.Command, args []string) error { @@ -226,7 +235,7 @@ func runFilesSearch(cmd *cobra.Command, args []string) error { return err } - return utils.PrintJSON(response) + return formatter.FormatFilesResults(*response, IsJSONOutput()) } func runBucketsSearch(cmd *cobra.Command, args []string) error { @@ -246,7 +255,7 @@ func runBucketsSearch(cmd *cobra.Command, args []string) error { return err } - return utils.PrintJSON(response) + return formatter.FormatBucketsResults(response, IsJSONOutput()) } func runBucketFilesSearch(cmd *cobra.Command, args []string) error { @@ -269,7 +278,7 @@ func runBucketFilesSearch(cmd *cobra.Command, args []string) error { return err } - return utils.PrintJSON(response) + return formatter.FormatBucketFilesResults(response, IsJSONOutput()) } func runShortlinksSearch(cmd *cobra.Command, args []string) error { @@ -294,7 +303,7 @@ func runShortlinksSearch(cmd *cobra.Command, args []string) error { return err } - return utils.PrintJSON(response) + return formatter.FormatShortlinksResults(response, IsJSONOutput()) } func runGeoSearch(cmd *cobra.Command, args []string) error { @@ -316,7 +325,7 @@ func runGeoSearch(cmd *cobra.Command, args []string) error { return err } - return utils.PrintJSON(response) + return formatter.FormatGeoResults(*response, IsJSONOutput()) } func runCrawl(cmd *cobra.Command, args []string) error { |