diff --git a/src/NzbDrone.Common.Test/ProcessProviderFixture.cs b/src/NzbDrone.Common.Test/ProcessProviderFixture.cs index f0802d10d..6951e2c42 100644 --- a/src/NzbDrone.Common.Test/ProcessProviderFixture.cs +++ b/src/NzbDrone.Common.Test/ProcessProviderFixture.cs @@ -1,4 +1,4 @@ -using System; +using System; using System.ComponentModel; using System.Diagnostics; using System.IO; @@ -102,6 +102,55 @@ namespace NzbDrone.Common.Test Subject.Exists(DummyApp.DUMMY_PROCCESS_NAME).Should().BeFalse(); } + [Test] + [Explicit] + public void Should_be_able_to_start_powershell() + { + WindowsOnly(); + + var tempDir = GetTempFilePath(); + var tempScript = Path.Combine(tempDir, "myscript.ps1"); + + Directory.CreateDirectory(tempDir); + + File.WriteAllText(tempScript, "Write-Output 'Hello There'\r\n"); + + try + { + var result = Subject.StartAndCapture(tempScript); + + result.Standard.First().Content.Should().Be("Hello There"); + } + catch (Win32Exception ex) when (ex.NativeErrorCode == 2) + { + Assert.Fail("No Powershell available?!?"); + } + } + + [Test] + public void Should_be_able_to_start_python() + { + WindowsOnly(); + + var tempDir = GetTempFilePath(); + var tempScript = Path.Combine(tempDir, "myscript.py"); + + Directory.CreateDirectory(tempDir); + + File.WriteAllText(tempScript, "print(\"Hello There\")\r\n"); + + try + { + var result = Subject.StartAndCapture(tempScript); + + result.Standard.First().Content.Should().Be("Hello There"); + } + catch (Win32Exception ex) when (ex.NativeErrorCode == 2) + { + Assert.Inconclusive("No Python available"); + } + } + [Test] [Platform(Exclude = "MacOsX")] [Retry(3)] diff --git a/src/NzbDrone.Common/Processes/ProcessProvider.cs b/src/NzbDrone.Common/Processes/ProcessProvider.cs index 103cdc183..b558c8a31 100644 --- a/src/NzbDrone.Common/Processes/ProcessProvider.cs +++ b/src/NzbDrone.Common/Processes/ProcessProvider.cs @@ -379,6 +379,16 @@ namespace NzbDrone.Common.Processes return ("cmd.exe", $"/c {path} {args}"); } + if (OsInfo.IsWindows && path.EndsWith(".ps1", StringComparison.InvariantCultureIgnoreCase)) + { + return ("powershell.exe", $"-ExecutionPolicy Bypass -NoProfile -File {path} {args}"); + } + + if (OsInfo.IsWindows && path.EndsWith(".py", StringComparison.InvariantCultureIgnoreCase)) + { + return ("python.exe", $"{path} {args}"); + } + return (path, args); } }