public static void ExpandValues(IHostContext context, IDictionary <string, string> source, IDictionary <string, string> target)
{
ArgUtil.NotNull(context, nameof(context));
ArgUtil.NotNull(source, nameof(source));
Tracing trace = context.GetTrace(nameof(VarUtil));
trace.Entering();
target = target ?? new Dictionary <string, string>();
// This algorithm does not perform recursive replacement.
// Process each key in the target dictionary.
foreach (string targetKey in target.Keys.ToArray())
{
trace.Verbose($"Processing expansion for: '{targetKey}'");
int startIndex = 0;
int prefixIndex;
int suffixIndex;
string targetValue = target[targetKey] ?? string.Empty;
// Find the next macro within the target value.
while (startIndex < targetValue.Length &&
(prefixIndex = targetValue.IndexOf(Constants.Variables.MacroPrefix, startIndex, StringComparison.Ordinal)) >= 0 &&
(suffixIndex = targetValue.IndexOf(Constants.Variables.MacroSuffix, prefixIndex + Constants.Variables.MacroPrefix.Length, StringComparison.Ordinal)) >= 0)
{
// A candidate was found.
string variableKey = targetValue.Substring(
startIndex: prefixIndex + Constants.Variables.MacroPrefix.Length,
length: suffixIndex - prefixIndex - Constants.Variables.MacroPrefix.Length);
trace.Verbose($"Found macro candidate: '{variableKey}'");
string variableValue;
if (!string.IsNullOrEmpty(variableKey) &&
TryGetValue(trace, source, variableKey, out variableValue))
{
// A matching variable was found.
// Update the target value.
trace.Verbose("Macro found.");
targetValue = string.Concat(
targetValue.Substring(0, prefixIndex),
variableValue ?? string.Empty,
targetValue.Substring(suffixIndex + Constants.Variables.MacroSuffix.Length));
// Bump the start index to prevent recursive replacement.
startIndex = prefixIndex + (variableValue ?? string.Empty).Length;
}
else
{
// A matching variable was not found.
trace.Verbose("Macro not found.");
startIndex = prefixIndex + 1;
}
}
target[targetKey] = targetValue ?? string.Empty;
}
}
public async Task CheckToolOutputAsync(string name, string fileName, string arguments)
{
_trace.Entering();
ArgUtil.NotNullOrEmpty(name, nameof(name));
ArgUtil.NotNullOrEmpty(fileName, nameof(fileName));
try
{
// Attempt to locate the tool.
string filePath = _whichUtil.Which(fileName);
if (string.IsNullOrEmpty(filePath))
{
return;
}
// Invoke the tool and capture the output.
var output = new StringBuilder();
using (var processInvoker = _hostContext.CreateService <IProcessInvoker>())
{
processInvoker.OutputDataReceived +=
(object sender, ProcessDataReceivedEventArgs args) =>
{
if (!string.IsNullOrEmpty(args.Data))
{
output.Append(args.Data);
}
};
await processInvoker.ExecuteAsync(
workingDirectory : string.Empty,
fileName : filePath,
arguments : arguments ?? string.Empty,
environment : null,
cancellationToken : _cancellationToken);
}
// Add the capability.
if (output.Length > 0)
{
string value = output.ToString();
_trace.Info($"Adding '{name}': '{value}'");
_capabilities.Add(new Capability(name, value));
}
}
catch (Exception ex) when(!(ex is OperationCanceledException))
{
_trace.Error(ex);
}
}
public static void ExpandEnvironmentVariables(IHostContext context, IDictionary <string, string> target)
{
ArgUtil.NotNull(context, nameof(context));
Tracing trace = context.GetTrace(nameof(VarUtil));
trace.Entering();
// Copy the environment variables into a dictionary that uses the correct comparer.
var source = new Dictionary <string, string>(EnvironmentVariableKeyComparer);
IDictionary environment = Environment.GetEnvironmentVariables();
foreach (DictionaryEntry entry in environment)
{
string key = entry.Key as string ?? string.Empty;
string val = entry.Value as string ?? string.Empty;
source[key] = val;
}
// Expand the target values.
ExpandValues(context, source, target);
}
public static void ExpandEnvironmentVariables(IHostContext context, IDictionary <string, string> target)
{
ArgUtil.NotNull(context, nameof(context));
Tracing trace = context.GetTrace(nameof(VarUtil));
trace.Entering();
// Determine which string comparer to use for the environment variable dictionary.
StringComparer comparer;
switch (Constants.Agent.Platform)
{
case Constants.OSPlatform.Linux:
case Constants.OSPlatform.OSX:
comparer = StringComparer.CurrentCulture;
break;
case Constants.OSPlatform.Windows:
comparer = StringComparer.CurrentCultureIgnoreCase;
break;
default:
throw new NotSupportedException();
}
// Copy the environment variables into a dictionary that uses the correct comparer.
var source = new Dictionary <string, string>(comparer);
IDictionary environment = Environment.GetEnvironmentVariables();
foreach (DictionaryEntry entry in environment)
{
string key = entry.Key as string ?? string.Empty;
string val = entry.Value as string ?? string.Empty;
source[key] = val;
}
// Expand the target values.
ExpandValues(context, source, target);
}
