nitro-renderer/src/core/asset/AssetManager.ts

import { BaseTexture, Resource, Texture } from '@pixi/core';
import { Loader, LoaderResource } from '@pixi/loaders';
import { Spritesheet } from '@pixi/spritesheet';
import { IAssetData, IAssetManager, IGraphicAsset, IGraphicAssetCollection, NitroLogger } from '../../api';
import { Disposable } from '../common';
import { ArrayBufferToBase64 } from '../utils';
import { GraphicAssetCollection } from './GraphicAssetCollection';
import { NitroBundle } from './NitroBundle';

export class AssetManager extends Disposable implements IAssetManager
{
    private _textures: Map<string, Texture<Resource>>;
    private _collections: Map<string, IGraphicAssetCollection>;

    constructor()
    {
        super();

        this._textures = new Map();
        this._collections = new Map();
    }

    public getTexture(name: string): Texture<Resource>
    {
        if (!name) return null;

        const existing = this._textures.get(name);

        if (!existing) return null;

        return existing;
    }

    public setTexture(name: string, texture: Texture<Resource>): void
    {
        if (!name || !texture) return;

        this._textures.set(name, texture);
    }

    public getAsset(name: string): IGraphicAsset
    {
        if (!name) return null;

        for (const collection of this._collections.values())
        {
            if (!collection) continue;

            const existing = collection.getAsset(name);

            if (!existing) continue;

            return existing;
        }

        return null;
    }

    public getCollection(name: string): IGraphicAssetCollection
    {
        if (!name) return null;

        const existing = this._collections.get(name);

        if (!existing) return null;

        return existing;
    }

    public createCollection(data: IAssetData, spritesheet: Spritesheet): IGraphicAssetCollection
    {
        if (!data) return null;

        const collection = new GraphicAssetCollection(data, spritesheet);

        if (collection)
        {
            for (const [name, texture] of collection.textures.entries()) this.setTexture(name, texture);

            this._collections.set(collection.name, collection);
        }

        return collection;
    }

    public downloadAsset(assetUrl: string, cb: (status: boolean) => void): void
    {
        this.downloadAssets([assetUrl], cb);
    }

    public downloadAssets(assetUrls: string[], cb: (status: boolean) => void): void
    {
        if (!assetUrls || !assetUrls.length)
        {
            cb(true);

            return;
        }

        const loader = new Loader();

        for (const url of assetUrls)
        {
            if (!url) continue;

            loader
                .add({
                    url,
                    crossOrigin: 'anonymous',
                    loadType: LoaderResource.LOAD_TYPE.XHR,
                    xhrType: LoaderResource.XHR_RESPONSE_TYPE.BUFFER
                });
        }

        let remaining = assetUrls.length;

        const onDownloaded = (status: boolean, url: string) =>
        {
            if (!status)
            {
                NitroLogger.error('Failed to download asset', url);

                loader.destroy();

                cb(false);

                return;
            }

            remaining--;

            if (!remaining)
            {
                loader.destroy();

                cb(true);

                return;
            }
        };

        loader.load((loader, resources) =>
        {
            for (const key in resources)
            {
                const resource = resources[key];

                if (!resource || resource.error || !resource.xhr)
                {
                    onDownloaded(false, resource.url);

                    return;
                }

                const resourceType = (resource.xhr.getResponseHeader('Content-Type') || 'application/octet-stream');

                if (resourceType === 'application/octet-stream')
                {
                    const nitroBundle = new NitroBundle(resource.data);

                    this.processAsset(nitroBundle.baseTexture, (nitroBundle.jsonFile as IAssetData), status =>
                    {
                        onDownloaded(status, resource.url);
                    });

                    continue;
                }

                if ((resourceType === 'image/png') || (resourceType === 'image/jpeg') || (resourceType === 'image/gif'))
                {
                    const base64 = ArrayBufferToBase64(resource.data);
                    const baseTexture = new BaseTexture(`data:${resourceType};base64,${base64}`);

                    if (baseTexture.valid)
                    {
                        const texture = new Texture(baseTexture);

                        this.setTexture(resource.name, texture);

                        onDownloaded(true, resource.url);
                    }
                    else
                    {
                        baseTexture.once('update', () =>
                        {
                            const texture = new Texture(baseTexture);

                            this.setTexture(resource.name, texture);

                            onDownloaded(true, resource.url);
                        });
                    }

                    continue;
                }

                onDownloaded(false, resource.url);
            }
        });
    }

    private processAsset(baseTexture: BaseTexture, data: IAssetData, onDownloaded: (status: boolean) => void): void
    {
        const spritesheetData = data.spritesheet;

        if (!baseTexture || !spritesheetData || !Object.keys(spritesheetData).length)
        {
            this.createCollection(data, null);

            onDownloaded(true);

            return;
        }

        const createAsset = () =>
        {
            const spritesheet = new Spritesheet(baseTexture, spritesheetData);

            spritesheet.parse(() =>
            {
                this.createCollection(data, spritesheet);

                onDownloaded(true);
            });
        };

        if (baseTexture.valid)
        {
            createAsset();
        }
        else
        {
            baseTexture.once('update', () => createAsset());
        }
    }

    public get collections(): Map<string, IGraphicAssetCollection>
    {
        return this._collections;
    }
}