Bnei-Baruch/mdb

package cmd

import (
    "fmt"
    "github.com/spf13/cobra"
    "io/ioutil"
)

var outFile string

const template = `# MDB generated config template
[server]
bind-address=":8080"
mode="debug"  # GIN mode. Either debug, release or test
rollbar-token=""
rollbar-environment="development"

[mdb]
url="postgres://localhost/mdb?sslmode=disable&user=postgres"
`

var configCmd = &cobra.Command{
    Use:   "config",
    Short: "Generate configuration file template",
    Long:  "Write default configuration to given file or stdout",
    Run:   configFn,
}

func init() {
    RootCmd.AddCommand(configCmd)
    configCmd.Flags().StringVarP(&outFile, "file", "f", "", "Path to generated config file (default is config.toml)")
}

func configFn(cmd *cobra.Command, args []string) {
    if outFile == "" && len(args) > 0 {
        outFile = args[0]
    }
    if outFile == "" {
        fmt.Print(template)
    } else {
        if err := ioutil.WriteFile(outFile, []byte(template), 0644); err != nil {
            panic(err)
        }
    }
}