Asset.SubAsset.cs

// Copyright (C) 2021-2023 Steffen Itterheim
// Refer to included LICENSE file for terms and conditions.
 
using System;
using System.Diagnostics.CodeAnalysis;
using UnityEditor;
using Object = UnityEngine.Object;
 
namespace CodeSmileEditor
{
    public sealed partial class Asset
    {
        public static class SubAsset
        {
            [ExcludeFromCodeCoverage] // simple relay
            public static Boolean Extract([NotNull] Object subAsset, [NotNull] Path destinationPath)
            {
                ThrowIf.ArgumentIsNull(subAsset, nameof(subAsset));
                ThrowIf.ArgumentIsNull(destinationPath, nameof(destinationPath));
 
                return Succeeded(AssetDatabase.ExtractAsset(subAsset, destinationPath));
            }

            public static void Add([NotNull] Object subAssetInstance, [NotNull] Object asset)
            {
                ThrowIf.ArgumentIsNull(subAssetInstance, nameof(subAssetInstance));
                ThrowIf.SubObjectIsGameObject(subAssetInstance);
                ThrowIf.AlreadyAnAsset(subAssetInstance);
                ThrowIf.ArgumentIsNull(asset, nameof(asset));
 
                AssetDatabase.AddObjectToAsset(subAssetInstance, asset);
            }

            public static void Remove([NotNull] Object subAsset)
            {
                ThrowIf.ArgumentIsNull(subAsset, nameof(subAsset));
 
                AssetDatabase.RemoveObjectFromAsset(subAsset);
            }

            public static void SetMain([NotNull] Object subAsset, [NotNull] Path path)
            {
                AssetDatabase.SetMainObject(subAsset, path);
 
                // SetMainObject only records which object should become the main one on the next import,
                // it does not write the asset file. On Unity 6000.7.0a5 and 6000.7.0a6, importing without
                // saving first re-reads the old main object from disk and the change is lost. Saving the
                // asset before the import makes the change take effect on all nine editors tested.
                AssetDatabase.SaveAssetIfDirty(subAsset);
                File.Import(path);
            }

            public static void SetMain([NotNull] Object subAsset, [NotNull] Object asset) => SetMain(subAsset, Path.Get(asset));

            public static Object[] LoadAll([NotNull] Path path) => AssetDatabase.LoadAllAssetsAtPath(path);

            public static Object[] LoadVisible([NotNull] Path path) => AssetDatabase.LoadAllAssetRepresentationsAtPath(path);
        }
    }
}