78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
"github.com/pterodactyl/wings/api"
|
||
|
|
"github.com/pterodactyl/wings/command"
|
||
|
|
"github.com/pterodactyl/wings/config"
|
||
|
|
"github.com/pterodactyl/wings/constants"
|
||
|
|
"github.com/pterodactyl/wings/control"
|
||
|
|
"github.com/pterodactyl/wings/utils"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
if err := command.RootCommand.Execute(); err != nil {
|
||
|
|
fmt.Println(err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 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) {
|
||
|
|
utils.InitLogging()
|
||
|
|
logrus.Info("Loading configuration...")
|
||
|
|
if err := config.LoadConfiguration(configPath); err != nil {
|
||
|
|
logrus.WithError(err).Fatal("Failed to find configuration file")
|
||
|
|
}
|
||
|
|
utils.ConfigureLogging()
|
||
|
|
|
||
|
|
logrus.Info(` ____`)
|
||
|
|
logrus.Info(`__ Pterodactyl _____/___/_______ _______ ______`)
|
||
|
|
logrus.Info(`\_____\ \/\/ / / / __ / ___/`)
|
||
|
|
logrus.Info(` \___\ / / / / /_/ /___ /`)
|
||
|
|
logrus.Info(` \___/\___/___/___/___/___ /______/`)
|
||
|
|
logrus.Info(` /_______/ v` + constants.Version)
|
||
|
|
logrus.Info()
|
||
|
|
|
||
|
|
logrus.Info("Configuration loaded successfully.")
|
||
|
|
|
||
|
|
logrus.Info("Loading configured servers...")
|
||
|
|
if err := control.LoadServerConfigurations(filepath.Join(viper.GetString(config.DataPath), constants.ServersPath)); err != nil {
|
||
|
|
logrus.WithError(err).Error("Failed to load configured servers.")
|
||
|
|
}
|
||
|
|
if amount := len(control.GetServers()); amount == 1 {
|
||
|
|
logrus.Info("Loaded 1 server.")
|
||
|
|
} else {
|
||
|
|
logrus.Info("Loaded " + strconv.Itoa(amount) + " servers.")
|
||
|
|
}
|
||
|
|
|
||
|
|
logrus.Info("Starting API Server...")
|
||
|
|
a := api.NewAPI()
|
||
|
|
a.Listen()
|
||
|
|
}
|