From 78563c8e44cedd8bd2ff1edfe86c81b64ec32ce8 Mon Sep 17 00:00:00 2001
From: Joao Matos <joao@tritao.eu>
Date: Thu, 16 Feb 2017 02:16:16 +0000
Subject: [PATCH] Convert enum handling in C# generator to visitor model

---
 src/Generator/Generators/CSharp/CSharpSources.cs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/Generator/Generators/CSharp/CSharpSources.cs b/src/Generator/Generators/CSharp/CSharpSources.cs
index 07638140..8ece1ebb 100644
--- a/src/Generator/Generators/CSharp/CSharpSources.cs
+++ b/src/Generator/Generators/CSharp/CSharpSources.cs
@@ -220,7 +220,7 @@ namespace CppSharp.Generators.CSharp
         public override bool VisitDeclContext(DeclarationContext context)
         {
             foreach (var @enum in context.Enums)
-                GenerateEnum(@enum);
+                @enum.Visit(this);
 
             foreach (var typedef in context.Typedefs)
                 GenerateTypedef(typedef);
@@ -3041,10 +3041,10 @@ namespace CppSharp.Generators.CSharp
             return true;
         }
 
-        public void GenerateEnum(Enumeration @enum)
+        public override bool VisitEnumDecl(Enumeration @enum)
         {
             if (@enum.IsIncomplete)
-                return;
+                return true;
 
             PushBlock(CSharpBlockKind.Enum);
             GenerateDeclarationCommon(@enum);
@@ -3085,6 +3085,8 @@ namespace CppSharp.Generators.CSharp
             WriteCloseBraceIndent();
 
             PopBlock(NewLineKind.BeforeNextBlock);
+
+            return true;
         }
 
         public static string GetMethodIdentifier(Method method)