namespace ProjectBrowserExtension
{
    using UnityEditor;
    using System.Reflection;
    using System.Linq;

    public class ProjectBrowserExtension
    {
        const string MenuItemName = "Assets/Highlight Current Directory";

        /// <summary>
        /// プロジェクトウィンドウでカレントディレクトリの親フォルダをハイライトする
        /// </summary>
        [MenuItem(MenuItemName)]
        static void FrameCurrentDirectory()
        {
            string currentDirectory = GetCurrentDirectory(); // カレントディレクトリ

            // カレントディレクトリ直下のファイルのパスを取得
            int depth = currentDirectory.Split('/').Length;
            string path = AssetDatabase.GetAllAssetPaths()
            .Where(assetPath => assetPath.Contains(currentDirectory)) // カレントディレクトリのファイルかどうか
            .FirstOrDefault(assetPath => assetPath.Split('/').Length == depth + 1); // カレントディレクトリ直下のファイルかどうか

            if (string.IsNullOrEmpty(path)) // カレントディレクトリにファイルが存在しない場合
            {
                FrameObjectInTwoColumnMode(GetInstanceID(currentDirectory), true, false);
                EditorApplication.delayCall += () =>
                {
                    ShowFolderContents(GetInstanceID(currentDirectory), true);
                };
            }
            else // カレントディレクトリにファイルが存在する場合
            {
                FrameObjectInTwoColumnMode(GetInstanceID(path), true, true);
            }
        }

        /// <summary>
        /// FrameCurrentDirectory()のValidate関数
        /// </summary>
        [MenuItem(MenuItemName, true, 20000)]
        static bool ValidateFrameCurrentDirectory()
        {
            return IsTwoColumn();
        }

        /// <summary>
        /// ProjectウィンドウがTwoColumnモードかどうか
        /// </summary>
        static bool IsTwoColumn()
        {
            var flag = BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static | BindingFlags.Instance;
            var asm = Assembly.Load("UnityEditor.dll");
            var typeProjectBrowser = asm.GetType("UnityEditor.ProjectBrowser");
            var projectBrowserWindow = EditorWindow.GetWindow(typeProjectBrowser);

            return typeProjectBrowser.GetField("m_ViewMode", flag).GetValue(projectBrowserWindow).ToString() == "TwoColumns";
        }

        /// <summary>
        /// pathからInstanceIDを取得
        /// </summary>
        static int GetInstanceID(string path)
        {
            return AssetDatabase.LoadAssetAtPath(path, typeof(UnityEngine.Object)).GetInstanceID();
        }

        /// <summary>
        /// Projectウィンドウでオブジェクトをハイライトする
        /// </summary>
        static void FrameObjectInTwoColumnMode(int instanceID, bool frame, bool ping)
        {
            var flag = BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static | BindingFlags.Instance;
            var asm = Assembly.Load("UnityEditor.dll");
            var typeProjectBrowser = asm.GetType("UnityEditor.ProjectBrowser");
            var projectBrowserWindow = EditorWindow.GetWindow(typeProjectBrowser);
            typeProjectBrowser.GetMethod("FrameObjectInTwoColumnMode", flag).Invoke(projectBrowserWindow, new object[] { instanceID, frame, ping });
        }

        /// <summary>
        /// Projectウィンドウでフォルダの中身を表示
        /// </summary>
        static void ShowFolderContents(int folderInstanceID, bool revealAndFrameInFolderTree)
        {
            var flag = BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static | BindingFlags.Instance;
            var asm = Assembly.Load("UnityEditor.dll");
            var typeProjectBrowser = asm.GetType("UnityEditor.ProjectBrowser");
            var projectBrowserWindow = EditorWindow.GetWindow(typeProjectBrowser);
            typeProjectBrowser.GetMethod("ShowFolderContents", flag).Invoke(projectBrowserWindow, new object[] { folderInstanceID, revealAndFrameInFolderTree });
        }

        /// <summary>
        /// Projectウィンドウで現在開いているディレクトリを取得
        /// </summary>
        static string GetCurrentDirectory()
        {
            var flag = BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static | BindingFlags.Instance;
            var asm = Assembly.Load("UnityEditor.dll");
            var typeProjectBrowser = asm.GetType("UnityEditor.ProjectBrowser");
            var projectBrowserWindow = EditorWindow.GetWindow(typeProjectBrowser);
            return (string)typeProjectBrowser.GetMethod("GetActiveFolderPath", flag).Invoke(projectBrowserWindow, null);
        }
    }
}