DTECommandExecutor.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Microsoft.VisualStudio.Shell;
using EnvDTE;
using EnvDTE80;

namespace VSCommandServer;

public class DTECommandExecutor(AsyncPackage package) : ICommandExecutor
{
	private readonly AsyncPackage _package = package ?? throw new ArgumentNullException(nameof(package));
	private DTE2 _dte;

	private async Task<DTE2> GetDTEAsync()
	{
		if (_dte != null)
			return _dte;

		await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();
		_dte = await _package.GetServiceAsync(typeof(DTE)) as DTE2;
		return _dte;
	}

	public async Task<object> ExecuteCommandAsync(string commandName, object[] args)
	{
		await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();
            
		var dte = await GetDTEAsync();
		if (dte == null)
			throw new InvalidOperationException("Could not access Visual Studio DTE service");

		try
		{
			var argString = args is { Length: > 0 } ? string.Join(" ", args) : string.Empty;
			dte.ExecuteCommand(commandName, argString);
			return $"Command '{commandName}' executed successfully";
		}
		catch (Exception ex)
		{
			throw new InvalidOperationException($"Failed to execute command '{commandName}': {ex.Message}", ex);
		}
	}

	public async Task<List<CommandInfo>> GetAvailableCommandsAsync()
	{
		await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();

		var dte = await GetDTEAsync();
		if (dte == null)
			throw new InvalidOperationException("Could not access Visual Studio DTE service");

		var commands = new List<CommandInfo>();
		try
		{
			foreach (Command cmd in dte.Commands)
			{
				try
				{
					if (!string.IsNullOrEmpty(cmd.Name))
					{
						commands.Add(new CommandInfo
						{
							Name = cmd.Name,
							DisplayName = cmd.LocalizedName
						});
					}
				}
				catch
				{
					// Skip commands that throw exceptions when accessed
				}
			}
		}
		catch (Exception ex)
		{
			throw new InvalidOperationException($"Failed to enumerate commands: {ex.Message}", ex);
		}

		return commands;
	}

	public async Task<string> GetSolutionPathAsync()
	{
		await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();

		try
		{
			var dte = await GetDTEAsync();
			if (dte?.Solution != null && !string.IsNullOrEmpty(dte.Solution.FullName))
			{
				return dte.Solution.FullName;
			}
		}
		catch
		{
			// Ignore errors
		}
            
		return string.Empty;
	}

	public async Task<string> GetVisualStudioVersionAsync()
	{
		await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();

		try
		{
			var dte = await GetDTEAsync();
			if (dte != null)
				return $"Visual Studio {dte.Version} ({dte.Edition})";
		}
		catch
		{
			// Ignore errors
		}
            
		return "Unknown";
	}
}