Support configuration file for export command

cmd/export.go

@@ -16,6 +16,7 @@ var exportCmd = &cobra.Command{
 	Run:   runExport,
 }
 
+var config string
 var channels []string
 var channelWithChildren []string
 var configChannels []string
@@ -27,6 +28,7 @@ var includeContainers bool
 var orgs []uint
 
 func init() {
+	exportCmd.Flags().StringVar(&config, "config", "", "Location of configuration file")
 	exportCmd.Flags().StringSliceVar(&channels, "channels", nil, "Channels to be exported")
 	exportCmd.Flags().StringSliceVar(&channelWithChildren, "channel-with-children", nil, "Channels to be exported")
 	exportCmd.Flags().StringVar(&outputDir, "outputDir", ".", "Location for generated data")
@@ -63,9 +65,10 @@ func runExport(cmd *cobra.Command, args []string) {
 		Containers:                includeContainers,
 		Orgs:                      orgs,
 	}
+	entityDumper.SetOptionsByConfig(config, &options)
 	entityDumper.DumpAllEntities(options)
 	var versionfile string
-	versionfile = path.Join(utils.GetAbsPath(outputDir), "version.txt")
+	versionfile = path.Join(utils.GetAbsPath(options.OutputFolder), "version.txt")
 	vf, err := os.Open(versionfile)
 	defer vf.Close()
 	if os.IsNotExist(err) {
@@ -78,5 +81,5 @@ func runExport(cmd *cobra.Command, args []string) {
 	version, product := utils.GetCurrentServerVersion(serverConfig)
 	vf.WriteString("product_name = " + product + "\n" + "version = " + version + "\n")
 
-	log.Info().Msgf("Export done. Directory: %s", outputDir)
+	log.Info().Msgf("Export done. Directory: %s", options.OutputFolder)
 }

entityDumper/dumper.go

@@ -3,10 +3,9 @@ package entityDumper
 import (
 	"bufio"
 	"compress/gzip"
-	"os"
-
 	"github.com/rs/zerolog/log"
 	"github.com/uyuni-project/inter-server-sync/schemareader"
+	"os"
 )
 
 func DumpAllEntities(options DumperOptions) {

entityDumper/exportConfigDumper.go

@@ -0,0 +1,79 @@
+package entityDumper
+
+import (
+	"encoding/json"
+	"github.com/rs/zerolog/log"
+	"github.com/uyuni-project/inter-server-sync/utils"
+	"io"
+	"os"
+)
+
+func SetOptionsByConfig(configPath string, options *DumperOptions) {
+	var config ExportConfig
+
+	if configPath == "" {
+		return
+	}
+
+	configFile, err := os.Open(utils.GetAbsPath(configPath))
+	if err != nil {
+		log.Panic().Err(err).Msg("failed to open config file")
+	}
+	defer func(configFile *os.File) {
+		err := configFile.Close()
+		if err != nil {
+			log.Warn().Err(err).Msg("failed to close config file")
+		}
+	}(configFile)
+
+	configBytes, err := io.ReadAll(configFile)
+	if err != nil {
+		log.Panic().Err(err).Msg("failed to read config file")
+	}
+
+	if err = json.Unmarshal(configBytes, &config); err != nil {
+		log.Panic().Err(err).Msg("failed to unmarshal config file")
+	}
+
+	// ServerConfig is set by cmd/root.go
+	// Should we change it anyway?
+	//if options.ServerConfig == "/etc/rhn/rhn.conf" && config.ServerConfig != "" {
+	//	options.ServerConfig = config.ServerConfig
+	//}
+
+	if options.OutputFolder == "." && config.OutputDir != "" {
+		options.OutputFolder = config.OutputDir
+	}
+
+	if len(options.ChannelLabels) == 0 {
+		options.ChannelLabels = config.Channels
+	}
+
+	if len(options.ChannelWithChildrenLabels) == 0 {
+		options.ChannelWithChildrenLabels = config.ChannelWithChildren
+	}
+
+	if len(options.ConfigLabels) == 0 {
+		options.ConfigLabels = config.ConfigChannels
+	}
+
+	if config.MetadataOnly {
+		options.MetadataOnly = config.MetadataOnly
+	}
+
+	if options.StartingDate == "" {
+		options.StartingDate = config.StartingDate
+	}
+
+	if len(options.Orgs) == 0 {
+		options.Orgs = config.Orgs
+	}
+
+	if config.IncludeImages {
+		options.OSImages = config.IncludeImages
+	}
+
+	if config.IncludeContainers {
+		options.Containers = config.IncludeContainers
+	}
+}

entityDumper/types.go

@@ -4,6 +4,19 @@ import (
 	"github.com/uyuni-project/inter-server-sync/utils"
 )
 
+type ExportConfig struct {
+	ServerConfig        string
+	Channels            []string
+	ChannelWithChildren []string
+	OutputDir           string
+	MetadataOnly        bool
+	StartingDate        string
+	ConfigChannels      []string
+	IncludeImages       bool
+	IncludeContainers   bool
+	Orgs                []uint
+}
+
 type DumperOptions struct {
 	ServerConfig              string
 	ChannelLabels             []string