Allow getting sponsors options for non-incremental generator

devlooped · Dec 3, 2024 · 3f72a9f · 3f72a9f
1 parent fb82cf3
commit 3f72a9f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/samples/dotnet/SponsorLink/SponsorLink.cs b/samples/dotnet/SponsorLink/SponsorLink.cs
@@ -107,6 +107,15 @@ public static IncrementalValueProvider<StatusOptions> GetStatusOptions(this Incr
         => context.GetSponsorAdditionalFiles().Combine(context.AnalyzerConfigOptionsProvider)
             .Select((source, _) => new StatusOptions(source.Left, source.Right.GlobalOptions));
 
+    /// <summary>
+    /// Gets the status options for use within a source generator, to avoid depending on 
+    /// analyzer runs. Used in combination with <see cref="DiagnosticsManager.GetOrSetStatus(StatusOptions)"/>.
+    /// </summary>
+    public static StatusOptions GetStatusOptions(this GeneratorExecutionContext context)
+        => new StatusOptions(
+            context.AdditionalFiles.Where(x => x.IsSponsorManifest(context.AnalyzerConfigOptions) || x.IsSponsorableAnalyzer(context.AnalyzerConfigOptions)).ToImmutableArray(),
+            context.AnalyzerConfigOptions.GlobalOptions);
+
     static bool IsSponsorManifest(this AdditionalText text, AnalyzerConfigOptionsProvider provider)
         => provider.GetOptions(text).TryGetValue("build_metadata.SponsorManifest.ItemType", out var itemType) &&
            itemType == "SponsorManifest" &&