esengine/packages/behavior-tree/src/Runtime/BehaviorTreeAssetManager.ts

import { BehaviorTreeData } from './BehaviorTreeData';
import { createLogger, IService } from '@esengine/ecs-framework';

const logger = createLogger('BehaviorTreeAssetManager');

/**
 * 行为树资产管理器（服务）
 *
 * 管理所有共享的BehaviorTreeData
 * 多个实例可以引用同一份数据
 *
 * 使用方式：
 * ```typescript
 * // 注册服务
 * Core.services.registerSingleton(BehaviorTreeAssetManager);
 *
 * // 使用服务
 * const assetManager = Core.services.resolve(BehaviorTreeAssetManager);
 * ```
 */
export class BehaviorTreeAssetManager implements IService {
    /**
     * 已加载的行为树资产
     */
    private assets: Map<string, BehaviorTreeData> = new Map();

    /**
     * 加载行为树资产
     */
    loadAsset(asset: BehaviorTreeData): void {
        if (this.assets.has(asset.id)) {
            logger.warn(`行为树资产已存在，将被覆盖: ${asset.id}`);
        }
        this.assets.set(asset.id, asset);
        logger.info(`行为树资产已加载: ${asset.name} (${asset.nodes.size}个节点)`);
    }

    /**
     * 获取行为树资产
     */
    getAsset(assetId: string): BehaviorTreeData | undefined {
        return this.assets.get(assetId);
    }

    /**
     * 检查资产是否存在
     */
    hasAsset(assetId: string): boolean {
        return this.assets.has(assetId);
    }

    /**
     * 卸载行为树资产
     */
    unloadAsset(assetId: string): boolean {
        const result = this.assets.delete(assetId);
        if (result) {
            logger.info(`行为树资产已卸载: ${assetId}`);
        }
        return result;
    }

    /**
     * 清空所有资产
     */
    clearAll(): void {
        this.assets.clear();
        logger.info('所有行为树资产已清空');
    }

    /**
     * 获取已加载资产数量
     */
    getAssetCount(): number {
        return this.assets.size;
    }

    /**
     * 获取所有资产ID
     */
    getAllAssetIds(): string[] {
        return Array.from(this.assets.keys());
    }

    /**
     * 释放资源（实现IService接口）
     */
    dispose(): void {
        this.clearAll();
    }
}