XRLib/Assets/Scripts/UVC/Factory/Alarm/AlarmManager.cs

#nullable enable

using SampleProject;
using System;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using UVC.Core;
using UVC.Data;
using UVC.Extention;
using UVC.Factory.Component;

namespace UVC.Factory.Alarm
{
    public class AlarmManager : SingletonScene<AlarmManager>
    {
        [Tooltip("알람 UI 프리팹입니다. 이 프리팹은 알람 정보를 표시하는 UI 요소를 포함해야 합니다.")]
        public GameObject alarmUIPrefab; // 알람 UI 프리팹 (아래에서 설명)
        private Dictionary<string, AlarmUIController> activeAlarmUIs = new Dictionary<string, AlarmUIController>();

        // FactoryDataManager에서 찾을 수 있도록 참조를 저장
        private FactoryObjectManager? dataManager;

        /// <summary>
        /// AlarmManager의 초기화 메서드입니다.
        /// Awake 메서드에서 호출되며, MonoBehaviour가 생성될 때 한 번만 실행됩니다.
        /// </summary>
        protected override void Init()
        {
            SceneMain.Instance.Initialized += OnSceneInitialized;
        }

        private void OnSceneInitialized()
        {
            dataManager = FactoryObjectManager.Instance;
        }

        /// <summary>
        /// Alarm 데이터를 수신하기 위한 MQTT 파이프라인을 설정합니다.
        /// </summary>
        public void Run()
        {
            //데이터를 어떤 형식으로 받을지 정의합니다.
            var dataMask = new DataMask();
            dataMask.ObjectName = "Alarm"; // Alarm 객체의 이름을 설정합니다.
            dataMask.ObjectIdKey = "ID"; // Alarm의 고유 식별자로 사용할 키를 설정합니다.
            dataMask["ID"] = "";
            dataMask["ALARM_TYPE"] = "";
            dataMask["LEVEL"] = "";
            dataMask["LOGISTIC"] = "";
            dataMask["STATE"] = "";
            dataMask["MESSAGE"] = "";
            dataMask["CODE"] = "";
            dataMask["ICON"] = "";
            dataMask["MACHINENAME"] = "";
            dataMask["SHOPNAME"] = "";
            dataMask["TRANSPORT_EQP_NAME"] = "";
            dataMask["TRANSPORT_UNIT_NAME"] = "";
            dataMask["TRANSPORT_EQP_ID"] = "";
            dataMask["TRANSPORT_UNIT_ID"] = "";
            dataMask["SET_TIME"] = DateTime.Now;
            dataMask["UPDATE_TIME"] = DateTime.Now;
            dataMask["TIMESTAMP"] = DateTime.Now;

            // MQTT 파이프라인 정보를 생성합니다.
            // 'ALARM' 토픽을 구독하고, 받은 데이터는 위에서 정의한 dataMask로 매핑하며,
            // 데이터 유효성 검사를 위해 DataValidator를 설정합니다.
            // 데이터가 업데이트되면 OnUpdateData 메서드를 호출하여 처리합니다.

            DataValidator validator = new DataValidator();
            validator.AddValidator("MACHINENAME", value => value != null);

            var pipelineInfo = new MQTTPipeLineInfo("ALARM")
                .setDataMapper(new DataMapper(dataMask))
                .setValidator(validator)
                .setHandler(OnUpdateData);

            // 생성한 파이프라인 정보를 전역 MQTT 파이프라인에 추가합니다.
            AppMain.Instance.MQTTPipeLine.Add(pipelineInfo);
        }
        /// <summary>
        /// 데이터 수신 시 호출되는 공개 핸들러입니다.
        /// 수신된 ALARM 데이터 배열을 비동기적으로 처리하여 씬에 반영합니다.
        /// 추가, 제거, 수정된 ALARM 데이터를 각각 구분하여 처리합니다.
        /// </summary>
        /// <param name="data">수신된 데이터 객체 (DataArray 형태)</param>
        public void OnUpdateData(IDataObject? data)
        {

            if (data == null) return;

            DataArray? arr = data as DataArray;
            if (arr == null) return;
            if (arr.Count == 0)
            {
                arr.ReturnToPool();
                return;
            }

            // 데이터 배열에서 추가, 제거, 수정된 항목 리스트를 가져옵니다.
            var AddedItems = arr.AddedItems;
            var RemovedItems = arr.RemovedItems;
            var ModifiedList = arr.ModifiedList;

            Debug.Log($"AlarmManager OnUpdateData: Added={AddedItems.Count}, Removed={RemovedItems.Count}, Modified={ModifiedList.Count}");

            // 새로 추가된 ALARM 처리
            foreach (var item in AddedItems.ToList())
            {
                HandleNewAlarm(item.Copy());
            }

            // 제거된 ALARM 처리
            foreach (var item in RemovedItems.ToList())
            {
                HandleClearedAlarm(item.Copy());
            }

            // 정보가 수정된 ALARM 처리
            foreach (var item in ModifiedList.ToList())
            {
                HandleModifyAlarm(item.Copy());
            }

            arr.ReturnToPool();
        }

        /// <summary>
        /// AlarmManager가 파괴될 때 호출됩니다.
        /// MQTT 파이프라인에서 'ALARM' 핸들러를 제거하여 메모리 누수를 방지합니다.
        /// </summary>
        protected override void OnDestroy()
        {
            base.OnDestroy();
            AppMain.Instance.MQTTPipeLine.Remove("ALARM");
        }

        public void HandleNewAlarm(DataObject data)
        {
            if (data.Id == null)
            {
                Debug.LogError($"New Alarm Received No data. {data}");
                data.ReturnToPool();
                return;
            }

            // 없으면 새로 생성
            FactoryObject? targetObject = dataManager!.FindById(data.GetString("MACHINENAME")!);
            if (targetObject != null)
            {
                GameObject newUIObject = Instantiate(alarmUIPrefab, transform); // 매니저 하위에 생성
                AlarmUIController newUiController = newUIObject.GetComponent<AlarmUIController>();
                newUiController.Initialize(targetObject, data);

                activeAlarmUIs.Add(data.Id!, newUiController);
            }

        }

        public void HandleModifyAlarm(DataObject data)
        {
            if (data.Id == null)
            {
                Debug.LogError($"Modify Alarm Received No data. {data}");
                data.ReturnToPool();
                return;
            }
            // 이미 해당 설비에 알람 UI가 떠 있는지 확인
            if (activeAlarmUIs.TryGetValue(data.Id!, out AlarmUIController uiController))
            {
                // 있으면 기존 UI에 알람 정보만 추가
                uiController.UpdateAlarm(data);
            }
        }

        public void HandleClearedAlarm(DataObject data)
        {
            if (data.Id.IsNullOrEmpty())
            {
                Debug.LogError($"Clear Alarm Received No data. {data}");
                data.ReturnToPool();
                return;
            }
            if (activeAlarmUIs.TryGetValue(data.Id!, out AlarmUIController uiController))
            {
                uiController.RemoveAlarm(data);
                if (uiController.GetAlarmCount() == 0)
                {
                    activeAlarmUIs.Remove(data.Id!);
                    Destroy(uiController.gameObject);
                }
                data.ReturnToPool();
            }
        }
    }

}