[Generator] Wrapper for Strings Constructed during Code Generation

src/bgen/CodeBlocks/BlockContainer.cs

@@ -0,0 +1,11 @@
+using System.IO;
+
+public class BlockContainer : CodeBlock {
+	public override void Print (TextWriter writer)
+	{
+		foreach (ICodeBlock block in Blocks) {
+			block.SetIndent (CurrentIndent);
+			block.Print (writer);
+		}
+	}
+}

src/bgen/CodeBlocks/CodeBlock.cs

@@ -0,0 +1,108 @@
+using System.Collections;
+using System.Collections.Generic;
+using System.IO;
+
+public class CodeBlock : ICodeBlock, IEnumerable<ICodeBlock> {
+	protected int CurrentIndent = 0;
+	protected int Indent = 4;
+	protected string HeaderText = string.Empty;
+	protected List<ICodeBlock> Blocks = new ();
+	protected const char StartBrace = '{';
+	protected const char EndBrace = '}';
+	protected const char NewLine = '\n';
+
+	public CodeBlock () { }
+
+	public CodeBlock (string text)
+	{
+		HeaderText = text;
+	}
+
+	public CodeBlock (string text, IList<ICodeBlock> blocks)
+	{
+		Blocks.AddRange (blocks);
+		HeaderText = text;
+	}
+
+	public CodeBlock (IList<ICodeBlock> blocks)
+	{
+		Blocks.AddRange (blocks);
+	}
+
+	public CodeBlock (string headerText, params string [] lines)
+	{
+		HeaderText = headerText;
+		foreach (string line in lines) {
+			Blocks.Add (new LineBlock (line));
+		}
+	}
+
+	public CodeBlock (params string [] lines)
+	{
+		foreach (string line in lines) {
+			Blocks.Add (new LineBlock (line));
+		}
+	}
+
+	public void Add (ICodeBlock block)
+	{
+		block.SetIndent (CurrentIndent + Indent);
+		Blocks.Add (block);
+	}
+
+	public void Add (string text)
+	{
+		LineBlock line = new LineBlock (text);
+		line.SetIndent (CurrentIndent + Indent);
+		Blocks.Add (line);
+	}
+
+	public void AddLine (string text)
+	{
+		LineBlock line = new LineBlock (text);
+		line.SetIndent (CurrentIndent + Indent);
+		Blocks.Add (line);
+	}
+
+	protected void WriteIndent (TextWriter writer)
+	{
+		for (var i = 0; i < CurrentIndent; i++)
+			writer.Write (' ');
+	}
+
+	protected virtual void WriteHeaderText (TextWriter writer)
+	{
+		WriteIndent (writer);
+		writer.Write (HeaderText);
+		writer.Write (NewLine);
+	}
+
+	public virtual void Print (TextWriter writer)
+	{
+		if (HeaderText != string.Empty)
+			WriteHeaderText (writer);
+
+		WriteIndent (writer);
+		writer.Write (StartBrace);
+		writer.Write (NewLine);
+
+		SetIndent (CurrentIndent + Indent);
+		foreach (ICodeBlock block in Blocks) {
+			block.SetIndent (CurrentIndent);
+			block.Print (writer);
+		}
+		SetIndent (CurrentIndent - Indent);
+
+		WriteIndent (writer);
+		writer.Write (EndBrace);
+		writer.Write (NewLine);
+	}
+
+	public void SetIndent (int indent)
+	{
+		CurrentIndent = indent;
+	}
+
+	public IEnumerator<ICodeBlock> GetEnumerator () => Blocks.GetEnumerator ();
+	IEnumerator IEnumerable.GetEnumerator () => GetEnumerator ();
+}

src/bgen/CodeBlocks/ICodeBlock.cs

@@ -0,0 +1,7 @@
+using System.IO;
+
+public interface ICodeBlock {
+	public void Print (TextWriter writer);
+	public void SetIndent (int indent);
+}
+

src/bgen/CodeBlocks/IfBlock.cs

@@ -0,0 +1,102 @@
+using System.Collections.Generic;
+using System.IO;
+
+public class IfBlock : CodeBlock {
+	List<IfBlock> ElseIfBlocks = new ();
+	CodeBlock? ElseBlock = null;
+	public IfBlock (string condition)
+	{
+		HeaderText = condition;
+	}
+
+	public IfBlock (string condition, List<ICodeBlock> blocks)
+	{
+		HeaderText = condition;
+		Blocks.AddRange (blocks);
+	}
+
+	public IfBlock (string condition, params string [] lines)
+	{
+		HeaderText = condition;
+		Blocks.AddRange (new CodeBlock (lines));
+	}
+
+	public IfBlock AddElseIf (string condition, List<ICodeBlock> blocks)
+	{
+		ElseIfBlocks.Add (new IfBlock (condition, blocks));
+		return this;
+	}
+
+	public IfBlock AddElseIf (string condition, params string [] lines)
+	{
+		ElseIfBlocks.Add (new IfBlock (condition, lines));
+		return this;
+	}
+
+	public IfBlock AddElse (List<ICodeBlock> blocks)
+	{
+		ElseBlock = new CodeBlock ("else", blocks);
+		return this;
+	}
+
+	public IfBlock AddElse (params string [] lines)
+	{
+		ElseBlock = new CodeBlock ("else", lines);
+		return this;
+	}
+
+	protected override void WriteHeaderText (TextWriter writer)
+	{
+		WriteIndent (writer);
+		writer.Write ("if (");
+		writer.Write (HeaderText);
+		writer.Write (')');
+		writer.Write (NewLine);
+	}
+
+	protected void WriteElseHeaderText (TextWriter writer)
+	{
+		WriteIndent (writer);
+		writer.Write ("else if (");
+		writer.Write (HeaderText);
+		writer.Write (')');
+		writer.Write (NewLine);
+	}
+
+	public override void Print (TextWriter writer)
+	{
+		if (HeaderText != string.Empty)
+			WriteHeaderText (writer);
+
+		WriteIndent (writer);
+		writer.Write (StartBrace);
+		writer.Write (NewLine);
+
+		SetIndent (CurrentIndent + Indent);
+		foreach (ICodeBlock block in Blocks) {
+			block.SetIndent (CurrentIndent);
+			block.Print (writer);
+		}
+		SetIndent (CurrentIndent - Indent);
+		WriteIndent (writer);
+		writer.Write (EndBrace);
+		writer.Write (NewLine);
+
+		foreach (IfBlock block in ElseIfBlocks) {
+			block.SetIndent (CurrentIndent);
+			block.WriteElseHeaderText (writer);
+			writer.Write (StartBrace);
+			writer.Write (NewLine);
+			foreach (ICodeBlock b in block.Blocks) {
+				b.SetIndent (CurrentIndent + Indent);
+				b.Print (writer);
+
+			}
+			writer.Write (EndBrace);
+			writer.Write (NewLine);
+		}
+
+		if (ElseBlock is not null)
+			ElseBlock.Print (writer);
+	}
+}

src/bgen/CodeBlocks/LineBlock.cs

@@ -0,0 +1,30 @@
+using System.IO;
+
+public class LineBlock : ICodeBlock {
+	readonly string line;
+	int currentIndent = 0;
+	const char newLine = '\n';
+
+	public LineBlock (string line)
+	{
+		this.line = line;
+	}
+
+	public void SetIndent (int indent)
+	{
+		currentIndent = indent;
+	}
+
+	public void WriteIndent (TextWriter writer)
+	{
+		for (var i = 0; i < currentIndent; i++)
+			writer.Write (' ');
+	}
+
+	public void Print (TextWriter writer)
+	{
+		WriteIndent (writer);
+		writer.Write (line);
+		writer.Write (newLine);
+	}
+}

src/bgen/CodeBlocks/MethodBlock.cs

@@ -0,0 +1,29 @@
+using System.Collections.Generic;
+using System.Text;
+
+public class MethodBlock : CodeBlock {
+	public MethodBlock (string methodSignature, params string [] parameters)
+	{
+		StringBuilder allParameters = new ();
+		for (int i = 0; i < parameters.Length; i++) {
+			allParameters.Append (parameters [i]);
+			if (i != parameters.Length - 1) {
+				allParameters.Append (", ");
+			}
+		}
+		HeaderText = methodSignature + "(" + allParameters + ")";
+	}
+
+	public MethodBlock (string methodSignature, List<ICodeBlock> blocks, params string [] parameters)
+	{
+		StringBuilder allParameters = new ();
+		for (int i = 0; i < parameters.Length; i++) {
+			allParameters.Append (parameters [i]);
+			if (i != parameters.Length - 1) {
+				allParameters.Append (", ");
+			}
+		}
+		HeaderText = methodSignature + "(" + allParameters + ")";
+		this.Blocks = blocks;
+	}
+}

src/generator.csproj

@@ -169,6 +169,12 @@
     <Compile Include="bgen\Models\MemberInformation.cs" />
     <Compile Include="bgen\Models\TrampolineInfo.cs" />
     <Compile Include="bgen\Models\WrapPropMemberInformation.cs" />
+    <Compile Include="bgen\CodeBlocks\ICodeBlock.cs" />
+    <Compile Include="bgen\CodeBlocks\CodeBlock.cs" />
+    <Compile Include="bgen\CodeBlocks\MethodBlock.cs" />
+    <Compile Include="bgen\CodeBlocks\IfBlock.cs" />
+    <Compile Include="bgen\CodeBlocks\LineBlock.cs" />
+    <Compile Include="bgen\CodeBlocks\BlockContainer.cs" />
     <Compile Include="Resources.Designer.cs">
       <DependentUpon>Resources.resx</DependentUpon>
     </Compile>

tests/bgen/bgen-tests.csproj

@@ -30,6 +30,9 @@
     <Compile Include="..\generator\BGenTool.cs">
       <Link>BGenTool.cs</Link>
     </Compile>
+      <Compile Include="..\generator\CodeBlockTests.cs">
+          <Link>CodeBlockTests.cs</Link>
+      </Compile>
     <Compile Include="..\generator\AttributeFactoryTests.cs">
       <Link>AttributeFactoryTests.cs</Link>
     </Compile>