41 lines
1.2 KiB
C#
41 lines
1.2 KiB
C#
using System;
|
|
using System.Linq;
|
|
|
|
namespace Microsoft.DotNet.Cli.Build.Framework
|
|
{
|
|
[AttributeUsage(AttributeTargets.Method, AllowMultiple = true, Inherited = false)]
|
|
public class EnvironmentAttribute : TargetConditionAttribute
|
|
{
|
|
private string _envVar;
|
|
private string[] _expectedVals;
|
|
|
|
public EnvironmentAttribute(string envVar, params string[] expectedVals)
|
|
{
|
|
if (string.IsNullOrEmpty(envVar))
|
|
{
|
|
throw new ArgumentNullException(nameof(envVar));
|
|
}
|
|
if (expectedVals == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(expectedVals));
|
|
}
|
|
|
|
_envVar = envVar;
|
|
_expectedVals = expectedVals;
|
|
}
|
|
|
|
public override bool EvaluateCondition()
|
|
{
|
|
var actualVal = Environment.GetEnvironmentVariable(_envVar);
|
|
|
|
if (_expectedVals.Any())
|
|
{
|
|
return _expectedVals.Any(ev => string.Equals(actualVal, ev, StringComparison.Ordinal));
|
|
}
|
|
else
|
|
{
|
|
return !string.IsNullOrEmpty(actualVal);
|
|
}
|
|
}
|
|
}
|
|
}
|