// 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.Generic;
using System.Collections.Immutable;
using System.Linq;
using System.Runtime.CompilerServices;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp.Syntax;
 
namespace Microsoft.CodeAnalysis.CSharp.Extensions;
 
internal static partial class MemberDeclarationSyntaxExtensions
{
    private static readonly ConditionalWeakTable<MemberDeclarationSyntax, Dictionary<string, ImmutableArray<SyntaxToken>>> s_declarationCache = new();
 
    public static LocalDeclarationMap GetLocalDeclarationMap(this MemberDeclarationSyntax member)
        => new(s_declarationCache.GetValue(member, static member =>
        {
            var dictionary = DeclarationFinder.GetAllDeclarations(member);
            return dictionary.ToDictionary(kvp => kvp.Key, kvp => kvp.Value.AsImmutable());
        }));
 
    public static SyntaxToken GetNameToken(this MemberDeclarationSyntax member)
    {
        if (member != null)
        {
            switch (member.Kind())
            {
                case SyntaxKind.EnumDeclaration:
                    return ((EnumDeclarationSyntax)member).Identifier;
                case SyntaxKind.ClassDeclaration:
                case SyntaxKind.RecordDeclaration:
                case SyntaxKind.InterfaceDeclaration:
                case SyntaxKind.StructDeclaration:
                case SyntaxKind.RecordStructDeclaration:
                    return ((TypeDeclarationSyntax)member).Identifier;
                case SyntaxKind.DelegateDeclaration:
                    return ((DelegateDeclarationSyntax)member).Identifier;
                case SyntaxKind.FieldDeclaration:
                    return ((FieldDeclarationSyntax)member).Declaration.Variables.First().Identifier;
                case SyntaxKind.EventFieldDeclaration:
                    return ((EventFieldDeclarationSyntax)member).Declaration.Variables.First().Identifier;
                case SyntaxKind.PropertyDeclaration:
                    return ((PropertyDeclarationSyntax)member).Identifier;
                case SyntaxKind.EventDeclaration:
                    return ((EventDeclarationSyntax)member).Identifier;
                case SyntaxKind.MethodDeclaration:
                    return ((MethodDeclarationSyntax)member).Identifier;
                case SyntaxKind.ConstructorDeclaration:
                    return ((ConstructorDeclarationSyntax)member).Identifier;
                case SyntaxKind.DestructorDeclaration:
                    return ((DestructorDeclarationSyntax)member).Identifier;
                case SyntaxKind.IndexerDeclaration:
                    return ((IndexerDeclarationSyntax)member).ThisKeyword;
                case SyntaxKind.OperatorDeclaration:
                    return ((OperatorDeclarationSyntax)member).OperatorToken;
            }
        }
 
        // Conversion operators don't have names.
        return default;
    }
 
    public static int GetArity(this MemberDeclarationSyntax member)
    {
        if (member != null)
        {
            switch (member.Kind())
            {
                case SyntaxKind.ClassDeclaration:
                case SyntaxKind.RecordDeclaration:
                case SyntaxKind.InterfaceDeclaration:
                case SyntaxKind.StructDeclaration:
                case SyntaxKind.RecordStructDeclaration:
                    return ((TypeDeclarationSyntax)member).Arity;
                case SyntaxKind.DelegateDeclaration:
                    return ((DelegateDeclarationSyntax)member).Arity;
                case SyntaxKind.MethodDeclaration:
                    return ((MethodDeclarationSyntax)member).Arity;
            }
        }
 
        return 0;
    }
 
    public static TypeParameterListSyntax GetTypeParameterList(this MemberDeclarationSyntax member)
    {
        if (member != null)
        {
            switch (member.Kind())
            {
                case SyntaxKind.ClassDeclaration:
                case SyntaxKind.RecordDeclaration:
                case SyntaxKind.InterfaceDeclaration:
                case SyntaxKind.StructDeclaration:
                case SyntaxKind.RecordStructDeclaration:
                    return ((TypeDeclarationSyntax)member).TypeParameterList;
                case SyntaxKind.DelegateDeclaration:
                    return ((DelegateDeclarationSyntax)member).TypeParameterList;
                case SyntaxKind.MethodDeclaration:
                    return ((MethodDeclarationSyntax)member).TypeParameterList;
            }
        }
 
        return null;
    }
 
    public static MemberDeclarationSyntax WithParameterList(
        this MemberDeclarationSyntax member,
        BaseParameterListSyntax parameterList)
    {
        if (member != null)
        {
            switch (member.Kind())
            {
                case SyntaxKind.DelegateDeclaration:
                    return ((DelegateDeclarationSyntax)member).WithParameterList((ParameterListSyntax)parameterList);
                case SyntaxKind.MethodDeclaration:
                    return ((MethodDeclarationSyntax)member).WithParameterList((ParameterListSyntax)parameterList);
                case SyntaxKind.ConstructorDeclaration:
                    return ((ConstructorDeclarationSyntax)member).WithParameterList((ParameterListSyntax)parameterList);
                case SyntaxKind.IndexerDeclaration:
                    return ((IndexerDeclarationSyntax)member).WithParameterList((BracketedParameterListSyntax)parameterList);
                case SyntaxKind.OperatorDeclaration:
                    return ((OperatorDeclarationSyntax)member).WithParameterList((ParameterListSyntax)parameterList);
                case SyntaxKind.ConversionOperatorDeclaration:
                    return ((ConversionOperatorDeclarationSyntax)member).WithParameterList((ParameterListSyntax)parameterList);
            }
        }
 
        return null;
    }
 
    public static TypeSyntax GetMemberType(this MemberDeclarationSyntax member)
    {
        if (member != null)
        {
            switch (member.Kind())
            {
                case SyntaxKind.DelegateDeclaration:
                    return ((DelegateDeclarationSyntax)member).ReturnType;
                case SyntaxKind.MethodDeclaration:
                    return ((MethodDeclarationSyntax)member).ReturnType;
                case SyntaxKind.OperatorDeclaration:
                    return ((OperatorDeclarationSyntax)member).ReturnType;
                case SyntaxKind.PropertyDeclaration:
                    return ((PropertyDeclarationSyntax)member).Type;
                case SyntaxKind.IndexerDeclaration:
                    return ((IndexerDeclarationSyntax)member).Type;
                case SyntaxKind.EventDeclaration:
                    return ((EventDeclarationSyntax)member).Type;
                case SyntaxKind.EventFieldDeclaration:
                    return ((EventFieldDeclarationSyntax)member).Declaration.Type;
                case SyntaxKind.FieldDeclaration:
                    return ((FieldDeclarationSyntax)member).Declaration.Type;
            }
        }
 
        return null;
    }
 
    public static bool HasMethodShape(this MemberDeclarationSyntax memberDeclaration)
        => memberDeclaration is BaseMethodDeclarationSyntax;
 
    public static BlockSyntax GetBody(this MemberDeclarationSyntax memberDeclaration)
        => memberDeclaration switch
        {
            BaseMethodDeclarationSyntax method => method.Body,
            _ => null,
        };
 
    public static ArrowExpressionClauseSyntax GetExpressionBody(this MemberDeclarationSyntax memberDeclaration)
        => memberDeclaration switch
        {
            BaseMethodDeclarationSyntax method => method.ExpressionBody,
            IndexerDeclarationSyntax indexer => indexer.ExpressionBody,
            PropertyDeclarationSyntax property => property.ExpressionBody,
            _ => null,
        };
 
    public static MemberDeclarationSyntax WithBody(this MemberDeclarationSyntax memberDeclaration, BlockSyntax body)
        => (memberDeclaration as BaseMethodDeclarationSyntax)?.WithBody(body);
}