From dabcce865ef52fcb53d369d1f7dcd0e32be774c7 Mon Sep 17 00:00:00 2001
From: Ice3man <nizamulrana@gmail.com>
Date: Fri, 14 Feb 2025 00:53:23 +0530
Subject: [PATCH] feat: fixed stats not working + misc changes

---
 internal/runner/runner.go |  3 ---
 pkg/output/stats/stats.go | 18 ++++++++++--------
 2 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/internal/runner/runner.go b/internal/runner/runner.go
index b9b0643a9..7a3b03020 100644
--- a/internal/runner/runner.go
+++ b/internal/runner/runner.go
@@ -325,9 +325,6 @@ func New(options *types.Options) (*Runner, error) {
 		}
 	}
 
-	// setup a proxy writer to automatically upload results to PDCP
-	runner.output = runner.setupPDCPUpload(outputWriter)
-
 	opts := interactsh.DefaultOptions(runner.output, runner.issuesClient, runner.progress)
 	opts.Debug = runner.options.Debug
 	opts.NoColor = runner.options.NoColor
diff --git a/pkg/output/stats/stats.go b/pkg/output/stats/stats.go
index 1e030a88d..048d1f2bf 100644
--- a/pkg/output/stats/stats.go
+++ b/pkg/output/stats/stats.go
@@ -105,14 +105,16 @@ func (t *Tracker) GetStats() *StatsOutput {
 func (t *Tracker) DisplayTopStats(noColor bool) {
 	stats := t.GetStats()
 
-	fmt.Printf("\n%s\n", aurora.Bold(aurora.Blue("Top Status Codes:")))
-	topStatusCodes := getTopN(stats.StatusCodeStats, 6)
-	for _, item := range topStatusCodes {
-		if noColor {
-			fmt.Printf("  %s: %d\n", item.Key, item.Value)
-		} else {
-			color := getStatusCodeColor(item.Key)
-			fmt.Printf("  %s: %d\n", aurora.Colorize(item.Key, color), item.Value)
+	if len(stats.StatusCodeStats) > 0 {
+		fmt.Printf("\n%s\n", aurora.Bold(aurora.Blue("Top Status Codes:")))
+		topStatusCodes := getTopN(stats.StatusCodeStats, 6)
+		for _, item := range topStatusCodes {
+			if noColor {
+				fmt.Printf("  %s: %d\n", item.Key, item.Value)
+			} else {
+				color := getStatusCodeColor(item.Key)
+				fmt.Printf("  %s: %d\n", aurora.Colorize(item.Key, color), item.Value)
+			}
 		}
 	}