From 068d5de355980c264c8f156995e9fd32939d8f50 Mon Sep 17 00:00:00 2001
From: CreepyCrafter24 <33260128+CreepyCrafter24@users.noreply.github.com>
Date: Wed, 11 Mar 2020 19:41:51 +0100
Subject: [PATCH] Fix mainBin when not specified

---
 UpTool build tool/Program.cs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/UpTool build tool/Program.cs b/UpTool build tool/Program.cs
index ab7231b..0289b40 100644
--- a/UpTool build tool/Program.cs	
+++ b/UpTool build tool/Program.cs	
@@ -57,7 +57,6 @@ namespace UpTool_build_tool
                 .Where(s => new[] { ".xml", ".pdb" }.Contains(Path.GetExtension(s)))
                 .ToList().ForEach(File.Delete);
             Console.WriteLine("Creating batch scripts...");
-            string programName = Path.GetFileNameWithoutExtension(mainBin);
             if (noShortcuts)
             {
                 File.WriteAllText(Path.Combine(tempPath, "Install.bat"), "@echo off\r\necho INSTALL\r\ntimeout /t 1");
@@ -65,6 +64,8 @@ namespace UpTool_build_tool
             }
             else
             {
+                mainBin = string.IsNullOrWhiteSpace(mainBin) ? Directory.GetFiles(binDir, "*.exe")[0] : mainBin;
+                string programName = Path.GetFileNameWithoutExtension(mainBin);
                 File.WriteAllText(Path.Combine(tempPath, "Install.bat"),
                     $"@echo off\r\necho INSTALL\r\npowershell \"$s=(New-Object -COM WScript.Shell).CreateShortcut('%appdata%\\Microsoft\\Windows\\Start Menu\\Programs\\{programName}.lnk');$s.TargetPath='%cd%\\{programName}.exe';$s.Save()\"\r\ntimeout /t 1");
                 File.WriteAllText(Path.Combine(tempPath, "Remove.bat"),