116 lines
2.6 KiB
Go
116 lines
2.6 KiB
Go
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)
|
|
}
|
|
}
|