package main

import (
	"fmt"
	"log"
	"strings"

	"github.com/naggie/dsnet"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

var (
	// Flags.
	hostname string

	// Commands.
	rootCmd = &cobra.Command{}

	initCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			dsnet.Init()
		},
		Use: "init",
		Short: fmt.Sprintf(
			"Create %s containing default configuration + new keys without loading. Edit to taste.",
			dsnet.CONFIG_FILE,
		),
	}

	addCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			dsnet.Add(hostname)
		},
		Use:   "add",
		Short: "Add a new peer + sync",
	}

	upCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			dsnet.Up()
		},
		Use:   "up",
		Short: "Create the interface, run pre/post up, sync",
	}

	syncCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			dsnet.Sync()
		},
		Use:   "sync",
		Short: fmt.Sprintf("Update wireguard configuration from %s after validating", dsnet.CONFIG_FILE),
	}

	reportCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			dsnet.Report()
		},
		Use:   "report",
		Short: fmt.Sprintf("Generate a JSON status report to the location configured in %s.", dsnet.CONFIG_FILE),
	}

	removeCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			dsnet.Remove()
		},
		Use:   "remove",
		Short: "Remove a peer by hostname provided as argument + sync",
	}

	downCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			dsnet.Down()
		},
		Use:   "down",
		Short: "Destroy the interface, run pre/post down",
	}

	versionCmd = &cobra.Command{
		Run: func(cmd *cobra.Command, args []string) {
			fmt.Printf("dsnet version %s\ncommit %s\nbuilt %s", dsnet.VERSION, dsnet.GIT_COMMIT, dsnet.BUILD_DATE)
		},
		Use:   "version",
		Short: "Print version",
	}
)

func main() {
	// Flags.
	rootCmd.PersistentFlags().StringP("output", "o", "wg-quick", "config file format: vyatta/wg-quick/nixos")
	addCmd.Flags().StringVarP(&hostname, "hostname", "n", "", "hostname of new peer")

	// Environment variable handling.
	viper.AutomaticEnv()
	viper.SetEnvPrefix("DSNET")
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))

	if err := viper.BindPFlag("output", rootCmd.PersistentFlags().Lookup("output")); err != nil {
		log.Fatal(err)
	}

	// Adds subcommands.
	rootCmd.AddCommand(initCmd)
	rootCmd.AddCommand(addCmd)
	rootCmd.AddCommand(upCmd)
	rootCmd.AddCommand(syncCmd)
	rootCmd.AddCommand(reportCmd)
	rootCmd.AddCommand(removeCmd)
	rootCmd.AddCommand(downCmd)
	rootCmd.AddCommand(versionCmd)

	if err := rootCmd.Execute(); err != nil {
		log.Fatal(err)
	}
}