using System;
using System.Diagnostics;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft;
using Microsoft.VisualStudio.Extensibility;
using Microsoft.VisualStudio.Extensibility.Commands;
using Microsoft.VisualStudio.ProjectSystem.Query;
namespace SqlAnalyzer;
/// <summary>
/// A command to execute analyzer on the current file selected in Solution Explorer.
/// </summary>
/// <remarks>
/// This command utilizes <see cref="CommandConfiguration.EnabledWhen"/> to describe when command state is enabled.
/// </remarks>
[VisualStudioContribution]
internal class RunAnalyzerOnCurrentFileCommand : Command
{
private readonly TraceSource logger;
#pragma warning disable CA2213 // This is an extension scoped service.
private readonly SqlAnalyzerDiagnosticsService diagnosticsProvider;
#pragma warning restore CA2213
/// <summary>
/// Initializes a new instance of the <see cref="RunAnalyzerOnCurrentFileCommand"/> class.
/// </summary>
/// <param name="traceSource">Logger instance that can be used to log extension actions.</param>
/// <param name="diagnosticsProvider">Local diagnostics provider service instance.</param>
public RunAnalyzerOnCurrentFileCommand(TraceSource traceSource, SqlAnalyzerDiagnosticsService diagnosticsProvider)
{
this.logger = Requires.NotNull(traceSource, nameof(traceSource));
this.diagnosticsProvider = Requires.NotNull(diagnosticsProvider, nameof(diagnosticsProvider));
this.logger.TraceEvent(TraceEventType.Information, 0, $"Initializing {nameof(RunAnalyzerOnCurrentFileCommand)} instance.");
}
/// <inheritdoc />
#pragma warning disable CA1303 // Do not pass literals as localized parameters
public override CommandConfiguration CommandConfiguration => new("%SqlAnalyzer.RunAnalyzerOnCurrentFileCommand.DisplayName%")
{
Placements = [CommandPlacement.KnownPlacements.ToolsMenu],
Icon = new(ImageMoniker.KnownValues.CodeInformationRule, IconSettings.IconAndText),
EnabledWhen = ActivationConstraint.ClientContext(ClientContextKey.Shell.ActiveSelectionFileName, ".+"),
};
#pragma warning restore CA1303 // Do not pass literals as localized parameters
/// <inheritdoc />
public override async Task ExecuteCommandAsync(IClientContext context, CancellationToken cancellationToken)
{
cancellationToken.ThrowIfCancellationRequested();
try
{
// Get the selected item URIs from IDE context that represents the state when command was executed.
Uri[] selectedItemPaths = [await context.GetSelectedPathAsync(cancellationToken)];
// Enumerate through each selection and run analyzer on each selected item.
foreach (var selectedItem in selectedItemPaths.Where(p => p.IsFile))
{
await this.diagnosticsProvider.ProcessFileAsync(selectedItem, cancellationToken);
}
}
catch (InvalidOperationException ex)
{
this.logger.TraceEvent(TraceEventType.Error, 0, ex.ToString());
}
}
}
