YooAsset/Assets/YooAsset/Runtime/CacheSystem/Operations/Internal/FindCacheFilesOperation.cs

using System;
using System.IO;
using System.Collections;
using System.Collections.Generic;

namespace YooAsset
{
	internal class FindCacheFilesOperation : AsyncOperationBase
	{
		private enum ESteps
		{
			None,
			FindPrepare,
			FindBundleFiles,
			FindRawFiles,
			Done,
		}

		private readonly string _packageName;
		private float _verifyStartTime;
		private IEnumerator<DirectoryInfo> _bundleFilesEnumerator = null;
		private IEnumerator<DirectoryInfo> _rawFilesEnumerator = null;
		private ESteps _steps = ESteps.None;

		/// <summary>
		/// 需要验证的元素
		/// </summary>
		public readonly List<VerifyCacheElement> VerifyElements = new List<VerifyCacheElement>(5000);

		public FindCacheFilesOperation(string packageName)
		{
			_packageName = packageName;
		}
		internal override void Start()
		{
			_steps = ESteps.FindPrepare;
			_verifyStartTime = UnityEngine.Time.realtimeSinceStartup;
		}
		internal override void Update()
		{
			if (_steps == ESteps.None || _steps == ESteps.Done)
				return;

			if (_steps == ESteps.FindPrepare)
			{
				// BundleFiles
				{
					string rootPath = PersistentTools.GetCachedBundleFileFolderPath(_packageName);
					DirectoryInfo rootDirectory = new DirectoryInfo(rootPath);
					if (rootDirectory.Exists)
					{
						var directorieInfos = rootDirectory.EnumerateDirectories();
						_bundleFilesEnumerator = directorieInfos.GetEnumerator();
					}
				}

				// RawFiles
				{
					string rootPath = PersistentTools.GetCachedRawFileFolderPath(_packageName);
					DirectoryInfo rootDirectory = new DirectoryInfo(rootPath);
					if (rootDirectory.Exists)
					{
						var directorieInfos = rootDirectory.EnumerateDirectories();
						_rawFilesEnumerator = directorieInfos.GetEnumerator();
					}
				}

				_steps = ESteps.FindBundleFiles;
			}

			if (_steps == ESteps.FindBundleFiles)
			{
				if (UpdateFindBundleFiles())
					return;

				_steps = ESteps.FindRawFiles;
			}

			if (_steps == ESteps.FindRawFiles)
			{
				if (UpdateFindRawFiles())
					return;

				// 注意：总是返回成功
				_steps = ESteps.Done;
				Status = EOperationStatus.Succeed;
				float costTime = UnityEngine.Time.realtimeSinceStartup - _verifyStartTime;
				YooLogger.Log($"Find cache files elapsed time {costTime:f1} seconds");
			}
		}

		private bool UpdateFindBundleFiles()
		{
			if (_bundleFilesEnumerator == null)
				return false;

			bool isFindItem;
			while (true)
			{
				isFindItem = _bundleFilesEnumerator.MoveNext();
				if (isFindItem == false)
					break;

				var rootFoder = _bundleFilesEnumerator.Current;
				var childDirectories = rootFoder.GetDirectories();
				foreach(var chidDirectory in childDirectories)
				{
					string cacheGUID = chidDirectory.Name;
					if (CacheSystem.IsCached(_packageName, cacheGUID))
						continue;

					// 创建验证元素类
					string fileRootPath = chidDirectory.FullName;
					string dataFilePath = $"{fileRootPath}/{ YooAssetSettings.CacheBundleDataFileName}";
					string infoFilePath = $"{fileRootPath}/{ YooAssetSettings.CacheBundleInfoFileName}";
					VerifyCacheElement element = new VerifyCacheElement(_packageName, cacheGUID, fileRootPath, dataFilePath, infoFilePath);
					VerifyElements.Add(element);
				}

				if (OperationSystem.IsBusy)
					break;
			}

			return isFindItem;
		}
		private bool UpdateFindRawFiles()
		{
			if (_rawFilesEnumerator == null)
				return false;

			bool isFindItem;
			while (true)
			{
				isFindItem = _rawFilesEnumerator.MoveNext();
				if (isFindItem == false)
					break;

				var rootFoder = _rawFilesEnumerator.Current;
				var childDirectories = rootFoder.GetDirectories();
				foreach (var chidDirectory in childDirectories)
				{
					string cacheGUID = chidDirectory.Name;
					if (CacheSystem.IsCached(_packageName, cacheGUID))
						continue;

					// 获取数据文件的后缀名
					string dataFileExtension = string.Empty;
					var fileInfos = chidDirectory.GetFiles();
					foreach (var fileInfo in fileInfos)
					{
						if (fileInfo.Extension == ".temp")
							continue;
						if (fileInfo.Name.StartsWith(YooAssetSettings.CacheBundleDataFileName))
						{
							dataFileExtension = fileInfo.Extension;
							break;
						}
					}

					// 创建验证元素类
					string fileRootPath = chidDirectory.FullName;
					string dataFilePath = $"{fileRootPath}/{ YooAssetSettings.CacheBundleDataFileName}{dataFileExtension}";
					string infoFilePath = $"{fileRootPath}/{ YooAssetSettings.CacheBundleInfoFileName}";
					VerifyCacheElement element = new VerifyCacheElement(_packageName, cacheGUID, fileRootPath, dataFilePath, infoFilePath);
					VerifyElements.Add(element);
				}

				if (OperationSystem.IsBusy)
					break;
			}

			return isFindItem;
		}
	}
}