XiuXianGame/Assets/TapSDK/Core/Runtime/Internal/Utils/ImageUtils.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using System.Security.Cryptography;
using System.Text;
using System.IO;
using UnityEngine;
using UnityEngine.Networking;
using TapSDK.Core.Internal.Log;

namespace TapSDK.Core.Internal.Utils {
    public class ImageUtils {
        private readonly static string OldCacheDirName = "tap-cache";
        private readonly static MD5 md5 = MD5.Create();

        private readonly static Dictionary<string, WeakReference<Texture>> cachedTextures = new Dictionary<string, WeakReference<Texture>>();

        public static async Task<Texture> LoadImage(string url, int timeout = 30, bool useMemoryCache = true) {
            if (string.IsNullOrEmpty(url)) {
                TapLog.Warning(string.Format($"ImageUtils Fetch image is null! url is null or empty!"));
                return null;
            }

            if (cachedTextures.TryGetValue(url, out WeakReference<Texture> refTex) &&
                refTex.TryGetTarget(out Texture tex)) {
                // 从内存加载
                return tex;
            } else {
                try {
                    // 从本地缓存加载
                    Texture cachedImage = await LoadCachedImaged(url, timeout);

                    if (useMemoryCache) {
                        cachedTextures[url] = new WeakReference<Texture>(cachedImage);
                    }

                    return cachedImage;
                } catch (Exception e) {
                    TapLog.Warning(e.Message);
                    try {
                        // 从网络加载
                        Texture2D newTex = await FetchImage(url, timeout);

                        if (useMemoryCache) {
                            cachedTextures[url] = new WeakReference<Texture>(newTex);
                        }

                        // 缓存到本地
                        _ = CacheImage(url, newTex);

                        return newTex;
                    } catch (Exception ex) {
                        TapLog.Warning(ex.Message);
                        return null;
                    }
                }
            }
        }

        public static async Task<Texture2D> FetchImage(string url, int timeout = 30) {
            using (UnityWebRequest request = UnityWebRequestTexture.GetTexture(url)) {
                request.timeout = timeout;
                UnityWebRequestAsyncOperation operation = request.SendWebRequest();
                while (!operation.isDone) {
                    await Task.Delay(30);
                }

                if (request.isNetworkError || request.isHttpError) {
                    throw new Exception("Fetch image error.");
                } else {
                    Texture2D texture = ((DownloadHandlerTexture)request.downloadHandler)?.texture;
                    if (texture == null) {
                        TapLog.Warning($"ImageUtils Fetch image is null! url: {url}");
                    }
                    return texture;
                }
            }
        }

        static async Task<Texture> LoadCachedImaged(string url, int timeout = 30) {
            string cachedImagePath = GetCachedPath(url);
            if (!File.Exists(cachedImagePath)) {
                throw new Exception("No cached image.");
            }
            string cachedImageUrl = $"file://{cachedImagePath}";
            using (UnityWebRequest request = UnityWebRequestTexture.GetTexture(cachedImageUrl)) {
                request.timeout = timeout;
                UnityWebRequestAsyncOperation operation = request.SendWebRequest();
                while (!operation.isDone) {
                    await Task.Delay(30);
                }

                if (request.isNetworkError || request.isHttpError) {
                    RemoveCachedImage(cachedImagePath);
                    throw new Exception("Load cache image error.");
                } else {
                    var texture = ((DownloadHandlerTexture)request.downloadHandler)?.texture;
                    if (texture == null) {
                        RemoveCachedImage(cachedImagePath);
                        throw new Exception("Cached image is invalid.");
                    }
                    return texture;
                }
            }
        }

        static async Task CacheImage(string url, Texture2D tex) {
            string cacheImagePath = GetCachedPath(url);
            // 写入缓存
            byte[] imageData = tex.EncodeToPNG();
            using (FileStream fileStream = new FileStream(cacheImagePath, FileMode.Create, FileAccess.Write, FileShare.None, bufferSize: 4096, useAsync: true)) {
                await fileStream.WriteAsync(imageData, 0, imageData.Length);
            }
        }

        static void RemoveCachedImage(string cachedImagePath) {
            try {
                File.Delete(cachedImagePath);
            } finally {

            }
        }

        static string ToHex(byte[] bytes) {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < bytes.Length; i++) {
                sb.Append(bytes[i].ToString("x2"));
            }
            return sb.ToString();
        }

        static string GetCachedPath(string url) {
            string cachedHashName = ToHex(md5.ComputeHash(Encoding.UTF8.GetBytes(url)));
            return Path.Combine(CacheDirPath, cachedHashName);
        }

        static string CacheDirPath {
            get {
                
                string newCacheDirPath = Path.Combine(Application.persistentDataPath, OldCacheDirName);
                if (TapTapSDK.taptapSdkOptions != null && !string.IsNullOrEmpty(TapTapSDK.taptapSdkOptions.clientId) ){
                    newCacheDirPath = Path.Combine(Application.persistentDataPath, OldCacheDirName + "_" + TapTapSDK.taptapSdkOptions.clientId);
                }
                if(!Directory.Exists(newCacheDirPath)) {
                    string oldPath = Path.Combine(Application.persistentDataPath, OldCacheDirName);
                    if (Directory.Exists(oldPath)) {
                        Directory.Move(oldPath, newCacheDirPath);
                    }
                }
               
                if (!Directory.Exists(newCacheDirPath)) {
                    Directory.CreateDirectory(newCacheDirPath);
                }
                return newCacheDirPath;
            }
        }
    }
}