res-avatar-unity/Assets/_PoiyomiShaders/Scripts/ModuleInstaller/Editor/ModuleInstaller.cs

#if UNITY_EDITOR
using System.Collections;
using System.Collections.Generic;
using UnityEditor;
using UnityEngine;
using System.IO;
using UnityEngine.UI;
using System;
using System.Text;
using UnityEngine.Assertions;
using System.Linq;
using System.Text.RegularExpressions;
using Poiyomi.ModularShaderSystem;

//using Poiyomi.ModularShaderSystem;

// PoiyomiPatreon.Scripts.poi_tools.Editor

namespace Poiyomi.AngryLabs
{
    [ExecuteInEditMode]
    public class ModuleInstaller : EditorWindow
    {
        [MenuItem("Poi/Module Installer")]
        static void Init()
        {
            //var window =  GetWindowWithRect<BlendshapeToTextures>(new Rect(0, 0, 200, 500));
            var window = GetWindow(typeof(ModuleInstaller));
            window.minSize = new Vector2(250, 300);
            window.Show();
        }

        private static void ShowWindow()
        {
            var window = GetWindow<ModuleInstaller>();
            window.titleContent = new GUIContent("Module Installer");
            window.Show();

            RefreshShaderList();
        }

        string CopyName { get; set; }
        string NewID { get; set; } = ".poiyomi/User Modules/";

        string OutputPath;

        static List<ModularShader> _shaders;
        static List<Shader> _compiledShaders;

        static IEnumerable<T> FindAssetsByType<T>(string searchFolder = null)
        {
            AssetDatabase.Refresh();
            string[] guids;
            if(searchFolder != null)
            {
                guids  = AssetDatabase.FindAssets($"t:{typeof(T).ToString().Replace("UnityEngine.", "")}", new string[] { searchFolder });
            }
            else
            {
                guids = AssetDatabase.FindAssets($"t:{typeof(T).ToString().Replace("UnityEngine.", "")}");
            }

            var paths = guids.Select(x => AssetDatabase.GUIDToAssetPath(x)).ToList();

            var oassets = paths
                .Select(path => AssetDatabase.LoadMainAssetAtPath(path))
                .Where(x => x != null)
                .ToList()
                ;

            List<T> assets = new List<T>(oassets.Cast<T>());
            return assets;
        }

        ModularShader _activeShader;
        ModularShader ActiveShader
        {
            get
            {
                return _activeShader;
            }
            set
            {
                if(ActiveShader == value)
                {
                    return;
                }

                _activeShader = value;
            }
        }

        /// <summary>
        /// List of modules to install
        /// </summary>
        List<ShaderModule> ToInstall = new List<ShaderModule>();

        bool AlreadyInstalled
        {
            get
            {
                if (ToInstall == null) return false;

                var modules = ActiveShader.BaseModules;

                var hash = new HashSet<string>(ToInstall.Select(x=>x?.Name).Where(x=>x!=null));

                return modules.Any(x => hash.Contains(x?.Name));
            }
        }

        bool ValidCopyName
        {
            get
            {
                if (_shaders == null) RefreshShaderList();
                return !_shaders.Any(x => x.Name == CopyName) && ! string.IsNullOrEmpty(CopyName);
            }
        }

        bool ValidNewID
        {
            get
            {
                if (_compiledShaders == null) RefreshShaderList();

                if (NewID.EndsWith("/"))
                {
                    return false;
                }

                Regex alreadyExists = new Regex($@".*{NewID}$");
                return !_compiledShaders.Any(x => alreadyExists.IsMatch(x.name)) && ! string.IsNullOrEmpty(NewID) ;
            }
        }

        Vector2 ScrollPosition { get; set; }

        bool _refreshing;
        private void OnFocus()
        {
            RefreshShaderList();
        }

        void OnGUI()
        {
            GUILayout.ExpandWidth(true);

            using (new GUILayout.HorizontalScope())
            {
                GUILayout.FlexibleSpace();
                EditorGUILayout.LabelField($"By: AngriestSCV");


                var sty = new GUIStyle(GUI.skin.button)
                {
                    padding = new RectOffset()
                    {
                        bottom = 8,
                        top = 8,
                        left = 25,
                        right = 25,
                    },
                    
                };

                if (GUILayout.Button("angriestscv.gumroad.com", sty))
                {
                    Application.OpenURL("https://angriestscv.gumroad.com");
                }
            }

            using(var sv = new GUILayout.ScrollViewScope(ScrollPosition))
            using (new GUILayout.VerticalScope())
            {
                ScrollPosition = sv.scrollPosition;

                if (_shaders == null || _compiledShaders == null)
                {
                    RefreshShaderList();
                }

                EditorGUILayout.LabelField($""); // provides the perfect amount of vertical space
                ActiveShader = (ModularShader) EditorGUILayout.ObjectField("Shader to install to", ActiveShader, typeof(ModularShader), true);

                if (ActiveShader == null)
                {
                    this.ShowNotification(new GUIContent("A Modular Shader target is required"));
                    return;
                }
                else
                {
                    RemoveNotification();
                }

                GUILayout.Space(20);

                CopyName = EditorGUILayout.TextField("Modular Shader Name", CopyName);
                NewID = EditorGUILayout.TextField("Output Shader Name", NewID);

                OutputPath = EditorGUILayout.TextField("Output Path", OutputPath);
                //EditorGUILayout.LabelField("Output Path", OutputPath);
                if (GUILayout.Button("Set Output Path"))
                {
                    string startPath = OutputPath;
                    if (!Directory.Exists(startPath??""))
                    {
                        startPath = "Assets";
                    }
                    OutputPath = EditorUtility.OpenFolderPanel("Output Path Picker", startPath, "");
                    if (OutputPath.StartsWith(Application.dataPath))
                    {
                        OutputPath = "Assets" + OutputPath.Substring(Application.dataPath.Length);
                    }
                    else
                    {
                        OutputPath = "Assets/";
                    }

                    if(!Directory.Exists(OutputPath))
                    {
                        OutputPath = "Assets/";
                    }

                }

                GUILayout.Space(20);
                using (new GUILayout.HorizontalScope())
                {
                    EditorGUILayout.LabelField("Modules to install");

                    if (GUILayout.Button("+"))
                    {
                        ToInstall.Add(null);
                    }
                    else if(GUILayout.Button("-"))
                    {
                        if(ToInstall.Count != 0)
                        {
                            ToInstall.RemoveAt(ToInstall.Count - 1);
                        }
                        else
                        {
                            Debug.Log("You are module-less.");
                        }
                    }
                }

                for(int i=0; i<ToInstall.Count; i++)
                {
                    ToInstall[i] = (ShaderModule) EditorGUILayout.ObjectField("", ToInstall[i], typeof(ShaderModule), true);
                }

                GUILayout.Space(20);
                if (!ValidCopyName || !ValidNewID)
                {
                    if (string.IsNullOrEmpty(CopyName))
                    {
                        GUILayout.Label("The output Modular Shader name is required.");
                    }
                    else if (!ValidCopyName)
                    {
                        GUILayout.Label("The Modular Shader asset already exists.");
                        if (GUILayout.Button("Delete it/them"))
                        {
                            DeleteModularShaderByName(CopyName);
                            RefreshShaderList();
                        }
                    }

                    if (string.IsNullOrEmpty(NewID))
                    {
                        GUILayout.Label("A new shader name is required");
                    }
                    else if (NewID.EndsWith("/"))
                    {
                        GUILayout.Label("'/' is not a valid final character");
                    }
                    else if (!ValidNewID)
                    {
                        GUILayout.Label("The shader name is already used.");

                        if (GUILayout.Button("Delete it"))
                        {
                            DeleteShaderByName(NewID);
                            RefreshShaderList();
                        }
                    }
                    else
                    {
                        GUILayout.Label("Valid NewID");
                    }
                }
                else if(ToInstall.All(x=>x==null))
                {
                    GUILayout.Label("There are no modules to install");
                }
                else if(string.IsNullOrEmpty(OutputPath))
                {
                    GUILayout.Label("An output path is required");
                }
                else if (!Directory.Exists(OutputPath))
                {
                    GUILayout.Label("The given output directory does not exist");
                }
                else 
                {

                    if (AlreadyInstalled)
                    {
                        GUILayout.Label("At least one of the modules to install is already in the active shader. Duplicates will not be reinstalled");
                    }

                    if (GUILayout.Button("Install Modules"))
                    {
                        Install();
                    }
                }
            }
        }

        private void DeleteModularShaderByName(string name)
        {
            AssetDatabase.Refresh();
            string[] guids;
            guids = AssetDatabase.FindAssets($"t:{typeof(ModularShader).ToString().Replace("UnityEngine.", "")}");
            
            IEnumerable<string> paths = guids.Select(x => AssetDatabase.GUIDToAssetPath(x)).ToList();
            var assets = paths
                .Select(path => new { Asset = AssetDatabase.LoadMainAssetAtPath(path) as ModularShader, Path = path })
                .Where(x => x.Asset != null)
                .Where(x => x.Asset.Name == name)
                .ToList();

            var toDel = assets
                .Select(x => x.Path)
                .ToList()
                ;

            foreach(string path in toDel)
            {
                AssetDatabase.DeleteAsset(path);
            }
        }

        private void DeleteShaderByName(string name)
        {
            AssetDatabase.Refresh();
            string[] guids;
            guids = AssetDatabase.FindAssets($"t:{typeof(Shader).ToString().Replace("UnityEngine.", "")}");
            
            IEnumerable<string> paths = guids.Select(x => AssetDatabase.GUIDToAssetPath(x)).ToList();
            var assets = paths
                .Select(path => new { Asset = AssetDatabase.LoadMainAssetAtPath(path) as Shader, Path = path })
                .Where(x => x.Asset != null)
                .ToList();

            Regex re = new Regex($@".*{name}$");

            string delAsset = assets
                .Where(x => re.IsMatch(x.Asset.name))
                .Select(x => x.Path)
                .FirstOrDefault()
                ;

            AssetDatabase.DeleteAsset(delAsset);
        }

        private void Install()
        {
            RefreshShaderList();
            if(!ValidCopyName || !ValidNewID)
            {
                return;
            }

            ModularShader shader = new ModularShader()
            {
                AdditionalModules = ActiveShader.AdditionalModules.ToList(),
                AdditionalSerializedData = ActiveShader.AdditionalSerializedData,
                Author = ActiveShader.Author,
                BaseModules = ActiveShader.BaseModules.Where(x=>x!=null).ToList(),
                CustomEditor = ActiveShader.CustomEditor,
                Description = ActiveShader.Description,
                hideFlags = ActiveShader.hideFlags,
                Name = CopyName,
                Properties = ActiveShader.Properties.ToList(),
                ShaderTemplate = ActiveShader.ShaderTemplate,
                ShaderPropertiesTemplate = ActiveShader.ShaderPropertiesTemplate,
                UseTemplatesForProperties = ActiveShader.UseTemplatesForProperties,
                Version = ActiveShader.Version,
                Id = ActiveShader.Id + "AudioBump",
                ShaderPath = NewID,
                LockBaseModules = ActiveShader.LockBaseModules,
                LastGeneratedShaders = new List<Shader>(),
            };

            string savePath = $"{OutputPath}/{CopyName}.asset";
            Debug.Log($"Creating new asset at [{savePath}]");
            AssetDatabase.CreateAsset(shader, savePath);

            var installedNames = new HashSet<string>(shader.BaseModules.Select(x => x.Name).Where(x => x != null));

            foreach(ShaderModule mod in ToInstall)
            {
                if (mod == null) continue;
                if (installedNames.Contains(mod.Name)) continue;

                installedNames.Add(mod.Name);
                shader.BaseModules.Add(mod);
            }

            AssetDatabase.SaveAssets();

            RefreshShaderList();
        }

        ShaderModule FindModuleById(string name, string searchPath)
        {
            var modules = FindAssetsByType<ShaderModule>(searchPath);

            return modules.FirstOrDefault(x => x?.Id == name);
        }
        

        private static void RefreshShaderList()
        {
            _shaders = FindAssetsByType<ModularShader>().ToList();
            _compiledShaders = FindAssetsByType<Shader>().ToList();
        }

    }
}
#endif