using System;
using System.Linq;
using Studio;
#if AI
using AIChara;
#endif

namespace KKAPI.Studio
{
    /// <summary>
    /// Useful extensions for studio metaobjects
    /// </summary>
    public static class StudioObjectExtensions
    {
        private static global::Studio.Studio Studio => global::Studio.Studio.Instance;

        /// <summary>
        /// Get the ObjectCtrlInfo object that uses this ObjectInfo.
        /// If the object was not found in current scene, null is returned.
        /// </summary>
        public static ObjectCtrlInfo GetObjectCtrlInfo(this ObjectInfo obj)
        {
            if (obj == null) throw new ArgumentNullException(nameof(obj));
            if (Studio == null) throw new InvalidOperationException("Studio is not initialized yet!");

            return Studio.dicObjectCtrl.Values.FirstOrDefault(x => x.objectInfo == obj);
        }

        /// <summary>
        /// Get the ID of this object as used in the currently loaded scene.
        /// If the object was not found in current scene, -1 is returned.
        /// </summary>
        public static int GetSceneId(this ObjectCtrlInfo obj)
        {
            if (obj == null) throw new ArgumentNullException(nameof(obj));
            if (Studio == null) throw new InvalidOperationException("Studio is not initialized yet!");

            foreach (var info in Studio.dicObjectCtrl)
            {
                if (info.Value == obj)
                    return info.Key;
            }
            return -1;
        }

        /// <summary>
        /// Get the ID of this object as used in the currently loaded scene.
        /// If the object was not found in current scene, -1 is returned.
        /// </summary>
        public static int GetSceneId(this ObjectInfo obj)
        {
            if (obj == null) throw new ArgumentNullException(nameof(obj));
            if (Studio == null) throw new InvalidOperationException("Studio is not initialized yet!");

            foreach (var info in Studio.dicObjectCtrl)
            {
                if (info.Value.objectInfo == obj)
                    return info.Key;
            }
            return -1;
        }

        /// <summary>
        /// Get GetOCIChar that is assigned to this character. Only works in CharaStudio, returns null elsewhere.
        /// </summary>
        public static OCIChar GetOCIChar(this ChaControl chaControl)
        {
            if (!StudioAPI.InsideStudio) return null;
            var infos = global::Studio.Studio.Instance.dicInfo;
            var charas = infos.Values.OfType<OCIChar>();
            return charas.FirstOrDefault(x => x.charInfo == chaControl);
        }
    }
}