aspnetcore/test/Cli.FunctionalTests/NuGetPackageSource.cs

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using System.Linq;

namespace Cli.FunctionalTests
{
    public class NuGetPackageSource
    {
        public static NuGetPackageSource None { get; } = new NuGetPackageSource
        {
            Name = nameof(None),
            SourceArgumentLazy = new Lazy<string>(string.Empty),
        };

        public static NuGetPackageSource NuGetOrg { get; } = new NuGetPackageSource
        {
            Name = nameof(NuGetOrg),
            SourceArgumentLazy = new Lazy<string>("--source https://api.nuget.org/v3/index.json"),
        };

        public static NuGetPackageSource DotNetCore { get; } = new NuGetPackageSource
        {
            Name = nameof(DotNetCore),
            SourceArgumentLazy = new Lazy<string>("--source https://dotnet.myget.org/F/dotnet-core/api/v3/index.json"),
        };

        public static NuGetPackageSource EnvironmentVariable { get; } = new NuGetPackageSource
        {
            Name = nameof(EnvironmentVariable),
            SourceArgumentLazy = new Lazy<string>(() => GetSourceArgumentFromEnvironment()),
        };

        public static NuGetPackageSource EnvironmentVariableAndNuGetOrg { get; } = new NuGetPackageSource
        {
            Name = nameof(EnvironmentVariableAndNuGetOrg),
            SourceArgumentLazy = new Lazy<string>(() => string.Join(" ", EnvironmentVariable.SourceArgument, NuGetOrg.SourceArgument)),
        };

        private NuGetPackageSource() { }

        public string Name { get; private set; }
        public string SourceArgument => SourceArgumentLazy.Value;
        private Lazy<string> SourceArgumentLazy { get; set; }

        public override string ToString() => Name;

        private static string GetSourceArgumentFromEnvironment()
        {
            var sourceString = Environment.GetEnvironmentVariable("NUGET_PACKAGE_SOURCE") ??
                throw new InvalidOperationException("Environment variable NUGET_PACKAGE_SOURCE is required but not set");

            // Split on pipe and remove blank entries
            var sources = sourceString.Split('|').Where(s => !string.IsNullOrWhiteSpace(s));

            return string.Join(" ", sources.Select(s => $"--source {s}"));
        }
    }
}