// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Aspire.Cli.Telemetry;
using Microsoft.Extensions.Logging.Abstractions;
 
namespace Aspire.Cli.Tests.Telemetry;
 
/// <summary>
/// Provides test helpers for creating telemetry instances.
/// </summary>
internal static class TestTelemetryHelper
{
    /// <summary>
    /// Creates and initializes an <see cref="AspireCliTelemetry"/> instance for testing.
    /// </summary>
    public static AspireCliTelemetry CreateInitializedTelemetry()
    {
        var provider = new TestMachineInformationProvider();
        var ciDetector = new TestCIEnvironmentDetector();
        var codingAgentDetector = new TestCodingAgentDetector();
        var internalMicrosoftDetector = new TestInternalMicrosoftDetector();
        var tagsSource = new TelemetryTagsSource(NullLogger<TelemetryTagsSource>.Instance);
        var telemetry = new AspireCliTelemetry(
            NullLogger<AspireCliTelemetry>.Instance,
            provider,
            ciDetector,
            codingAgentDetector,
            internalMicrosoftDetector,
            new TelemetryConfiguration { ReportedTelemetryEnabled = true },
            AspireCliTelemetry.ReportedActivitySourceName,
            AspireCliTelemetry.DiagnosticsActivitySourceName,
            CreateExecutionContext(),
            tagsSource);
        telemetry.Initialize();
        tagsSource.TagsTask.GetAwaiter().GetResult();
        return telemetry;
    }
 
    /// <summary>
    /// Creates and initializes an <see cref="AspireCliTelemetry"/> instance for testing with custom activity source names.
    /// </summary>
    public static AspireCliTelemetry CreateInitializedTelemetry(string reportedSourceName, string diagnosticsSourceName)
    {
        var provider = new TestMachineInformationProvider();
        var ciDetector = new TestCIEnvironmentDetector();
        var codingAgentDetector = new TestCodingAgentDetector();
        var internalMicrosoftDetector = new TestInternalMicrosoftDetector();
        var tagsSource = new TelemetryTagsSource(NullLogger<TelemetryTagsSource>.Instance);
        var telemetry = new AspireCliTelemetry(NullLogger<AspireCliTelemetry>.Instance, provider, ciDetector, codingAgentDetector, internalMicrosoftDetector, reportedSourceName, diagnosticsSourceName, CreateExecutionContext(), tagsSource);
        telemetry.Initialize();
        tagsSource.TagsTask.GetAwaiter().GetResult();
        return telemetry;
    }
 
    private static CliExecutionContext CreateExecutionContext()
        => Utils.TestExecutionContextHelper.CreateExecutionContext(new DirectoryInfo(AppContext.BaseDirectory));
 
    private sealed class TestMachineInformationProvider : IMachineInformationProvider
    {
        public Task<string?> GetOrCreateDeviceId() => Task.FromResult<string?>("test-device-id");
        public Task<string> GetMacAddressHash() => Task.FromResult("test-mac-hash");
    }
 
    private sealed class TestCIEnvironmentDetector : ICIEnvironmentDetector
    {
        public bool IsCIEnvironment() => false;
    }
 
    private sealed class TestCodingAgentDetector : ICodingAgentDetector
    {
        public string? GetCodingAgent() => null;
    }
 
    private sealed class TestInternalMicrosoftDetector : IInternalMicrosoftDetector
    {
        public Task<InternalMicrosoftDetectionResult> IsInternalMicrosoftMachineAsync(CancellationToken cancellationToken = default)
            => Task.FromResult(new InternalMicrosoftDetectionResult(IsInternalMicrosoft: false, Source: null, Alias: null, Domain: null));
    }
}