HeaderParser.cs

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

namespace WowClientDB2MySQLTableGenerator
{
    public sealed class HeaderParser
    {
        public string FileName { get; init; }
        public List<CStructureInfo> Structures { get; } = [];

        private readonly StringBuilder CommonTypes = new StringBuilder()
            .AppendLine("struct int64 { };")
            .AppendLine("struct int32 { };")
            .AppendLine("struct int16 { };")
            .AppendLine("struct int8 { };")
            .AppendLine("struct uint64 { };")
            .AppendLine("struct uint32 { };")
            .AppendLine("struct uint16 { };")
            .AppendLine("struct uint8 { };")
            .AppendLine("struct LocalizedString { };")
            .AppendLine("struct DBCPosition2D { float X, Y; };")
            .AppendLine("struct DBCPosition3D { float X, Y, Z; };")
            .AppendLine("namespace std { template<typename T, int N> using array = T[N]; }")
            .AppendLine("namespace Trinity { template<typename T> using RaceMask = T; }")
            .AppendLine("template<typename T> using EnumFlag = T;")
            .AppendLine("enum AreaFlags { };")
            .AppendLine("enum AreaFlags2 { };")
            .AppendLine("enum AreaMountFlags { };")
            .AppendLine("enum AreaTriggerActionSetFlag { };")
            .AppendLine("enum AreaTriggerShapeType { };")
            .AppendLine("using flag128 = int32[4];")
            .AppendLine("using BattlegroundBracketId = uint32;")
            .AppendLine("enum BattlePetSpeciesFlags { };")
            .AppendLine("enum BattlemasterListFlags { };")
            .AppendLine("enum BattlemasterType { };")
            .AppendLine("enum CampaignFlags { };")
            .AppendLine("enum CfgCategoriesCharsets { };")
            .AppendLine("enum CfgCategoriesFlags { };")
            .AppendLine("enum ChatChannelFlags { };")
            .AppendLine("enum ChatChannelRuleset { };")
            .AppendLine("enum ChrCustomizationReqFlag { };")
            .AppendLine("enum ChrRacesFlag { };")
            .AppendLine("enum ChrSpecializationFlag { };")
            .AppendLine("enum ChrSpecializationRole { };")
            .AppendLine("enum ContentTuningFlag { };")
            .AppendLine("enum CreatureModelDataFlags { };")
            .AppendLine("enum CriteriaFlags { };")
            .AppendLine("enum CriteriaTreeFlags { };")
            .AppendLine("enum CurrencyTypesFlags { };")
            .AppendLine("enum CurrencyTypesFlagsB { };")
            .AppendLine("enum FriendshipReputationFlags { };")
            .AppendLine("enum MapFlags { };")
            .AppendLine("enum MapFlags2 { };")
            .AppendLine("enum MapFlags3 { };")
            .AppendLine("enum MapDifficultyFlags { };")
            .AppendLine("enum MountFlags { };")
            .AppendLine("enum PathPropertyIndex { };")
            .AppendLine("enum PhaseEntryFlags { };")
            .AppendLine("enum PlayerConditionFlags { };")
            .AppendLine("enum PlayerDataElementType { };")
            .AppendLine("enum PowerTypeFlags { };")
            .AppendLine("enum QuestLineXQuestFlags { };")
            .AppendLine("enum SkillLineFlags { };")
            .AppendLine("enum SkillLineAbilityAcquireMethod { };")
            .AppendLine("enum SkillLineAbilityFlags { };")
            .AppendLine("enum SpellCategoryFlags { };")
            .AppendLine("enum SpellEffectAttributes { };")
            .AppendLine("enum SpellItemEnchantmentFlags { };")
            .AppendLine("enum SpellShapeshiftFormFlags { };")
            .AppendLine("enum SummonPropertiesFlags { };")
            .AppendLine("enum TaxiNodeFlags { };")
            .AppendLine("enum TraitCondFlags { };")
            .AppendLine("enum TraitConditionType { };")
            .AppendLine("enum TraitCurrencyType { };")
            .AppendLine("enum TraitPointsOperationType { };")
            .AppendLine("enum TraitNodeType { };")
            .AppendLine("enum TraitNodeEntryType { };")
            .AppendLine("enum TraitTreeFlag { };")
            .AppendLine("enum TraitSystemVariationType { };")
            .AppendLine("enum TransmogIllusionFlags { };")
            .AppendLine("enum TransmogOutfitEntryFlags { };")
            .AppendLine("enum TransmogOutfitEntrySource { };")
            .AppendLine("enum TransmogOutfitSetType { };")
            .AppendLine("enum TransmogOutfitSlot { };")
            .AppendLine("enum TransmogOutfitSlotFlags { };")
            .AppendLine("enum TransmogOutfitSlotOption { };")
            .AppendLine("enum TransmogOutfitSlotOptionFlags { };")
            .AppendLine("enum TransmogSituation { };")
            .AppendLine("enum TransmogSituationFlags { };")
            .AppendLine("enum TransmogSituationTrigger { };")
            .AppendLine("enum TransmogSituationTriggerFlags { };")
            .AppendLine("enum UiMapFlag { };")
            .AppendLine("enum UnitConditionFlags { };")
            .AppendLine("enum VehicleSeatFlags {};")
            .AppendLine("enum VehicleSeatFlagsB {};")
            .AppendLine("enum VehicleSeatFlagsC {};")
            .AppendLine("enum VignetteFlags {};")
            .AppendLine("enum WarbandSceneFlags {};")
            .AppendLine("enum WMOAreaTableFlags {};")
            .AppendLine("#define MAX_ITEM_PROTO_FLAGS 5")
            .AppendLine("#define MAX_ITEM_PROTO_ZONES 2")
            .AppendLine("#define MAX_ITEM_PROTO_SOCKETS 3")
            .AppendLine("#define MAX_ITEM_PROTO_STATS 10")
            .AppendLine("#define MAX_SPELL_AURA_INTERRUPT_FLAGS 2");

        public void Parse()
        {
            using var index = ClangSharp.Index.Create(false, true);
            var args = new[]
            {
                "-x",
                "c++",
                "--std=c++11"
            };

            // prepare input file
            // comment out existing includes
            // append required typedefs
            var hdr = CommonTypes.Append(File.ReadAllText(FileName).Replace("#include", "//#include")).ToString();

            var unsavedFiles = new[]
            {
                CXUnsavedFile.Create(FileName, hdr)
            };

            var result = CXTranslationUnit.TryParse(index.Handle, FileName, args, unsavedFiles, CXTranslationUnit_Flags.CXTranslationUnit_SkipFunctionBodies, out CXTranslationUnit translationUnitHandle);
            if (result != CXErrorCode.CXError_Success)
                return;

            using var translationUnit = TranslationUnit.GetOrCreate(translationUnitHandle);

            foreach (var cursorChild in translationUnit.TranslationUnitDecl.CursorChildren)
                VisitStruct(cursorChild);
        }

        private void VisitStruct(Cursor structCursor)
        {
            if (structCursor.Location.IsInSystemHeader)
                return;

            if (structCursor.CursorKind != CXCursorKind.CXCursor_StructDecl)
                return;

            if (!structCursor.Spelling.Contains("Entry"))
                return;

            var structure = new CStructureInfo();
            structure.Name = structCursor.Spelling;
            var suffixIndex = structure.Name.LastIndexOf("Entry", StringComparison.Ordinal);
            if (suffixIndex != -1)
                structure.Name = structure.Name.Remove(suffixIndex);

            structure.NormalizedName = structure.Name
                .Replace("GameObject", "Gameobject")
                .Replace("NPC", "Npc")
                .Replace("PvP", "Pvp")
                .Replace("PVP", "Pvp")
                .Replace("QuestXP", "QuestXp")
                .Replace("WMO", "Wmo")
                .Replace("AddOn", "Addon")
                .Replace("LFG", "Lfg")
                .Replace("POI", "Poi")
                .Replace("UI", "Ui")
                .Replace("_", "");

            foreach (var structMemberCursor in structCursor.CursorChildren)
                VisitField(structMemberCursor, structure);

            structure.Members.Add(new CStructureMemberInfo
            {
                TypeName = "int32",
                Name = "VerifiedBuild"
            });

            Structures.Add(structure);
            var localeStruct = structure.CreateLocaleTable();
            if (localeStruct != null)
                Structures.Add(localeStruct);
        }

        private static void VisitField(Cursor cursor, CStructureInfo structure)
        {
            if (cursor.CursorKind != CXCursorKind.CXCursor_FieldDecl)
                return;

            var typeInfo = ExtractFieldType(cursor);
            structure.Members.Add(new CStructureMemberInfo
            {
                TypeName = typeInfo.Type.Spelling.ToString(),
                Name = cursor.Spelling.Replace("_lang", ""),
                ArraySize = typeInfo.ArraySize
            });
        }

        private static (CXType Type, long ArraySize) ExtractFieldType(Cursor cursor)
        {
            var cxType = Desugar(cursor.Handle.Type);
            var arraySize = cxType.ArraySize;
            if (arraySize > 0)
                cxType = Desugar(cxType.ArrayElementType);

            return (cxType, Math.Max(arraySize, 1L));
        }

        private static CXType Desugar(CXType type)
        {
            while (type.IsSugared)
                type = type.Desugar;

            return type;
        }
    }
}