Fix lack of error handling when process isn't open

MechaDragonX · Mar 28, 2020 · eb7ac7a · eb7ac7a
1 parent f5d0880
commit eb7ac7a
Showing 1 changed file with 19 additions and 4 deletions.
diff --git a/Program.cs b/Program.cs
@@ -21,7 +21,10 @@ class Program
         private static void Main(string[] args)
         {
             Initialize();
-            while(true)
+            if (Process.GetProcesses().Where(x => x.ProcessName.StartsWith("DOSBox")).Count() == 0)
+                return;
+
+            while (true)
             {
                 Update();
                 if(Process.GetProcesses().Where(x => x.ProcessName.StartsWith("DOSBox")).Count() == 0)
@@ -89,9 +92,16 @@ private static void Deinitialize()
 
         private static void OnUpdate()
         {
-            if(Process.GetProcesses().Where(x => x.ProcessName.StartsWith("DOSBox")).ToList()[0].MainWindowTitle != windowTitle)
+            Process process;
+            try
             {
-                dos = Process.GetProcesses().Where(x => x.ProcessName.StartsWith("DOSBox")).ToList()[0];
+                process = Process.GetProcesses().Where(x => x.ProcessName.StartsWith("DOSBox")).ToList()[0];
+            }
+            catch(IndexOutOfRangeException) { return;  }
+
+            if(process.MainWindowTitle != windowTitle)
+            {
+                dos = process;
                 windowTitle = dos.MainWindowTitle;
                 setNewPresence();
             }
@@ -107,7 +117,12 @@ private static void setNewPresence()
                     i--;
                 }
             }
-            string status = new StringBuilder($"v{titleParts[0].Split(' ')[1]}").AppendFormat(", {0}, {1}", titleParts[1], titleParts[2]).ToString();
+            string status;
+            try
+            {
+                status = new StringBuilder($"v{titleParts[0].Split(' ')[1]}").AppendFormat(", {0}, {1}", titleParts[1], titleParts[2]).ToString();
+            }
+            catch(IndexOutOfRangeException) { return; }
 
             client.SetPresence(new RichPresence
             {