lanpartytools/LANCommander
lanpartytools/LANCommander
1
0
Fork 0
Code Issues Pull requests Projects Releases 17 Packages Activity

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

Browse source
This commit is contained in:
Pat Hartl 2023-01-15 04:29:47 -06:00
parent 2d80226693
commit 0681d0a75b
6 changed files with 97 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
LANCommander.Playnite.Extension/InstallController.cs
View file

@ -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);
}
}
}

46
LANCommander.Playnite.Extension/PowerShellRuntime.cs
View file

@ -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))
throw new FileNotFoundException(scriptPath);
var path = Path.Combine(game.InstallDirectory, filenames[type]);
RunScript(scriptPath);
}
if (File.Exists(path))
{
var contents = File.ReadAllText(path);
public void RunUninstallScript(Game game)
{
var scriptPath = Path.Combine(game.InstallDirectory, "_uninstall.ps1");
if (!File.Exists(scriptPath))
throw new FileNotFoundException(scriptPath);
RunScript(scriptPath);
if (contents.StartsWith("# Requires Admin"))
RunScriptAsAdmin(path);
else
RunScript(path);
}
}
}
}

3
LANCommander.Playnite.Extension/UninstallController.cs
View file

@ -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 { }

10
LANCommander.SDK/Enums/ScriptType.cs Normal file
View file

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

1
LANCommander.SDK/Models/Game.cs
View file

@ -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; }
}
}

17
LANCommander.SDK/Models/Script.cs Normal file
View file

@ -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; }
}
}