// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Threading;
using Microsoft.AspNetCore.Razor.Language;
using Microsoft.AspNetCore.Razor.Language.TagHelpers.Producers;
using Microsoft.CodeAnalysis;

namespace Microsoft.AspNetCore.Mvc.Razor.Extensions.Version1_X;

internal sealed partial class ViewComponentTagHelperProducer : TagHelperProducer
{
    private readonly ViewComponentTagHelperDescriptorFactory _factory;
    private readonly INamedTypeSymbol _viewComponentAttributeType;
    private readonly INamedTypeSymbol? _nonViewComponentAttributeType;

    private ViewComponentTagHelperProducer(
        ViewComponentTagHelperDescriptorFactory factory,
        INamedTypeSymbol viewComponentAttributeType,
        INamedTypeSymbol? nonViewComponentAttributeType)
    {
        _factory = factory;
        _viewComponentAttributeType = viewComponentAttributeType;
        _nonViewComponentAttributeType = nonViewComponentAttributeType;
    }

    public override TagHelperProducerKind Kind => TagHelperProducerKind.Mvc1_X_ViewComponent;

    public override bool SupportsTypes => true;
    public override bool SupportsNestedTypes => true;

    public override bool IsCandidateType(INamedTypeSymbol type)
        => type.IsViewComponent(_viewComponentAttributeType, _nonViewComponentAttributeType);

    public override void AddTagHelpersForType(
        INamedTypeSymbol type,
        ref TagHelperCollection.RefBuilder results,
        CancellationToken cancellationToken)
    {
        if (_factory.CreateDescriptor(type) is { } descriptor)
        {
            results.Add(descriptor);
        }
    }
}