COM3D2.MeidoPhotoStudio/COM3D2.MeidoPhotoStudio.Plugin/MeidoPhotoStudio/GUI/Windows/SceneModalWindow.cs

using System;
using UnityEngine;

namespace COM3D2.MeidoPhotoStudio.Plugin
{
    internal class SceneModalWindow : BaseWindow
    {
        private static Texture2D infoHighlight = Utility.MakeTex(2, 2, new Color(0f, 0f, 0f, 0.8f));
        private SceneManager sceneManager;
        public override Rect WindowRect
        {
            set
            {
                value.width = Mathf.Clamp(Screen.width * 0.3f, 360f, 500f);
                value.height = directoryMode ? 150f : Mathf.Clamp(Screen.height * 0.4f, 240f, 380f);
                base.WindowRect = value;
            }
        }
        private bool visible;
        public override bool Visible
        {
            get => visible;
            set
            {
                visible = value;
                if (value)
                {
                    WindowRect = WindowRect;
                    windowRect.x = MiddlePosition.x;
                    windowRect.y = MiddlePosition.y;
                }
            }
        }

        private Button okButton;
        private Button cancelButton;
        private Button deleteButton;
        private Button overwriteButton;
        private string deleteDirectoryMessage
            = "Are you sure you want to permanently delete '{0}' and all of its files?";
        private string deleteSceneMessage = "Are you sure you want to permanently delete '{0}'?";
        private string directoryDeleteCommit = "Delete Folder";
        private string sceneDeleteCommit = "Delete Scene";
        private string sceneLoadCommit = "Load Scene";
        private bool directoryMode = false;
        private bool deleteScene = false;

        public SceneModalWindow(SceneManager sceneManager)
        {
            this.sceneManager = sceneManager;

            windowRect.x = MiddlePosition.x;
            windowRect.y = MiddlePosition.y;
            this.okButton = new Button(sceneLoadCommit);
            this.okButton.ControlEvent += (s, a) => Commit();

            this.cancelButton = new Button("Cancel");
            this.cancelButton.ControlEvent += (s, a) => Cancel();

            this.deleteButton = new Button("Delete");
            this.deleteButton.ControlEvent += (s, a) =>
            {
                okButton.Label = sceneDeleteCommit;
                deleteScene = true;
            };

            this.overwriteButton = new Button("Overwrite");
            this.overwriteButton.ControlEvent += (s, a) =>
            {
                sceneManager.OverwriteScene();
                Visible = false;
            };
        }

        public override void Draw()
        {
            GUILayout.BeginArea(new Rect(10f, 10f, WindowRect.width - 20f, WindowRect.height - 20f));

            // thumbnail
            if (!directoryMode)
            {
                SceneManager.Scene scene = sceneManager.CurrentScene;
                Texture2D thumb = scene.Thumbnail;

                float scale = Mathf.Min(
                    (WindowRect.width - 20f) / (float)thumb.width, (WindowRect.height - 110f) / (float)thumb.height
                );
                float width = Mathf.Min(thumb.width, (float)thumb.width * scale);
                float height = Mathf.Min(thumb.height, (float)thumb.height * scale);

                GUILayout.BeginHorizontal();
                GUILayout.FlexibleSpace();

                MiscGUI.DrawTexture(thumb, GUILayout.Width(width), GUILayout.Height(height));

                GUIStyle labelStyle = new GUIStyle(GUI.skin.label);
                labelStyle.fontSize = Utility.GetPix(12);
                labelStyle.alignment = TextAnchor.MiddleCenter;
                labelStyle.normal.background = infoHighlight;

                Rect labelBox = GUILayoutUtility.GetLastRect();

                if (scene.NumberOfMaids != SceneManager.Scene.initialNumberOfMaids)
                {
                    string infoString = scene.NumberOfMaids == MeidoPhotoStudio.kankyoMagic
                        ? "Kankyo"
                        : $"{scene.NumberOfMaids} Maids";

                    Vector2 labelSize = labelStyle.CalcSize(new GUIContent(infoString));

                    labelBox = new Rect(
                        labelBox.x + 10, labelBox.y + labelBox.height - (labelSize.y + 10),
                        labelSize.x + 10, labelSize.y + 2
                    );

                    GUI.Label(labelBox, infoString, labelStyle);
                }

                GUILayout.FlexibleSpace();
                GUILayout.EndHorizontal();
            }

            // message
            string currentMessage = string.Empty;
            string context = string.Empty;

            if (directoryMode)
            {
                currentMessage = deleteDirectoryMessage;
                context = sceneManager.CurrentDirectoryName;
            }
            else
            {
                if (deleteScene)
                {
                    currentMessage = deleteSceneMessage;
                    context = sceneManager.CurrentScene.FileInfo.Name;
                }
                else
                {
                    currentMessage = sceneManager.CurrentScene.FileInfo.Name;
                    context = currentMessage;
                }
            }

            GUIStyle messageStyle = new GUIStyle(GUI.skin.label);
            messageStyle.alignment = TextAnchor.MiddleCenter;
            messageStyle.fontSize = Utility.GetPix(12);

            GUILayout.FlexibleSpace();

            GUILayout.Label(string.Format(currentMessage, context), messageStyle);

            GUILayout.FlexibleSpace();

            // Buttons

            GUIStyle buttonStyle = new GUIStyle(GUI.skin.button);
            buttonStyle.fontSize = Utility.GetPix(12);

            GUILayoutOption buttonHeight = GUILayout.Height(Utility.GetPix(20));

            GUILayout.BeginHorizontal();

            if (directoryMode || deleteScene)
            {
                GUILayout.FlexibleSpace();
                okButton.Draw(buttonStyle, buttonHeight, GUILayout.ExpandWidth(false));
                cancelButton.Draw(buttonStyle, buttonHeight, GUILayout.Width(100));
            }
            else
            {
                deleteButton.Draw(buttonStyle, buttonHeight, GUILayout.ExpandWidth(false));
                overwriteButton.Draw(buttonStyle, buttonHeight, GUILayout.ExpandWidth(false));

                GUILayout.FlexibleSpace();

                okButton.Draw(buttonStyle, buttonHeight, GUILayout.ExpandWidth(false));
                cancelButton.Draw(buttonStyle, buttonHeight, GUILayout.Width(100));
            }

            GUILayout.EndHorizontal();

            GUILayout.EndArea();
        }

        public void ShowDirectoryDialogue()
        {
            okButton.Label = directoryDeleteCommit;
            directoryMode = true;
            Modal.Show(this);
        }

        public void ShowSceneDialogue()
        {
            directoryMode = false;
            okButton.Label = sceneLoadCommit;
            Modal.Show(this);
        }

        private void Commit()
        {
            if (directoryMode)
            {
                sceneManager.DeleteDirectory();
                Modal.Close();
            }
            else
            {
                if (deleteScene)
                {
                    sceneManager.DeleteScene();
                    deleteScene = false;
                }
                else sceneManager.LoadScene();

                Modal.Close();
            }
        }

        private void Cancel()
        {
            if (directoryMode) Modal.Close();
            else
            {
                if (deleteScene) deleteScene = false;
                else Modal.Close();
            }
            okButton.Label = sceneLoadCommit;
        }
    }
}