add configure command

cmd/configure.go

@@ -0,0 +1,113 @@
+package cmd
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"os"
+	"path"
+
+	"github.com/AlecAivazis/survey/v2"
+	"github.com/AlecAivazis/survey/v2/terminal"
+	"github.com/pterodactyl/wings/config"
+	"github.com/spf13/cobra"
+)
+
+var (
+	configureArgs struct {
+		PanelURL string
+		Token    string
+		Override bool
+	}
+)
+
+var configureCmd = &cobra.Command{
+	Use:   "configure",
+	Short: "Use a token to configure wings automatically",
+
+	Run: configureCmdRun,
+}
+
+func init() {
+	configureCmd.PersistentFlags().StringVarP(&configureArgs.PanelURL, "panel-url", "p", "", "the baseurl of the pterodactyl panel to fetch the configuration from")
+	configureCmd.PersistentFlags().StringVarP(&configureArgs.Token, "token", "t", "", "the auto-deploy token to use")
+	configureCmd.PersistentFlags().BoolVar(&configureArgs.Override, "override", false, "override an existing configuration")
+}
+
+func configureCmdRun(cmd *cobra.Command, args []string) {
+	_, err := os.Stat("config.yml")
+	if err != os.ErrNotExist && !configureArgs.Override {
+		survey.AskOne(&survey.Confirm{Message: "Override existing configuration file"}, &configureArgs.Override)
+		if !configureArgs.Override {
+			fmt.Println("Aborted.")
+			os.Exit(1)
+		}
+	}
+
+	questions := []*survey.Question{}
+	if configureArgs.PanelURL == "" {
+		questions = append(questions, &survey.Question{
+			Name:   "PanelURL",
+			Prompt: &survey.Input{Message: "Panel URL: "},
+			Validate: func(ans interface{}) error {
+				if str, ok := ans.(string); ok {
+					_, err := url.ParseRequestURI(str)
+					return err
+				}
+				return nil
+			},
+		})
+	}
+	if configureArgs.Token == "" {
+		questions = append(questions, &survey.Question{
+			Name:   "Token",
+			Prompt: &survey.Input{Message: "Token: "},
+			Validate: func(ans interface{}) error {
+				if str, ok := ans.(string); ok {
+					if len(str) != 32 {
+						return fmt.Errorf("the token needs to have 32 characters")
+					}
+				}
+				return nil
+			},
+		})
+	}
+
+	err = survey.Ask(questions, &configureArgs)
+	if err == terminal.InterruptErr {
+		return
+	}
+	if err != nil {
+		panic(err)
+	}
+
+	url, err := url.Parse(configureArgs.PanelURL)
+	url.Path = path.Join(url.Path, "daemon/configure", configureArgs.Token)
+	req, err := http.NewRequest("GET", url.String(), nil)
+	if err != nil {
+		panic(err)
+	}
+	req.Header.Add("Accept", "application/json")
+	res, err := http.DefaultClient.Do(req)
+	if err != nil {
+		fmt.Println("Couldn't fetch configuration from panel.\n", err.Error())
+		os.Exit(1)
+	}
+	defer res.Body.Close()
+	if res.StatusCode == http.StatusForbidden {
+		fmt.Println("The provided token is invalid.")
+		os.Exit(1)
+	}
+	configJSON, err := ioutil.ReadAll(res.Body)
+
+	cfg := config.Configuration{}
+	json.Unmarshal(configJSON, &cfg)
+	err = cfg.WriteToDisk()
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println("Successfully configured wings.")
+}

cmd/root.go

@@ -31,13 +31,14 @@ var root = &cobra.Command{
 func init() {
 	root.PersistentFlags().StringVar(&configPath, "config", "config.yml", "set the location for the configuration file")
 	root.PersistentFlags().BoolVar(&debug, "debug", false, "pass in order to run wings in debug mode")
+
+	root.AddCommand(configureCmd)
 }
 
 func rootCmdRun(cmd *cobra.Command, args []string) {
 	c, err := config.ReadConfiguration(configPath)
 	if err != nil {
 		panic(err)
-		return
 	}
 
 	if debug {