dotnet-installer/src/dotnet/commands/dotnet-restore/Program.cs

// Copyright (c) .NET Foundation and contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

using System.Collections.Generic;
using Microsoft.DotNet.Cli.CommandLine;
using Microsoft.DotNet.Cli.Utils;
using Microsoft.DotNet.Tools.MSBuild;
using Microsoft.DotNet.Cli;
using System.Diagnostics;

namespace Microsoft.DotNet.Tools.Restore
{
    public class RestoreCommand
    {
        private MSBuildForwardingApp _forwardingApp;

        public RestoreCommand(IEnumerable<string> msbuildArgs, string msbuildPath = null)
        {
            _forwardingApp = new MSBuildForwardingApp(msbuildArgs, msbuildPath);
        }

        public static RestoreCommand FromArgs(string[] args, string msbuildPath = null)
        {
            DebugHelper.HandleDebugSwitch(ref args);

            CommandLineApplication cmd = new CommandLineApplication(throwOnUnexpectedArg: false)
            {
                Name = "restore",
                FullName = LocalizableStrings.AppFullName,
                Description = LocalizableStrings.AppDescription,
                HandleRemainingArguments = true,
                ArgumentSeparatorHelpText = HelpMessageStrings.MSBuildAdditionalArgsHelpText
            };

            cmd.HelpOption("-h|--help");

            var argRoot = cmd.Argument(
                    $"[{LocalizableStrings.CmdArgument}]",
                    LocalizableStrings.CmdArgumentDescription,
                    multipleValues: true);

            var sourceOption = cmd.Option(
                    $"-s|--source <{LocalizableStrings.CmdSourceOption}>",
                    LocalizableStrings.CmdSourceOptionDescription,
                    CommandOptionType.MultipleValue);

            var runtimeOption = cmd.Option(
                    $"-r|--runtime <{LocalizableStrings.CmdRuntimeOption}>",
                    LocalizableStrings.CmdRuntimeOptionDescription,
                    CommandOptionType.MultipleValue);

            var packagesOption = cmd.Option(
                    $"--packages <{LocalizableStrings.CmdPackagesOption}>",
                    LocalizableStrings.CmdPackagesOptionDescription,
                    CommandOptionType.SingleValue);

            var disableParallelOption = cmd.Option(
                    "--disable-parallel",
                    LocalizableStrings.CmdDisableParallelOptionDescription,
                    CommandOptionType.NoValue);

            var configFileOption = cmd.Option(
                    $"--configfile <{LocalizableStrings.CmdConfigFileOption}>",
                    LocalizableStrings.CmdConfigFileOptionDescription,
                    CommandOptionType.SingleValue);

            var noCacheOption = cmd.Option(
                    "--no-cache",
                    LocalizableStrings.CmdNoCacheOptionDescription,
                    CommandOptionType.NoValue);

            var ignoreFailedSourcesOption = cmd.Option(
                    "--ignore-failed-sources",
                    LocalizableStrings.CmdIgnoreFailedSourcesOptionDescription,
                    CommandOptionType.NoValue);

            var noDependenciesOption = cmd.Option(
                "--no-dependencies",
                LocalizableStrings.CmdNoDependenciesOptionDescription,
                CommandOptionType.NoValue);

            CommandOption verbosityOption = MSBuildForwardingApp.AddVerbosityOption(cmd);

            List<string> msbuildArgs = null;
            cmd.OnExecute(() =>
            {
                msbuildArgs = new List<string>()
                {
                     "/NoLogo", 
                     "/t:Restore", 
                     "/ConsoleLoggerParameters:Verbosity=Minimal" 
                };

                if (sourceOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RestoreSources={string.Join("%3B", sourceOption.Values)}");
                }

                if (runtimeOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RuntimeIdentifiers={string.Join("%3B", runtimeOption.Values)}");
                }

                if (packagesOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RestorePackagesPath={packagesOption.Value()}");
                }

                if (disableParallelOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RestoreDisableParallel=true");
                }

                if (configFileOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RestoreConfigFile={configFileOption.Value()}");
                }

                if (noCacheOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RestoreNoCache=true");
                }

                if (ignoreFailedSourcesOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RestoreIgnoreFailedSources=true");
                }

                if (noDependenciesOption.HasValue())
                {
                    msbuildArgs.Add($"/p:RestoreRecursive=false");
                }

                if (verbosityOption.HasValue())
                {
                    msbuildArgs.Add($"/verbosity:{verbosityOption.Value()}");
                }

                // Add in arguments
                msbuildArgs.AddRange(argRoot.Values);

                // Add remaining arguments that the parser did not understand
                msbuildArgs.AddRange(cmd.RemainingArguments);

                return 0;
            });

            int exitCode = cmd.Execute(args);
            if (msbuildArgs == null)
            {
                throw new CommandCreationException(exitCode);
            }

            return new RestoreCommand(msbuildArgs, msbuildPath);
        }

        public static int Run(string[] args)
        {
            DebugHelper.HandleDebugSwitch(ref args);

            RestoreCommand cmd;
            try
            {
                cmd = FromArgs(args);
            }
            catch (CommandCreationException e)
            {
                return e.ExitCode;
            }

            return cmd.Execute();
        }

        public ProcessStartInfo GetProcessStartInfo()
        {
            return _forwardingApp.GetProcessStartInfo();
        }

        public int Execute()
        {
            return GetProcessStartInfo().Execute();
        }
    }
}