GFramework/GFramework.Core/Architectures/GameContext.cs

// Copyright (c) 2025-2026 GeWuYou
// SPDX-License-Identifier: Apache-2.0

using System.Collections.Concurrent;
using System.Threading;
using GFramework.Core.Abstractions.Architectures;

namespace GFramework.Core.Architectures;

/// <summary>
///     游戏上下文管理类，用于管理当前活动的架构上下文实例及其兼容类型别名。
/// </summary>
public static class GameContext
{
    private static readonly ConcurrentDictionary<Type, IArchitectureContext> ArchitectureDictionary
        = new();
    private static IArchitectureContext? _currentArchitectureContext;


    /// <summary>
    ///     获取所有已注册的架构上下文类型别名映射。
    ///     该只读视图会反映当前并发状态，不保证是稳定快照。
    /// </summary>
    public static IReadOnlyDictionary<Type, IArchitectureContext> ArchitectureReadOnlyDictionary =>
        ArchitectureDictionary;

    /// <summary>
    ///     绑定指定类型的架构上下文到管理器中。
    ///     同一时刻只允许存在一个活动上下文实例，但可以为其绑定多个兼容类型别名。
    /// </summary>
    /// <param name="architectureType">架构类型</param>
    /// <param name="context">架构上下文实例</param>
    /// <exception cref="ArgumentNullException">
    ///     <paramref name="architectureType" /> 或 <paramref name="context" /> 为 <see langword="null" />。
    /// </exception>
    /// <exception cref="InvalidOperationException">当指定类型的架构上下文已存在，或尝试绑定第二个不同上下文实例时抛出。</exception>
    public static void Bind(Type architectureType, IArchitectureContext context)
    {
        ArgumentNullException.ThrowIfNull(architectureType);
        ArgumentNullException.ThrowIfNull(context);

        var currentContext = Interlocked.CompareExchange(ref _currentArchitectureContext, context, comparand: null);
        if (currentContext != null && !ReferenceEquals(currentContext, context))
            throw new InvalidOperationException(
                $"GameContext already tracks active context '{currentContext.GetType().Name}'. " +
                $"Cannot bind a different context '{context.GetType().Name}'.");

        if (!ArchitectureDictionary.TryAdd(architectureType, context))
            throw new InvalidOperationException(
                $"Architecture context for '{architectureType.Name}' already exists");
    }

    /// <summary>
    ///     获取当前活动的架构上下文。
    ///     该方法保留原有名称以兼容存量调用方，但语义已经收敛为“当前上下文”，而不是任意字典首项。
    /// </summary>
    /// <returns>当前活动的架构上下文实例。</returns>
    /// <exception cref="InvalidOperationException">当当前没有活动上下文时抛出。</exception>
    public static IArchitectureContext GetFirstArchitectureContext()
    {
        if (_currentArchitectureContext is { } context)
            return context;

        throw new InvalidOperationException("No active architecture context is currently bound.");
    }

    /// <summary>
    ///     根据类型获取对应的架构上下文。
    ///     兼容层会优先查找显式绑定的类型别名，然后回退到当前上下文的类型兼容判断。
    /// </summary>
    /// <param name="type">要查找的架构类型</param>
    /// <returns>返回指定类型的架构上下文实例</returns>
    /// <exception cref="ArgumentNullException"><paramref name="type" /> 为 <see langword="null" />。</exception>
    /// <exception cref="InvalidOperationException">当指定类型的架构上下文不存在时抛出</exception>
    public static IArchitectureContext GetByType(Type type)
    {
        ArgumentNullException.ThrowIfNull(type);

        if (ArchitectureDictionary.TryGetValue(type, out var context))
            return context;

        if (_currentArchitectureContext != null && type.IsInstanceOfType(_currentArchitectureContext))
            return _currentArchitectureContext;

        throw new InvalidOperationException(
            $"Architecture context for '{type.Name}' not found");
    }


    /// <summary>
    ///     获取指定类型的架构上下文实例
    /// </summary>
    /// <typeparam name="T">架构上下文类型，必须实现IArchitectureContext接口</typeparam>
    /// <returns>指定类型的架构上下文实例</returns>
    /// <exception cref="InvalidOperationException">当指定类型的架构上下文不存在时抛出</exception>
    public static T Get<T>() where T : class, IArchitectureContext
    {
        if (_currentArchitectureContext is T currentContext)
            return currentContext;

        if (ArchitectureDictionary.TryGetValue(typeof(T), out var ctx))
            return (T)ctx;

        throw new InvalidOperationException(
            $"Architecture context '{typeof(T).Name}' not found");
    }

    /// <summary>
    ///     尝试获取指定类型的架构上下文实例
    /// </summary>
    /// <typeparam name="T">架构上下文类型，必须实现IArchitectureContext接口</typeparam>
    /// <param name="context">输出参数，如果找到则返回对应的架构上下文实例，否则返回null</param>
    /// <returns>如果找到指定类型的架构上下文则返回true，否则返回false</returns>
    public static bool TryGet<T>(out T? context)
        where T : class, IArchitectureContext
    {
        if (_currentArchitectureContext is T currentContext)
        {
            context = currentContext;
            return true;
        }

        if (ArchitectureDictionary.TryGetValue(typeof(T), out var ctx))
        {
            context = (T)ctx;
            return true;
        }

        context = null;
        return false;
    }

    /// <summary>
    ///     移除指定类型的架构上下文绑定
    /// </summary>
    /// <param name="architectureType">要移除的架构类型</param>
    public static void Unbind(Type architectureType)
    {
        ArgumentNullException.ThrowIfNull(architectureType);

        if (!ArchitectureDictionary.TryRemove(architectureType, out var removedContext))
            return;

        if (_currentArchitectureContext == null || !ReferenceEquals(_currentArchitectureContext, removedContext))
            return;

        if (!ArchitectureDictionary.Values.Any(current => ReferenceEquals(current, removedContext)))
            Interlocked.CompareExchange(ref _currentArchitectureContext, null, removedContext);
    }


    /// <summary>
    ///     清空所有架构上下文绑定
    /// </summary>
    public static void Clear()
    {
        ArchitectureDictionary.Clear();
        Interlocked.Exchange(ref _currentArchitectureContext, null);
    }
}