/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under the license found in the
 * LICENSE file in the root directory of this source tree.
 */

using System.Text;
using UnityEngine;
using UnityEngine.UI;
using Meta.WitAi.TTS.Data;
using Meta.WitAi.TTS.Utilities;

namespace Meta.Voice.Samples.TTSVoices
{
    /// <summary>
    /// A simple script for providing clip information to a Text label
    /// </summary>
    public class TTSSpeakerStatusLabel : TTSSpeakerObserver
    {
        [SerializeField] private Text _label;

        protected override void OnEnable()
        {
            base.OnEnable();
            RefreshLabel();
        }
        protected override void OnLoadBegin(TTSSpeaker speaker, TTSClipData clipData)
        {
            RefreshLabel();
        }
        protected override void OnLoadAbort(TTSSpeaker speaker, TTSClipData clipData)
        {
            RefreshLabel();
        }
        protected override void OnLoadFailed(TTSSpeaker speaker, TTSClipData clipData, string error)
        {
            RefreshLabel();
        }
        protected override void OnLoadSuccess(TTSSpeaker speaker, TTSClipData clipData)
        {
            RefreshLabel();
        }
        protected override void OnPlaybackReady(TTSSpeaker speaker, TTSClipData clipData)
        {
            RefreshLabel();
        }
        protected override void OnPlaybackStart(TTSSpeaker speaker, TTSClipData clipData)
        {
            RefreshLabel();
        }
        protected override void OnPlaybackCancelled(TTSSpeaker speaker, TTSClipData clipData, string reason)
        {
            RefreshLabel();
        }
        protected override void OnPlaybackComplete(TTSSpeaker speaker, TTSClipData clipData)
        {
            RefreshLabel();
        }
        private void RefreshLabel()
        {
            StringBuilder status = new StringBuilder();
            int count = 1;
            if (Speaker.IsSpeaking)
            {
                status.Append($"Speaking[{count}]: '{Speaker.SpeakingClip.textToSpeak}'");
                status.AppendLine("\n");
                count++;
            }
            foreach (var clip in Speaker.QueuedClips)
            {
                status.Append($"{clip.loadState.ToString()}[{count}]: '{clip.textToSpeak}'");
                status.AppendLine("\n");
                count++;
            }
            _label.text = status.ToString();
            _label.rectTransform.SetSizeWithCurrentAnchors(RectTransform.Axis.Vertical, _label.preferredHeight);
        }
    }
}