// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
namespace Microsoft.VisualStudioCode.Razor.IntegrationTests.Services;
 
/// <summary>
/// Services for navigation (Go to Definition, Find References) operations in integration tests.
/// </summary>
public class NavigationServices(IntegrationTestServices testServices) : ServiceBase(testServices)
{
    /// <summary>
    /// Triggers Go to Definition and waits for navigation.
    /// </summary>
    /// <param name="expectedFileName">If provided, waits until a file containing this name is opened.</param>
    /// <param name="timeout">Timeout for waiting.</param>
    public async Task GoToDefinitionAsync(string? expectedFileName = null, TimeSpan? timeout = null)
    {
        timeout ??= TestServices.Settings.LspTimeout;
        var originalFile = await TestServices.Editor.GetCurrentFileNameAsync();
        var originalPosition = await TestServices.Editor.GetCursorPositionAsync();
 
        await TestServices.Input.PressAsync(SpecialKey.F12);
 
        if (expectedFileName != null)
        {
            // Wait for navigation to the expected file
            await Helper.WaitForConditionAsync(
                TestServices.Editor.GetCurrentFileNameAsync,
                fileName => fileName?.Contains(expectedFileName, StringComparison.OrdinalIgnoreCase) == true,
                timeout.Value);
        }
        else
        {
            // Wait for any navigation: file name changes, cursor position changes, or peek definition appears
            await Helper.WaitForConditionAsync(
                async () =>
                {
                    var currentFile = await TestServices.Editor.GetCurrentFileNameAsync();
                    var currentPosition = await TestServices.Editor.GetCursorPositionAsync();
                    var peekVisible = await TestServices.Playwright.Page.Locator(".peekview-widget").CountAsync() > 0;
 
                    // Check if file changed, position changed, or peek appeared
                    var fileChanged = currentFile != originalFile;
                    var positionChanged = originalPosition != null && currentPosition != null &&
                        (originalPosition.Value.Line != currentPosition.Value.Line ||
                         originalPosition.Value.Column != currentPosition.Value.Column);
 
                    return fileChanged || positionChanged || peekVisible;
                },
                timeout.Value);
        }
    }
 
    /// <summary>
    /// Triggers Find All References (Shift+F12) and waits for references panel.
    /// </summary>
    public async Task FindAllReferencesAsync(TimeSpan? timeout = null)
    {
        timeout ??= TestServices.Settings.LspTimeout;
 
        await TestServices.Input.PressWithShiftAsync(SpecialKey.F12);
 
        // Wait for the references panel or peek view to appear
        await Helper.WaitForConditionAsync(
            async () =>
            {
                var peekViewCount = await TestServices.Playwright.Page.Locator(".peekview-widget").CountAsync();
                var referencesPanelCount = await TestServices.Playwright.Page.Locator("[id='workbench.panel.referencesView']").CountAsync();
                return peekViewCount > 0 || referencesPanelCount > 0;
            },
            timeout.Value);
    }
 
    /// <summary>
    /// Gets the count of references shown in the peek view or references panel.
    /// </summary>
    public async Task<int> GetReferencesCountAsync()
    {
        // Try to count references in the peek view
        var peekItemsCount = await TestServices.Playwright.Page.Locator(".peekview-widget .monaco-list-row").CountAsync();
        if (peekItemsCount > 0)
        {
            return peekItemsCount;
        }
 
        // Try the references panel
        var panelItemsCount = await TestServices.Playwright.Page.Locator("[id='workbench.panel.referencesView'] .monaco-list-row").CountAsync();
        return panelItemsCount;
    }
}