// 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.CodeAnalysis;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.Diagnostics;
using Microsoft.CodeAnalysis.Operations;
using Microsoft.NetCore.Analyzers.Runtime;

namespace Microsoft.NetCore.CSharp.Analyzers.Runtime
{
    [DiagnosticAnalyzer(LanguageNames.CSharp)]
    public sealed class CSharpPreferDictionaryContainsMethods : PreferDictionaryContainsMethods
    {
        private protected override bool TryGetPropertyReferenceOperation(IInvocationOperation containsInvocation, [NotNullWhen(true)] out IPropertySymbol? propertySymbol)
        {
            IMethodSymbol method = containsInvocation.TargetMethod;
            IOperation? receiver = null;

            //  In C#, the "this" argument is included in the argument list for extension methods.
            if (method.IsExtensionMethod && method.Parameters.Length == 2)
            {
                receiver = containsInvocation.Arguments[0].Value;
            }
            else if (method.Parameters.Length == 1)
            {
                receiver = containsInvocation.Instance;
            }

            //  The receiver may be a conversion operation if the invocation is an extension method.
            receiver = receiver is IConversionOperation conversion ? conversion.Operand : receiver;
            propertySymbol = (receiver as IPropertyReferenceOperation)?.Property;

            return propertySymbol is not null;
        }
    }
}