Generate enum from file instead of string

zompinc · Dec 21, 2023 · 33ba4d0 · 33ba4d0
1 parent e3de5d6
commit 33ba4d0
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 22 deletions.
diff --git a/src/Zomp.SyncMethodGenerator/CollectionTypes.cs b/src/Zomp.SyncMethodGenerator/CollectionTypes.cs
@@ -1,28 +1,29 @@
-namespace Zomp.SyncMethodGenerator;
-
-/// <summary>
-/// All types that an IAsyncEnumerable can be converted into.
-/// </summary>
-[Flags]
-public enum CollectionTypes
+namespace Zomp.SyncMethodGenerator
 {
     /// <summary>
-    /// Type for System.Collections.Generic.IEnumerable .
+    /// All types that an IAsyncEnumerable can be converted into.
     /// </summary>
-    IEnumerable = 1,
+    [Flags]
+    public enum CollectionTypes
+    {
+        /// <summary>
+        /// Type for System.Collections.Generic.IEnumerable .
+        /// </summary>
+        IEnumerable = 1,
 
-    /// <summary>
-    /// Type for System.Collections.Generic.IList .
-    /// </summary>
-    IList = 2,
+        /// <summary>
+        /// Type for System.Collections.Generic.IList .
+        /// </summary>
+        IList = 2,
 
-    /// <summary>
-    /// Type for System.ReadOnlySpan .
-    /// </summary>
-    ReadOnlySpan = 4,
+        /// <summary>
+        /// Type for System.ReadOnlySpan .
+        /// </summary>
+        ReadOnlySpan = 4,
 
-    /// <summary>
-    /// Type for System.Span .
-    /// </summary>
-    Span = 8,
+        /// <summary>
+        /// Type for System.Span .
+        /// </summary>
+        Span = 8,
+    }
 }
diff --git a/src/Zomp.SyncMethodGenerator/SyncMethodSourceGenerator.cs b/src/Zomp.SyncMethodGenerator/SyncMethodSourceGenerator.cs
@@ -28,6 +28,10 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
             Debugger.Launch();
         }
 #endif
+
+        context.RegisterPostInitializationOutput(ctx => ctx.AddSource(
+            $"CollectionTypes.g.cs", SourceText.From(GetResourceText("Zomp.SyncMethodGenerator.tests.CollectionTypes.cs"), Encoding.UTF8)));
+
         context.RegisterPostInitializationOutput(ctx => ctx.AddSource(
             $"{CreateSyncVersionAttribute}.g.cs", SourceText.From(SourceGenerationHelper.CreateSyncVersionAttributeSource, Encoding.UTF8)));
 
@@ -249,4 +253,14 @@ private static List<MethodToGenerate> GetTypesToGenerate(SourceProductionContext
 
         return methodsToGenerate;
     }
+
+    private string GetResourceText(string name)
+    {
+        using var stream = GetType().Assembly.GetManifestResourceStream(name);
+        using var streamReader = new StreamReader(stream);
+        return $"""
+                // <auto-generated/>
+                {streamReader.ReadToEnd()}
+                """;
+    }
 }
diff --git a/src/Zomp.SyncMethodGenerator/Zomp.SyncMethodGenerator.csproj b/src/Zomp.SyncMethodGenerator/Zomp.SyncMethodGenerator.csproj
@@ -24,6 +24,12 @@
     <None Include="$(OutputPath)\$(AssemblyName).dll" Pack="true" PackagePath="analyzers/dotnet/cs" Visible="false" />
     <None Include="..\..\README.md" Pack="true" PackagePath="\" />
   </ItemGroup>
+
+  <ItemGroup>
+    <EmbeddedResource Include="..\..\src\Zomp.SyncMethodGenerator\CollectionTypes.cs" Link="CollectionTypes.cs">
+      <LogicalName>Zomp.SyncMethodGenerator.tests.CollectionTypes.cs</LogicalName>
+    </EmbeddedResource>
+  </ItemGroup>
 
   <!-- https://stackoverflow.com/a/59893520/6461844 -->
   <Target Name="CopyProjectReferencesToPackage" DependsOnTargets="BuildOnlySettings;ResolveReferences">

diff --git a/tests/Generator.Tests/TestHelper.cs b/tests/Generator.Tests/TestHelper.cs
@@ -126,7 +126,7 @@ partial class Class
         var target = new RunResultWithIgnoreList
         {
             Result = driver.GetRunResult(),
-            IgnoredFiles = { $"{SyncMethodSourceGenerator.CreateSyncVersionAttribute}.g.cs", $"{SyncMethodSourceGenerator.ReplaceWithAttribute}.g.cs" },
+            IgnoredFiles = { $"{SyncMethodSourceGenerator.CreateSyncVersionAttribute}.g.cs", $"{SyncMethodSourceGenerator.ReplaceWithAttribute}.g.cs", $"CollectionTypes.g.cs" },
         };
 
         var verifier = Verifier