AssetRipper_AssetRipper/Source/AssetRipper.Export.UnityProjects/ExportCollection.cs

using AssetRipper.Assets;
using AssetRipper.Assets.Collections;
using AssetRipper.IO.Files.SerializedFiles;
using AssetRipper.SourceGenerated.Classes.ClassID_1102;
using AssetRipper.SourceGenerated.Classes.ClassID_1107;
using AssetRipper.SourceGenerated.Classes.ClassID_1109;
using AssetRipper.SourceGenerated.Classes.ClassID_1113;
using AssetRipper.SourceGenerated.Classes.ClassID_121;
using AssetRipper.SourceGenerated.Classes.ClassID_134;
using AssetRipper.SourceGenerated.Classes.ClassID_158;
using AssetRipper.SourceGenerated.Classes.ClassID_1953259897;
using AssetRipper.SourceGenerated.Classes.ClassID_200;
using AssetRipper.SourceGenerated.Classes.ClassID_206;
using AssetRipper.SourceGenerated.Classes.ClassID_21;
using AssetRipper.SourceGenerated.Classes.ClassID_221;
using AssetRipper.SourceGenerated.Classes.ClassID_240;
using AssetRipper.SourceGenerated.Classes.ClassID_28;
using AssetRipper.SourceGenerated.Classes.ClassID_319;
using AssetRipper.SourceGenerated.Classes.ClassID_48;
using AssetRipper.SourceGenerated.Classes.ClassID_62;
using AssetRipper.SourceGenerated.Classes.ClassID_74;
using AssetRipper.SourceGenerated.Classes.ClassID_84;
using AssetRipper.SourceGenerated.Classes.ClassID_850595691;
using AssetRipper.SourceGenerated.Classes.ClassID_89;
using AssetRipper.SourceGenerated.Classes.ClassID_91;
using AssetRipper.Yaml;
using System.Text;

namespace AssetRipper.Export.UnityProjects;

public abstract class ExportCollection : IExportCollection
{
	public virtual UnityGuid GUID => throw new NotSupportedException();

	protected static void ExportMeta(IExportContainer container, Meta meta, string filePath, FileSystem fileSystem)
	{
		string metaPath = $"{filePath}{MetaExtension}";
		using Stream fileStream = fileSystem.File.Create(metaPath);
		using InvariantStreamWriter streamWriter = new InvariantStreamWriter(fileStream, new UTF8Encoding(false));

		YamlWriter writer = new();
		writer.IsWriteDefaultTag = false;
		writer.IsWriteVersion = false;
		writer.IsFormatKeys = true;
		YamlDocument doc = meta.ExportYamlDocument(container);
		writer.AddDocument(doc);
		writer.Write(streamWriter);
	}

	public abstract bool Export(IExportContainer container, string projectDirectory, FileSystem fileSystem);
	public abstract bool Contains(IUnityObjectBase asset);
	public abstract long GetExportID(IExportContainer container, IUnityObjectBase asset);
	public abstract MetaPtr CreateExportPointer(IExportContainer container, IUnityObjectBase asset, bool isLocal);

	protected void ExportAsset(IExportContainer container, IUnityObjectBase importer, IUnityObjectBase asset, string path, string name, FileSystem fileSystem)
	{
		if (!fileSystem.Directory.Exists(path))
		{
			fileSystem.Directory.Create(path);
		}

		string fullName = $"{name}.{GetExportExtension(asset)}";
		string uniqueName = fileSystem.GetUniqueName(path, fullName, FileSystem.MaxFileNameLength - MetaExtension.Length);
		string filePath = fileSystem.Path.Join(path, uniqueName);
		AssetExporter.Export(container, asset, filePath, fileSystem);
		Meta meta = new Meta(GUID, importer);
		ExportMeta(container, meta, filePath, fileSystem);
	}

	protected string GetUniqueFileName(IUnityObjectBase asset, string dirPath, FileSystem fileSystem)
	{
		string fileName = asset.GetBestName();
		fileName = FileSystem.RemoveCloneSuffixes(fileName);
		fileName = FileSystem.RemoveInstanceSuffixes(fileName);
		fileName = fileName.Trim();
		if (string.IsNullOrEmpty(fileName))
		{
			fileName = asset.ClassName;
		}
		else
		{
			fileName = FileSystem.FixInvalidFileNameCharacters(fileName);
		}

		fileName = $"{fileName}.{GetExportExtension(asset)}";
		return GetUniqueFileName(dirPath, fileName, fileSystem);
	}

	protected static string GetUniqueFileName(string directoryPath, string fileName, FileSystem fileSystem)
	{
		return fileSystem.GetUniqueName(directoryPath, fileName, FileSystem.MaxFileNameLength - MetaExtension.Length);
	}

	protected virtual string GetExportExtension(IUnityObjectBase asset)
	{
		//https://docs.unity3d.com/Manual/BuiltInImporters.html
		return asset switch
		{
			IShader => "shader",
			IMaterial => "mat",
			IAnimationClip => "anim",
			IAnimatorController => "controller",
			IAnimatorOverrideController => "overrideController",
			IAudioMixer => "mixer",
			IAvatarMask => "mask",
			IShaderVariantCollection => "shadervariants",
			ICubemap => "cubemap",
			ITexture2D => "texture2D",
			IFlare => "flare",
			ILightingSettings => "lighting",
			ILightmapParameters => "giparams",
			IPhysicsMaterial => "physicMaterial",
			IPhysicsMaterial2D => "physicsMaterial2D",
			IRenderTexture => "renderTexture",
			ITerrainLayer => "terrainlayer",
			IWebCamTexture => "webCamTexture",
			IAnimatorState => "state",
			IAnimatorStateMachine => "statemachine",
			IAnimatorTransition => "transition",
			IBlendTree => "blendtree",
			_ => AssetExtension
		};
	}

	public abstract IAssetExporter AssetExporter { get; }
	public abstract AssetCollection File { get; }
	public virtual TransferInstructionFlags Flags => TransferInstructionFlags.NoTransferInstructionFlags;
	public abstract IEnumerable<IUnityObjectBase> Assets { get; }
	public virtual IEnumerable<IUnityObjectBase> ExportableAssets => Assets;
	public abstract string Name { get; }

	private const string MetaExtension = ".meta";
	protected const string AssetExtension = "asset";
	public const string AssetsKeyword = "Assets";
}