public override RuntimeValue Evaluate(IVariableFunctionContext context)
{
if (!(context is IOperationExecutionContext execContext))
{
throw new NotSupportedException("This function can currently only be used within an execution.");
}
var job = new ExecutePowerShellJob
{
CollectOutput = true,
ScriptText = this.ScriptText,
Variables = PowerShellScriptRunner.ExtractVariables(this.ScriptText, execContext)
};
var jobExecuter = execContext.Agent.GetService <IRemoteJobExecuter>();
var result = (ExecutePowerShellJob.Result)jobExecuter.ExecuteJobAsync(job, execContext.CancellationToken).GetAwaiter().GetResult();
if (result.Output.Count == 1)
{
return(result.Output[0]);
}
else
{
return(new RuntimeValue(result.Output));
}
}
public override async Task ExecuteAsync(IOperationExecutionContext context)
{
if (context.Simulation && !this.RunOnSimulation)
{
this.LogInformation("Executing PowerShell script...");
return;
}
var jobRunner = context.Agent.GetService <IRemoteJobExecuter>();
var job = new ExecutePowerShellJob
{
ScriptText = this.ScriptText,
DebugLogging = this.DebugLogging,
VerboseLogging = this.VerboseLogging,
CollectOutput = false,
LogOutput = true,
Variables = PowerShellScriptRunner.ExtractVariables(this.ScriptText, context),
Isolated = this.Isolated,
WorkingDirectory = context.WorkingDirectory
};
job.MessageLogged += (s, e) => this.Log(e.Level, e.Message);
job.ProgressUpdate += (s, e) => Interlocked.Exchange(ref this.currentProgress, e);
var result = (ExecutePowerShellJob.Result) await jobRunner.ExecuteJobAsync(job, context.CancellationToken);
PSUtil.LogExit(this, result.ExitCode, this.SuccessExitCode);
}
public override async Task <PersistedConfiguration> CollectAsync(IOperationCollectionContext context)
{
if (!this.ValidateConfiguration())
{
return(null);
}
ExecutePowerShellJob.Result result;
if (!string.IsNullOrWhiteSpace(this.CollectScriptAsset))
{
result = await PSUtil.ExecuteScriptAsync(
logger : this,
context : context,
fullScriptName : this.CollectScriptAsset,
arguments : this.CollectScriptParams ?? new Dictionary <string, RuntimeValue>(),
outArguments : new Dictionary <string, RuntimeValue>(),
collectOutput : !this.UseExitCode,
progressUpdateHandler : (s, e) => Interlocked.Exchange(ref this.currentProgress, e)
);
}
else
{
var jobRunner = context.Agent.GetService <IRemoteJobExecuter>();
var job = new ExecutePowerShellJob
{
ScriptText = this.CollectScript,
DebugLogging = this.DebugLogging,
VerboseLogging = this.VerboseLogging,
CollectOutput = !this.UseExitCode,
LogOutput = this.UseExitCode,
Variables = PowerShellScriptRunner.ExtractVariables(this.CollectScript, context)
};
job.MessageLogged += (s, e) => this.Log(e.Level, e.Message);
job.ProgressUpdate += (s, e) => Interlocked.Exchange(ref this.currentProgress, e);
result = await jobRunner.ExecuteJobAsync(job, context.CancellationToken) as ExecutePowerShellJob.Result;
}
if (result.ExitCode != null)
{
this.LogDebug("Script exit code: " + result.ExitCode);
}
return(new KeyValueConfiguration
{
Key = this.ConfigurationKey,
Value = this.UseExitCode ? result.ExitCode?.ToString() : string.Join(", ", result.Output)
});
}
