From 07277ac4af24a9386c25b19f55ac8068c00707e2 Mon Sep 17 00:00:00 2001
From: Rob Reynolds <ferventcoder@gmail.com>
Date: Sun, 24 Apr 2016 18:09:03 -0500
Subject: [PATCH] (GH-709) Get Exit code from PowerShell Host

When interacting with a PowerShell host, ensure it returns the proper
exit code by calling the right method that will set it for both system
powershell and the internal host.
---
 .../chocolatey.resources.csproj               |  3 +++
 .../functions/Set-PowerShellExitCode.ps1      | 23 +++++++++++++++++++
 .../services/PowershellService.cs             |  4 ++--
 3 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 src/chocolatey.resources/helpers/functions/Set-PowerShellExitCode.ps1

diff --git a/src/chocolatey.resources/chocolatey.resources.csproj b/src/chocolatey.resources/chocolatey.resources.csproj
index bc2ef12896..226eecc882 100644
--- a/src/chocolatey.resources/chocolatey.resources.csproj
+++ b/src/chocolatey.resources/chocolatey.resources.csproj
@@ -137,6 +137,9 @@
   <ItemGroup>
     <EmbeddedResource Include="helpers\ChocolateyTabExpansion.ps1" />
   </ItemGroup>
+  <ItemGroup>
+    <EmbeddedResource Include="helpers\functions\Set-PowerShellExitCode.ps1" />
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it.
        Other similar extension points exist, see Microsoft.Common.targets.
diff --git a/src/chocolatey.resources/helpers/functions/Set-PowerShellExitCode.ps1 b/src/chocolatey.resources/helpers/functions/Set-PowerShellExitCode.ps1
new file mode 100644
index 0000000000..0132b0174a
--- /dev/null
+++ b/src/chocolatey.resources/helpers/functions/Set-PowerShellExitCode.ps1
@@ -0,0 +1,23 @@
+# Copyright © 2011 - Present RealDimensions Software, LLC
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# 
+# You may obtain a copy of the License at
+# 
+#   http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+Function Set-PowerShellExitCode {
+param (
+  [int]$exitCode
+)
+
+  $host.SetShouldExit($exitCode); 
+  $env:ChocolateyExitCode = $exitCode;
+}
\ No newline at end of file
diff --git a/src/chocolatey/infrastructure.app/services/PowershellService.cs b/src/chocolatey/infrastructure.app/services/PowershellService.cs
index 272d25187a..d3204a7b40 100644
--- a/src/chocolatey/infrastructure.app/services/PowershellService.cs
+++ b/src/chocolatey/infrastructure.app/services/PowershellService.cs
@@ -564,11 +564,11 @@ private PowerShellExecutionResults run_host(ChocolateyConfiguration config, stri
                             case PipelineState.Failed:
                             case PipelineState.Stopping:
                             case PipelineState.Stopped:
-                                host.SetShouldExit(1);
+                                if (host.ExitCode == 0) host.SetShouldExit(1);
                                 host.HostException = pipeline.PipelineStateInfo.Reason;
                                 break;
                             case PipelineState.Completed:
-                                host.SetShouldExit(0);
+                                if (host.ExitCode == -1) host.SetShouldExit(0);
                                 break;
                         }