CppSharp/src/Generator/Utils/TextGenerator.cs

using System;
using System.Text;

namespace CppSharp
{
    public interface ITextGenerator
    {
        void Write(string msg, params object[] args);
        void WriteLine(string msg, params object[] args);
        void WriteLineIndent(string msg, params object[] args);
        void NewLine();
        void NewLineIfNeeded();
        void NeedNewLine();
        void ResetNewLine();
        void Indent(uint indentation = TextGenerator.DefaultIndentation);
        void Unindent();
        void WriteOpenBraceAndIndent();
        void UnindentAndWriteCloseBrace();
    }

    public class TextGenerator : ITextGenerator
    {
        public const uint DefaultIndentation = 4;

        public StringBuilder StringBuilder = new StringBuilder();
        public bool IsStartOfLine { get; set; }
        public bool NeedsNewLine { get; set; }
        public uint CurrentIndentation { get; set; }

        public TextGenerator()
        {
        }

        public TextGenerator(TextGenerator generator)
        {
            StringBuilder = new StringBuilder(generator);
            IsStartOfLine = generator.IsStartOfLine;
            NeedsNewLine = generator.NeedsNewLine;
            CurrentIndentation = generator.CurrentIndentation;
        }

        public TextGenerator Clone()
        {
            return new TextGenerator(this);
        }

        public void Write(string msg, params object[] args)
        {
            if (string.IsNullOrEmpty(msg))
                return;

            if (args.Length > 0)
                msg = string.Format(msg, args);

            if (IsStartOfLine && !string.IsNullOrWhiteSpace(msg))
                StringBuilder.Append(new string(' ',
                    (int) (CurrentIndentation * DefaultIndentation)));

            if (msg.Length > 0)
                IsStartOfLine = msg.EndsWith(Environment.NewLine);

            StringBuilder.Append(msg);
        }

        public void WriteLine(string msg, params object[] args)
        {
            Write(msg, args);
            NewLine();
        }

        public void WriteLineIndent(string msg, params object[] args)
        {
            Indent();
            WriteLine(msg, args);
            Unindent();
        }

        public void NewLine()
        {
            StringBuilder.AppendLine(string.Empty);
            IsStartOfLine = true;
        }

        public void NewLineIfNeeded()
        {
            if (!NeedsNewLine) return;

            NewLine();
            NeedsNewLine = false;
        }

        public void NeedNewLine()
        {
            NeedsNewLine = true;
        }

        public void ResetNewLine()
        {
            NeedsNewLine = false;
        }

        public void Indent(uint indentation = 1)
        {
            CurrentIndentation++;
        }

        public void Unindent()
        {
            CurrentIndentation--;
        }

        public void WriteOpenBraceAndIndent()
        {
            WriteLine("{");
            Indent();
        }

        public void UnindentAndWriteCloseBrace()
        {
            Unindent();
            WriteLine("}");
        }

        public override string ToString()
        {
            return StringBuilder.ToString();
        }

        public static implicit operator string(TextGenerator tg)
        {
            return tg.ToString();
        }
    }
}