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)
	}
}