Fishing2/Packages/com.jbooth.microverse/Scripts/Editor/ContentBrowser/ContentItemPopup.cs

using System.Collections.Generic;
using UnityEditor;
using UnityEngine;

namespace JBooth.MicroVerseCore.Browser
{
    public class ContentItemPopup : PopupWindowContent
    {
        private GUIContent prepareMoveContent = new GUIContent("Prepare Move", "Select a preset item for movement. Use Insert Here to effectively move the preset item.\n\nShortcut: ctrl + x");
        private GUIContent insertBeforeContent = new GUIContent("Insert Before", "Inserts a preset item that was previously specified with 'Prepare Move' before current preset item.\n\nShorcut: ctrl + v");

        private enum Feature
        {
            UpdateThumbnail,
            PingCollection,
            PingHeightmap,
            PingPrefab,
            Rename,
            PrepareMove,
            InsertBefore,
            Close
        }

        private ContentBrowser browser;
        private PresetItem presetItem;

        private List<Feature> features;

        public ContentItemPopup(ContentBrowser browser, PresetItem presetItem)
        {
            if (presetItem == null || presetItem.collection == null)
                return;

            this.browser = browser;
            this.presetItem = presetItem;

            // set features per preset content type
            features = new List<Feature>();

            switch(presetItem.collection.contentType)
            {
                case ContentType.Height:
                    //features.Add(Feature.UpdateThumbnail);
                    features.Add(Feature.PingCollection);
                    features.Add(Feature.PingHeightmap);
                    //features.Add(Feature.Rename);
                    features.Add(Feature.Close);
                    break;

                default:
                    features.Add(Feature.UpdateThumbnail);
                    features.Add(Feature.PingCollection);
                    features.Add(Feature.PingPrefab);
                    features.Add(Feature.Rename);
                    features.Add(Feature.PrepareMove);
                    features.Add(Feature.InsertBefore);
                    features.Add(Feature.Close);
                    break;
            }
        }

        public override Vector2 GetWindowSize()
        {
            int menuRows = features.Count;
            return new Vector2(180, 21 * menuRows);
        }

        public override void OnGUI(Rect rect)
        {
            if (features.Contains(Feature.UpdateThumbnail))
            {
                if (GUILayout.Button("Update Thumbnail"))
                {
                    ThumbnailCreator.SaveIcon( presetItem);
                }
            }

            if (features.Contains(Feature.PingCollection))
            {
                if (GUILayout.Button("Ping Collection"))
                {
                    PingCollection();

                }
            }

            if (features.Contains(Feature.PingHeightmap))
            {
                if (GUILayout.Button("Ping Heightmap"))
                {
                    PingHeightmap();

                }
            }

            if (features.Contains(Feature.PingPrefab))
            {
                if (GUILayout.Button("Ping Prefab"))
                {
                    PingPrefab();

                }
            }

            if (features.Contains(Feature.Rename))
            {
                if (GUILayout.Button("Rename"))
                {
                    Rename();
                }
            }


            if (features.Contains(Feature.PrepareMove))
            {
                if (GUILayout.Button(prepareMoveContent))
                {
                    ContentSelectionGridMovement.PrepareMove(presetItem);
                    editorWindow.Close();

                }
            }

            bool moveHereEnabled = ContentSelectionGridMovement.MoveHereEnabled(presetItem);

            GUI.enabled = moveHereEnabled;
            if (features.Contains(Feature.InsertBefore))
            {
                if (GUILayout.Button(insertBeforeContent))
                {
                    ContentSelectionGridMovement.InsertBefore(presetItem);
                    editorWindow.Close();

                }
            }
            GUI.enabled = true;

            if (features.Contains(Feature.Close))
            {
                if (GUILayout.Button("Close"))
                {
                    editorWindow.Close();
                }
            }
        }

        public override void OnOpen()
        {
        }

        public override void OnClose()
        {
        }



        void Rename()
        {
            if (presetItem == null)
                return;

            ContentData item = presetItem.content;

            if (item.prefab == null)
            {
                Debug.LogError("Prefab missing");
                return;
            }

            // close the current popup, we don't need it anymore
            // note: we can't close it later after PopupWindow.show ... for some reason the code execution stops there
            this.editorWindow.Close();

            // popup window parameters
            Vector2 position = Event.current.mousePosition;
            Rect popupRect = new Rect(position, Vector2.zero); // 2nd parameter is the offset from mouse position

            // show rename popup
            PresetItemPopup renamePopup = new PresetItemPopup(presetItem);
            UnityEditor.PopupWindow.Show(popupRect, renamePopup);

        }

        void PingCollection()
        {
            if (presetItem == null)
                return;

            EditorGUIUtility.PingObject(presetItem.collection);
        }

        void PingHeightmap()
        {
            if (presetItem == null)
                return;

            ContentCollection collection = presetItem.collection;
            int index = presetItem.collectionIndex;
            ContentData data = collection.contents[index];

            Texture2D heightmap = AssetDatabase.LoadAssetAtPath<Texture2D>(AssetDatabase.GUIDToAssetPath(new GUID(data.stamp)));
            EditorGUIUtility.PingObject(heightmap);

        }

        void PingPrefab()
        {
            if (presetItem == null)
                return;

            ContentCollection collection = presetItem.collection;
            int index = presetItem.collectionIndex;
            ContentData data = collection.contents[index];

            EditorGUIUtility.PingObject(data.prefab);
        }
    }
}