fix(weaver): fix #796, reload assemblies after initial import (#1106)

* fix: fix #796, reload assemblies after initial import * Do not weave assemblies that aren't compiled yet * Update CompilationFinishedHook.cs * Update CompilationFinishedHook.cs * Use SessionState instead of env variable
MirageNet · Sep 24, 2019 · d91b387 · d91b387
1 parent 747fba0
commit d91b387
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/Assets/Mirror/Editor/Weaver/CompilationFinishedHook.cs b/Assets/Mirror/Editor/Weaver/CompilationFinishedHook.cs
@@ -47,6 +47,27 @@ static void HandleError(string msg)
         static void OnInitializeOnLoad()
         {
             CompilationPipeline.assemblyCompilationFinished += OnCompilationFinished;
+
+            // We only need to run this once per session
+            // after that, all assemblies will be weaved by the event
+            if (!SessionState.GetBool("MIRROR_WEAVED", false) )
+            {
+                SessionState.SetBool("MIRROR_WEAVED", true);
+                WeaveExisingAssemblies();
+            }
+        }
+
+        static void WeaveExisingAssemblies()
+        {
+            foreach (UnityAssembly assembly in CompilationPipeline.GetAssemblies())
+            {
+                if (File.Exists(assembly.outputPath))
+                {
+                    OnCompilationFinished(assembly.outputPath, new CompilerMessage[0]);
+                }
+            }
+
+            UnityEditorInternal.InternalEditorUtility.RequestScriptReload();
         }
 
         static string FindMirrorRuntime()