Refactor helper binaries to save 161MB of disk space when the agent is installed and reduce RPM by 48MB

cmd/amazon-cloudwatch-agent-config-wizard/wizard.go

@@ -4,139 +4,22 @@
 package main
 
 import (
-	"bufio"
 	"flag"
-	"fmt"
+	"log"
 	"os"
 
-	"github.com/aws/amazon-cloudwatch-agent/tool/data"
-	"github.com/aws/amazon-cloudwatch-agent/tool/processors"
-	"github.com/aws/amazon-cloudwatch-agent/tool/processors/basicInfo"
-	"github.com/aws/amazon-cloudwatch-agent/tool/processors/migration/linux"
-	"github.com/aws/amazon-cloudwatch-agent/tool/processors/migration/windows"
-	"github.com/aws/amazon-cloudwatch-agent/tool/processors/serialization"
-	"github.com/aws/amazon-cloudwatch-agent/tool/processors/tracesconfig"
-	"github.com/aws/amazon-cloudwatch-agent/tool/runtime"
-	"github.com/aws/amazon-cloudwatch-agent/tool/stdin"
-	"github.com/aws/amazon-cloudwatch-agent/tool/testutil"
-	"github.com/aws/amazon-cloudwatch-agent/tool/util"
+	"github.com/aws/amazon-cloudwatch-agent/tool/cmdwrapper"
+	"github.com/aws/amazon-cloudwatch-agent/tool/wizard/flags"
 )
 
-type IMainProcessor interface {
-	VerifyProcessor(processor interface{})
-}
-type MainProcessorStruct struct{}
-
-var MainProcessorGlobal IMainProcessor = &MainProcessorStruct{}
-
-var isNonInteractiveWindowsMigration *bool
-
-var configOutputPath *string
-
-var isNonInteractiveXrayMigration *bool
-
 func main() {
-	// Parse command line args for non-interactive Windows migration
-	isNonInteractiveWindowsMigration = flag.Bool("isNonInteractiveWindowsMigration", false,
-		"If true, it will use command line args to bypass the wizard. Default value is false.")
-
-	isNonInteractiveLinuxMigration := flag.Bool("isNonInteractiveLinuxMigration", false,
-		"If true, it will do the linux config migration. Default value is false.")
-
-	tracesOnly := flag.Bool("tracesOnly", false, "If true, only trace configuration will be generated")
-	useParameterStore := flag.Bool("useParameterStore", false,
-		"If true, it will use the parameter store for the migrated config storage.")
-	isNonInteractiveXrayMigration = flag.Bool("nonInteractiveXrayMigration", false, "If true, then this is part of non Interactive xray migration tool.")
-	configFilePath := flag.String("configFilePath", "",
-		fmt.Sprintf("The path of the old config file. Default is %s on Windows or %s on Linux", windows.DefaultFilePathWindowsConfiguration, linux.DefaultFilePathLinuxConfiguration))
-
-	configOutputPath = flag.String("configOutputPath", "", "Specifies where to write the configuration file generated by the wizard")
-	parameterStoreName := flag.String("parameterStoreName", "", "The parameter store name. Default is AmazonCloudWatch-windows")
-	parameterStoreRegion := flag.String("parameterStoreRegion", "", "The parameter store region. Default is us-east-1")
+	log.Printf("Starting config-wizard, this will map back to a call to amazon-cloudwatch-agent")
 
+	translatorFlags := cmdwrapper.AddFlags("", flags.WizardFlags)
 	flag.Parse()
 
-	if *isNonInteractiveWindowsMigration {
-		addWindowsMigrationInputs(*configFilePath, *parameterStoreName, *parameterStoreRegion, *useParameterStore)
-	} else if *isNonInteractiveLinuxMigration {
-		ctx := new(runtime.Context)
-		config := new(data.Config)
-		ctx.HasExistingLinuxConfig = true
-		ctx.ConfigFilePath = *configFilePath
-		if ctx.ConfigFilePath == "" {
-			ctx.ConfigFilePath = linux.DefaultFilePathLinuxConfiguration
-		}
-		process(ctx, config, linux.Processor, serialization.Processor)
-		return
-	} else if *tracesOnly {
-		ctx := new(runtime.Context)
-		config := new(data.Config)
-		ctx.TracesOnly = true
-		ctx.ConfigOutputPath = *configOutputPath
-		if *isNonInteractiveXrayMigration {
-			ctx.NonInteractiveXrayMigration = true
-		}
-		process(ctx, config, tracesconfig.Processor, serialization.Processor)
-		return
-	}
-
-	startProcessing()
-}
-
-func init() {
-	stdin.Scanln = func(a ...interface{}) (n int, err error) {
-		scanner := bufio.NewScanner(os.Stdin)
-		scanner.Scan()
-		if len(a) > 0 {
-			*a[0].(*string) = scanner.Text()
-			n = len(*a[0].(*string))
-		}
-		err = scanner.Err()
-		return
+	err := cmdwrapper.ExecuteAgentCommand(flags.Command, translatorFlags)
+	if err != nil {
+		os.Exit(1)
 	}
-	processors.StartProcessor = basicInfo.Processor
-}
-
-func addWindowsMigrationInputs(configFilePath string, parameterStoreName string, parameterStoreRegion string, useParameterStore bool) {
-	inputChan := testutil.SetUpTestInputStream()
-	if useParameterStore {
-		testutil.Type(inputChan, "2", "1", "2", "1", configFilePath, "1", parameterStoreName, parameterStoreRegion, "1")
-	} else {
-		testutil.Type(inputChan, "2", "1", "2", "1", configFilePath, "2")
-	}
-}
-
-func process(ctx *runtime.Context, config *data.Config, processors ...processors.Processor) {
-	for _, processor := range processors {
-		processor.Process(ctx, config)
-	}
-}
-
-func startProcessing() {
-	ctx := new(runtime.Context)
-	config := new(data.Config)
-	ctx.ConfigOutputPath = *configOutputPath
-	var processor interface{}
-	processor = processors.StartProcessor
-	if *isNonInteractiveWindowsMigration {
-		ctx.WindowsNonInteractiveMigration = true
-	}
-	if *isNonInteractiveXrayMigration {
-		ctx.NonInteractiveXrayMigration = true
-	}
-	for {
-		if processor == nil {
-			if util.CurOS() == util.OsTypeWindows && !*isNonInteractiveWindowsMigration {
-				util.EnterToExit()
-			}
-			fmt.Println("Program exits now.")
-			break
-		}
-		MainProcessorGlobal.VerifyProcessor(processor) // For testing purposes
-		processor.(processors.Processor).Process(ctx, config)
-		processor = processor.(processors.Processor).NextProcessor(ctx, config)
-	}
-}
-
-func (p *MainProcessorStruct) VerifyProcessor(processor interface{}) {
 }

cmd/amazon-cloudwatch-agent/amazon-cloudwatch-agent.go

@@ -41,13 +41,20 @@ import (
 	"github.com/aws/amazon-cloudwatch-agent/internal/version"
 	cwaLogger "github.com/aws/amazon-cloudwatch-agent/logger"
 	"github.com/aws/amazon-cloudwatch-agent/logs"
-	_ "github.com/aws/amazon-cloudwatch-agent/plugins"
+	_ "github.com/aws/amazon-cloudwatch-agent/plugins" // do not remove, necessary for telegraf to know what plugins are used
 	"github.com/aws/amazon-cloudwatch-agent/profiler"
 	"github.com/aws/amazon-cloudwatch-agent/receiver/adapter"
 	"github.com/aws/amazon-cloudwatch-agent/service/configprovider"
 	"github.com/aws/amazon-cloudwatch-agent/service/defaultcomponents"
 	"github.com/aws/amazon-cloudwatch-agent/service/registry"
+	"github.com/aws/amazon-cloudwatch-agent/tool/cmdwrapper"
+	"github.com/aws/amazon-cloudwatch-agent/tool/downloader"
+	downloaderflags "github.com/aws/amazon-cloudwatch-agent/tool/downloader/flags"
 	"github.com/aws/amazon-cloudwatch-agent/tool/paths"
+	"github.com/aws/amazon-cloudwatch-agent/tool/wizard"
+	wizardflags "github.com/aws/amazon-cloudwatch-agent/tool/wizard/flags"
+	"github.com/aws/amazon-cloudwatch-agent/translator/cmdutil"
+	translatorflags "github.com/aws/amazon-cloudwatch-agent/translator/flags"
 	"github.com/aws/amazon-cloudwatch-agent/translator/tocwconfig/toyamlconfig"
 )
 
@@ -95,6 +102,11 @@ var fRunAsConsole = flag.Bool("console", false, "run as console application (win
 var fSetEnv = flag.String("setenv", "", "set an env in the configuration file in the format of KEY=VALUE")
 var fStartUpErrorFile = flag.String("startup-error-file", "", "file to touch if agent can't start")
 
+// sub-commands
+var fConfigTranslator = flag.Bool(translatorflags.TranslatorCommand, false, "run in config-translator mode")
+var fConfigDownloader = flag.Bool(downloaderflags.Command, false, "run in config-downloader mode")
+var fConfigWizard = flag.Bool(wizardflags.Command, false, "run in config-wizard mode")
+
 var stop chan struct{}
 
 func reloadLoop(
@@ -485,6 +497,10 @@ func (p *program) Stop(_ service.Service) error {
 
 func main() {
 	flag.Var(&fOtelConfigs, configprovider.OtelConfigFlagName, "YAML configuration files to run OTel pipeline")
+	translatorFlags := cmdwrapper.AddFlags(translatorflags.TranslatorCommand, translatorflags.TranslatorFlags)
+	downloaderFlags := cmdwrapper.AddFlags(downloaderflags.Command, downloaderflags.DownloaderFlags)
+	wizardFlags := cmdwrapper.AddFlags(wizardflags.Command, wizardflags.WizardFlags)
+
 	flag.Parse()
 	if len(fOtelConfigs) == 0 {
 		_ = fOtelConfigs.Set(paths.YamlConfigPath)
@@ -607,6 +623,24 @@ func main() {
 			}
 		}
 		return
+	case *fConfigTranslator:
+		err := cmdutil.RunTranslator(translatorFlags)
+		if err != nil {
+			log.Fatalf("E! Failed to initialize config translator: %v", err)
+		}
+		return
+	case *fConfigDownloader:
+		err := downloader.RunDownloaderFromFlags(downloaderFlags)
+		if err != nil {
+			log.Fatalf("E! Failed to initialize config downloader: %v", err)
+		}
+		return
+	case *fConfigWizard:
+		err := wizard.RunWizardFromFlags(wizardFlags)
+		if err != nil {
+			log.Fatalf("E! Failed to run config wizard: %v", err)
+		}
+		return
 	}
 
 	if runtime.GOOS == "windows" && windowsRunAsService() {