PUSH

-> Added a logger for easy debug
-> Renamed the rpc file
-> Added a ascii art for the start of the app
-> Some more debugging helpers
-> Added a hooker for PteroConsole LIB

Forms/FrmServerController.cs

@@ -1,18 +1,47 @@
-namespace PteroController.Forms
+using Newtonsoft.Json;
+using PteroController.PteroConsoleHook;
+using PteroController.Pterodactyl;
+using System.Net;
+
+namespace PteroController.Forms
 {
     public partial class FrmServerController : Form
     {
         private string? serverIdentifier;
-
         public FrmServerController(string serverIdentifier)
         {
             InitializeComponent();
             this.serverIdentifier = serverIdentifier;
+
+        }
+        private async void initPteroConsole()
+        {
+            try
+            {
+                var console = new PteroConsole.NET.PteroConsole();
+                console.RequestToken += pteroConsole =>
+                {
+                    Console.WriteLine("Revoking token");
+                    var wc = new WebClient();
+                    wc.Headers.Add("Authorization", "Bearer " + Pterodactyl.User.Info.panel_api_key);
+                    var raw = wc.DownloadString($"{Pterodactyl.User.Info.panel_url}/api/client/servers/{serverIdentifier}/websocket");
+                    var data = JsonConvert.DeserializeObject<WebsocketDataResource>(raw).Data;
+                    return data.Token;
+                };
+                var wc = new WebClient();
+                wc.Headers.Add("Authorization", "Bearer " + Pterodactyl.User.Info.panel_api_key);
+                var raw = wc.DownloadString($"{Pterodactyl.User.Info.panel_url}/api/client/servers/{serverIdentifier}/websocket");
+                var data = JsonConvert.DeserializeObject<WebsocketDataResource>(raw).Data;
+                await console.Connect(Pterodactyl.User.Info.panel_url, data.Socket, data.Token);
+            } catch (Exception ex)
+            {
+                Console.WriteLine(ex.ToString());
+                Program.Alert("We are sorry but we can't launch the stats", FrmAlert.enmType.Warning);
+            }
         }
-
         private void FrmServerController_Load(object sender, EventArgs e)
         {
-
+            initPteroConsole();
         }
 
         private void lblminimize_Click(object sender, EventArgs e)
@@ -38,6 +67,11 @@ private void btnservers_Click(object sender, EventArgs e)
             x.Show();
             this.Hide();
         }
+
+        private void pblblpanellogo_Click(object sender, EventArgs e)
+        {
+
+        }
     }
 
 }

Forms/FrmServerController.resx

@@ -121,7 +121,7 @@
   <data name="btnservers.Image" type="System.Drawing.Bitmap, System.Drawing" mimetype="application/x-microsoft.net.object.bytearray.base64">
     <value>
         iVBORw0KGgoAAAANSUhEUgAAAEIAAABCCAYAAADjVADoAAAABGdBTUEAALGPC/xhBQAAAAlwSFlzAAAL
-        DQAACw0B7QfALAAAAzdJREFUeF7tm02IjVEcxu/MhBoLWRhlwoKUlAWasJgy2RiRj0lShs3MRpGS2FBE
+        DAAACwwBP0AiyAAAAzdJREFUeF7tm02IjVEcxu/MhBoLWRhlwoKUlAWasJgy2RiRj0lShs3MRpGS2FBE
         1EQWForNYImVDQ0pdmPnY8dGSigWvvJxPf/zf+ad933dxft5nXM6v3o259z/c57/ae69877nvY1AIBAI
         BNpHs9nsgrZDV6FH0GQb9QC6BPUzzv8BAdZAzyEbkE1ZzGjtA4sOQF8kgUW8hZYxYv1gsR7og6xsIc+g
         mYxaL1hI3pc2c4BR6wOLzIA+muXsZZJx6wOL9OtaVvMHmsfI9YAFzpil7GcbI9cDFniq61jPBUauHpjP
@@ -141,7 +141,7 @@
   <data name="btnlogout.Image" type="System.Drawing.Bitmap, System.Drawing" mimetype="application/x-microsoft.net.object.bytearray.base64">
     <value>
         iVBORw0KGgoAAAANSUhEUgAAAEAAAABACAYAAACqaXHeAAAABGdBTUEAALGPC/xhBQAAAAlwSFlzAAAL
-        DQAACw0B7QfALAAAArpJREFUeF7tmktPFFEQhUfjBl1BcOdCwKDga0E0blzjz9BfQDRujMYY16Ib/RPu
+        DAAACwwBP0AiyAAAArpJREFUeF7tmktPFFEQhUfjBl1BcOdCwKDga0E0blzjz9BfQDRujMYY16Ib/RPu
         fPwRE18s3EHia6NC1IWOp24dMeCtMXTf7mrp+pKTMN01p85UmDu3e2YQBEEQBEEQFGQ4HB6A5qGFQhKv
         /bTvLgh5DnoKfYNKI55PoLNs1y0Q7Cb0E2oa6XGDbbsBAl1N0drlMtv7giBT0PcUqV2k5wxj+IEQ91Ic
         H5YZww+EeKVZXHjBGH4gxIZmyfIRelNT4mGxwRh+MIjFFMsqA49ptcrDMj+Yw+IQyyojHmqVh2V+MIdF

Forms/FrmSessions.cs

@@ -28,7 +28,7 @@ private void PopulateListBoxWithSessions()
         }
         private void FrmSessions_Load(object sender, EventArgs e)
         {
-            RPCHandler discordRPC = new RPCHandler();
+            DiscordRPCHandler discordRPC = new DiscordRPCHandler();
             discordRPC.InitializeRPC();
         }
 

Handlers/RPCHandler.cs → Handlers/DiscordRPCHandler.cs

@@ -3,7 +3,7 @@
 
 namespace PteroController.Handlers
 {
-    public class RPCHandler
+    public class DiscordRPCHandler
     {
         public DiscordRpcClient? client;
 
@@ -46,7 +46,7 @@ public void InitializeRPC()
             }
             catch (Exception ex)
             {
-                Console.WriteLine("Failed to load discord rpc: " + ex.Message);
+                Program.logger.Log(Managers.LogType.Error, "[Handlers.DiscordRPCHandler.cs]: \n" + ex.Message);
             }
         }
     }

Handlers/RegistryHandler.cs

@@ -11,55 +11,78 @@ public class RegistryHandler
 
         public void CreateAccountData(string sessionName, string url, string password, string apiKey)
         {
-            using (RegistryKey key = Registry.CurrentUser.CreateSubKey($"{RegistryAccountBasePath}\\{sessionName}"))
+            try
             {
-                key.SetValue("panel_url", url);
-                key.SetValue("panel_pwd", password);
-                key.SetValue("api_key", apiKey);
+                using (RegistryKey key = Registry.CurrentUser.CreateSubKey($"{RegistryAccountBasePath}\\{sessionName}"))
+                {
+                    key.SetValue("panel_url", url);
+                    key.SetValue("panel_pwd", password);
+                    key.SetValue("api_key", apiKey);
+                }
+            } catch (Exception ex)
+            {
+                Program.logger.Log(Managers.LogType.Error, "[Handlers.RegistryHandler.cs]: \n" + ex.Message);
             }
         }
         public List<string> ListSessions()
         {
-            List<string> sessionInfo = new List<string>();
 
-            using (RegistryKey key = Registry.CurrentUser.OpenSubKey(RegistryAccountBasePath))
+            try
             {
-                if (key != null)
+                List<string> sessionInfo = new List<string>();
+
+                using (RegistryKey key = Registry.CurrentUser.OpenSubKey(RegistryAccountBasePath))
                 {
-                    foreach (string sessionName in key.GetSubKeyNames())
+                    if (key != null)
                     {
-                        string panelUrl = key.OpenSubKey(sessionName)?.GetValue("panel_url")?.ToString();
-                        if (!string.IsNullOrEmpty(panelUrl))
+                        foreach (string sessionName in key.GetSubKeyNames())
                         {
-                            string displayText = $"{sessionName} ({panelUrl})";
-                            sessionInfo.Add(displayText);
+                            string panelUrl = key.OpenSubKey(sessionName)?.GetValue("panel_url")?.ToString();
+                            if (!string.IsNullOrEmpty(panelUrl))
+                            {
+                                string displayText = $"{sessionName} ({panelUrl})";
+                                sessionInfo.Add(displayText);
+                            }
                         }
                     }
                 }
-            }
 
-            return sessionInfo;
+                return sessionInfo;
+            }
+            catch (Exception ex)
+            {
+                Program.logger.Log(Managers.LogType.Error, "[Handlers.RegistryHandler.cs]: \n" + ex.Message);
+                return null;
+            }
         }
         public SessionInfo GetSessionInfo(string sessionName)
         {
-            using (RegistryKey key = Registry.CurrentUser.OpenSubKey($"{RegistryAccountBasePath}\\{sessionName}"))
+            try
             {
-                if (key != null)
+                using (RegistryKey key = Registry.CurrentUser.OpenSubKey($"{RegistryAccountBasePath}\\{sessionName}"))
                 {
-                    string panelUrl = key.GetValue("panel_url")?.ToString();
-                    string panelPwd = key.GetValue("panel_pwd")?.ToString();
-                    string apiKey = key.GetValue("api_key")?.ToString();
-
-                    return new SessionInfo
+                    if (key != null)
                     {
-                        PanelUrl = panelUrl,
-                        PanelPwd = panelPwd,
-                        ApiKey = apiKey
-                    };
+                        string panelUrl = key.GetValue("panel_url")?.ToString();
+                        string panelPwd = key.GetValue("panel_pwd")?.ToString();
+                        string apiKey = key.GetValue("api_key")?.ToString();
+
+                        return new SessionInfo
+                        {
+                            PanelUrl = panelUrl,
+                            PanelPwd = panelPwd,
+                            ApiKey = apiKey
+                        };
+                    }
                 }
-            }
 
-            return null;
+                return null;
+            }
+            catch (Exception ex)
+            {
+                Program.logger.Log(Managers.LogType.Error, "[Handlers.RegistryHandler.cs]: \n" + ex.Message);
+                return null;
+            }
         }
         public void DeleteAccountData(string sessionName)
         {

Managers/LoggerManager.cs

@@ -0,0 +1,94 @@
+namespace PteroController.Managers
+{
+    public enum LogType
+    {
+        Info,
+        Warning,
+        Error
+    }
+
+    public class LoggerManager
+    {
+        private string logFilePath;
+
+        public LoggerManager()
+        {
+            string logDirectory = "logs";
+            string logFileName = "log.txt";
+            string logDirectoryPath = Path.Combine(Directory.GetCurrentDirectory(), logDirectory);
+            logFilePath = Path.Combine(logDirectoryPath, logFileName);
+            Directory.CreateDirectory(logDirectoryPath);
+            RenameLogFile();
+        }
+
+        public void Log(LogType type, string message)
+        {
+            string timestamp = DateTime.Now.ToString("HH:mm:ss");
+            string logText = $"[{timestamp}] [{type.ToString()}] {message}";
+
+            ConsoleColor color = ConsoleColor.White;
+            switch (type)
+            {
+                case LogType.Info:
+                    color = ConsoleColor.Blue;
+                    break;
+                case LogType.Warning:
+                    color = ConsoleColor.Yellow;
+                    break;
+                case LogType.Error:
+                    color = ConsoleColor.Red;
+                    break;
+            }
+            Console.ForegroundColor = color;
+            Console.WriteLine(logText);
+            Console.ResetColor();
+            AppendToFile(logText);
+        }
+
+        private void AppendToFile(string logText)
+        {
+            try
+            {
+                using StreamWriter writer = File.AppendText(logFilePath);
+                writer.WriteLine(logText);
+            }
+            catch (Exception ex)
+            {
+                string timestamp = DateTime.Now.ToString("HH:mm:ss");
+                Console.ForegroundColor = ConsoleColor.Red;
+                Console.WriteLine($"[{timestamp}] [Error] Error writing to log file: {ex.Message}");
+                Console.ResetColor();
+            }
+        }
+
+        private void RenameLogFile()
+        {
+            if (File.Exists(logFilePath))
+            {
+                string logFileNameWithoutExtension = Path.GetFileNameWithoutExtension(logFilePath);
+                string logFileExtension = Path.GetExtension(logFilePath);
+#pragma warning disable
+                string logDirectoryPath = Path.GetDirectoryName(logFilePath);
+                string newLogFileName = GetUniqueLogFileName(logDirectoryPath, logFileNameWithoutExtension, logFileExtension);
+#pragma warning restore
+                string newLogFilePath = Path.Combine(logDirectoryPath, newLogFileName);
+                File.Move(logFilePath, newLogFilePath);
+            }
+        }
+
+        private string GetUniqueLogFileName(string directoryPath, string fileNameWithoutExtension, string fileExtension)
+        {
+            string uniqueFileName = $"{fileNameWithoutExtension}-{DateTime.Now:yyyy-MM-dd}{fileExtension}";
+            string uniqueFilePath = Path.Combine(directoryPath, uniqueFileName);
+
+            int counter = 1;
+            while (File.Exists(uniqueFilePath))
+            {
+                uniqueFileName = $"{fileNameWithoutExtension}-{counter++}-{DateTime.Now:yyyy-MM-dd}{fileExtension}";
+                uniqueFilePath = Path.Combine(directoryPath, uniqueFileName);
+            }
+
+            return uniqueFileName;
+        }
+    }
+}

Program.cs

@@ -1,5 +1,5 @@
 using PteroController.Forms;
-using System.IO;
+using PteroController.Managers;
 
 namespace PteroController
 {
@@ -9,23 +9,34 @@ internal static class Program
         ///  The main entry point for the application.
         /// </summary>
         public static string AppVersion = "1.0.0";
+        public static LoggerManager logger = new LoggerManager();
+        public static string mcascii = @" 
+  _____  _                  _____            _             _ _           
+ |  __ \| |                / ____|          | |           | | |          
+ | |__) | |_ ___ _ __ ___ | |     ___  _ __ | |_ _ __ ___ | | | ___ _ __ 
+ |  ___/| __/ _ \ '__/ _ \| |    / _ \| '_ \| __| '__/ _ \| | |/ _ \ '__|
+ | |    | ||  __/ | | (_) | |___| (_) | | | | |_| | | (_) | | |  __/ |   
+ |_|     \__\___|_|  \___/ \_____\___/|_| |_|\__|_|  \___/|_|_|\___|_|   
+                                                                                 
+    ";
         [STAThread]
         public static void Main()
         {
             ApplicationConfiguration.Initialize();
             if (!OperatingSystem.IsWindows())
             {
-                Console.WriteLine("We are sorry, but we blocked this app from running on any other OS than Windows!");
+                logger.Log(LogType.Error, "We are sorry, but we blocked this app from running on any other OS than Windows!");
                 Application.Exit();
             }
             if (!File.Exists(Application.StartupPath + @"\PteroConsole.exe"))
             {
                 Application.Run(new FrmInstallTools());
-            } else
+            }
+            else
             {
                 Application.Run(new FrmLoading());
             }
-            
+
         }
         public static void Alert(string msg, FrmAlert.enmType type)
         {

PteroConsoleHook/WebsocketData.cs

@@ -0,0 +1,13 @@
+using Newtonsoft.Json;
+
+namespace PteroController.PteroConsoleHook
+{
+    public class WebsocketData
+    {
+        [JsonProperty("token")]
+        public string? Token { get; set; }
+
+        [JsonProperty("socket")]
+        public string? Socket { get; set; }
+    }
+}

PteroConsoleHook/WebsocketDataResource.cs

@@ -0,0 +1,11 @@
+using Newtonsoft.Json;
+
+namespace PteroController.PteroConsoleHook
+{
+    public class WebsocketDataResource
+    {
+        [JsonProperty("data")]
+        public WebsocketData? Data { get; set; }
+    }
+
+}