diff --git a/src/Generators/GlobalRegistryGenerator.cs b/src/Generators/GlobalRegistryGenerator.cs index 239be67..dc5ec0e 100644 --- a/src/Generators/GlobalRegistryGenerator.cs +++ b/src/Generators/GlobalRegistryGenerator.cs @@ -20,6 +20,7 @@ public class GlobalRegistryGenerator : ISourceGenerator .AppendLine("using System.Reflection;") .AppendLine("using Polonium.Attributes;") .AppendLine("using Polonium;") + .AppendLine("using Polonium.Interfaces;") .AppendLine("public static partial class GlobalRegistry") .AppendLine("{") .AppendLine(" public static void Start()") @@ -50,6 +51,8 @@ public class GlobalRegistryGenerator : ISourceGenerator .AppendLine(" }") .AppendLine(" }") .AppendLine(" public static PoloniumRegistry PoloniumRegistry => PoloniumRegistry.Instance;") + .AppendLine(" public static bool Paused { get; set; }") + .AppendLine(" public static HashSet TimeConsumers { get; } = new ();") .AppendLine("}"); context.AddSource("GlobalRegistry.g.cs", sb.ToString()); }