Disable logging when using the programmatic interface (closes #126)

DataDog · Jun 16, 2022 · bf7a2ab · bf7a2ab
1 parent 14bba84
commit bf7a2ab
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/cmd/stratus/main.go b/cmd/stratus/main.go
@@ -3,13 +3,17 @@ package main
 import (
 	_ "github.com/datadog/stratus-red-team/internal/attacktechniques"
 	"github.com/spf13/cobra"
+	"log"
+	"os"
 )
 
 var rootCmd = &cobra.Command{
 	Use: "stratus",
 }
 
 func init() {
+	setupLogging()
+
 	listCmd := buildListCmd()
 	showCmd := buildShowCmd()
 	warmupCmd := buildWarmupCmd()
@@ -29,6 +33,10 @@ func init() {
 	rootCmd.AddCommand(versionCmd)
 }
 
+func setupLogging() {
+	log.SetOutput(os.Stdout)
+}
+
 func main() {
 	rootCmd.Execute()
 }
diff --git a/pkg/stratus/loader/main.go b/pkg/stratus/loader/main.go
@@ -2,4 +2,11 @@ package loader
 
 import (
 	_ "github.com/datadog/stratus-red-team/internal/attacktechniques" // Required for programmatic usage
+	"io"
+	"log"
 )
+
+func init() {
+	// Disable logging for programmatic usage
+	log.SetOutput(io.Discard)
+}