// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.ComponentModel;
using System.ComponentModel.Design;
 
namespace System.Windows.Forms.Design;
 
internal class BindingSourceDesigner : ComponentDesigner
{
    private bool _bindingUpdatedByUser;
 
    public bool BindingUpdatedByUser
    {
        set => _bindingUpdatedByUser = value;
    }
 
    public override void Initialize(IComponent component)
    {
        base.Initialize(component);
 
        if (TryGetService(out IComponentChangeService? componentChangeService))
        {
            componentChangeService.ComponentChanged += OnComponentChanged;
            componentChangeService.ComponentRemoving += OnComponentRemoving;
        }
    }
 
    protected override void Dispose(bool disposing)
    {
        if (disposing)
        {
            if (TryGetService(out IComponentChangeService? componentChangeService))
            {
                componentChangeService.ComponentChanged -= OnComponentChanged;
                componentChangeService.ComponentRemoving -= OnComponentRemoving;
            }
        }
 
        base.Dispose(disposing);
    }
 
    private void OnComponentChanged(object? sender, ComponentChangedEventArgs e)
    {
        if (_bindingUpdatedByUser && e.Component == Component &&
            e.Member is not null && (e.Member.Name == "DataSource" || e.Member.Name == "DataMember"))
        {
            _bindingUpdatedByUser = false;
 
            DataSourceProviderService? dataSourceProviderService = GetService<DataSourceProviderService>();
            dataSourceProviderService?.NotifyDataSourceComponentAdded(Component);
        }
    }
 
    private void OnComponentRemoving(object? sender, ComponentEventArgs e)
    {
        if (Component is BindingSource bindingSource && bindingSource.DataSource == e.Component)
        {
            IComponentChangeService? componentChangeService = GetService<IComponentChangeService>();
            string previousDataMember = bindingSource.DataMember;
 
            PropertyDescriptorCollection propertyDescriptorCollection = TypeDescriptor.GetProperties(bindingSource);
            PropertyDescriptor? propertyDescriptor = propertyDescriptorCollection?["DataMember"];
 
            if (propertyDescriptor is not null)
            {
                componentChangeService?.OnComponentChanging(bindingSource, propertyDescriptor);
            }
 
            bindingSource.DataSource = null;
 
            if (propertyDescriptor is not null)
            {
                componentChangeService?.OnComponentChanged(bindingSource, propertyDescriptor, previousDataMember, string.Empty);
            }
        }
    }
}