// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
#nullable disable
 
using System.Collections.Immutable;
using System.Runtime.CompilerServices;
 
namespace Microsoft.CodeAnalysis.CodeGeneration;
 
internal class CodeGenerationDestructorInfo
{
    private static readonly ConditionalWeakTable<IMethodSymbol, CodeGenerationDestructorInfo> s_destructorToInfoMap = new();
 
    private readonly string _typeName;
    private readonly ImmutableArray<SyntaxNode> _statements;
 
    private CodeGenerationDestructorInfo(
        string typeName,
        ImmutableArray<SyntaxNode> statements)
    {
        _typeName = typeName;
        _statements = statements;
    }
 
    public static void Attach(
        IMethodSymbol destructor,
        string typeName,
        ImmutableArray<SyntaxNode> statements)
    {
        var info = new CodeGenerationDestructorInfo(typeName, statements);
        s_destructorToInfoMap.Add(destructor, info);
    }
 
    private static CodeGenerationDestructorInfo GetInfo(IMethodSymbol method)
    {
        s_destructorToInfoMap.TryGetValue(method, out var info);
        return info;
    }
 
    public static ImmutableArray<SyntaxNode> GetStatements(IMethodSymbol destructor)
        => GetStatements(GetInfo(destructor));
 
    public static string GetTypeName(IMethodSymbol destructor)
        => GetTypeName(GetInfo(destructor), destructor);
 
    private static ImmutableArray<SyntaxNode> GetStatements(CodeGenerationDestructorInfo info)
        => info?._statements ?? default;
 
    private static string GetTypeName(CodeGenerationDestructorInfo info, IMethodSymbol constructor)
        => info == null ? constructor.ContainingType.Name : info._typeName;
}