// NOTE: Taken from https://github.com/dilmerv/UnityARFoundationEssentials/blob/master/Assets/Scripts/TrackedImageInfoMultipleManager.cs
// This has been modified to fit the need of the UDRI XRTeam's use-cases

using System.Collections.Generic;
using System.Linq;
using Unity.Mathematics;
using UnityEngine;
using UnityEngine.XR.ARFoundation;

namespace MAVRIC.GEEEKCup.Samples.QRCode
{
    public class TrackedImageChanged : MonoBehaviour
    { 
        [SerializeField] private ARTrackedImageManager trackedImageManager;
        [SerializeField] private GameObject[] arPrefabs;

        [SerializeField] private float3 scaleFactor = new (0.1f, 0.1f, 0.1f);

        private Dictionary<string, GameObject> arObjects = new ();

        private void Awake()
        {
            if (trackedImageManager == null)
            {
                if (!TryGetComponent(out trackedImageManager))
                {
                    Debug.LogError("Failed to find ARTrackedImageManager", this.gameObject);
                }
            }
        }

        private void OnEnable()
        {
            if (trackedImageManager == null) return;
            
            trackedImageManager.trackedImagesChanged += OnTrackedImagesChanged;
            
            SpawnAllARObjects();
        }


        private void OnDisable()
        {
            if (trackedImageManager == null) return;
            
            trackedImageManager.trackedImagesChanged -= OnTrackedImagesChanged;

            for (int i = arObjects.Keys.Count - 1; i >= 0; i--)
            {
                var key = arObjects.Keys.ElementAt(i);
                if (arObjects.Remove(key, out var value))
                {
                    Destroy(value);
                }
            }
        }
        
        private void SpawnAllARObjects()
        {
            // setup all game objects in dictionary
            foreach (GameObject arObject in arPrefabs)
            {
                GameObject newARObject = Instantiate(arObject, Vector3.zero, Quaternion.identity);
                newARObject.SetActive(false);

                newARObject.name = arObject.name;
                arObjects.Add(arObject.name, newARObject);
            }
        }

        private void OnTrackedImagesChanged(ARTrackedImagesChangedEventArgs eventArgs)
        {
            foreach (ARTrackedImage trackedImage in eventArgs.added)
            {
                UpdateARImage(trackedImage);
            }

            foreach (ARTrackedImage trackedImage in eventArgs.updated)
            {
                UpdateARImage(trackedImage);
            }

            foreach (ARTrackedImage trackedImage in eventArgs.removed)
            {
                arObjects[trackedImage.name].SetActive(false);
            }
        }

        private void UpdateARImage(ARTrackedImage trackedImage)
        {
            // Assign and Place Game Object
            AssignGameObject(trackedImage.referenceImage.name, trackedImage.transform.position, scaleFactor, trackedImage.transform.rotation);

            Debug.Log($"trackedImage.referenceImage.name: {trackedImage.referenceImage.name}");
        }

        private void AssignGameObject(string name, Vector3 newPosition, Vector3 newScale, Quaternion newRotation)
        {
            if (arPrefabs != null)
            {
                GameObject goARObject = arObjects[name];
                goARObject.SetActive(true);
                
                goARObject.transform.position = newPosition;
                goARObject.transform.localScale = newScale;
                goARObject.transform.localRotation = newRotation;
                
                foreach (GameObject go in arObjects.Values)
                {
                    Debug.Log($"Go in arObjects.Values: {go.name}");
                    if (go.name != name)
                    {
                        go.SetActive(false);
                    }
                }
            }
        }
    }
}