Remove the library and move attributes into generator.

And a few other changes: - Make dumping the tree optional as it's not necessary; - Remove dependency on external dependencies for the generator.
GGG-KILLER · Jan 29, 2024 · cf98c70 · cf98c70
1 parent 677a95a
commit cf98c70
Show file tree

Hide file tree

Showing 30 changed files with 565 additions and 482 deletions.
diff --git a/Tsu.Trees.RedGreen/Tsu.Trees.RedGreen.sln b/Tsu.Trees.RedGreen/Tsu.Trees.RedGreen.sln
@@ -5,9 +5,7 @@ VisualStudioVersion = 17.0.31903.59
 MinimumVisualStudioVersion = 10.0.40219.1
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Tsu.Trees.RedGreen.Sample", "sample\Tsu.Trees.RedGreen.Sample.csproj", "{D00F090D-145E-462E-BE75-F2805ADB572A}"
 EndProject
-Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Tsu.Trees.RedGreen.SourceGenerator", "sourcegen\Tsu.Trees.RedGreen.SourceGenerator.csproj", "{A0A6DBE6-9899-42C6-A3EC-90153F844109}"
-EndProject
-Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Tsu.Trees.RedGreen", "src\Tsu.Trees.RedGreen.csproj", "{ABE13963-6F78-47DF-A652-303BD6565A93}"
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Tsu.Trees.RedGreen", "src\Tsu.Trees.RedGreen.csproj", "{A0A6DBE6-9899-42C6-A3EC-90153F844109}"
 EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
@@ -26,9 +24,5 @@ Global
 		{A0A6DBE6-9899-42C6-A3EC-90153F844109}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{A0A6DBE6-9899-42C6-A3EC-90153F844109}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{A0A6DBE6-9899-42C6-A3EC-90153F844109}.Release|Any CPU.Build.0 = Release|Any CPU
-		{ABE13963-6F78-47DF-A652-303BD6565A93}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
-		{ABE13963-6F78-47DF-A652-303BD6565A93}.Debug|Any CPU.Build.0 = Debug|Any CPU
-		{ABE13963-6F78-47DF-A652-303BD6565A93}.Release|Any CPU.ActiveCfg = Release|Any CPU
-		{ABE13963-6F78-47DF-A652-303BD6565A93}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 EndGlobal
diff --git a/...edGreen.SourceGenerator/Tsu.Trees.RedGreen.SourceGenerator.Generator/GreenNode.Debug.g.cs b/...edGreen.SourceGenerator/Tsu.Trees.RedGreen.SourceGenerator.Generator/GreenNode.Debug.g.cs
diff --git a/...es.RedGreen.SourceGenerator/Tsu.Trees.RedGreen.SourceGenerator.Generator/Initialized.g.cs b/...es.RedGreen.SourceGenerator/Tsu.Trees.RedGreen.SourceGenerator.Generator/Initialized.g.cs
diff --git a/Tsu.Trees.RedGreen/src/GreenNodeAttribute.cs → ...nerator.Generator/GreenNodeAttribute.g.cs b/Tsu.Trees.RedGreen/src/GreenNodeAttribute.cs → ...nerator.Generator/GreenNodeAttribute.g.cs
@@ -1,4 +1,4 @@
-namespace Tsu.Trees.RedGreen;
+namespace Tsu.Trees.RedGreen;
 
 /// <summary>
 /// Marks this class as a green node.
@@ -8,7 +8,7 @@ namespace Tsu.Trees.RedGreen;
 /// This should be the value on the enum (e.g.: SyntaxKind.ClassDeclarationSyntax)
 /// </param>
 [AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
-public sealed class GreenNodeAttribute(params object[] kinds) : Attribute
+internal sealed class GreenNodeAttribute(params object[] kinds) : Attribute
 {
     /// <summary>
     /// This node's Kind.

diff --git a/...es.RedGreen/src/GreenTreeRootAttribute.cs → ...tor.Generator/GreenTreeRootAttribute.g.cs b/...es.RedGreen/src/GreenTreeRootAttribute.cs → ...tor.Generator/GreenTreeRootAttribute.g.cs
@@ -1,10 +1,10 @@
-namespace Tsu.Trees.RedGreen;
+namespace Tsu.Trees.RedGreen;
 
 /// <summary>
 /// An attribute that marks the given class as the base class for all nodes in a green node tree.
 /// </summary>
 [AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
-public sealed class GreenTreeRootAttribute(Type redBase, string suffix, Type kindEnum) : Attribute
+internal sealed class GreenTreeRootAttribute(Type redBase, string suffix, Type kindEnum) : Attribute
 {
     /// <summary>
     /// The suffix for nodes in this tree.
@@ -36,14 +36,8 @@ public sealed class GreenTreeRootAttribute(Type redBase, string suffix, Type kin
     /// </summary>
     public bool CreateRewriter { get; set; }
 
-    // NOTE: Make cache configurable?
-    // /// <summary>
-    // /// Whether to enable the green cache or not.
-    // /// </summary>
-    // public bool CacheEnabled { get; set; } = true;
-
-    // /// <summary>
-    // /// The amount of bits from the hash to use for caching.
-    // /// </summary>
-    // public int CacheSizeBits { get; set; } = 16;
+    /// <summary>
+    /// Whether to generate a file with only comments dumping the entire tree structure.
+    /// </summary>
+    public bool DebugDump { get; set; }
 }
diff --git a/...eGenerator.Generator/Sample.Internal.g.cs → ...eGenerator.Generator/Sample.Internal.g.cs b/...eGenerator.Generator/Sample.Internal.g.cs → ...eGenerator.Generator/Sample.Internal.g.cs
@@ -8,7 +8,7 @@
 
 namespace Tsu.Trees.RedGreen.Sample.Internal
 {
-    abstract partial class GreenNode : global::Tsu.Trees.RedGreen.Internal.IGreenNode<global::Tsu.Trees.RedGreen.Sample.Internal.GreenNode, global::Tsu.Trees.RedGreen.Sample.SampleNode, global::Tsu.Trees.RedGreen.Sample.SampleKind>
+    abstract partial class GreenNode
     {
         private readonly global::Tsu.Trees.RedGreen.Sample.SampleKind _kind;
         private byte _slotCount;

diff --git a/...ourceGenerator.Generator/Sample.Main.g.cs → ...ourceGenerator.Generator/Sample.Main.g.cs b/...ourceGenerator.Generator/Sample.Main.g.cs → ...ourceGenerator.Generator/Sample.Main.g.cs
diff --git a/...SourceGenerator.Generator/Sample.Red.g.cs → ...SourceGenerator.Generator/Sample.Red.g.cs b/...SourceGenerator.Generator/Sample.Red.g.cs → ...SourceGenerator.Generator/Sample.Red.g.cs
@@ -8,7 +8,7 @@
 
 namespace Tsu.Trees.RedGreen.Sample
 {
-    abstract partial class SampleNode : global::Tsu.Trees.RedGreen.IRedNode<global::Tsu.Trees.RedGreen.Sample.SampleNode, global::Tsu.Trees.RedGreen.Sample.SampleKind>
+    abstract partial class SampleNode
     {
         private readonly global::Tsu.Trees.RedGreen.Sample.SampleNode? _parent;
 

diff --git a/Tsu.Trees.RedGreen/sample/Tsu.Trees.RedGreen.Sample.csproj b/Tsu.Trees.RedGreen/sample/Tsu.Trees.RedGreen.Sample.csproj
@@ -20,8 +20,7 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <ProjectReference Include="$(RepositoryRootDirectory)Tsu.Trees.RedGreen/src/Tsu.Trees.RedGreen.csproj" />
-    <ProjectReference Include="$(RepositoryRootDirectory)Tsu.Trees.RedGreen/sourcegen/Tsu.Trees.RedGreen.SourceGenerator.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" />
+    <ProjectReference Include="$(RepositoryRootDirectory)Tsu.Trees.RedGreen/src/Tsu.Trees.RedGreen.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" />
   </ItemGroup>
 
   <ItemGroup>

diff --git a/Tsu.Trees.RedGreen/sourcegen/Tsu.Trees.RedGreen.SourceGenerator.csproj b/Tsu.Trees.RedGreen/sourcegen/Tsu.Trees.RedGreen.SourceGenerator.csproj
diff --git a/Tsu.Trees.RedGreen/src/Code.cs b/Tsu.Trees.RedGreen/src/Code.cs
@@ -0,0 +1,74 @@
+namespace Tsu.Trees.RedGreen.SourceGenerator;
+
+internal static class Code
+{
+    public const string GreenNodeAttributeName = "Tsu.Trees.RedGreen.GreenNodeAttribute";
+
+    public const string GreenNodeAttributeCode = """
+    namespace Tsu.Trees.RedGreen;
+
+    /// <summary>
+    /// Marks this class as a green node.
+    /// </summary>
+    /// <param name="kinds">
+    /// The kinds this node could have.
+    /// This should be the value on the enum (e.g.: SyntaxKind.ClassDeclarationSyntax)
+    /// </param>
+    [AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
+    internal sealed class GreenNodeAttribute(params object[] kinds) : Attribute
+    {
+        /// <summary>
+        /// This node's Kind.
+        /// </summary>
+        public object[] Kinds { get; } = kinds;
+    }
+    """;
+
+    public const string GreenTreeRootAttributeName = "Tsu.Trees.RedGreen.GreenTreeRootAttribute";
+
+    public const string GreenTreeRootAttributeCode = """
+    namespace Tsu.Trees.RedGreen;
+
+    /// <summary>
+    /// An attribute that marks the given class as the base class for all nodes in a green node tree.
+    /// </summary>
+    [AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
+    internal sealed class GreenTreeRootAttribute(Type redBase, string suffix, Type kindEnum) : Attribute
+    {
+        /// <summary>
+        /// The suffix for nodes in this tree.
+        /// </summary>
+        public string Suffix { get; } = suffix;
+
+        /// <summary>
+        /// The base node type for all nodes in the red tree.
+        /// </summary>
+        public Type RedBase { get; } = redBase;
+
+        /// <summary>
+        /// The enum type that contains the definitions for the node kinds.
+        /// </summary>
+        public Type KindEnum { get; } = kindEnum;
+
+        /// <summary>
+        /// Whether to create base visitor implementations for this tree.
+        /// </summary>
+        public bool CreateVisitors { get; set; }
+
+        /// <summary>
+        /// Whether to create a base walker implementation for this tree.
+        /// </summary>
+        public bool CreateWalker { get; set; }
+
+        /// <summary>
+        /// Whether to generate a base rewriter implementation for this tree.
+        /// </summary>
+        public bool CreateRewriter { get; set; }
+
+        /// <summary>
+        /// Whether to generate a file with only comments dumping the entire tree structure.
+        /// </summary>
+        public bool DebugDump { get; set; }
+    }
+    """;
+}
diff --git a/Tsu.Trees.RedGreen/sourcegen/Generator.cs → Tsu.Trees.RedGreen/src/Generator.cs b/Tsu.Trees.RedGreen/sourcegen/Generator.cs → Tsu.Trees.RedGreen/src/Generator.cs
@@ -34,11 +34,15 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
 
         context.RegisterPostInitializationOutput(ctx =>
         {
-            ctx.AddSource("Initialized.g.cs", "// <auto-generated />");
+            ctx.AddSource("GreenNodeAttribute.g.cs", Code.GreenNodeAttributeCode);
+            ctx.AddSource("GreenTreeRootAttribute.g.cs", Code.GreenTreeRootAttributeCode);
         });
 
         context.RegisterSourceOutput(trees, (ctx, tree) =>
         {
+            if (!tree.DebugDump)
+                return;
+
             var builder = new StringBuilder();
 
             builder.AppendLine($"// GreenBase = {tree.GreenBase.ToCSharpString()}");
@@ -70,7 +74,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                     queue.Push((node.Item1 + 1, derived));
             }
 
-            ctx.AddSource($"{tree.GreenBase.Name}.Debug.g.cs", builder.ToSourceText());
+            ctx.AddSource($"{tree.Suffix}.Debug.g.cs", builder.ToSourceText());
         });
 
         context.RegisterGreenOutput(trees);

diff --git a/....RedGreen/sourcegen/GreenTreeGenerator.cs → Tsu.Trees.RedGreen/src/GreenTreeGenerator.cs b/....RedGreen/sourcegen/GreenTreeGenerator.cs → Tsu.Trees.RedGreen/src/GreenTreeGenerator.cs
@@ -83,11 +83,7 @@ public static void RegisterGreenOutput(this IncrementalGeneratorInitializationCo
 
     private static void WriteGreenRoot(this IndentedTextWriter writer, Tree tree, Node root)
     {
-        writer.WriteLine("abstract partial class {0} : global::Tsu.Trees.RedGreen.Internal.IGreenNode<{1}, {2}, {3}>",
-            root.TypeSymbol.Name,
-            tree.GreenBase.ToCSharpString(false),
-            tree.RedBase.ToCSharpString(false),
-            tree.KindEnum.ToCSharpString(false));
+        writer.WriteLine("abstract partial class {0}", root.TypeSymbol.Name);
         writer.WriteLine('{');
         writer.Indent++;
         {