AddFileCommand.cs

#region Copyright notice and license

// Copyright 2019 The gRPC Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#endregion

using System;
using System.CommandLine;
using System.CommandLine.Invocation;
using System.CommandLine.Parsing;
using Grpc.Dotnet.Cli.Internal;
using Grpc.Dotnet.Cli.Options;
using Grpc.Dotnet.Cli.Properties;
using Microsoft.Build.Evaluation;

namespace Grpc.Dotnet.Cli.Commands;

internal class AddFileCommand : CommandBase
{
    public AddFileCommand(IConsole console, string? projectPath, HttpClient httpClient)
        : base(console, projectPath, httpClient) { }

    public static Command Create(HttpClient httpClient)
    {
        var command = new Command(
            name: "add-file",
            description: CoreStrings.AddFileCommandDescription);

        var projectOption = CommonOptions.ProjectOption();
        var serviceOption = CommonOptions.ServiceOption();
        var additionalImportDirsOption = CommonOptions.AdditionalImportDirsOption();
        var accessOption = CommonOptions.AccessOption();
        var recursiveOption = CommonOptions.RecursiveOption();
        var filesArgument = new Argument<string[]>
        {
            Name = "files",
            Description = CoreStrings.AddFileCommandArgumentDescription,
            Arity = ArgumentArity.OneOrMore
        };

        command.AddOption(projectOption);
        command.AddOption(serviceOption);
        command.AddOption(accessOption);
        command.AddOption(additionalImportDirsOption);
        command.AddOption(recursiveOption);
        command.AddArgument(filesArgument);

        command.SetHandler(
            async (context) =>
            {
                var project = context.ParseResult.GetValueForOption(projectOption);
                var services = context.ParseResult.GetValueForOption(serviceOption);
                var access = context.ParseResult.GetValueForOption(accessOption);
                var additionalImportDirs = context.ParseResult.GetValueForOption(additionalImportDirsOption);
                var searchOption = context.ParseResult.HasOption(recursiveOption) ? SearchOption.AllDirectories : SearchOption.TopDirectoryOnly;
                var files = context.ParseResult.GetValueForArgument(filesArgument);

                try
                {
                    var command = new AddFileCommand(context.Console, project, httpClient);
                    await command.AddFileAsync(services, access, additionalImportDirs, files, searchOption);

                    context.ExitCode = 0;
                }
                catch (CLIToolException e)
                {
                    context.Console.LogError(e);

                    context.ExitCode = -1;
                }
            });

        return command;
    }

    public async Task AddFileAsync(Services services, Access access, string? additionalImportDirs, string[] files, SearchOption searchOption)
    {
        var resolvedServices = ResolveServices(services);
        await EnsureNugetPackagesAsync(resolvedServices);
        files = GlobReferences(files, searchOption);

        foreach (var file in files)
        {
            Console.Log(CoreStrings.LogAddFileReference, file);
            AddProtobufReference(resolvedServices, additionalImportDirs, access, file, string.Empty);
        }

        Project.Save();
    }
}