added requirement to PSEnsure to use AHEXECMODE

Inedo · Feb 18, 2021 · ba4e5db · ba4e5db
1 parent 95c90db
commit ba4e5db
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/Scripting/InedoExtension/PowerShell/PSUtil.cs b/Scripting/InedoExtension/PowerShell/PSUtil.cs
@@ -88,6 +88,15 @@ void setOutputVariable(string oe)
                 variables = arguments.ToDictionary(a => a.Key, a => a.Value);
             }
 
+            if (executionMode == PsExecutionMode.Configure && string.IsNullOrEmpty(scriptInfo?.ExecutionModeVariableName))
+            {
+                logger.LogError(
+                    ".AHEXECMODE additional help was not detected. When using PSEnsure to remediate drift, you must specify the name of " +
+                    "a variable that will capture \"Collect\" or \"Configure\" in the .AHEXECMODE help."
+                );
+                return null;
+            }
+
             var jobRunner = context.Agent.GetService<IRemoteJobExecuter>();
 
             var job = new ExecutePowerShellJob