update

Author: DCFApixels

Editor/NoteUtility.cs

@@ -1,6 +1,8 @@
 #if UNITY_EDITOR
+using System.IO;
 using System.Reflection;
 using UnityEditor;
+using UnityEditorInternal;
 using UnityEngine;
 
 namespace DCFApixels.Notes.Editors
@@ -9,40 +11,35 @@ namespace DCFApixels.Notes.Editors
     internal static class NoteUtility
     {
         private static string _gizmosPath;
-        private static GameObject FindRoot(string name)
-        {
-            GameObject root = GameObject.Find(name);
-            if (root == null)
-            {
-                root = new GameObject(name);
-                root.tag = EDITOR_NAME_TAG;
-            }
-            return root;
-        }
+        
 
         #region CreateLazyNote
         [MenuItem("GameObject/" + ASSET_SHORT_NAME + "/Create " + nameof(LazyNote) + " with arrow")]
         public static void CreateLazyNoteWithArrow(MenuCommand menuCommand)
         {
-            GameObject go = CreateLazyNoteInternal(menuCommand);
-            go.AddComponent<NoteArrow>();
-            Undo.RegisterCreatedObjectUndo(go, "Create " + go.name);
+            CreateLazyNoteInternal(menuCommand, true);
         }
         [MenuItem("GameObject/" + ASSET_SHORT_NAME + "/Create " + nameof(LazyNote))]
         public static void CreateLazyNote(MenuCommand menuCommand)
         {
-            GameObject go = CreateLazyNoteInternal(menuCommand);
-            Undo.RegisterCreatedObjectUndo(go, "Create " + go.name);
+            CreateLazyNoteInternal(menuCommand, false);
         }
-        private static GameObject CreateLazyNoteInternal(MenuCommand menuCommand)
+        private static GameObject CreateLazyNoteInternal(MenuCommand menuCommand, bool isWithArrow)
         {
             GameObject go = new GameObject(nameof(LazyNote));
             go.tag = EDITOR_NAME_TAG;
             go.AddComponent<LazyNote>();
             GameObjectUtility.SetParentAndAlign(go, menuCommand.context as GameObject);
             if (go.transform.parent == null)
-                go.transform.parent = FindRoot(NOTES_ROOT_NAME).transform;
+            {
+                go.transform.parent = FindRootTransform().transform;
+            }
             Selection.activeObject = go;
+            if (isWithArrow)
+            {
+                go.AddComponent<NoteArrow>();
+            }
+            Undo.RegisterCreatedObjectUndo(go, "Create " + go.name);
             return go;
         }
         #endregion
@@ -51,29 +48,34 @@ private static GameObject CreateLazyNoteInternal(MenuCommand menuCommand)
         [MenuItem("GameObject/" + ASSET_SHORT_NAME + "/Create " + nameof(Note) + " with arrow")]
         public static void CreateNoteWithArrow(MenuCommand menuCommand)
         {
-            GameObject go = CreateNoteInternal(menuCommand);
-            go.AddComponent<NoteArrow>();
-            Undo.RegisterCreatedObjectUndo(go, "Create " + go.name);
+            CreateNoteInternal(menuCommand, true);
         }
         [MenuItem("GameObject/" + ASSET_SHORT_NAME + "/Create " + nameof(Note))]
         public static void CreateNote(MenuCommand menuCommand)
         {
-            GameObject go = CreateNoteInternal(menuCommand);
-            Undo.RegisterCreatedObjectUndo(go, "Create " + go.name);
+            CreateNoteInternal(menuCommand, false);
         }
-        private static GameObject CreateNoteInternal(MenuCommand menuCommand)
+        private static GameObject CreateNoteInternal(MenuCommand menuCommand, bool isWithArrow)
         {
             GameObject go = new GameObject(nameof(Note));
             go.tag = EDITOR_NAME_TAG;
             go.AddComponent<Note>();
             GameObjectUtility.SetParentAndAlign(go, menuCommand.context as GameObject);
             if (go.transform.parent == null)
-                go.transform.parent = FindRoot(NOTES_ROOT_NAME).transform;
+            {
+                go.transform.parent = FindRootTransform().transform;
+            }
             Selection.activeObject = go;
+            if (isWithArrow)
+            {
+                go.AddComponent<NoteArrow>();
+            }
+            Undo.RegisterCreatedObjectUndo(go, "Create " + go.name);
             return go;
         }
         #endregion
 
+        #region Draw
         [DrawGizmo(GizmoType.Selected | GizmoType.NonSelected | GizmoType.Pickable)]
         private static void DrawLazyNote(LazyNote note, GizmoType gizmoType)
         {
@@ -97,29 +99,59 @@ private static void DrawNote(Note note, GizmoType gizmoType)
             string sceneNote = GetSceneNote(note.Text, note.DrawIcon);
             Handles.Label(note.transform.position, sceneNote, EditorStyles.whiteBoldLabel);
         }
+        internal static string GetSceneNote(string fullNote, bool isNeedSpacing)
+        {
+            int index = fullNote.IndexOf(NOTE_SEPARATOR);
+            if (index < 0) return string.Empty;
+            string result = fullNote.Substring(0, index);
+            return isNeedSpacing ? "\r\n" + result : result;
+        }
+        #endregion
 
+        #region Utils
+        private static Transform FindRootTransform(string name = NOTES_ROOT_NAME)
+        {
+            GameObject root = GameObject.Find(name);
+            if (root == null)
+            {
+                root = new GameObject(name);
+                root.tag = EDITOR_NAME_TAG;
+                root.transform.position = Vector3.zero;
+                root.transform.rotation = Quaternion.identity;
+                root.transform.localScale = Vector3.one;
+            }
+            return root.transform;
+        }
         internal static string GetGizmosPath()
         {
             if (string.IsNullOrEmpty(_gizmosPath))
             {
                 var assembly = Assembly.GetExecutingAssembly();
                 string packagePath = null;
                 if (assembly != null)
+                {
                     packagePath = UnityEditor.PackageManager.PackageInfo.FindForAssembly(assembly)?.assetPath;
+                }
                 if (string.IsNullOrEmpty(packagePath))
-                    packagePath = "Assets";
+                {
+                    var guids = AssetDatabase.FindAssets($"Notes-Unity t:AssemblyDefinitionAsset");
+                    for (var i = 0; i < guids.Length; i++)
+                    {
+                        var guid = guids[i];
+                        var path = AssetDatabase.GUIDToAssetPath(guid);
+                        var asmdef = AssetDatabase.LoadAssetAtPath<AssemblyDefinitionAsset>(path);
+                        if (asmdef != null && asmdef.name == "Notes-Unity")
+                        {
+                            packagePath = path.Substring(0, path.LastIndexOf("/"));
+                            break;
+                        }
+                    }
+                }
                 _gizmosPath = packagePath + "/Gizmos";
-
             }
             return _gizmosPath;
         }
-        internal static string GetSceneNote(string fullNote, bool isNeedSpacing)
-        {
-            int index = fullNote.IndexOf(NOTE_SEPARATOR);
-            if (index < 0) return string.Empty;
-            string result = fullNote.Substring(0, index);
-            return isNeedSpacing ? "\r\n" + result : result;
-        }
+        #endregion
     }
 }
 #endif

Runtime/Note.cs

@@ -23,9 +23,9 @@ internal class Note : MonoBehaviour, INote
         [SerializeField]
         private bool _drawIcon = true;
 
-        [SerializeField, HideInInspector]
+        [SerializeField]
         private int _authorID;
-        [SerializeField, HideInInspector]
+        [SerializeField]
         private int _typeID;
 
         private AuthorInfo _author;