// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Diagnostics;
using System.Diagnostics.CodeAnalysis;
using Microsoft.AspNetCore.Razor.Language.Components;
 
namespace Microsoft.AspNetCore.Razor.Language.Intermediate;
 
public sealed class ReferenceCaptureIntermediateNode : IntermediateNode
{
    private const string DefaultFieldTypeName = $"global::{ComponentsApi.ElementReference.FullTypeName}";
    private const string DefaultTypeName = $"global::System.Action<{DefaultFieldTypeName}>";
 
    private string? _componentCaptureTypeName;
    private string _fieldTypeName;
    private string _typeName;
 
    [MemberNotNullWhen(true, nameof(ComponentCaptureTypeName))]
    public bool IsComponentCapture { get; }
 
    public IntermediateToken IdentifierToken { get; }
 
    /// <remarks>
    /// This is not <c>global::</c>-prefixed, for that consider using <see cref="FieldTypeName"/> instead.
    /// </remarks>
    public string? ComponentCaptureTypeName => _componentCaptureTypeName;
 
    public string FieldTypeName => _fieldTypeName;
 
    public string TypeName => _typeName;
 
    public override IntermediateNodeCollection Children => IntermediateNodeCollection.ReadOnly;
 
    public ReferenceCaptureIntermediateNode(IntermediateToken identifierToken)
    {
        ArgHelper.ThrowIfNull(identifierToken);
 
        IdentifierToken = identifierToken;
        Source = IdentifierToken.Source;
 
        _fieldTypeName = DefaultFieldTypeName;
        _typeName = DefaultTypeName;
    }
 
    public ReferenceCaptureIntermediateNode(IntermediateToken identifierToken, string componentCaptureTypeName)
        : this(identifierToken)
    {
        IsComponentCapture = true;
        UpdateComponentCaptureTypeName(componentCaptureTypeName);
    }
 
    public void UpdateComponentCaptureTypeName(string componentCaptureTypeName)
    {
        ArgHelper.ThrowIfNullOrEmpty(componentCaptureTypeName);
 
        Debug.Assert(IsComponentCapture);
 
        _componentCaptureTypeName = componentCaptureTypeName;
        _fieldTypeName = TypeNameHelper.GetGloballyQualifiedNameIfNeeded(componentCaptureTypeName);
        _typeName = $"global::System.Action<{_fieldTypeName}>";
    }
 
    public override void Accept(IntermediateNodeVisitor visitor)
    {
        visitor.VisitReferenceCapture(this);
    }
 
    public override void FormatNode(IntermediateNodeFormatter formatter)
    {
        formatter.WriteContent(IdentifierToken.Content);
 
        formatter.WriteProperty(nameof(IdentifierToken), IdentifierToken.Content);
        formatter.WriteProperty(nameof(ComponentCaptureTypeName), ComponentCaptureTypeName);
    }
}