From 03ce9899ce35b5be6024fdc221642dc00febc04d Mon Sep 17 00:00:00 2001
From: poi-vrc <77053052+poi-vrc@users.noreply.github.com>
Date: Tue, 23 Apr 2024 12:06:09 +0800
Subject: [PATCH] feat(ma/ndmf): inconsistency script recompiler

---
 Editor/Detail/NDMF/ScriptRecompiler.cs      | 116 ++++++++++++++++++++
 Editor/Detail/NDMF/ScriptRecompiler.cs.meta |  11 ++
 2 files changed, 127 insertions(+)
 create mode 100644 Editor/Detail/NDMF/ScriptRecompiler.cs
 create mode 100644 Editor/Detail/NDMF/ScriptRecompiler.cs.meta

diff --git a/Editor/Detail/NDMF/ScriptRecompiler.cs b/Editor/Detail/NDMF/ScriptRecompiler.cs
new file mode 100644
index 0000000..c8a25e4
--- /dev/null
+++ b/Editor/Detail/NDMF/ScriptRecompiler.cs
@@ -0,0 +1,116 @@
+/*
+ * Copyright (c) 2024 chocopoi
+ * 
+ * This file is part of DressingFramework.
+ * 
+ * DressingFramework is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
+ * 
+ * DressingFramework is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License along with DressingFramework. If not, see <https://www.gnu.org/licenses/>.
+ */
+#if UNITY_EDITOR
+using System;
+using System.IO;
+using System.Linq;
+using UnityEditor;
+using UnityEditor.Compilation;
+using UnityEditor.PackageManager;
+using UnityEditor.PackageManager.Requests;
+using UnityEngine;
+
+namespace Chocopoi.DressingFramework.Detail.NDMF
+{
+    [InitializeOnLoad]
+    public static class ScriptRecompiler
+    {
+        private const string NDMFPackageName = "nadena.dev.ndmf";
+        private const string MAPackageName = "nadena.dev.modular-avatar";
+        private const string TempFileName = "Temp/com.chocopoi.vrc.dressingframework.ScriptRecompiler.lastRecompileTime.txt";
+
+        private static ListRequest s_listReq = null;
+
+        static ScriptRecompiler()
+        {
+            s_listReq = Client.List();
+            EditorApplication.update += OnEditorUpdate;
+        }
+
+        private static bool CheckNDMFInconsistency(PackageCollection pkgs)
+        {
+#if DK_NDMF
+            return !pkgs.Any(p => p.name == NDMFPackageName);
+#else
+            return pkgs.Any(p => p.name == NDMFPackageName);
+#endif
+        }
+
+        private static bool CheckMAInconsistency(PackageCollection pkgs)
+        {
+#if DK_MA
+            return !pkgs.Any(p => p.name == MAPackageName);
+#else
+            return pkgs.Any(p => p.name == MAPackageName);
+#endif
+        }
+
+        private static void WriteLastRecompileTime()
+        {
+            var writer = new StreamWriter(TempFileName);
+            writer.WriteLine(DateTime.Now.ToString());
+            writer.Close();
+        }
+
+        private static bool IsLastRecompileMinuteAgo()
+        {
+            if (!File.Exists(TempFileName))
+            {
+                return false;
+            }
+
+            var reader = new StreamReader(TempFileName);
+            var dateTimeLine = reader.ReadLine();
+            reader.Close();
+            try
+            {
+                var dateTime = DateTime.Parse(dateTimeLine);
+                if ((DateTime.Now - dateTime).TotalMinutes > 1)
+                {
+                    return true;
+                }
+            }
+            catch
+            {
+                Debug.Log("Unable to parse data time from last recompile time file.");
+                File.Delete(TempFileName);
+                return false;
+            }
+            return false;
+        }
+
+        private static void OnEditorUpdate()
+        {
+            if (!s_listReq.IsCompleted)
+            {
+                return;
+            }
+
+            if (CheckNDMFInconsistency(s_listReq.Result) || CheckMAInconsistency(s_listReq.Result))
+            {
+                if (IsLastRecompileMinuteAgo())
+                {
+                    Debug.LogError("[DressingFramework] Version defines and package presence inconsistency detected, but last recompile was just a minute ago. Probably there is a compilation error? Recompilation is not requested to prevent endless loop.");
+                }
+                else
+                {
+                    Debug.Log("[DressingFramework] Version defines and package presence inconsistency detected, requesting recompilation");
+                    WriteLastRecompileTime();
+                    CompilationPipeline.RequestScriptCompilation();
+                }
+            }
+
+            EditorApplication.update -= OnEditorUpdate;
+        }
+    }
+}
+#endif
diff --git a/Editor/Detail/NDMF/ScriptRecompiler.cs.meta b/Editor/Detail/NDMF/ScriptRecompiler.cs.meta
new file mode 100644
index 0000000..3a78b3f
--- /dev/null
+++ b/Editor/Detail/NDMF/ScriptRecompiler.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 75cd7586c92b5b840bf32e2e3202a123
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: