Save all scripts on a game and allow them to run as admin.

LANCommander.Playnite.Extension/InstallController.cs

@@ -1,6 +1,7 @@
 using Playnite.SDK;
 using Playnite.SDK.Models;
 using Playnite.SDK.Plugins;
+using LANCommander.SDK.Enums;
 using LANCommander.SDK.Extensions;
 using System;
 using System.Collections.Generic;
@@ -13,6 +14,7 @@ using ICSharpCode.SharpZipLib.Zip;
 using ICSharpCode.SharpZipLib.Core;
 using YamlDotNet.Serialization;
 using YamlDotNet.Serialization.NamingConventions;
+using LANCommander.SDK.Models;
 
 namespace LANCommander.PlaynitePlugin
 {
@@ -20,9 +22,9 @@ namespace LANCommander.PlaynitePlugin
     {
         private LANCommanderLibraryPlugin Plugin;
         private PowerShellRuntime PowerShellRuntime;
-        private Game PlayniteGame;
+        private Playnite.SDK.Models.Game PlayniteGame;
 
-        public LANCommanderInstallController(LANCommanderLibraryPlugin plugin, Game game) : base(game)
+        public LANCommanderInstallController(LANCommanderLibraryPlugin plugin, Playnite.SDK.Models.Game game) : base(game)
         {
             Name = "Install using LANCommander";
             Plugin = plugin;
@@ -49,9 +51,14 @@ namespace LANCommander.PlaynitePlugin
 
             File.WriteAllText(Path.Combine(installDirectory, "_manifest.yml"), GetManifest(gameId));
 
+            SaveScript(game, installDirectory, ScriptType.Install);
+            SaveScript(game, installDirectory, ScriptType.Uninstall);
+            SaveScript(game, installDirectory, ScriptType.NameChange);
+            SaveScript(game, installDirectory, ScriptType.KeyChange);
+
             try
             {
-                PowerShellRuntime.RunInstallScript(PlayniteGame);
+                PowerShellRuntime.RunScript(PlayniteGame, ScriptType.Install);
             }
             catch { }
 
@@ -169,5 +176,33 @@ namespace LANCommander.PlaynitePlugin
 
             return yaml;
         }
+
+        private void SaveScript(LANCommander.SDK.Models.Game game, string installationDirectory, ScriptType type)
+        {
+            var script = game.Scripts.FirstOrDefault(s => s.Type == type);
+
+            if (script == null)
+                return;
+
+            if (script.RequiresAdmin)
+                script.Contents = "# Requires Admin" + "\r\n\r\n" + script.Contents;
+
+            Dictionary<ScriptType, string> filenames = new Dictionary<ScriptType, string>() {
+                { ScriptType.Install, "_install.ps1" },
+                { ScriptType.Uninstall, "_uninstall.ps1" },
+                { ScriptType.NameChange, "_changename.ps1" },
+                { ScriptType.KeyChange, "_changekey.ps1" }
+            };
+
+            if (!filenames.ContainsKey(type))
+                return;
+
+            var filename = Path.Combine(installationDirectory, filenames[type]);
+
+            if (File.Exists(filename))
+                File.Delete(filename);
+
+            File.WriteAllText(filename, script.Contents);
+        }
     }
 }

LANCommander.Playnite.Extension/PowerShellRuntime.cs

@@ -1,4 +1,5 @@
-using Playnite.SDK.Models;
+using LANCommander.SDK.Enums;
+using Playnite.SDK.Models;
 using System;
 using System.Collections.Generic;
 using System.Diagnostics;
@@ -12,35 +13,46 @@ namespace LANCommander.PlaynitePlugin
 {
     internal class PowerShellRuntime
     {
-        public void RunScript(string script)
+        public void RunScript(string path)
+        {
+            var process = new Process();
+            process.StartInfo.FileName = "powershell.exe";
+            process.StartInfo.Arguments = $@"-File ""{path}""";
+            process.Start();
+            process.WaitForExit();
+        }
+
+        public void RunScriptAsAdmin(string path)
         {
             var process = new Process();
             process.StartInfo.FileName = "powershell.exe";
             process.StartInfo.UseShellExecute = true;
             process.StartInfo.Verb = "runas";
-            process.StartInfo.Arguments = $@"-ExecutionPolicy Unrestricted -File ""{script}""";
+            process.StartInfo.Arguments = $@"-ExecutionPolicy Unrestricted -File ""{path}""";
             process.Start();
             process.WaitForExit();
         }
 
-        public void RunInstallScript(Game game)
+        public void RunScript(Game game, ScriptType type)
         {
-            var scriptPath = Path.Combine(game.InstallDirectory, "_install.ps1");
+            Dictionary<ScriptType, string> filenames = new Dictionary<ScriptType, string>() {
+                { ScriptType.Install, "_install.ps1" },
+                { ScriptType.Uninstall, "_uninstall.ps1" },
+                { ScriptType.NameChange, "_changename.ps1" },
+                { ScriptType.KeyChange, "_changekey.ps1" }
+            };
 
-            if (!File.Exists(scriptPath))
+            var path = Path.Combine(game.InstallDirectory, filenames[type]);
-                throw new FileNotFoundException(scriptPath);
 
-            RunScript(scriptPath);
+            if (File.Exists(path))
-        }
+            {
+                var contents = File.ReadAllText(path);
 
-        public void RunUninstallScript(Game game)
+                if (contents.StartsWith("# Requires Admin"))
-        {
+                    RunScriptAsAdmin(path);
-            var scriptPath = Path.Combine(game.InstallDirectory, "_uninstall.ps1");
+                else
-
+                    RunScript(path);
-            if (!File.Exists(scriptPath))
+            }
-                throw new FileNotFoundException(scriptPath);
-
-            RunScript(scriptPath);
         }
     }
 }

LANCommander.Playnite.Extension/UninstallController.cs

@@ -11,6 +11,7 @@ using System.Threading;
 using System.Threading.Tasks;
 using ICSharpCode.SharpZipLib.Zip;
 using ICSharpCode.SharpZipLib.Core;
+using LANCommander.SDK.Enums;
 
 namespace LANCommander.PlaynitePlugin
 {
@@ -30,7 +31,7 @@ namespace LANCommander.PlaynitePlugin
         {
             try
             {
-                PowerShellRuntime.RunUninstallScript(Game);
+                PowerShellRuntime.RunScript(Game, ScriptType.Uninstall);
             }
             catch { }
 

LANCommander.SDK/Enums/ScriptType.cs

@@ -0,0 +1,10 @@
+namespace LANCommander.SDK.Enums
+{
+    public enum ScriptType
+    {
+        Install,
+        Uninstall,
+        NameChange,
+        KeyChange
+    }
+}

LANCommander.SDK/Models/Game.cs

@@ -14,5 +14,6 @@ namespace LANCommander.SDK.Models
         public virtual Company Publisher { get; set; }
         public virtual Company Developer { get; set; }
         public virtual IEnumerable<Archive> Archives { get; set; }
+        public virtual IEnumerable<Script> Scripts { get; set; }
     }
 }

LANCommander.SDK/Models/Script.cs

@@ -0,0 +1,17 @@
+using LANCommander.SDK.Enums;
+using System;
+using System.Collections.Generic;
+using System.Text;
+
+namespace LANCommander.SDK.Models
+{
+    public class Script : BaseModel
+    {
+        public ScriptType Type { get; set; }
+        public string Name { get; set; }
+        public string Description { get; set; }
+        public bool RequiresAdmin { get; set; }
+        public string Contents { get; set; }
+        public virtual Game Game { get; set; }
+    }
+}