Change shortcuts based on which launcher is downloaded

Author: WumboSpasm

src/Common.cs

@@ -168,6 +168,9 @@ public static string SourcePath2
                 set { Main.SourcePath2.Text = value; }
             }
 
+            // Path to launcher executable based on selection
+            public static string LauncherPath { get; set; } = "";
+
             // Flag to control how operation window will function
             // 0 is for downloading Flashpoint
             // 1 is for adding/removing components
@@ -246,6 +249,7 @@ public static TreeNode AddNodeToList(XmlNode child, TreeNodeCollection parent, b
                     Component component = new Component(child);
 
                     listNode.Text = component.Title;
+                    listNode.Name = component.ID;
 
                     if (component.ID.StartsWith("required"))
                     {
@@ -259,6 +263,7 @@ public static TreeNode AddNodeToList(XmlNode child, TreeNodeCollection parent, b
                     Category category = new Category(child);
 
                     listNode.Text = category.Title;
+                    listNode.Name = category.ID;
 
                     if (category.ID.StartsWith("required"))
                     {

src/Forms/FinishOperation.cs

@@ -27,11 +27,20 @@ private void FinishDownload_Exit(object sender, EventArgs e)
             {
                 if (RunOnClose.Checked)
                 {
-                    var flashpointProcess = new Process();
-                    flashpointProcess.StartInfo.UseShellExecute = true;
-                    flashpointProcess.StartInfo.FileName = "Flashpoint.exe";
-                    flashpointProcess.StartInfo.WorkingDirectory = Path.Combine(FPM.DestinationPath, @"Launcher");
-                    flashpointProcess.Start();
+                    if (FPM.LauncherPath != "")
+                    {
+                        string[] path = FPM.LauncherPath.Split('\\');
+
+                        var flashpointProcess = new Process();
+                        flashpointProcess.StartInfo.UseShellExecute = true;
+                        flashpointProcess.StartInfo.FileName = path[1];
+                        flashpointProcess.StartInfo.WorkingDirectory = Path.Combine(FPM.DestinationPath, path[0]);
+                        flashpointProcess.Start();
+                    }
+                    else
+                    {
+                        Process.Start("explorer.exe", FPM.DestinationPath);
+                    }
                 }
             }
 

src/Forms/MainDownload.cs

@@ -100,6 +100,24 @@ private void DownloadButton_Click(object sender, EventArgs e)
 
             FPM.CheckDependencies(ComponentList);
 
+            FPM.LauncherPath = "";
+            foreach (TreeNode launcher in ComponentList.Nodes["launcher"].Nodes)
+            {
+                if (launcher.Checked)
+                {
+                    if (launcher.Name == "launcher-electron")
+                    {
+                        FPM.LauncherPath = @"Launcher\Flashpoint.exe";
+                        break;
+                    }
+                    else if (launcher.Name == "launcher-winforms")
+                    {
+                        FPM.LauncherPath = @"SharpLauncher\SharpLauncher.exe";
+                        break;
+                    }
+                }
+            }
+
             FPM.OperateMode = 0;
 
             var operationWindow = new Operation();

src/Forms/Operation.cs

@@ -301,8 +301,17 @@ await Task.Run(() =>
                     foreach (string path in shortcutPaths)
                     {
                         var shortcut = new IWshRuntimeLibrary.WshShell().CreateShortcut(Path.Combine(path, "Flashpoint.lnk"));
-                        shortcut.TargetPath = Path.Combine(FPM.DestinationPath, @"Launcher\Flashpoint.exe");
-                        shortcut.WorkingDirectory = Path.Combine(FPM.DestinationPath, @"Launcher");
+
+                        if (FPM.LauncherPath != "")
+                        {
+                            shortcut.TargetPath = Path.Combine(FPM.DestinationPath, FPM.LauncherPath);
+                            shortcut.WorkingDirectory = Path.Combine(FPM.DestinationPath, FPM.LauncherPath.Split('\\')[0]);
+                        }
+                        else
+                        {
+                            shortcut.TargetPath = FPM.DestinationPath;
+                        }
+
                         shortcut.Description = "Shortcut to Flashpoint";
                         shortcut.Save();
                     }