Allow multiple file extensions

Author: András Kurai

UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/IResourceData.cs

@@ -1,9 +1,11 @@
-namespace AutSoft.UnityResourceGenerator.Editor.Generation
+using System.Collections.Generic;
+
+namespace AutSoft.UnityResourceGenerator.Editor.Generation
 {
     public interface IResourceData
     {
         string ClassName { get; }
-        string FileExtension { get; }
+        IReadOnlyList<string> FileExtensions { get; }
         bool IsResource { get; }
     }
 }

UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Modules/AllResources.cs

@@ -27,8 +27,9 @@ public static partial class {data.ClassName}
             // ReSharper disable once MissingIndent
             const string classEnd = "        }";
 
-            var values = Directory
-                .EnumerateFiles(context.AssetsFolder, data.FileExtension, SearchOption.AllDirectories)
+            var values = data
+                .FileExtensions
+                .SelectMany(ext => Directory.EnumerateFiles(context.AssetsFolder, ext, SearchOption.AllDirectories))
                 .Select(filePath =>
                 {
                     var (canLoad, baseFolder) = GetBaseFolder(filePath, data.IsResource, context);

UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/ResourceGeneratorSettings.cs

@@ -12,22 +12,22 @@ public sealed class ResourceGeneratorSettings : ScriptableObject
         public sealed class ResourceData : IResourceData
         {
             [SerializeField] private string _className = default;
-            [SerializeField] private string _fileExtension = default;
+            [SerializeField] private string[] _fileExtensions = default;
             [SerializeField] private bool _isResource = default;
 
             public ResourceData()
             {
             }
 
-            public ResourceData(string className, string fileExtension, bool isResource)
+            public ResourceData(string className, string[] fileExtensions, bool isResource)
             {
                 _className = className;
-                _fileExtension = fileExtension;
+                _fileExtensions = fileExtensions;
                 _isResource = isResource;
             }
 
             public string ClassName => _className;
-            public string FileExtension => _fileExtension;
+            public IReadOnlyList<string> FileExtensions => _fileExtensions;
             public bool IsResource => _isResource;
         }
 
@@ -42,14 +42,14 @@ public ResourceData(string className, string fileExtension, bool isResource)
 
         [SerializeField] private bool _logInfo;
         [SerializeField] private bool _logError;
-        [SerializeField] private ResourceData[] _data;
+        [SerializeField] private List<ResourceData> _data;
 
         public string FolderPath => _folderPath;
         public string BaseNamespace => _baseNamespace;
         public string ClassName => _className;
         public bool LogInfo => _logInfo;
         public bool LogError => _logError;
-        public ResourceData[] Data => _data;
+        public IReadOnlyList<ResourceData> Data => _data;
 
         public static ResourceGeneratorSettings GetOrCreateSettings
         (
@@ -71,11 +71,11 @@ public static ResourceGeneratorSettings GetOrCreateSettings
             settings._logInfo = logInfo ?? false;
             settings._logError = logError ?? true;
 
-            settings._data = new[]
+            settings._data = new List<ResourceData>
             {
-                new ResourceData("Scenes", "*.unity", false),
-                new ResourceData("Prefabs", "*.prefab", true),
-                new ResourceData("Materials", "*.mat", true),
+                new ResourceData("Scenes", new[]{"*.unity"}, false),
+                new ResourceData("Prefabs", new[]{"*.prefab"}, true),
+                new ResourceData("Materials", new[]{"*.mat"}, true),
             };
 
             AssetDatabase.CreateAsset(settings, SettingsPath);

UnityResourceGenerator/Assets/ResourceGenerator.asset

@@ -19,11 +19,14 @@ MonoBehaviour:
   _logError: 1
   _data:
   - _className: Scenes
-    _fileExtension: '*.unity'
+    _fileExtensions:
+    - '*.unity'
     _isResource: 0
   - _className: Prefabs
-    _fileExtension: '*.prefab'
+    _fileExtensions:
+    - '*.prefab'
     _isResource: 1
   - _className: Materials
-    _fileExtension: '*.mat'
+    _fileExtensions:
+    - '*.mat'
     _isResource: 1