using Content.Shared.MachineLinking;
using Robust.Client.AutoGenerated;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.Controls;
using Robust.Client.UserInterface.CustomControls;
using Robust.Client.UserInterface.XAML;
using Robust.Client.Graphics;
using Robust.Shared.Prototypes;

namespace Content.Client.MachineLinking.UI
{
    [GenerateTypedNameReferences]
    public sealed partial class SignalPortSelectorMenu : DefaultWindow
    {
        private SignalPortSelectorBoundUserInterface _bui;
        private LinksRender _links;

        private ButtonGroup _buttonGroup = new();
        private IPrototypeManager _protoMan;

        public SignalPortSelectorMenu(SignalPortSelectorBoundUserInterface boundUserInterface)
        {
            RobustXamlLoader.Load(this);
            _bui = boundUserInterface;
            _links = new(ButtonContainerLeft, ButtonContainerRight);
            ContainerMiddle.AddChild(_links);
            ButtonClear.OnPressed += _ => _bui.OnClearPressed();
            ButtonLinkDefault.OnPressed += _ => _bui.OnLinkDefaultPressed();
            _protoMan = IoCManager.Resolve<IPrototypeManager>();
        }

        public void UpdateState(SignalPortsState state)
        {
            HeaderLeft.SetMessage(state.TransmitterName);
            ButtonContainerLeft.DisposeAllChildren();
            foreach (var port in state.TransmitterPorts)
            {
                var proto = _protoMan.Index<TransmitterPortPrototype>(port);
                var portButton = new Button()
                {
                    Text = Loc.GetString(proto.Name),
                    ToolTip = Loc.GetString(proto.Description),
                    ToggleMode = true,
                    Group = _buttonGroup
                };
                portButton.OnPressed += _ => _bui.OnTransmitterPortSelected(port);
                ButtonContainerLeft.AddChild(portButton);
            }

            HeaderRight.SetMessage(state.ReceiverName);
            ButtonContainerRight.DisposeAllChildren();
            foreach (var port in state.ReceiverPorts)
            {
                var proto = _protoMan.Index<ReceiverPortPrototype>(port);
                var portButton = new Button()
                {
                    Text = Loc.GetString(proto.Name),
                    ToolTip = Loc.GetString(proto.Description),
                    ToggleMode = true,
                    Group = _buttonGroup
                };
                portButton.OnPressed += _ => _bui.OnReceiverPortSelected(port);
                ButtonContainerRight.AddChild(portButton);
            }

            _links.Links = state.Links;
        }

        private sealed class LinksRender : Control
        {
            public List<(int, int)> Links = new();
            public BoxContainer LeftButton;
            public BoxContainer RightButton;

            public LinksRender(BoxContainer leftButton, BoxContainer rightButton)
            {
                LeftButton = leftButton;
                RightButton = rightButton;
            }

            protected override void Draw(DrawingHandleScreen handle)
            {
                var leftOffset = LeftButton.PixelPosition.Y;
                var rightOffset = RightButton.PixelPosition.Y;
                foreach (var (left, right) in Links)
                {
                    var leftChild = LeftButton.GetChild(left);
                    var rightChild = RightButton.GetChild(right);
                    var y1 = leftChild.PixelPosition.Y + leftChild.PixelHeight / 2 + leftOffset;
                    var y2 = rightChild.PixelPosition.Y + rightChild.PixelHeight / 2 + rightOffset;
                    handle.DrawLine((0, y1), (PixelWidth, y2), Color.Cyan);
                }
            }
        }
    }
}