wings/command/root.go

package command

import (
	"path/filepath"
	"strconv"

	"github.com/spf13/viper"

	"github.com/Pterodactyl/wings/api"
	"github.com/Pterodactyl/wings/config"
	"github.com/Pterodactyl/wings/constants"
	"github.com/Pterodactyl/wings/control"
	"github.com/Pterodactyl/wings/utils"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
)

// RootCommand is the root command of wings
var RootCommand = &cobra.Command{
	Use:   "wings",
	Short: "",
	Long:  "",
	Run:   run,
}

var configPath string

func init() {
	RootCommand.Flags().StringVarP(&configPath, "config", "c", "./config.yml", "Allows to set the path of the configuration file.")
}

// Execute registers the RootCommand
func Execute() {
	RootCommand.Execute()
}

func run(cmd *cobra.Command, args []string) {
	tools.InitLogging()
	log.Info("Loading configuration...")
	if err := config.LoadConfiguration(configPath); err != nil {
		log.WithError(err).Fatal("Failed to find configuration file")
	}
	tools.ConfigureLogging()

	log.Info(`                     ____`)
	log.Info(`__ Pterodactyl _____/___/_______ _______ ______`)
	log.Info(`\_____\    \/\/    /   /       /  __   /   ___/`)
	log.Info(`   \___\          /   /   /   /  /_/  /___   /`)
	log.Info(`        \___/\___/___/___/___/___    /______/`)
	log.Info(`                            /_______/ v` + constants.Version)
	log.Info()

	log.Info("Configuration loaded successfully.")

	log.Info("Loading configured servers...")
	if err := control.LoadServerConfigurations(filepath.Join(viper.GetString(config.DataPath), constants.ServersPath)); err != nil {
		log.WithError(err).Error("Failed to load configured servers.")
	}
	if amount := len(control.GetServers()); amount == 1 {
		log.Info("Loaded 1 server.")
	} else {
		log.Info("Loaded " + strconv.Itoa(amount) + " servers.")
	}

	log.Info("Starting api webserver...")
	api := api.NewAPI()
	api.Listen()
}