SuggestionStore.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.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Text;
using System.Threading.Tasks;

namespace System.CommandLine.Suggest
{
    public class SuggestionStore : ISuggestionStore
    {
        public string GetCompletions(string exeFileName, string suggestionTargetArguments, TimeSpan timeout)
        {
            if (string.IsNullOrWhiteSpace(exeFileName))
            {
                throw new ArgumentException("Value cannot be null, empty, or consist entirely of whitespace.", nameof(exeFileName));
            }

            if (string.IsNullOrWhiteSpace(suggestionTargetArguments))
            {
                throw new ArgumentException("Value cannot be null, empty, or consist entirely of whitespace.", nameof(suggestionTargetArguments));
            }

            string result = "";

            try
            {
                // Invoke target with args
                var processStartInfo = new ProcessStartInfo(
                                           exeFileName, 
                                           suggestionTargetArguments)
                                       {
                                           UseShellExecute = false,
                                           RedirectStandardOutput = true
                                       };

                using (var process = new Process
                                     {
                                         StartInfo = processStartInfo
                                     })
                {
                    process.Start();
                    process.BeginOutputReadLine();
                    
                    var stringBuilder = new StringBuilder();
                    process.OutputDataReceived += (sender, eventArgs) =>
                    {
                        if (eventArgs.Data != null)
                        {
                            stringBuilder.AppendLine(eventArgs.Data);
                        }
                    };
                    
                    if (process.WaitForExit(timeout) && process.ExitCode == 0)
                    {
                        result = stringBuilder.ToString();
                    }
                    else
                    {
                        process.Kill();
                    }
                }
            }
            catch (Win32Exception exception)
            {
                // We don't check for the existence of exeFileName until the exception in case
                // it is a command that start process can resolve to a file name.
                if (!File.Exists(exeFileName))
                {
                    var message = $"Unable to find the file '{exeFileName}'";

#if DEBUG
                    Program.LogDebug($"exception: {message}");
#endif

                    throw new ArgumentException(
                        message, nameof(exeFileName), exception);
                }
            }
            return result;
        }
    }
}