IAvatarElementStorage.cs

using CommunicationData.URLHelpers;
using DCL.AvatarRendering.Loading.Components;
using DCL.AvatarRendering.Loading.DTO;
using DCL.AvatarRendering.Wearables.Components;
using DCL.Diagnostics;
using DCL.Optimization.PerformanceBudgeting;
using System.Collections.Generic;

namespace DCL.AvatarRendering.Loading
{
    /// <summary>
    ///     Avatar elements cache, each implementation should be thread safe.
    /// </summary>
    public interface IAvatarElementStorage<TElement, in TDTO> where TElement: IAvatarAttachment<TDTO> where TDTO: AvatarAttachmentDTO
    {
        /// <summary>
        ///     Attempts to retrieve an element from the catalog.
        /// </summary>
        /// <param name="urn">The URN identifier.</param>
        /// <param name="element">The element instance if found.</param>
        /// <returns>Returns true if the element exists; otherwise, false.</returns>
        bool TryGetElement(URN urn, out TElement element);

        void Set(URN urn, TElement element);

        /// <summary>
        ///     Retrieves an element by its DTO or adds a new one if it doesn't exist.
        /// </summary>
        /// <param name="dto">The wearable DTO</param>
        /// <param name="qualifiedForUnloading">Determines if the wearable should be unloaded when memory is full</param>
        /// <returns>An instance of the <see cref="TElement" /> type.</returns>
        TElement GetOrAddByDTO(TDTO dto, bool qualifiedForUnloading = true);

        /// <summary>
        ///     Unloads the wearable from the catalog by a frame time budget provider.
        /// </summary>
        /// <param name="frameTimeBudget">The frame time budget provider.</param>
        void Unload(IPerformanceBudget frameTimeBudget);

        void SetOwnedNft(URN urn, NftBlockchainOperationEntry operation);

        bool TryGetOwnedNftRegistry(URN nftUrn, out IReadOnlyDictionary<URN, NftBlockchainOperationEntry> registry);

        int GetOwnedNftCount(URN nftUrn);

        void ClearOwnedNftRegistry();

        /// <summary>
        /// Clears the owned NFT entries for a specific base URN.
        /// Call this before repopulating with fresh data from the API.
        /// </summary>
        void ClearOwnedNftForUrn(URN nftUrn);
    }

    public static class AvatarElementCache
    {
        public static bool TryGetElementWithLogs<TElement, TDTO>(this IAvatarElementStorage<TElement, TDTO> storage, TDTO assetDTO, string reportCategory, out TElement? element)
            where TElement: IAvatarAttachment<TDTO>
            where TDTO: AvatarAttachmentDTO
        {
            string? id = assetDTO.Metadata?.id;

            bool inCatalog = storage.TryGetElement(id ?? string.Empty, out element);

            //An element that has a DTO request should already have an empty representation in the catalog at this point
            if (inCatalog == false)
                ReportHub.LogError(reportCategory, $"Requested {typeof(TDTO).Name} '{id}' is not in the catalog");

            return inCatalog;
        }
    }
}