HDRobotics/Assets/Scripts/View/InteractionView.cs

using System;
using System.Collections;
using UnityEngine;
using UnityEngine.XR;
using UnityEngine.XR.Interaction.Toolkit;
using UnityEngine.XR.Interaction.Toolkit.Interactors;

public enum HandSide { Left, Right }

public class InteractionView : MonoBehaviour
{
    public event Action<bool> OnRobotHoverStateChanged;
    public event Action OnRobotGrabStart;
    public event Action<Vector3, Quaternion> OnRobotGrabbed;
    public event Action<RobotData> OnRobotReleased;
    public event Action<int, Vector3> OnPointClicked;
    public event Action<int> OnPointDragStart;
    public event Action<int, Vector3, Quaternion> OnPointDragUpdate;
    public event Action<int> OnPointDragEnd;

    [SerializeField]private NearFarInteractor nearFarInteractor;
    public HandSide handSide;
    private XRBaseInteractor baseInteractor;
    private IXRRayProvider rayProvider;
    private InputDevice hapticDevice;
    private bool isHapticDeviceFound = false;

    [SerializeField]
    [Tooltip("<22>巡<EFBFBD><E5B7A1> <20><> <20><>ġ<EFBFBD><C4A1> ǥ<><C7A5><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>κ<EFBFBD> <20><><EFBFBD><EFBFBD>")]
    private GameObject ghostRobot;
    [SerializeField]
    [Tooltip("<22>巡<EFBFBD>׿<EFBFBD> <20><><EFBFBD>콺 <20>̹<EFBFBD><CCB9><EFBFBD>")]
    private GameObject dragArrow;
    [SerializeField] private float clickTimeThreshold = 0.5f;
    [SerializeField] private float dragMovementThreshold = 0.05f;
    private Coroutine clickOrDragCoroutine = null;
    private Vector3 startGrabPosition;

    private bool isInitialized = false;
    private bool isGrabbingPoint = false;
    private int currentGrabbedPointIndex = -1;

    Vector3 currentTargetPosition = Vector3.zero;
    Quaternion currentTargetRotation = Quaternion.identity;

    public bool isGrabbingRobot = false;

    void Update()
    {
        if (!isInitialized || rayProvider == null)
        {
            if (nearFarInteractor == null)
            {
                nearFarInteractor = GetComponentInChildren<NearFarInteractor>(true);
                baseInteractor = nearFarInteractor as XRBaseInteractor;
                rayProvider = nearFarInteractor as IXRRayProvider;
            }
            if (nearFarInteractor != null)
            {
                InitializeInteraction();
            }
        }
        else
        {
            currentTargetPosition = rayProvider.rayEndPoint;
            currentTargetRotation = baseInteractor.attachTransform.rotation;

            if (isGrabbingRobot)
            {
                OnRobotGrabbed?.Invoke(currentTargetPosition, currentTargetRotation);
                return;
            }

            if (isGrabbingPoint)
            {
                OnPointDragUpdate?.Invoke(currentGrabbedPointIndex, currentTargetPosition, currentTargetRotation);
            }
            else if (clickOrDragCoroutine != null)
            {
                float distance = Vector3.Distance(startGrabPosition, currentTargetPosition);

                if (distance > dragMovementThreshold)
                {
                    StartDragMode();
                }
            }
        }
    }

    private void InitializeInteraction()
    {
        // XRI <20>̺<EFBFBD>Ʈ <20><><EFBFBD><EFBFBD>
        baseInteractor.selectEntered.AddListener(HandleGrabStart);
        baseInteractor.selectExited.AddListener(HandleGrabEnd);
        baseInteractor.hoverEntered.AddListener(OnHoverEntered);
        baseInteractor.hoverExited.AddListener(OnHoverExited);

        TryInitializeHapticDevice();

        isInitialized = true;
        AppManager.Instance.RegisterView(this);
    }

    private void OnDestroy()
    {
        if (baseInteractor != null)
        {
            baseInteractor.selectEntered.RemoveListener(HandleGrabStart);
            baseInteractor.selectExited.RemoveListener(HandleGrabEnd);
        }
    }

    private void TryInitializeHapticDevice()
    {
        XRNode node = (this.handSide == HandSide.Left) ? XRNode.LeftHand : XRNode.RightHand;
        InputDevice device = InputDevices.GetDeviceAtXRNode(node);

        if (device.isValid)
        {
            hapticDevice = device;
            isHapticDeviceFound = true;
        }
    }

    /// <summary>
    /// InputDevice<63><65> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><>ƽ
    /// </summary>
    public void SendHapticImpulse(float amplitude, float duration)
    {
        // <20><>Ʈ<EFBFBD>ѷ<EFBFBD><D1B7><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>ʾҰų<D2B0>(isValid) <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><> <20><><EFBFBD><EFBFBD><EFBFBD>Ƿ<EFBFBD> <20>Ź<EFBFBD> Ȯ<><C8AE>
        if (!isHapticDeviceFound || !hapticDevice.isValid)
        {
            TryInitializeHapticDevice(); // <20>ٽ<EFBFBD> ã<><C3A3> <20>õ<EFBFBD>
            if (!isHapticDeviceFound) return; // <20>׷<EFBFBD><D7B7><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>ߴ<EFBFBD>
        }

        // <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>ִ<EFBFBD><D6B4><EFBFBD> Ȯ<><C8AE>
        HapticCapabilities capabilities;
        if (hapticDevice.TryGetHapticCapabilities(out capabilities))
        {
            if (capabilities.supportsImpulse)
            {
                // <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD>
                hapticDevice.SendHapticImpulse(0, amplitude, duration);
            }
        }
    }

    // --- XRI <20>̺<EFBFBD>Ʈ <20>ڵ鷯 ---

    private void HandleGrabStart(SelectEnterEventArgs args)
    {
        GameObject grabbedGO = args.interactableObject.transform.gameObject;

        RobotPoint point = grabbedGO.GetComponent<RobotPoint>();
        if (point != null)
        {
            isGrabbingPoint = true;
            isGrabbingRobot = false;
            currentGrabbedPointIndex = point.pointIndex;

            // Ÿ<≯<EFBFBD> <20><><EFBFBD><EFBFBD>
            startGrabPosition = rayProvider.rayEndPoint;
            if (clickOrDragCoroutine != null) 
                StopCoroutine(clickOrDragCoroutine);
            clickOrDragCoroutine = StartCoroutine(ClickOrDragTimer());

            //OnPointDragStart?.Invoke(currentGrabbedPointIndex);
        }
        else if (grabbedGO.CompareTag("RobotArm"))
        {
            isGrabbingPoint = false;
            isGrabbingRobot = true;
            currentGrabbedPointIndex = -1;

            OnRobotGrabStart?.Invoke();
            OnPointDragStart?.Invoke(currentGrabbedPointIndex);
        }
    }

    private void HandleGrabEnd(SelectExitEventArgs args)
    {
        if (isGrabbingRobot)
        {
            OnRobotReleased?.Invoke(new RobotData());
        }
        else
        {
            if (clickOrDragCoroutine != null)
            {
                StopCoroutine(clickOrDragCoroutine);
                clickOrDragCoroutine = null;

                OnPointClicked?.Invoke(currentGrabbedPointIndex, currentTargetPosition);
            }
            else if (isGrabbingPoint)
            {
                OnPointDragEnd?.Invoke(currentGrabbedPointIndex);
            }
        }

        // <20><><EFBFBD><EFBFBD> <20>ʱ<EFBFBD>ȭ
        clickOrDragCoroutine = null;
        isGrabbingPoint = false;
        isGrabbingRobot = false;
        currentGrabbedPointIndex = -1;
    }

    private void OnHoverEntered(HoverEnterEventArgs args)
    {
        // ȣ<><C8A3><EFBFBD><EFBFBD> <20><>ü<EFBFBD><C3BC> '<27>κ<EFBFBD>'<27><><EFBFBD><EFBFBD> Ȯ<><C8AE>
        if (args.interactableObject.transform.CompareTag("RobotArm"))
        {
            OnRobotHoverStateChanged?.Invoke(true);
        }
    }

    // ȣ<><C8A3> <20><><EFBFBD><EFBFBD> <20><> ȣ<><C8A3>
    private void OnHoverExited(HoverExitEventArgs args)
    {
        if (args.interactableObject.transform.CompareTag("RobotArm"))
        {
            OnRobotHoverStateChanged?.Invoke(false);
        }
    }

    // Ŭ<><C5AC>/<2F>巡<EFBFBD><E5B7A1> <20><><EFBFBD><EFBFBD><EFBFBD>ϴ<EFBFBD> Ÿ<≯<EFBFBD> <20>ڷ<EFBFBD>ƾ
    private IEnumerator ClickOrDragTimer()
    {
        yield return new WaitForSeconds(clickTimeThreshold);

        if (clickOrDragCoroutine != null)
        {
            Debug.Log("<22>ð<EFBFBD> <20>ʰ<EFBFBD><CAB0><EFBFBD> <20>巡<EFBFBD><E5B7A1> <20><><EFBFBD><EFBFBD> (OnPointDragStart)");
            StartDragMode();
        }
    }

    // Ÿ<≯<EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD>ϰ<EFBFBD> <20>巡<EFBFBD><E5B7A1> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><>ȯ
    private void StartDragMode()
    {
        if (clickOrDragCoroutine != null)
        {
            StopCoroutine(clickOrDragCoroutine);
            clickOrDragCoroutine = null;
        }

        isGrabbingPoint = true;
        OnPointDragStart?.Invoke(currentGrabbedPointIndex);
    }

    // <20><><EFBFBD><EFBFBD>Ʈ <20>κ<EFBFBD><CEBA><EFBFBD> <20><>ġ/ȸ<><C8B8> <20><><EFBFBD><EFBFBD>
    public void ShowGhostRobot()
    {
        if (ghostRobot == null)
        {
            Debug.LogWarning("Ghost Robot<6F><74> Inspector<6F><72> <20>Ҵ<EFBFBD><D2B4><EFBFBD><EFBFBD><EFBFBD> <20>ʾҽ<CABE><D2BD>ϴ<EFBFBD>.");
            return;
        }

        ghostRobot.SetActive(true);
    }

    // <20><><EFBFBD><EFBFBD>Ʈ <20>κ<EFBFBD> <20><>Ȱ<EFBFBD><C8B0>ȭ
    public void HideGhostRobot()
    {
        if (ghostRobot != null)
        {
            ghostRobot.SetActive(false);
        }
    }

    // <20>巡<EFBFBD><E5B7A1> ȭ<><C8AD>ǥ Ȱ<><C8B0>ȭ <20><> <20><>ġ <20><><EFBFBD><EFBFBD>
    public void ShowDragArrow(Vector3 position)
    {
        if (dragArrow == null)
        {
            Debug.LogWarning("Drag Arrow<6F><77> Inspector<6F><72> <20>Ҵ<EFBFBD><D2B4><EFBFBD><EFBFBD><EFBFBD> <20>ʾҽ<CABE><D2BD>ϴ<EFBFBD>.");
            return;
        }

        dragArrow.SetActive(true);
        dragArrow.transform.position = position;
    }

    // <20>巡<EFBFBD><E5B7A1> ȭ<><C8AD>ǥ <20><>Ȱ<EFBFBD><C8B0>ȭ
    public void HideDragArrow()
    {
        if (dragArrow != null)
        {
            dragArrow.SetActive(false);
        }
    }

    // <20><><EFBFBD><EFBFBD> <20><> <20><>ġ <20><>ȯ
    public Vector3 GetCurrentRayPosition()
    {
        if (rayProvider != null)
        {
            return rayProvider.rayEndPoint;
        }

        if (baseInteractor != null)
        {
            return baseInteractor.attachTransform.position;
        }

        return transform.position;
    }
}