rename ServerManagerFromContext from ExtractServerManager

This commit is contained in:
Jakob Schrettenbrunner 2021-01-10 02:18:02 +00:00
parent abeb9655f9
commit c9c949a091
7 changed files with 9 additions and 9 deletions

View File

@ -100,7 +100,7 @@ func (m *Middleware) WithServerManager() gin.HandlerFunc {
} }
} }
func ServerManagerFromContext(c *gin.Context) server.Manager { func ExtractServerManager(c *gin.Context) server.Manager {
if s, ok := c.Get("servermanager"); ok { if s, ok := c.Get("servermanager"); ok {
if srvs, ok := s.(server.Manager); ok { if srvs, ok := s.(server.Manager); ok {
return srvs return srvs

View File

@ -14,7 +14,7 @@ import (
// Handle a download request for a server backup. // Handle a download request for a server backup.
func getDownloadBackup(c *gin.Context) { func getDownloadBackup(c *gin.Context) {
serverManager := ServerManagerFromContext(c) serverManager := ExtractServerManager(c)
token := tokens.BackupPayload{} token := tokens.BackupPayload{}
if err := tokens.ParseToken([]byte(c.Query("token")), &token); err != nil { if err := tokens.ParseToken([]byte(c.Query("token")), &token); err != nil {
@ -59,7 +59,7 @@ func getDownloadBackup(c *gin.Context) {
// Handles downloading a specific file for a server. // Handles downloading a specific file for a server.
func getDownloadFile(c *gin.Context) { func getDownloadFile(c *gin.Context) {
serverManager := ServerManagerFromContext(c) serverManager := ExtractServerManager(c)
token := tokens.FilePayload{} token := tokens.FilePayload{}
if err := tokens.ParseToken([]byte(c.Query("token")), &token); err != nil { if err := tokens.ParseToken([]byte(c.Query("token")), &token); err != nil {

View File

@ -191,7 +191,7 @@ func postServerReinstall(c *gin.Context) {
// Deletes a server from the wings daemon and dissociate it's objects. // Deletes a server from the wings daemon and dissociate it's objects.
func deleteServer(c *gin.Context) { func deleteServer(c *gin.Context) {
s := ExtractServer(c) s := ExtractServer(c)
sm := ServerManagerFromContext(c) sm := ExtractServerManager(c)
// Immediately suspend the server to prevent a user from attempting // Immediately suspend the server to prevent a user from attempting
// to start it while this process is running. // to start it while this process is running.

View File

@ -497,7 +497,7 @@ func postServerChmodFile(c *gin.Context) {
} }
func postServerUploadFiles(c *gin.Context) { func postServerUploadFiles(c *gin.Context) {
serverManager := ServerManagerFromContext(c) serverManager := ExtractServerManager(c)
token := tokens.UploadPayload{} token := tokens.UploadPayload{}
if err := tokens.ParseToken([]byte(c.Query("token")), &token); err != nil { if err := tokens.ParseToken([]byte(c.Query("token")), &token); err != nil {

View File

@ -12,7 +12,7 @@ import (
// Upgrades a connection to a websocket and passes events along between. // Upgrades a connection to a websocket and passes events along between.
func getServerWebsocket(c *gin.Context) { func getServerWebsocket(c *gin.Context) {
serverManager := ServerManagerFromContext(c) serverManager := ExtractServerManager(c)
s := serverManager.Get(c.Param("server")) s := serverManager.Get(c.Param("server"))
handler, err := websocket.GetHandler(s, c.Writer, c.Request) handler, err := websocket.GetHandler(s, c.Writer, c.Request)
if err != nil { if err != nil {

View File

@ -27,7 +27,7 @@ func getSystemInformation(c *gin.Context) {
// Returns all of the servers that are registered and configured correctly on // Returns all of the servers that are registered and configured correctly on
// this wings instance. // this wings instance.
func getAllServers(c *gin.Context) { func getAllServers(c *gin.Context) {
serverManager := ServerManagerFromContext(c) serverManager := ExtractServerManager(c)
c.JSON(http.StatusOK, serverManager.GetAll()) c.JSON(http.StatusOK, serverManager.GetAll())
} }
@ -52,7 +52,7 @@ func postCreateServer(c *gin.Context) {
// Plop that server instance onto the request so that it can be referenced in // Plop that server instance onto the request so that it can be referenced in
// requests from here-on out. // requests from here-on out.
serverManager := ServerManagerFromContext(c) serverManager := ExtractServerManager(c)
serverManager.Add(install.Server()) serverManager.Add(install.Server())
// Begin the installation process in the background to not block the request // Begin the installation process in the background to not block the request

View File

@ -323,7 +323,7 @@ func postTransfer(c *gin.Context) {
i.Server().Events().Publish(server.TransferLogsEvent, output) i.Server().Events().Publish(server.TransferLogsEvent, output)
} }
serverManager := ServerManagerFromContext(c) serverManager := ExtractServerManager(c)
// Mark the server as transferring to prevent problems later on during the process and // Mark the server as transferring to prevent problems later on during the process and
// then push the server into the global server collection for this instance. // then push the server into the global server collection for this instance.