Possibility to filter and create valid ScriptableObjects

Author: arimger

Assets/Editor Toolbox/Editor/Windows/ScriptableObjectCreationWindow.cs

@@ -2,17 +2,38 @@
 using System.Reflection;
 
 using UnityEditor;
-using UnityEditor.IMGUI.Controls;
 using UnityEngine;
+using Object = UnityEngine.Object;
 
 namespace Toolbox.Editor.Windows
 {
+    using Toolbox.Editor.Internal;
+
     public class ScriptableObjectCreationWindow : EditorWindow
     {
-        private SearchField searchField;
+        private class TypeConstraintScriptableObject : TypeConstraintStandard
+        {
+            public TypeConstraintScriptableObject() : base(typeof(ScriptableObject), TypeSettings.Class, false, false)
+            { }
+
+
+            public override bool IsSatisfied(Type type)
+            {
+                return Attribute.IsDefined(type, typeof(CreateAssetMenuAttribute)) && base.IsSatisfied(type);
+            }
+        }
+
 
+        private static readonly TypeConstraintContext sharedConstraint = new TypeConstraintScriptableObject();
+        private static readonly TypeAppearanceContext sharedAppearance = new TypeAppearanceContext(sharedConstraint, TypeGrouping.None, true);
+        private static readonly TypeField typeField = new TypeField(sharedConstraint, sharedAppearance);
 
-        [MenuItem("Assets/Create/ScriptableObject Creation Window", priority = 5)]
+        private Type activeType;
+        private int instancesCount = 1;
+        private string baseName;
+        private Object defaultObject;
+
+        [MenuItem("Assets/Create/Toolbox/ScriptableObject Creation Window", priority = 5)]
         internal static void Initialize()
         {
             var window = GetWindow<ScriptableObjectCreationWindow>();
@@ -21,41 +42,65 @@ internal static void Initialize()
         }
 
 
-        private void OnEnable()
-        {
-            searchField = new SearchField();
-        }
-
         private void OnGUI()
         {
-            using (new EditorGUILayout.HorizontalScope())
+            using (new EditorGUILayout.VerticalScope(EditorStyles.helpBox))
             {
-                using (new EditorGUILayout.VerticalScope())
-                {
-                    DrawSearchPanel();
-                }
-
-                using (new EditorGUILayout.VerticalScope())
-                {
-                    DrawCreatePanel();
-                }
+                DrawSettingsPanel();
             }
+
+            DrawCreatePanel();
         }
 
-        private void DrawSearchPanel()
+
+        private void DrawSettingsPanel()
         {
-            var rect = GUILayoutUtility.GetRect(100.0f, 16.0f);
-            searchField.OnGUI(rect, string.Empty);
+            EditorGUILayout.LabelField("Settings", EditorStyles.boldLabel);
+            var rect = EditorGUILayout.GetControlRect(true);
+            typeField.OnGui(rect, true, OnTypeSelected, activeType);
+            if (activeType == null)
+            {
+                return;
+            }
+
+            instancesCount = EditorGUILayout.IntField("Instances To Create", instancesCount);
+            instancesCount = Mathf.Min(instancesCount, 1);
+            baseName = EditorGUILayout.TextField("Base Name", baseName);
+            EditorGUI.BeginChangeCheck();
+            defaultObject = EditorGUILayout.ObjectField(new GUIContent("Default Object"), defaultObject, activeType, false);
+            if (EditorGUI.EndChangeCheck())
+            {
+                //TODO: validate default Object
+            }
         }
 
         private void DrawCreatePanel()
         {
             if (GUILayout.Button("Create"))
             {
-                Debug.Log(GetActiveFolderPath());
+                CreateObjects();
             }
         }
 
+        private void CreateObjects()
+        {
+            CreateObjects(activeType, defaultObject);
+        }
+
+        private void CreateObjects(Type targetType, Object defaultObject)
+        {
+            //TODO: validate default Object
+            for (var i = 0; i < instancesCount; i++)
+            {
+                //TODO: instantiate SOs
+            }
+        }
+
+        private Object CreateObject(Type targetType, Object defaultObject)
+        {
+            return defaultObject != null ? Instantiate(defaultObject) : CreateInstance(targetType);
+        }
+
         private static string GetActiveFolderPath()
         {
             Type projectWindowUtilType = typeof(ProjectWindowUtil);
@@ -64,5 +109,19 @@ private static string GetActiveFolderPath()
             string pathToCurrentFolder = obj.ToString();
             return pathToCurrentFolder;
         }
+
+        private void OnTypeSelected(Type type)
+        {
+            activeType = type;
+            var attribute = type?.GetCustomAttribute<CreateAssetMenuAttribute>();
+            if (attribute != null)
+            {
+                baseName = attribute.fileName;
+            }
+            else
+            {
+                baseName = string.Empty;
+            }
+        }
     }
 }

Assets/Examples/Scripts/SampleScriptableObject.cs

@@ -1,6 +1,6 @@
-using UnityEngine;
+using UnityEngine;
 
-[CreateAssetMenu]
+[CreateAssetMenu(fileName = "Sample Scriptable Object")]
 public class SampleScriptableObject : ScriptableObject
 {
     public bool var1;

Assets/New Sample Sample SO.asset

@@ -0,0 +1,16 @@
+%YAML 1.1
+%TAG !u! tag:unity3d.com,2011:
+--- !u!114 &11400000
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 0}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: 15c192e004cb32d4b86a9cea8b49d83b, type: 3}
+  m_Name: New Sample Sample SO
+  m_EditorClassIdentifier: 
+  var1: 0
+  var2: 0

Assets/New Sample Sample SO.asset.meta

@@ -0,0 +1,7 @@
+using UnityEngine;
+
+[CreateAssetMenu]
+public class SampleSampleSO : SampleScriptableObject
+{
+
+}