smart-bow-infrared/Assets/AppUI/Scripts/View/Home/DeviceView/DeviceView_ItemShow.cs

using AppUI.Bluetooth;
using AppUI.Localization;
using AppUI.Manager;
using AppUI.Manager.View;
using AppUI.View.Home.Main;
using InfraredManager;
using SmartBowSDK;
using System;
using TMPro;
using UnityEngine;
using UnityEngine.UI;
using BLEAimhandler = AppUI.Bluetooth.AimHandler;
using SdkAimDeviceInfo = SmartBowSDK.AimDeviceInfo;

namespace AppUI.View.Home.DeviceView
{
    public class DeviceView_ItemShow : MonoBehaviour
    {
        [Tooltip("当前选择的用户设备")]
        [SerializeField] BluetoothPlayer bluetoothPlayer = BluetoothPlayer.FIRST_PLAYER;
        [SerializeField] Image selectPanel;
        [SerializeField] Button btnBg;
        [SerializeField] TMP_Text titleText;

        [Header("操作按钮（连接/屏幕定位 + 切换设备）")]
        [SerializeField] Button btnPrimaryAction;
        [SerializeField] Image btnPrimaryActionBg;
        [SerializeField] TMP_Text btnPrimaryActionLabel;
        [SerializeField] Button btnSwitchDevice;
        [SerializeField] Image btnSwitchDeviceBg;
        [SerializeField] TMP_Text btnSwitchDeviceLabel;

        [SerializeField] GameObject bettery;
        [SerializeField] GameObject betteryBar;
        [SerializeField] GameObject betteryValue;

        [SerializeField] Image bowBg;

        float countingTime1 = 5;

        [SerializeField] int deviceIndex;

        bool bShowInfo;

        string omitText;
        int dotCount;
        float interval = 0.5f;
        float timer;

        [SerializeField] TMP_Text curConnectStatus;
        [SerializeField] TMP_Text curScreenPositioningStatus;
        const string KeyScreenPositioningDone = "appui-sideslip-screen-positioning-done";
        const string KeyScreenPositioningIncomplete = "appui-sideslip-screen-positioning-incomplete";
        const string KeyStatusNotConnected = "appui-device-status-not-connected";
        const string KeyStatusConnecting = "appui-device-status-connecting";
        const string KeyStatusConnected = "appui-device-status-connected";
        const string KeyBtnConnect = "appui-device-btn-connect";
        const string KeyBtnDisconnect = "appui-device-btn-disconnect";
        const string KeyBtnSwitch = "appui-device-btn-switch";
        const string KeyBtnScreenPositioning = "appui-device-btn-screen-positioning";
        const string KeyTipInfraredConnect = "appui-device-tip-infrared-connect";

        static readonly Color32 BlackBg = new Color32(0, 0, 0, 255);
        static readonly Color32 WhiteBg = new Color32(255, 255, 255, 255);
        static readonly Color32 WhiteText = new Color32(255, 255, 255, 255);
        static readonly Color32 BlackText = new Color32(50, 50, 50, 255);
        static readonly Color32 StatusCompleteColor = new Color32(0, 0, 0, 255);
        static readonly Color32 StatusIncompleteColor = new Color32(145, 144, 144, 255);

        SmartBowDeviceHub Hub => SmartBowDeviceHub.ins;

        public BluetoothPlayer Player => bluetoothPlayer;

        /// <summary>连接 / 屏幕定位按钮节点。</summary>
        public Button BtnPrimaryAction => btnPrimaryAction;

        /// <summary>切换设备按钮节点。</summary>
        public Button BtnSwitchDevice => btnSwitchDevice;

        public GameObject BtnPrimaryActionNode => btnPrimaryAction != null ? btnPrimaryAction.gameObject : null;

        public GameObject BtnSwitchDeviceNode => btnSwitchDevice != null ? btnSwitchDevice.gameObject : null;

        enum PrimaryActionKind
        {
            ConnectSideSlip,
            InfraredView,
            Disconnect,
        }

        PrimaryActionKind _primaryActionKind = PrimaryActionKind.ConnectSideSlip;

        bool IsConnectSuccess()
        {
            return Hub != null && Hub.IsConnected(bluetoothPlayer);
        }

        bool IsConnecting()
        {
            return Hub != null && Hub.IsConnecting(bluetoothPlayer);
        }

        bool IsScreenPositioned()
        {
            InfraredCameraHelper helper = InfraredCameraHelper.GetInstance();
            return helper != null && helper.IsScreenLoateOK();
        }

        void Awake()
        {
            ResolveUiReferences();
            if (Hub?.Aim != null)
                Hub.Aim.aimDeviceInfoChangeEvent += onShowDeviceInfo;
        }

        void ResolveUiReferences()
        {
            if (titleText == null)
            {
                Transform title = transform.Find("Panel/title");
                if (title != null)
                    titleText = title.GetComponent<TMP_Text>();
            }

            ResolveActionButtonReferences();
        }

        void ResolveActionButtonReferences()
        {
            if (btnPrimaryAction == null)
            {
                Transform screen = transform.Find("connected/bottom/Buttons/Screen");
                if (screen != null)
                {
                    btnPrimaryAction = screen.GetComponent<Button>();
                    btnPrimaryActionBg = screen.GetComponent<Image>();
                    btnPrimaryActionLabel = screen.GetComponentInChildren<TMP_Text>(true);
                }
            }

            if (btnSwitchDevice == null)
            {
                Transform switchBtn = transform.Find("connected/bottom/Buttons/Switch");
                if (switchBtn != null)
                {
                    btnSwitchDevice = switchBtn.GetComponent<Button>();
                    btnSwitchDeviceBg = switchBtn.GetComponent<Image>();
                    btnSwitchDeviceLabel = switchBtn.GetComponentInChildren<TMP_Text>(true);
                }
            }
        }

        void SetDeviceTitle(string text)
        {
            if (titleText != null)
                titleText.text = text;
        }

        void SetDeviceSprite(Sprite sprite)
        {
            if (bowBg != null && sprite != null)
                bowBg.sprite = sprite;
        }

        void OnEnable()
        {
            AppUILocalization.OnLanguageChanged += HandleLanguageChanged;
        }

        void OnDisable()
        {
            AppUILocalization.OnLanguageChanged -= HandleLanguageChanged;
        }

        void HandleLanguageChanged(LanguageEnum _)
        {
            onUpdateStatusInfo();
            RefreshActionButtons();
        }

        void Start()
        {
            ResolveUiReferences();
            btnBg?.onClick.AddListener(OnSelectEvent);
            btnPrimaryAction?.onClick.AddListener(OnClickPrimaryAction);
            btnSwitchDevice?.onClick.AddListener(OnClickSwitchDevice);

            if (Hub != null)
            {
                Hub.OnStateChanged += OnHubStateChanged;
                if (IsConnectSuccess())
                {
                    int battery = Hub.GetBattery(bluetoothPlayer);
                    if (battery > 0)
                        RenderBattery(-1, battery);
                }
            }
            onUpdateStatusInfo();
            RefreshActionButtons();

            //开始选择一次
            if(bluetoothPlayer == BluetoothPlayer.FIRST_PLAYER)
            {
                OnSelectEvent();
            }
        }

        void OnDestroy()
        {
            if (Hub != null)
                Hub.OnStateChanged -= OnHubStateChanged;
            if (Hub?.Aim != null)
                Hub.Aim.aimDeviceInfoChangeEvent -= onShowDeviceInfo;
        }

        void OnHubStateChanged(SmartBowDeviceState state)
        {
            onUpdateStatusInfo();
            RefreshActionButtons();
        }

        void Update()
        {
            if (Hub == null) return;

            if (IsConnectSuccess())
            {
                if (countingTime1 < 5)
                    countingTime1 += Time.deltaTime;
                else
                {
                    countingTime1 = 0;
                    RenderBattery(-1, Hub.GetBattery(bluetoothPlayer));
                }
            }
            else if (!IsConnecting())
            {
                onShowDeviceInfo();
                RenderBattery(-1, 0);
            }
            else
            {
                RenderBattery(-1, 0);
                onShowDeviceInfo();
            }

           onUpdateStatusInfo();


        }

        void UpdateLoadingText()
        {
            timer += Time.deltaTime;
            if (timer >= interval)
            {
                dotCount = (dotCount + 1) % 4;
                omitText = new string('.', dotCount);
                timer = 0f;
            }
        }

        public void StopLoading()
        {
            dotCount = 0;
            omitText = "";
        }

        public void RenderBattery(int deviceID, int value)
        {
            if (betteryBar == null || betteryValue == null) return;
            Image img = betteryBar.GetComponent<Image>();
            Text txt = betteryValue.GetComponent<Text>();
            if (img == null || txt == null) return;
            img.fillAmount = value / 100f;
            txt.text = value + "%";
        }

        bool TryGetDeviceInfo(out int type, out bool bInitMac)
        {
            type = -1;
            bInitMac = false;
            var aim = Hub?.Aim;
            if (aim == null) return false;
            var player = (BluetoothPlayer)deviceIndex;
            SdkAimDeviceInfo info = aim.GetAimDeviceInfo(player);
            if (info == null) return false;
            type = aim.GetDeviceType(player);
            bInitMac = info.bInitMac;
            return true;
        }

        public void onShowDeviceInfo()
        {
            if (!TryGetDeviceInfo(out int deviceType, out bool bInitMac) || deviceType == -1 || !bInitMac)
            {
                bShowInfo = false;
                RefreshActionButtons();
                return;
            }

            bShowInfo = true;
            var deviceTypeEnum = (AimDeviceType)deviceType;
            if (Hub == null || !Hub.TryGetDeviceDisplay(deviceTypeEnum, out Sprite sprite, out string displayName))
            {
                bShowInfo = false;
                RefreshActionButtons();
                return;
            }

            SetDeviceTitle(displayName);
            SetDeviceSprite(sprite);
            RefreshActionButtons();
        }

        public void onUpdateStatusInfo()
        {
            if (Hub == null || curConnectStatus == null) return;

            if (IsConnecting())
                UpdateLoadingText();
            else
                StopLoading();

            string statusKey = ResolveStatusTextKey(Hub.GetStatusTextKey(bluetoothPlayer));
            curConnectStatus.text = AppUILocalization.GetTextByKey(statusKey, curConnectStatus) + omitText;
            curConnectStatus.color = IsConnectSuccess() ? StatusCompleteColor : StatusIncompleteColor;
            RefreshScreenPositioningStatusText();
        }

        void RefreshScreenPositioningStatusText()
        {
            if (curScreenPositioningStatus == null) return;
            if (!bInfraredDevices())
            {
                curScreenPositioningStatus.gameObject.SetActive(false);
                return;
            }
            curScreenPositioningStatus.gameObject.SetActive(true);
            bool positioned = IsScreenPositioned();
            curScreenPositioningStatus.text = AppUILocalization.GetTextByKey(positioned ? KeyScreenPositioningDone : KeyScreenPositioningIncomplete, curScreenPositioningStatus);
            curScreenPositioningStatus.color = positioned ? StatusCompleteColor : StatusIncompleteColor;
        }

        /// <summary>刷新主操作按钮与切换按钮的文案、样式与可见性。</summary>
        public void RefreshActionButtons()
        {
            ResolveActionButtonReferences();
            RefreshSwitchButton();
            RefreshPrimaryActionButton();
            RefreshScreenPositioningStatusText();
        }

        void RefreshSwitchButton()
        {
            if (btnSwitchDevice == null) return;
            bool show = bShowInfo;
            if (btnSwitchDevice.gameObject.activeSelf != show)
                btnSwitchDevice.gameObject.SetActive(show);
            if (!show) return;

            ApplyButtonStyle(btnSwitchDeviceBg, btnSwitchDeviceLabel, false, KeyBtnSwitch);
        }

        void RefreshPrimaryActionButton()
        {
            if (btnPrimaryAction == null) return;

            bool connected = IsConnectSuccess();
            bool screenPositioned = IsScreenPositioned();
            bool isInfrared = bInfraredDevices();
            bool isNineAxis = bNineAxisDevices();

            if (!bShowInfo)
            {
                btnPrimaryAction.gameObject.SetActive(false);
                return;
            }

            btnPrimaryAction.gameObject.SetActive(true);

            if (isInfrared)
            {
                if (connected)
                {
                    _primaryActionKind = PrimaryActionKind.InfraredView;
                    ApplyButtonStyle(
                        btnPrimaryActionBg,
                        btnPrimaryActionLabel,
                        !screenPositioned,
                        KeyBtnScreenPositioning);
                }
                else
                {
                    _primaryActionKind = PrimaryActionKind.ConnectSideSlip;
                    ApplyButtonStyle(btnPrimaryActionBg, btnPrimaryActionLabel, true, KeyBtnConnect);
                }
                return;
            }

            if (isNineAxis && connected)
            {
                _primaryActionKind = PrimaryActionKind.Disconnect;
                ApplyButtonStyle(btnPrimaryActionBg, btnPrimaryActionLabel, false, KeyBtnDisconnect);
                return;
            }

            _primaryActionKind = PrimaryActionKind.ConnectSideSlip;
            ApplyButtonStyle(btnPrimaryActionBg, btnPrimaryActionLabel, true, KeyBtnConnect);
        }

        static string ResolveStatusTextKey(string legacyKey)
        {
            return legacyKey switch
            {
                "Connect_BLE_NotConnected" => KeyStatusNotConnected,
                "Connect_BLE_Connecting" => KeyStatusConnecting,
                "Connect_BLE_Connected" => KeyStatusConnected,
                _ => legacyKey
            };
        }

        static void ApplyButtonStyle(Image bg, TMP_Text label, bool blackBackground, string textKey)
        {
            if (bg != null)
                bg.color = blackBackground ? BlackBg : WhiteBg;
            if (label != null)
            {
                label.text = AppUILocalization.GetTextByKey(textKey);
                label.color = blackBackground ? WhiteText : BlackText;
            }
        }

        void OnClickPrimaryAction()
        {
            AudioMgr.ins.PlayBtn();
            if (!getPanelStatus()) return;

            Hub?.SetActivePlayer(bluetoothPlayer);
            Hub?.Aim?.SetAimDeviceSelectIndex(deviceIndex);

            if (_primaryActionKind == PrimaryActionKind.InfraredView)
            {
                if (!IsConnectSuccess())
                {
                    PopupMgr.ins.ShowTipTop(AppUILocalization.GetTextByKey(KeyTipInfraredConnect));
                    return;
                }
                ViewManager.ShowView(UIViewType.InfraredView);
                return;
            }

            if (_primaryActionKind == PrimaryActionKind.Disconnect)
            {
                OnDisConnectedEvent();
                return;
            }

            Hub?.Aim?.PrepareDirectSideSlipConnect(deviceIndex);
            ModuleViewMgr.ins.ShowSideSlip(guideDeviceType: ResolveGuideDeviceTypeForSideSlip());
        }

        AimDeviceType? ResolveGuideDeviceTypeForSideSlip()
        {
            BLEAimhandler aim = Hub?.Aim;
            if (aim == null)
                return null;

            int deviceType = aim.GetDeviceType(bluetoothPlayer);
            if (deviceType == (int)AimDeviceType.NONE)
                return null;

            return (AimDeviceType)deviceType;
        }

        void OnClickSwitchDevice()
        {
            AudioMgr.ins.PlayBtn();
            if (!getPanelStatus()) return;

            Hub?.SetActivePlayer(bluetoothPlayer);
            Hub?.Aim?.SetAimDeviceSelectIndex(deviceIndex);

            if (IsConnectSuccess())
                OnDisConnectedEvent();

            Hub?.CancelConnecting();
            Hub?.Aim?.onCreateTempAimDeviceInfo();

            DeviceView deviceView = GetComponentInParent<DeviceView>();
            if (deviceView != null)
            {
                deviceView.BeginDeviceSelectionFlow();
                return;
            }

            if (GlobalData.MyDeviceMode == DeviceMode.Gun)
                ViewManager.ShowView(UIViewType.DeviceGunView);
            else
                ViewManager.ShowView(UIViewType.DeviceArcheryView);
        }

        void CloseAllPanelStatus()
        {
            DeviceView root = GetComponentInParent<DeviceView>();
            if (root == null) return;
            foreach (DeviceView_ItemShow item in root.GetComponentsInChildren<DeviceView_ItemShow>(true))
            {
                if (item.selectPanel != null)
                    item.selectPanel.enabled = false;
            }
        }

        public void setPanelStatus(bool active)
        {
            if (selectPanel != null)
                selectPanel.enabled = active;
            if (!active) return;

            Hub?.SetActivePlayer(bluetoothPlayer);
            Hub?.Aim?.SetAimDeviceSelectIndex(deviceIndex);
            Debug.Log("当前选择的操作蓝牙用户 deviceSelectIndex：" + deviceIndex);
        }

        public bool getPanelStatus()
        {
            bool enable = selectPanel != null && selectPanel.enabled;
            CloseAllPanelStatus();
            setPanelStatus(true);
            return enable;
        }

        public bool getCurrentPanelEnable() => selectPanel != null && selectPanel.enabled;

        public bool getBLEConnectState() => IsConnectSuccess();

        public void OnConnectedEvent()
        {
            if (!getPanelStatus()) return;

            Hub?.Aim?.SetAimDeviceSelectIndex(deviceIndex);
            Hub?.Aim?.onCreateAimDeviceInfoById();

            if (IsConnectSuccess())
            {
                Hub?.ConnectActivePlayer();
                return;
            }

            if (bluetoothPlayer == BluetoothPlayer.FIRST_PLAYER)
                Hub?.Connect1P();
            else
                Hub?.Connect2P();
        }

        public void OnDisConnectedEvent()
        {
            if (!getPanelStatus()) return;

            Hub?.Aim?.SetAimDeviceSelectIndex(deviceIndex);
            Hub?.Aim?.onCreateAimDeviceInfoById();

            if (bluetoothPlayer == BluetoothPlayer.FIRST_PLAYER)
            {
                if (Hub == null) return;
                if (IsConnectSuccess())
                    Hub.Connect1P();
                else
                    Hub.SetConnectCanceled(true);
            }
            else if (IsConnectSuccess())
            {
                Hub?.Set2PNoNeedToReconnect(false);
                Hub?.Connect2P();
            }
        }

        public void OnSelectEvent()
        {
            if (getCurrentPanelEnable()) return;
            AudioMgr.ins.PlayBtn();
            Hub?.Aim?.SetAimDeviceSelectIndex(deviceIndex);
            CloseAllPanelStatus();
            setPanelStatus(true);
        }

        public bool getCurrentItemVisible() => selectPanel.enabled && bShowInfo;

        public bool bInfraredDevices()
        {
            if (!TryGetDeviceInfo(out int deviceType, out bool bInitMac) || deviceType == -1 || !bInitMac)
                return false;
            switch ((AimDeviceType)deviceType)
            {
                case AimDeviceType.HOUYIPRO:
                case AimDeviceType.Gun:
                case AimDeviceType.ARTEMISPRO:
                case AimDeviceType.PistolM17:
                case AimDeviceType.RifleM416:
                    return true;
                default:
                    return false;
            }
        }

        public bool bNineAxisDevices()
        {
            if (!TryGetDeviceInfo(out int deviceType, out bool bInitMac) || deviceType == -1 || !bInitMac)
                return false;
            return deviceType == (int)AimDeviceType.HOUYI
                || deviceType == (int)AimDeviceType.HOUYI2
                || deviceType == (int)AimDeviceType.ARTEMIS;
        }
    }
}