wings/router/router_system.go

128 lines
4.0 KiB
Go
Raw Normal View History

2020-04-06 01:00:33 +00:00
package router
import (
"context"
"errors"
2021-01-10 01:22:39 +00:00
"net/http"
"strings"
2020-06-13 17:26:35 +00:00
"github.com/apex/log"
2020-04-06 01:00:33 +00:00
"github.com/gin-gonic/gin"
"github.com/pterodactyl/wings/config"
2020-04-06 01:00:33 +00:00
"github.com/pterodactyl/wings/installer"
2021-01-26 04:28:24 +00:00
"github.com/pterodactyl/wings/router/middleware"
"github.com/pterodactyl/wings/server"
2020-04-06 01:00:33 +00:00
"github.com/pterodactyl/wings/system"
)
// Returns information about the system that wings is running on.
func getSystemInformation(c *gin.Context) {
i, err := system.GetSystemInformation()
if err != nil {
2020-12-16 05:08:00 +00:00
NewTrackedError(err).Abort(c)
2020-04-06 01:00:33 +00:00
return
}
c.JSON(http.StatusOK, i)
}
// Returns all of the servers that are registered and configured correctly on
// this wings instance.
func getAllServers(c *gin.Context) {
2021-04-02 19:32:30 +00:00
servers := middleware.ExtractManager(c).All()
2021-04-04 17:20:27 +00:00
out := make([]server.APIResponse, len(servers), len(servers))
for i, v := range servers {
2021-04-04 17:20:27 +00:00
out[i] = v.ToAPIResponse()
2021-04-02 19:32:30 +00:00
}
c.JSON(http.StatusOK, out)
2020-04-06 01:00:33 +00:00
}
// Creates a new server on the wings daemon and begins the installation process
// for it.
func postCreateServer(c *gin.Context) {
manager := middleware.ExtractManager(c)
2020-04-06 01:00:33 +00:00
details := installer.ServerDetails{}
if err := c.BindJSON(&details); err != nil {
return
}
install, err := installer.New(c.Request.Context(), manager, details)
2020-04-06 01:00:33 +00:00
if err != nil {
if installer.IsValidationError(err) {
c.AbortWithStatusJSON(http.StatusUnprocessableEntity, gin.H{
"error": "The data provided in the request could not be validated.",
})
return
}
2021-03-04 05:00:58 +00:00
middleware.CaptureAndAbort(c, err)
2020-04-06 01:00:33 +00:00
return
}
// Plop that server instance onto the request so that it can be referenced in
// requests from here-on out.
2021-01-26 04:28:24 +00:00
manager.Add(install.Server())
2020-04-06 01:00:33 +00:00
// Begin the installation process in the background to not block the request
// cycle. If there are any errors they will be logged and communicated back
// to the Panel where a reinstall may take place.
go func(i *installer.Installer) {
if err := i.Server().CreateEnvironment(); err != nil {
i.Server().Log().WithField("error", err).Error("failed to create server environment during install process")
return
}
2020-04-06 01:00:33 +00:00
if err := i.Server().Install(false); err != nil {
log.WithFields(log.Fields{"server": i.Server().ID(), "error": err}).Error("failed to run install process for server")
return
}
if i.StartOnCompletion {
log.WithField("server_id", i.Server().ID()).Debug("starting server after successful installation")
if err := i.Server().HandlePowerAction(server.PowerActionStart, 30); err != nil {
if errors.Is(err, context.DeadlineExceeded) {
log.WithFields(log.Fields{"server_id": i.Server().ID(), "action": "start"}).Warn("could not acquire a lock while attempting to perform a power action")
} else {
log.WithFields(log.Fields{"server_id": i.Server().ID(), "action": "start", "error": err}).Error("encountered error processing a server power action in the background")
}
}
} else {
log.WithField("server_id", i.Server().ID()).Debug("skipping automatic start after successful server installation")
2020-04-06 01:00:33 +00:00
}
}(install)
c.Status(http.StatusAccepted)
}
2021-01-15 04:32:38 +00:00
// Updates the running configuration for this Wings instance.
func postUpdateConfiguration(c *gin.Context) {
2021-01-15 04:32:38 +00:00
cfg := config.Get()
2020-05-29 15:44:49 +00:00
if err := c.BindJSON(&cfg); err != nil {
return
}
2022-01-18 03:22:13 +00:00
// Keep the SSL certificates the same since the Panel will send through Lets Encrypt
// default locations. However, if we picked a different location manually we don't
// want to override that.
//
// If you pass through manual locations in the API call this logic will be skipped.
if strings.HasPrefix(cfg.Api.Ssl.KeyFile, "/etc/letsencrypt/live/") {
2022-01-18 03:22:13 +00:00
cfg.Api.Ssl.KeyFile = config.Get().Api.Ssl.KeyFile
cfg.Api.Ssl.CertificateFile = config.Get().Api.Ssl.CertificateFile
}
2022-01-18 03:22:13 +00:00
2021-01-15 04:32:38 +00:00
// Try to write this new configuration to the disk before updating our global
// state with it.
if err := config.WriteToDisk(cfg); err != nil {
WithError(c, err)
return
}
2021-01-15 04:32:38 +00:00
// Since we wrote it to the disk successfully now update the global configuration
// state to use this new configuration struct.
config.Set(cfg)
c.Status(http.StatusNoContent)
2020-05-29 15:44:49 +00:00
}