ET9_WebGL_Dev/Packages/cn.etetet.yooassets/Editor/AssetBundleBuilder/BuildSystem/BuildContext.cs

using System;
using System.Collections;
using System.Collections.Generic;

namespace YooAsset.Editor
{
    public class BuildContext
    {
        private readonly Dictionary<System.Type, IContextObject> _contextObjects = new Dictionary<System.Type, IContextObject>();

        /// <summary>
        /// 清空所有上下文对象
        /// </summary>
        public void ClearAllContext()
        {
            _contextObjects.Clear();
        }

        /// <summary>
        /// 设置上下文对象
        /// </summary>
        public void SetContextObject(IContextObject contextObject)
        {
            if (contextObject == null)
                throw new ArgumentNullException("contextObject");

            var type = contextObject.GetType();
            if (_contextObjects.ContainsKey(type))
                throw new Exception($"Context object {type} is already existed.");

            _contextObjects.Add(type, contextObject);
        }

        /// <summary>
        /// 获取上下文对象
        /// </summary>
        public T GetContextObject<T>() where T : IContextObject
        {
            var type = typeof(T);
            if (_contextObjects.TryGetValue(type, out IContextObject contextObject))
            {
                return (T)contextObject;
            }
            else
            {
                throw new Exception($"Not found context object : {type}");
            }
        }

        /// <summary>
        /// 获取上下文对象
        /// </summary>
        public T TryGetContextObject<T>() where T : IContextObject
        {
            var type = typeof(T);
            if (_contextObjects.TryGetValue(type, out IContextObject contextObject))
            {
                return (T)contextObject;
            }
            else
            {
                return default;
            }
        }
    }
}