fix: re-set global usings on reset

rdavisau · Feb 5, 2023 · 6095abc · 6095abc
1 parent 10ccc7c
commit 6095abc
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/components/tbc.host/Components/IncrementalCompiler/IncrementalCompiler.cs b/src/components/tbc.host/Components/IncrementalCompiler/IncrementalCompiler.cs
@@ -98,6 +98,11 @@ IEnumerable<ICompilationFixer> fixers
             GlobalUsingResolution =
                 _globalUsingsResolver.ResolveGlobalUsings(new ResolveGlobalUsingsRequest(options.GlobalUsingsSources)).Result;
 
+            SetGlobalUsings();
+        }
+
+        private void SetGlobalUsings()
+        {
             if (GlobalUsingResolution.Usings.Any() && GlobalUsingResolution.UsingsSource is { } gus)
                 CurrentCompilation = CurrentCompilation.AddSyntaxTrees(
                     CSharpSyntaxTree.ParseText(
@@ -108,7 +113,6 @@ IEnumerable<ICompilationFixer> fixers
                            .WithPreprocessorSymbols(_options.PreprocessorSymbols.ToArray()),
                         path: "",
                         Encoding.Default));
-
         }
 
         private ImmutableDictionary<SourceGeneratorReference, ResolveSourceGeneratorsResponse> ResolveSourceGenerators()
@@ -360,15 +364,19 @@ private T WithCompilation<T>(Func<CSharpCompilation, (CSharpCompilation, T)> act
                 return c.RemoveAllReferences();
             });
 
-        public void ClearTrees() =>
+        public void ClearTrees()
+        {
             WithCompilation(c =>
             {
                 Logger.LogInformation("Clearing syntax trees");
-                
+
                 RawTrees.Clear();
                 return c.RemoveAllSyntaxTrees();
             });
-
+
+            SetGlobalUsings();
+        }
+
         public void PrintTrees(bool withDetail)
         {
             if (_rootPath is null)