104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
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)
|
|
}
|
|
}
|