CppSharp/src/Generator/Generators/Registrable/Lua/Sol/LuaSolHeaders.cs

using System.Collections.Generic;
using CppSharp.AST;

namespace CppSharp.Generators.Registrable.Lua.Sol
{
    public class LuaSolHeaders : LuaSolSources
    {
        public LuaSolHeaders(LuaSolGenerator generator, IEnumerable<TranslationUnit> units)
            : base(generator, units)
        {
        }

        public override string FileExtension => "h";

        protected override bool TemplateAllowed { get { return true; } }

        public override void Process()
        {
            GenerateFilePreamble(CommentKind.BCPL);

            PushBlock();
            WriteLine("#pragma once");
            PopBlock(NewLineKind.BeforeNextBlock);

            TranslationUnit.Visit(this);
        }

        #region TranslationUnit

        public virtual void GenerateTranslationUnitRegistrationFunctionDeclaration(TranslationUnit translationUnit)
        {
            NewLine();
            GenerateTranslationUnitRegistrationFunctionSignature(translationUnit);
            WriteLine(";");
            NewLine();
        }

        public override void GenerateTranslationUnit(TranslationUnit translationUnit)
        {
            GenerateTranslationUnitNamespaceBegin(translationUnit);
            GenerateTranslationUnitRegistrationFunctionBody(translationUnit);
            GenerateTranslationUnitRegistrationFunctionDeclaration(translationUnit);
            GenerateTranslationUnitNamespaceEnd(translationUnit);
        }

        public override bool CanGenerateTranslationUnit(TranslationUnit unit)
        {
            if (AlreadyVisited(unit))
            {
                return false;
            }
            return true;
        }

        public override bool VisitTranslationUnit(TranslationUnit unit)
        {
            if (!CanGenerateTranslationUnit(unit))
            {
                return false;
            }

            GenerateTranslationUnit(unit);

            return true;
        }

        #endregion

        public virtual void GenerateMain()
        {
            VisitNamespace(TranslationUnit);
        }

        public virtual void GenerateIncludes()
        {
            if (Generator.GeneratorOptions.BaseInclude != null)
            {
                WriteLineIndent(Generator.GeneratorOptions.BaseInclude.ToString());
            }
        }
    }
}