Brotli-Builder/BrotliLib/Markers/MarkerRoot.cs

using System.Collections;
using System.Collections.Generic;
using System.IO;
using System.Text;

namespace BrotliLib.Markers{
    public sealed class MarkerRoot : IEnumerable<MarkerNode>{
        public int TotalBits => children.Count == 0 ? 0 : children[^1].Marker.IndexEnd;

        private readonly List<MarkerNode> children = new List<MarkerNode>();

        internal void AddChild(MarkerNode added){
            children.Add(added);
        }

        public string BuildText(bool includeBitCounts){
            using var writer = new StringWriter{
                NewLine = "\n"
            };

            WriteText(writer, includeBitCounts);
            return writer.ToString();
        }

        public void WriteText(TextWriter writer, bool includeBitCounts){
            var build = new StringBuilder(256);

            foreach(MarkerNode node in this){
                node.WriteSelf(build, includeBitCounts);
                writer.WriteLine(build.ToString()); // TODO there is an overload for StringBuilder, which is officially documented for .NET Core 3.0, but the compiler finds sweet FA
                build.Clear();
            }
        }

        public IEnumerator<MarkerNode> GetEnumerator(){
            foreach(MarkerNode child in children){
                foreach(MarkerNode node in child){
                    yield return node;
                }
            }
        }

        IEnumerator IEnumerable.GetEnumerator(){
            return GetEnumerator();
        }

        public override string ToString(){
            return "Root";
        }
    }
}