fix cli log config issue (#830)

projectdiscovery · Jun 27, 2024 · d835f94 · d835f94
1 parent 0776a79
commit d835f94
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 20 deletions.
diff --git a/internal/runner/runner.go b/internal/runner/runner.go
@@ -56,7 +56,7 @@ func New(options *types.Options) (*Runner, error) {
 		}
 		options.URLs = mapsutil.GetKeys(runnerState.InFlightUrls.GetAll())
 	}
-
+	options.ConfigureOutput()
 	showBanner()
 
 	if options.Version {

diff --git a/pkg/types/crawler_options.go b/pkg/types/crawler_options.go
@@ -6,15 +6,12 @@ import (
 	"time"
 
 	"github.com/projectdiscovery/fastdialer/fastdialer"
-	"github.com/projectdiscovery/gologger"
-	"github.com/projectdiscovery/gologger/levels"
 	"github.com/projectdiscovery/katana/pkg/output"
 	"github.com/projectdiscovery/katana/pkg/utils/extensions"
 	"github.com/projectdiscovery/katana/pkg/utils/filters"
 	"github.com/projectdiscovery/katana/pkg/utils/scope"
 	"github.com/projectdiscovery/ratelimit"
 	errorutil "github.com/projectdiscovery/utils/errors"
-	logutil "github.com/projectdiscovery/utils/log"
 	urlutil "github.com/projectdiscovery/utils/url"
 	wappalyzer "github.com/projectdiscovery/wappalyzergo"
 )
@@ -42,7 +39,7 @@ type CrawlerOptions struct {
 // NewCrawlerOptions creates a new crawler options structure
 // from user specified options.
 func NewCrawlerOptions(options *Options) (*CrawlerOptions, error) {
-	configureOutput(options)
+	options.ConfigureOutput()
 	extensionsValidator := extensions.NewValidator(options.ExtensionsMatch, options.ExtensionFilter)
 
 	dialerOpts := fastdialer.DefaultOptions
@@ -152,18 +149,3 @@ func (c *CrawlerOptions) ValidateScope(absURL, rootHostname string) (bool, error
 	}
 	return true, nil
 }
-
-// configureOutput configures the output logging levels to be displayed on the screen
-func configureOutput(options *Options) {
-	if options.Silent {
-		gologger.DefaultLogger.SetMaxLevel(levels.LevelSilent)
-	} else if options.Verbose {
-		gologger.DefaultLogger.SetMaxLevel(levels.LevelWarning)
-	} else if options.Debug {
-		gologger.DefaultLogger.SetMaxLevel(levels.LevelDebug)
-	} else {
-		gologger.DefaultLogger.SetMaxLevel(levels.LevelInfo)
-	}
-
-	logutil.DisableDefaultLogger()
-}
diff --git a/pkg/types/options.go b/pkg/types/options.go
@@ -6,8 +6,11 @@ import (
 	"time"
 
 	"github.com/projectdiscovery/goflags"
+	"github.com/projectdiscovery/gologger"
+	"github.com/projectdiscovery/gologger/levels"
 	"github.com/projectdiscovery/katana/pkg/output"
 	fileutil "github.com/projectdiscovery/utils/file"
+	logutil "github.com/projectdiscovery/utils/log"
 )
 
 // OnResultCallback (output.Result)
@@ -194,3 +197,18 @@ func (options *Options) ParseHeadlessOptionalArguments() map[string]string {
 func (options *Options) ShouldResume() bool {
 	return options.Resume != "" && fileutil.FileExists(options.Resume)
 }
+
+// ConfigureOutput configures the output logging levels to be displayed on the screen
+func (options *Options) ConfigureOutput() {
+	if options.Silent {
+		gologger.DefaultLogger.SetMaxLevel(levels.LevelSilent)
+	} else if options.Verbose {
+		gologger.DefaultLogger.SetMaxLevel(levels.LevelWarning)
+	} else if options.Debug {
+		gologger.DefaultLogger.SetMaxLevel(levels.LevelDebug)
+	} else {
+		gologger.DefaultLogger.SetMaxLevel(levels.LevelInfo)
+	}
+
+	logutil.DisableDefaultLogger()
+}