using System;
using System.IO;
using Microsoft.MixedReality.Toolkit.UI;
using UnityEngine;
using UnityEngine.Windows.WebCam;
#if WINDOWS_UWP && !UNITY_EDITOR
using System.Collections;
using System.Threading.Tasks;
using Windows.Storage;
#endif

namespace WebViewStream
{
    public class VideoCaptureHandler : MonoBehaviour
    {
        [SerializeField]
        private GameObject videoCaptureButton = null;

        private Interactable videoCaptureButtonInteractable = null;
        private VideoCapture videoCapture = null;

#if WINDOWS_UWP && !UNITY_EDITOR
        private const string freeSpace = "System.FreeSpace";
        private const UInt64 minAvailableSpace = 5UL * 1024 * 1024 * 1024; // 5GB

        private IEnumerator CheckAvailableStorageSpace()
        {
            while (videoCapture != null && videoCapture.IsRecording)
            {
                yield return CheckSpaceAndHandleRecording();
                yield return new WaitForSeconds(5);
            }
        }

        private async Task CheckSpaceAndHandleRecording()
        {
            try
            {
                StorageFolder folder = ApplicationData.Current.TemporaryFolder;
                var props = await folder.Properties.RetrievePropertiesAsync(new string[] { freeSpace });
                UInt64 availableSpace = (UInt64)props[freeSpace];

                if (availableSpace < minAvailableSpace)
                {
                    Debug.LogWarning("Not enough storage space to continue recording. Saving video.");
                    StopRecordingVideo();
                }
            }
            catch (Exception ex)
            {
                Debug.LogError("Error checking storage space: " + ex.Message);
            }
        }
#endif

        /// <summary>
        /// Starts recording a video.
        /// </summary>
        public void StartRecordingVideo()
        {
#if WINDOWS_UWP && !UNITY_EDITOR
            VideoCapture.CreateAsync(true, OnVideoCaptureCreated);
#else
            VideoCapture.CreateAsync(false, OnVideoCaptureCreated);
#endif
            if (videoCaptureButtonInteractable == null)
            {
                videoCaptureButtonInteractable = videoCaptureButton.GetComponent<Interactable>();
            }
            videoCaptureButtonInteractable.IsToggled = true;
        }

        /// <summary>
        /// Stops recording a video.
        /// </summary>
        public void StopRecordingVideo()
        {
            videoCapture.StopRecordingAsync(OnStoppedRecordingVideo);
            videoCaptureButtonInteractable.IsToggled = false;
        }

        /// <summary>
        /// Toggles the recording of a video.
        /// </summary>
        public void ToggleRecordingVideo()
        {
            if (videoCapture == null)
            {
                StartRecordingVideo();
            }
            else if (videoCapture.IsRecording)
            {
                StopRecordingVideo();
            }
        }

        private void OnVideoCaptureCreated(VideoCapture videoCapture)
        {
            if (videoCapture != null)
            {
                this.videoCapture = videoCapture;

                Resolution cameraResolution = new Resolution();
                foreach (Resolution resolution in VideoCapture.SupportedResolutions)
                {
                    if (
                        resolution.width * resolution.height
                        > cameraResolution.width * cameraResolution.height
                    )
                    {
                        cameraResolution = resolution;
                    }
                }

                float cameraFramerate = 0.0f;
                foreach (
                    float framerate in VideoCapture.GetSupportedFrameRatesForResolution(cameraResolution)
                )
                {
                    if (framerate > cameraFramerate)
                    {
                        cameraFramerate = framerate;
                    }
                }

                CameraParameters cameraParameters = new CameraParameters();
                cameraParameters.hologramOpacity = 0.75f;
                cameraParameters.frameRate = cameraFramerate;
                cameraParameters.cameraResolutionWidth = cameraResolution.width;
                cameraParameters.cameraResolutionHeight = cameraResolution.height;
                cameraParameters.pixelFormat = CapturePixelFormat.BGRA32;

                this.videoCapture.StartVideoModeAsync(
                    cameraParameters,
                    VideoCapture.AudioState.ApplicationAndMicAudio,
                    OnStartedVideoCaptureMode
                );
            }
            else
            {
                Debug.LogError("Failed to create VideoCapture instance");
            }
        }

#if WINDOWS_UWP && !UNITY_EDITOR
        private async void OnStartedVideoCaptureMode(VideoCapture.VideoCaptureResult result)
#else
        private void OnStartedVideoCaptureMode(VideoCapture.VideoCaptureResult result)
#endif
        {
            if (result.success)
            {
                string filename = string.Format(
                    "WebView_{0}.mp4",
                    DateTime.UtcNow.ToString("yyyy-MM-ddTHHmmssZ")
                );
#if WINDOWS_UWP && !UNITY_EDITOR
                StorageLibrary videosLibrary = await StorageLibrary.GetLibraryAsync(KnownLibraryId.Videos);
                string filepath = Path.Combine(videosLibrary.SaveFolder.Path, filename);
#else
                string filepath = Path.Combine(Application.persistentDataPath, filename);
#endif
                Debug.Log("Saving video to: " + filepath);

                videoCapture.StartRecordingAsync(filepath, OnStartedRecordingVideo);
            }
        }

        private void OnStartedRecordingVideo(VideoCapture.VideoCaptureResult result)
        {
            Debug.Log("Started recording video");
#if WINDOWS_UWP && !UNITY_EDITOR
            StartCoroutine(CheckAvailableStorageSpace());
#endif
        }

        private void OnStoppedRecordingVideo(VideoCapture.VideoCaptureResult result)
        {
            Debug.Log("Stopped recording video");
            videoCapture.StopVideoModeAsync(OnStoppedVideoCaptureMode);
        }

        private void OnStoppedVideoCaptureMode(VideoCapture.VideoCaptureResult result)
        {
            videoCapture.Dispose();
            videoCapture = null;
        }
    }
}