haunt/cmd/init.go

package cmd

import (
	"fmt"
	"log"
	"os"
	"strings"

	"github.com/abs3ntdev/haunt/src/haunt"
	"github.com/spf13/cobra"
)

// initCmd represents the init command
var initCmd = &cobra.Command{
	Use:   "init",
	Short: "Generates a haunt config file using sane defaults",
	Long:  "Generates a haunt config file using sane defaults, haunt will look for a main.go file and any directories inside the relative path 'cmd' and add them all as projects",
	RunE:  defaultConfig,
}

func init() {
	rootCmd.AddCommand(initCmd)
}

func defaultConfig(cmd *cobra.Command, args []string) error {
	h := haunt.NewHaunt()
	write := true
	if _, err := os.Stat(haunt.HFile); err == nil {
		fmt.Print(h.Prefix("Config file exists. Overwire? " + haunt.Magenta.Bold("[y/n] ") + haunt.Green.Bold("(n) ")))
		var overwrite string
		fmt.Scanf("%s", &overwrite)
		write = false
		switch strings.ToLower(overwrite) {
		case "y", "ye", "yes":
			write = true
		}
	}
	if write {
		h.SetDefaults()
		err := h.Settings.Write(h)
		if err != nil {
			return err
		}
		log.Println(h.Prefix(
			"Config file has successfully been saved at .haunt.yaml",
		))
		log.Println(h.Prefix(
			"Run haunt add --help to see how to add more projects",
		))
		return nil
	}
	return nil
}