import { assetManager, AssetManager } from "cc";

/**
 * 分包管理器
 * 提供分包加载、获取、移除功能。
 */
class BundleMgr {
    /** 私有构造函数*/
    private constructor() { }

    /** 单例实例 */
    public static readonly instance: BundleMgr = new BundleMgr();

    /**
     * 获取指定分包，如果未加载则进行加载。
     * @param nameOrUrl - 分包名称或URL。
     * @param onProgress - 进度回调函数。
     * @returns Promise<AssetManager.Bundle | null> - 加载完成后的Promise。
     */
    public async getBundle(nameOrUrl: string, onProgress?: (progress: number) => void): Promise<AssetManager.Bundle | null> {
        const bundle = assetManager.getBundle(nameOrUrl);
        if (bundle) return bundle;

        try {
            const loadedBundle = await this.loadBundle(nameOrUrl);
            if (onProgress) {
                await this.loadAssetsWithProgress(loadedBundle, onProgress);
            }
            return loadedBundle;
        } catch (error) {
            console.error(`分包 ${nameOrUrl} 加载失败`, error.message);
            return null;
        }
    }

    /**
     * 加载指定分包。
     * @param nameOrUrl - 分包名称或URL。
     * @returns Promise<AssetManager.Bundle> - 加载完成后的Promise。
     */
    private loadBundle(nameOrUrl: string): Promise<AssetManager.Bundle> {
        return new Promise((resolve, reject) => {
            assetManager.loadBundle(nameOrUrl, (err, loadedBundle) => {
                if (err) {
                    reject(err);
                } else {
                    resolve(loadedBundle);
                }
            });
        });
    }

    /**
     * 加载分包中的资源并提供进度反馈。
     * @param bundle - 已加载的分包。
     * @param onProgress - 进度回调函数。
     * @returns Promise<void> - 加载完成后的Promise。
     */
    private loadAssetsWithProgress(bundle: AssetManager.Bundle, onProgress: (progress: number) => void): Promise<void> {
        return new Promise((resolve, reject) => {
            const assets = bundle.getDirWithPath('');
            const totalAssets = assets.length;
            let loadedAssets = 0;

            if (totalAssets === 0) {
                onProgress(1);
                resolve();
                return;
            }

            assets.forEach((asset) => {
                bundle.load(asset.path, (err) => {
                    if (err) {
                        reject(err);
                        return;
                    }
                    loadedAssets++;
                    onProgress(loadedAssets / totalAssets);
                    if (loadedAssets === totalAssets) {
                        resolve();
                    }
                });
            });
        });
    }
}

/** 分包管理器实例 */
export const bundleMgr = BundleMgr.instance;