created pretty printing for all commandsv1.0.0

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 {