snapshot of https://github.com/centrifugal/centrifugo.git (master, 2026-09-16; blobs > 131072B omitted)

internal/cli/defaultconfig.go

@@ -0,0 +1,103 @@
+package cli
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/centrifugal/centrifugo/v6/internal/config"
+	"github.com/centrifugal/centrifugo/v6/internal/tools"
+
+	"github.com/pelletier/go-toml/v2"
+	"github.com/spf13/cobra"
+	"gopkg.in/yaml.v3"
+)
+
+func DefaultConfig() *cobra.Command {
+	var defaultConfigFile string
+	var baseConfigFile string
+	var dryRun bool
+	var defaultConfigCmd = &cobra.Command{
+		Use:   "defaultconfig",
+		Short: "Generate full configuration file with defaults",
+		Long:  `Generate full Centrifugo configuration file with defaults`,
+		Run: func(cmd *cobra.Command, args []string) {
+			defaultConfig(defaultConfigFile, baseConfigFile, dryRun)
+		},
+	}
+	defaultConfigCmd.Flags().StringVarP(&defaultConfigFile, "config", "c", "config.json", "path to default config file to generate")
+	defaultConfigCmd.Flags().StringVarP(&baseConfigFile, "base", "b", "", "path to the base config file to use")
+	defaultConfigCmd.Flags().BoolVarP(&dryRun, "dry-run", "d", false, "dry run mode (do not write file and just print generated config to stdout)")
+	return defaultConfigCmd
+}
+
+func defaultConfig(configFile string, baseFile string, dryRun bool) {
+	if !dryRun {
+		exists, err := tools.PathExists(configFile)
+		if err != nil {
+			fmt.Printf("error: %v\n", err)
+			os.Exit(1)
+		}
+		if exists {
+			fmt.Printf("error: target file already exists\n")
+			os.Exit(1)
+		}
+	}
+	conf, _, err := config.GetConfig(nil, baseFile)
+	if err != nil {
+		fmt.Printf("error: %v\n", err)
+		os.Exit(1)
+	}
+	if err = conf.Validate(); err != nil {
+		fmt.Printf("error: %v\n", err)
+		os.Exit(1)
+	}
+
+	var b []byte
+
+	ext := filepath.Ext(configFile)
+	if len(ext) > 1 {
+		ext = ext[1:]
+	}
+
+	supportedExtensions := []string{"json", "toml", "yaml", "yml"}
+
+	// This is an unreleased feature so we remove RedisNats from generated config.
+	conf.Broker.RedisNats = nil
+
+	jsonBytes, err := json.MarshalIndent(conf, "", "  ")
+	if err != nil {
+		fmt.Printf("error: %v\n", err)
+		os.Exit(1)
+	}
+
+	switch ext {
+	case "json":
+		b = jsonBytes
+	case "toml":
+		b, err = toml.Marshal(conf)
+	case "yaml", "yml":
+		b, err = yaml.Marshal(conf)
+	default:
+		err = errors.New("output config file must have one of supported extensions: " + strings.Join(supportedExtensions, ", "))
+	}
+
+	if err != nil {
+		fmt.Printf("error: %v\n", err)
+		os.Exit(1)
+	}
+
+	if dryRun {
+		fmt.Println(string(b))
+		return
+	}
+
+	err = os.WriteFile(configFile, b, 0600)
+	if err != nil {
+		fmt.Printf("error: %v\n", err)
+		os.Exit(1)
+	}
+}