From 0c17e240f400b59bda424c653a082850d79dad8a Mon Sep 17 00:00:00 2001 From: antony1060 Date: Wed, 24 Mar 2021 10:26:03 +0100 Subject: [PATCH 1/3] Added app name --- config/config.go | 2 ++ server/console.go | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/config/config.go b/config/config.go index 7f28d8f..137b921 100644 --- a/config/config.go +++ b/config/config.go @@ -247,6 +247,8 @@ type Configuration struct { // if the debug flag is passed through the command line arguments. Debug bool + AppName string `json:"app_name" yaml:"app_name"` + // A unique identifier for this node in the Panel. Uuid string diff --git a/server/console.go b/server/console.go index 1a41f2b..23043a4 100644 --- a/server/console.go +++ b/server/console.go @@ -127,6 +127,6 @@ func (s *Server) Throttler() *ConsoleThrottler { func (s *Server) PublishConsoleOutputFromDaemon(data string) { s.Events().Publish( ConsoleOutputEvent, - colorstring.Color(fmt.Sprintf("[yellow][bold][Pterodactyl Daemon]:[default] %s", data)), + colorstring.Color(fmt.Sprintf("[yellow][bold][%s Daemon]:[default] %s", config.Get().AppName, data)), ) } From 52fcf1e37f38dd37a564f92da623d256a696f850 Mon Sep 17 00:00:00 2001 From: Antony Date: Wed, 24 Mar 2021 11:24:54 +0100 Subject: [PATCH 2/3] Added defaults Co-authored-by: Jakob --- config/config.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/config/config.go b/config/config.go index 137b921..eb1e2c0 100644 --- a/config/config.go +++ b/config/config.go @@ -247,7 +247,7 @@ type Configuration struct { // if the debug flag is passed through the command line arguments. Debug bool - AppName string `json:"app_name" yaml:"app_name"` + AppName string `default:"Pterodactyl" json:"app_name" yaml:"app_name"` // A unique identifier for this node in the Panel. Uuid string @@ -619,4 +619,4 @@ func getSystemName() (string, error) { return "", err } return release["ID"], nil -} \ No newline at end of file +} From 1b1eaa3171801ff733ce3230c0a7391d9df17193 Mon Sep 17 00:00:00 2001 From: Dane Everitt Date: Sat, 3 Apr 2021 11:11:36 -0700 Subject: [PATCH 3/3] Avoid expensive copies of the config for every line output --- server/console.go | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/server/console.go b/server/console.go index 23043a4..5689191 100644 --- a/server/console.go +++ b/server/console.go @@ -13,6 +13,11 @@ import ( "github.com/pterodactyl/wings/system" ) +// appName is a local cache variable to avoid having to make expensive copies of +// the configuration every time we need to send output along to the websocket for +// a server. +var appName string + var ErrTooMuchConsoleData = errors.New("console is outputting too much data") type ConsoleThrottler struct { @@ -122,11 +127,14 @@ func (s *Server) Throttler() *ConsoleThrottler { return s.throttler } -// Sends output to the server console formatted to appear correctly as being sent -// from Wings. +// PublishConsoleOutputFromDaemon sends output to the server console formatted +// to appear correctly as being sent from Wings. func (s *Server) PublishConsoleOutputFromDaemon(data string) { + if appName == "" { + appName = config.Get().AppName + } s.Events().Publish( ConsoleOutputEvent, - colorstring.Color(fmt.Sprintf("[yellow][bold][%s Daemon]:[default] %s", config.Get().AppName, data)), + colorstring.Color(fmt.Sprintf("[yellow][bold][%s Daemon]:[default] %s", appName, data)), ) }