// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
#pragma warning disable ASPIREAZURE003 // Type is for evaluation purposes only and is subject to change or removal in future updates. Suppress this diagnostic to proceed.
 
using Aspire.Hosting.ApplicationModel;
using Azure.Provisioning.Primitives;
using Azure.Provisioning.Search;
 
namespace Aspire.Hosting.Azure;
 
/// <summary>
/// Represents an Azure AI Search resource.
/// </summary>
/// <param name="name">The name of the resource</param>
/// <param name="configureInfrastructure">Callback to configure the Azure AI Search resource.</param>
public class AzureSearchResource(string name, Action<AzureResourceInfrastructure> configureInfrastructure)
    : AzureProvisioningResource(name, configureInfrastructure), IResourceWithConnectionString, IAzurePrivateEndpointTarget
{
    /// <summary>
    /// Gets the "connectionString" output reference from the Azure AI Search resource.
    /// </summary>
    /// <remarks>
    /// This connection string will assume you're deploying to public Azure.
    /// </remarks>
    public BicepOutputReference ConnectionString => new("connectionString", this);
 
    /// <summary>
    /// Gets the "name" output reference for the resource.
    /// </summary>
    public BicepOutputReference NameOutputReference => new("name", this);
 
    /// <summary>
    /// Gets the "id" output reference for the resource.
    /// </summary>
    public BicepOutputReference Id => new("id", this);
 
    /// <summary>
    /// Gets the service endpoint URI expression for the Azure AI Search resource.
    /// </summary>
    /// <remarks>
    /// Format: <c>https://{name}.search.windows.net</c>.
    /// </remarks>
    public BicepOutputReference Endpoint => new("endpoint", this);
 
    /// <summary>
    /// Gets the service endpoint URI expression for the Azure AI Search resource.
    /// </summary>
    /// <remarks>
    /// Format: <c>https://{name}.search.windows.net</c>.
    /// </remarks>
    public ReferenceExpression UriExpression => ReferenceExpression.Create($"{Endpoint}");
 
    /// <summary>
    /// Gets the connection string template for the manifest for the resource.
    /// </summary>
    public ReferenceExpression ConnectionStringExpression =>
        ReferenceExpression.Create($"{ConnectionString}");
 
    /// <inheritdoc/>
    public override ProvisionableResource AddAsExistingResource(AzureResourceInfrastructure infra)
    {
        var bicepIdentifier = this.GetBicepIdentifier();
        var resources = infra.GetProvisionableResources();
        
        // Check if a SearchService with the same identifier already exists
        var existingStore = resources.OfType<SearchService>().SingleOrDefault(store => store.BicepIdentifier == bicepIdentifier);
        
        if (existingStore is not null)
        {
            return existingStore;
        }
        
        // Create and add new resource if it doesn't exist
        var store = SearchService.FromExisting(bicepIdentifier);
 
        if (!TryApplyExistingResourceAnnotation(
            this,
            infra,
            store))
        {
            store.Name = NameOutputReference.AsProvisioningParameter(infra);
        }
 
        infra.Add(store);
        return store;
    }
 
    IEnumerable<KeyValuePair<string, ReferenceExpression>> IResourceWithConnectionString.GetConnectionProperties()
    {
        yield return new("Uri", UriExpression);
    }
 
    BicepOutputReference IAzurePrivateEndpointTarget.Id => Id;
 
    IEnumerable<string> IAzurePrivateEndpointTarget.GetPrivateLinkGroupIds() => ["searchService"];
 
    string IAzurePrivateEndpointTarget.GetPrivateDnsZoneName() => "privatelink.search.windows.net";
}