using AcDream.App.UI;

namespace AcDream.App.Input;

internal interface IInputCaptureSource
{
    bool WantCaptureMouse { get; }
    bool WantCaptureKeyboard { get; }
    bool DevToolsWantCaptureKeyboard { get; }
}

internal sealed class DevToolsInputCaptureSource
{
    private readonly bool _enabled;

    public DevToolsInputCaptureSource(bool enabled) => _enabled = enabled;

    public bool WantCaptureMouse =>
        _enabled && ImGuiNET.ImGui.GetIO().WantCaptureMouse;

    public bool WantCaptureKeyboard =>
        _enabled && ImGuiNET.ImGui.GetIO().WantCaptureKeyboard;
}

internal sealed class RetainedUiInputCaptureSlot
{
    public UiRoot? Root { get; private set; }

    public IDisposable Bind(UiRoot root)
    {
        ArgumentNullException.ThrowIfNull(root);
        if (Root is not null)
            throw new InvalidOperationException("Retained UI input capture is already bound.");
        Root = root;
        return new Binding(this, root);
    }

    public bool WantCaptureMouse => Root?.WantsMouse ?? false;
    public bool WantCaptureKeyboard => Root?.WantsKeyboard ?? false;

    private void Unbind(UiRoot expected)
    {
        if (ReferenceEquals(Root, expected))
            Root = null;
    }

    private sealed class Binding : IDisposable
    {
        private RetainedUiInputCaptureSlot? _owner;
        private readonly UiRoot _expected;

        public Binding(RetainedUiInputCaptureSlot owner, UiRoot expected)
        {
            _owner = owner;
            _expected = expected;
        }

        public void Dispose() =>
            Interlocked.Exchange(ref _owner, null)?.Unbind(_expected);
    }
}

internal sealed class CompositeInputCaptureSource : IInputCaptureSource
{
    private readonly DevToolsInputCaptureSource _devTools;
    private readonly RetainedUiInputCaptureSlot _retained;

    public CompositeInputCaptureSource(
        DevToolsInputCaptureSource devTools,
        RetainedUiInputCaptureSlot retained)
    {
        _devTools = devTools ?? throw new ArgumentNullException(nameof(devTools));
        _retained = retained ?? throw new ArgumentNullException(nameof(retained));
    }

    public bool WantCaptureMouse =>
        _devTools.WantCaptureMouse || _retained.WantCaptureMouse;

    public bool WantCaptureKeyboard =>
        DevToolsWantCaptureKeyboard || _retained.WantCaptureKeyboard;

    public bool DevToolsWantCaptureKeyboard =>
        _devTools.WantCaptureKeyboard;
}