configure class generation by data

András Kurai · András Kurai · commit f129c2ab0d5f · 2021-10-13T10:04:02.000+02:00
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions.meta b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions.meta
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/EnumerableExtensions.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/EnumerableExtensions.cs
@@ -2,7 +2,7 @@
 using System.Collections.Generic;
 using System.Linq;
 
-namespace AutSoft.UnityResourceGenerator.Editor.Generation
+namespace AutSoft.UnityResourceGenerator.Editor.Generation.Extensions
 {
     internal static class EnumerableExtensions
     {
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/EnumerableExtensions.cs.meta b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/EnumerableExtensions.cs.meta
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/StringBuilderExtensions.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/StringBuilderExtensions.cs
@@ -0,0 +1,18 @@
+﻿using System.Collections.Generic;
+using System.Text;
+
+namespace AutSoft.UnityResourceGenerator.Editor.Generation.Extensions
+{
+    internal static class StringBuilderExtensions
+    {
+        public static StringBuilder AppendMultipleLines(this StringBuilder builder, IEnumerable<string> lines)
+        {
+            foreach (var line in lines)
+            {
+                builder.AppendLine(line);
+            }
+
+            return builder;
+        }
+    }
+}
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/StringBuilderExtensions.cs.meta b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Extensions/StringBuilderExtensions.cs.meta
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/IResourceData.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/IResourceData.cs
@@ -0,0 +1,9 @@
+﻿namespace AutSoft.UnityResourceGenerator.Editor.Generation
+{
+    public interface IResourceData
+    {
+        string ClassName { get; }
+        string FileExtension { get; }
+        bool IsResource { get; }
+    }
+}
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/IResourceData.cs.meta b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/IResourceData.cs.meta
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Modules/AllResources.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/Modules/AllResources.cs
@@ -1,4 +1,5 @@
-﻿using System;
+﻿using AutSoft.UnityResourceGenerator.Editor.Generation.Extensions;
+using System;
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
@@ -10,29 +11,27 @@ public sealed class AllResources : IModuleGenerator
     {
         public string Generate(ResourceContext context) =>
             new StringBuilder()
-                .AppendLine(Generate(context, "Scenes", "*.unity", false))
-                .AppendLine(Generate(context, "Prefabs", "*.prefab", true))
-                .AppendLine(Generate(context, "Materials", "*.mat", true))
+                .AppendMultipleLines(context.Data.Select(d => Generate(context, d)))
                 .ToString();
 
-        private static string Generate(ResourceContext context, string className, string fileExtension, bool isResource)
+        private static string Generate(ResourceContext context, IResourceData data)
         {
-            context.Info($"Started generating {className}");
+            context.Info($"Started generating {data.ClassName}");
 
             // ReSharper disable once MissingIndent
             var classBegin =
 $@"
-        public static partial class {className}
+        public static partial class {data.ClassName}
         {{
 ";
             // ReSharper disable once MissingIndent
             const string classEnd = "        }";
 
             var values = Directory
-                .EnumerateFiles(context.AssetsFolder, fileExtension, SearchOption.AllDirectories)
+                .EnumerateFiles(context.AssetsFolder, data.FileExtension, SearchOption.AllDirectories)
                 .Select(filePath =>
                 {
-                    var (canLoad, baseFolder) = GetBaseFolder(filePath, isResource, context);
+                    var (canLoad, baseFolder) = GetBaseFolder(filePath, data.IsResource, context);
                     if (!canLoad) return (null, null);
 
                     var resourcePath = filePath
@@ -75,7 +74,7 @@ public static partial class {className}
                 .AppendLine(classEnd)
                 .ToString();
 
-            context.Info($"Finished generating {className}");
+            context.Info($"Finished generating {data.ClassName}");
 
             return output;
         }
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/ResourceContext.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/ResourceContext.cs
@@ -1,4 +1,5 @@
 ﻿using System;
+using System.Collections.Generic;
 
 namespace AutSoft.UnityResourceGenerator.Editor.Generation
 {
@@ -11,14 +12,16 @@ public ResourceContext
             string baseNamespace,
             string className,
             Action<string> info,
-            Action<string> error)
+            Action<string> error,
+            IReadOnlyList<IResourceData> data)
         {
             AssetsFolder = assetsFolder;
             FolderPath = folderPath;
             BaseNamespace = baseNamespace;
             ClassName = className;
             Info = info;
             Error = error;
+            Data = data;
         }
 
         public string AssetsFolder { get; }
@@ -27,5 +30,7 @@ public ResourceContext
         public string ClassName { get; }
         public Action<string> Info { get; }
         public Action<string> Error { get; }
+
+        public IReadOnlyList<IResourceData> Data { get; }
     }
 }
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/ResourceFileGenerator.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/Generation/ResourceFileGenerator.cs
@@ -1,4 +1,5 @@
-﻿using System;
+﻿using AutSoft.UnityResourceGenerator.Editor.Generation.Extensions;
+using System;
 using System.Linq;
 using System.Text;
 
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/ResourceFileMenu.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/ResourceFileMenu.cs
@@ -22,7 +22,8 @@ public static void GenerateResources()
                 settings.BaseNamespace,
                 settings.ClassName,
                 settings.LogInfo ? Debug.Log : LogEmpty,
-                settings.LogError ? Debug.LogError : LogEmpty
+                settings.LogError ? Debug.LogError : LogEmpty,
+                settings.Data
             );
 
             context.Info("Resource Path generation started");
diff --git a/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/ResourceGeneratorSettings.cs b/UnityResourceGenerator/Assets/AutSoft.UnityResourceGenerator/Editor/ResourceGeneratorSettings.cs
@@ -1,11 +1,36 @@
-﻿using System.Collections.Generic;
+﻿using AutSoft.UnityResourceGenerator.Editor.Generation;
+using System;
+using System.Collections.Generic;
 using UnityEditor;
 using UnityEngine;
 
 namespace AutSoft.UnityResourceGenerator.Editor
 {
     public sealed class ResourceGeneratorSettings : ScriptableObject
     {
+        [Serializable]
+        public sealed class ResourceData : IResourceData
+        {
+            [SerializeField] private string _className = default;
+            [SerializeField] private string _fileExtension = default;
+            [SerializeField] private bool _isResource = default;
+
+            public ResourceData()
+            {
+            }
+
+            public ResourceData(string className, string fileExtension, bool isResource)
+            {
+                _className = className;
+                _fileExtension = fileExtension;
+                _isResource = isResource;
+            }
+
+            public string ClassName => _className;
+            public string FileExtension => _fileExtension;
+            public bool IsResource => _isResource;
+        }
+
         private const string SettingsPath = "Assets/ResourceGenerator.asset";
 
         [SerializeField] private string _baseNamespace;
@@ -17,12 +42,14 @@ public sealed class ResourceGeneratorSettings : ScriptableObject
 
         [SerializeField] private bool _logInfo;
         [SerializeField] private bool _logError;
+        [SerializeField] private 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 static ResourceGeneratorSettings GetOrCreateSettings()
         {
@@ -37,6 +64,13 @@ public static ResourceGeneratorSettings GetOrCreateSettings()
             settings._logInfo = false;
             settings._logError = true;
 
+            settings._data = new[]
+            {
+                new ResourceData("Scenes", "*.unity", false),
+                new ResourceData("Prefabs", "*.prefab", true),
+                new ResourceData("Materials", "*.mat", true),
+            };
+
             AssetDatabase.CreateAsset(settings, SettingsPath);
             AssetDatabase.SaveAssets();
 
@@ -57,6 +91,7 @@ public static SettingsProvider CreateSettingsProvider() =>
                     EditorGUILayout.PropertyField(settings.FindProperty(nameof(_className)), new GUIContent("Class name"));
                     EditorGUILayout.PropertyField(settings.FindProperty(nameof(_logInfo)), new GUIContent("Log Infos"));
                     EditorGUILayout.PropertyField(settings.FindProperty(nameof(_logError)), new GUIContent("Log Errors"));
+                    EditorGUILayout.PropertyField(settings.FindProperty(nameof(_data)), new GUIContent("Data"));
 
                     settings.ApplyModifiedProperties();
                 },
diff --git a/UnityResourceGenerator/Assets/ResourceGenerator.asset b/UnityResourceGenerator/Assets/ResourceGenerator.asset
@@ -15,5 +15,15 @@ MonoBehaviour:
   _baseNamespace: AutSoft.UnityResourceGenerator.Sample
   _className: ResourcePaths
   _folderPath: AutSoft.UnityResourceGenerator.Sample
-  _logInfo: 1
+  _logInfo: 0
   _logError: 1
+  _data:
+  - _className: Scenes
+    _fileExtension: '*.unity'
+    _isResource: 0
+  - _className: Prefabs
+    _fileExtension: '*.prefab'
+    _isResource: 1
+  - _className: Materials
+    _fileExtension: '*.mat'
+    _isResource: 1

Original file line number	Diff line number	Diff line change
`@@ -2,7 +2,7 @@`
`2`	`2`	`using System.Collections.Generic;`
`3`	`3`	`using System.Linq;`
`4`	`4`
`5`		`-namespace AutSoft.UnityResourceGenerator.Editor.Generation`
	`5`	`+namespace AutSoft.UnityResourceGenerator.Editor.Generation.Extensions`
`6`	`6`	`{`
`7`	`7`	`internal static class EnumerableExtensions`
`8`	`8`	`{`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`using System;`
	`2`	`+using System.Collections.Generic;`
`2`	`3`
`3`	`4`	`namespace AutSoft.UnityResourceGenerator.Editor.Generation`
`4`	`5`	`{`
`@@ -11,14 +12,16 @@ public ResourceContext`
`11`	`12`	`string baseNamespace,`
`12`	`13`	`string className,`
`13`	`14`	`Action<string> info,`
`14`		`- Action<string> error)`
	`15`	`+ Action<string> error,`
	`16`	`+ IReadOnlyList<IResourceData> data)`
`15`	`17`	`{`
`16`	`18`	`AssetsFolder = assetsFolder;`
`17`	`19`	`FolderPath = folderPath;`
`18`	`20`	`BaseNamespace = baseNamespace;`
`19`	`21`	`ClassName = className;`
`20`	`22`	`Info = info;`
`21`	`23`	`Error = error;`
	`24`	`+ Data = data;`
`22`	`25`	`}`
`23`	`26`
`24`	`27`	`public string AssetsFolder { get; }`
`@@ -27,5 +30,7 @@ public ResourceContext`
`27`	`30`	`public string ClassName { get; }`
`28`	`31`	`public Action<string> Info { get; }`
`29`	`32`	`public Action<string> Error { get; }`
	`33`	`+`
	`34`	`+ public IReadOnlyList<IResourceData> Data { get; }`
`30`	`35`	`}`
`31`	`36`	`}`